Add position type and query on it

4 files changed, 87 insertions, 2 deletions

diff --git a/src/graphql/query.rs b/src/graphql/query.rs
index 0e19771..a875d25 100644
--- a/src/graphql/query.rs
+++ b/src/graphql/query.rs
@@ -1,4 +1,4 @@
-use crate::{errors::AppError, graphql::types::user};
+use crate::graphql::types::{position, user};
 use async_graphql::{Context, Object};
 
 pub struct Query;
@@ -18,4 +18,14 @@ impl Query {
     ) -> Result<Option<Vec<user::User>>, String> {
         user::get_users(ctx, limit, offset).await
     }
+
+    /// Returns all the positions
+    async fn positions<'ctx>(
+        &self,
+        ctx: &Context<'ctx>,
+        #[graphql(desc = "Limit results")] limit: Option<i64>,
+        #[graphql(desc = "Offset results")] offset: Option<i64>,
+    ) -> Result<Option<Vec<position::Position>>, String> {
+        position::get_positions(ctx, limit, offset).await
+    }
 }
diff --git a/src/graphql/types/jwt.rs b/src/graphql/types/jwt.rs
index c118622..475b1bd 100644
--- a/src/graphql/types/jwt.rs
+++ b/src/graphql/types/jwt.rs
@@ -32,7 +32,7 @@ impl Keys {
 /// Claims struct.
 #[derive(Serialize, Deserialize, Debug, Clone)]
 pub struct Claims {
-    user_id: i32,
+    pub user_id: i32,
     exp: usize,
 }
 
diff --git a/src/graphql/types/mod.rs b/src/graphql/types/mod.rs
index d7cdece..a77cf8c 100644
--- a/src/graphql/types/mod.rs
+++ b/src/graphql/types/mod.rs
@@ -1,2 +1,3 @@
 pub mod jwt;
+pub mod position;
 pub mod user;
diff --git a/src/graphql/types/position.rs b/src/graphql/types/position.rs
new file mode 100644
index 0000000..e5ea8cd
--- /dev/null
+++ b/src/graphql/types/position.rs
@@ -0,0 +1,74 @@
+use crate::{dates::GraphQLDate, state::AppState};
+use async_graphql::{Context, Object};
+use chrono::Utc;
+use serde::{Deserialize, Serialize};
+
+use super::jwt::Authentication;
+
+#[derive(Clone, Debug, Serialize, Deserialize)]
+pub struct Position {
+    pub id: i32,
+    pub user_id: i32,
+    pub created_at: GraphQLDate,
+    pub latitude: f64,
+    pub longitude: f64,
+}
+
+#[Object]
+impl Position {
+    async fn id(&self) -> i32 {
+        self.id
+    }
+
+    async fn user_id(&self) -> i32 {
+        self.user_id
+    }
+
+    async fn created_at(&self) -> GraphQLDate {
+        self.created_at.clone()
+    }
+
+    async fn latitude(&self) -> f64 {
+        self.latitude
+    }
+
+    async fn longitude(&self) -> f64 {
+        self.longitude
+    }
+}
+
+pub async fn get_positions<'ctx>(
+    ctx: &Context<'ctx>,
+    limit: Option<i64>,
+    offset: Option<i64>,
+) -> Result<Option<Vec<Position>>, String> {
+    let state = ctx.data::<AppState>().expect("Can't connect to db");
+    let client = &*state.client;
+    let auth: &Authentication = ctx.data().unwrap();
+    match auth {
+        Authentication::NotLogged => Err("Unauthorized".to_string()),
+        Authentication::Logged(claims) => {
+            let rows = client.query("
+                SELECT id, user_id, created_at, ST_Y(location::geometry) AS latitude, ST_X(location::geometry) AS longitude
+                FROM positions
+                WHERE user_id = $1
+                ORDER BY id DESC
+                LIMIT $2
+                OFFSET $3",
+                &[&claims.user_id, &limit.unwrap_or(20), &offset.unwrap_or(0)]).await.unwrap();
+
+            let positions: Vec<Position> = rows
+                .iter()
+                .map(|row| Position {
+                    id: row.get("id"),
+                    user_id: row.get("user_id"),
+                    created_at: GraphQLDate(Utc::now()),
+                    latitude: row.get("latitude"),
+                    longitude: row.get("longitude"),
+                })
+                .collect();
+
+            Ok(Some(positions))
+        }
+    }
+}