add projects feature

Author: ivinjabraham

migrations/20250124095543_create_project_table.sql

@@ -0,0 +1,6 @@
+CREATE TABLE Project (
+    project_id SERIAL PRIMARY KEY,
+    member_id INT NOT NULL,
+    title TEXT,
+    CONSTRAINT fkey_member FOREIGN KEY (member_id) REFERENCES Member(member_id) ON DELETE CASCADE
+);

src/graphql/mod.rs

@@ -1,14 +1,24 @@
 use async_graphql::MergedObject;
-use mutations::{AttendanceMutations, MemberMutations, StreakMutations};
-use queries::{AttendanceQueries, MemberQueries, StreakQueries};
+use mutations::{AttendanceMutations, MemberMutations, ProjectMutations, StreakMutations};
+use queries::{AttendanceQueries, MemberQueries, ProjectQueries, StreakQueries};
 
 pub mod mutations;
 pub mod queries;
 
 // This is our main query or QueryRoot. It is made up of structs representing sub-queries, one for each table in the DB. The fields of a relation are exposed via the [`async_graphql::SimpleObject`] directive on the [`models`] themselves. Specific queries, such as getting a member by ID or getting the streak of a member is defined as methods of the sub-query struct. Complex queries, such as those getting related data from multiple tables like querying all members and the streaks of each member, are defined via the [`async_graphql::ComplexObject`] directive on the [`models`] and can be found in the corresponding sub-query module.
 #[derive(MergedObject, Default)]
-pub struct Query(MemberQueries, AttendanceQueries, StreakQueries);
+pub struct Query(
+    MemberQueries,
+    AttendanceQueries,
+    StreakQueries,
+    ProjectQueries,
+);
 
 // Mutations work the same as Queries, sub-modules for each relation in the DB. However, all methods are directly defined on these sub-module structs. But they use slightly modified versions of the [`models`], marked by the Input in the name, to get input.
 #[derive(MergedObject, Default)]
-pub struct Mutation(MemberMutations, AttendanceMutations, StreakMutations);
+pub struct Mutation(
+    MemberMutations,
+    AttendanceMutations,
+    StreakMutations,
+    ProjectMutations,
+);

src/graphql/mutations/mod.rs

@@ -1,7 +1,9 @@
 pub mod attendance_mutations;
 pub mod member_mutations;
+pub mod project_mutations;
 pub mod streak_mutations;
 
 pub use attendance_mutations::AttendanceMutations;
 pub use member_mutations::MemberMutations;
+pub use project_mutations::ProjectMutations;
 pub use streak_mutations::StreakMutations;

src/graphql/mutations/project_mutations.rs

@@ -0,0 +1,29 @@
+use std::sync::Arc;
+
+use async_graphql::{Context, Error, Object, Result};
+use sqlx::PgPool;
+
+use crate::models::project::{Project, SetProjectInput};
+
+#[derive(Default)]
+pub struct ProjectMutations;
+
+#[Object]
+impl ProjectMutations {
+    #[graphql(name = "setProject")]
+    // input is member_id
+    async fn set_project(&self, ctx: &Context<'_>, input: SetProjectInput) -> Result<Project> {
+        let pool = ctx
+            .data::<Arc<PgPool>>()
+            .expect("Pool must be found in context");
+
+        let project = sqlx::query_as::<_, Project>(
+            "INSERT INTO Project (member_id, project_title) VALUES ($1, $2) RETURNING * ",
+        )
+        .bind(input.member_id)
+        .bind(input.title)
+        .fetch_one(pool.as_ref())
+        .await?;
+        Ok(project)
+    }
+}

src/graphql/queries/mod.rs

@@ -1,7 +1,9 @@
 pub mod attendance_queries;
 pub mod member_queries;
+pub mod project_queries;
 pub mod streak_queries;
 
 pub use attendance_queries::AttendanceQueries;
 pub use member_queries::MemberQueries;
+pub use project_queries::ProjectQueries;
 pub use streak_queries::StreakQueries;

src/graphql/queries/project_queries.rs

@@ -0,0 +1,23 @@
+use std::sync::Arc;
+
+use crate::models::project::Project;
+use async_graphql::{Context, Object, Result};
+use sqlx::PgPool;
+
+/// Sub-query for the [`Project`] table. The queries are:
+/// * projects - get all projects
+#[derive(Default)]
+pub struct ProjectQueries;
+
+#[Object]
+impl ProjectQueries {
+    pub async fn projects(&self, ctx: &Context<'_>) -> Result<Vec<Project>> {
+        let pool = ctx.data::<Arc<PgPool>>().expect("Pool must be in context.");
+
+        let projects = sqlx::query_as::<_, Project>("SELECT * FROM Project")
+            .fetch_all(pool.as_ref())
+            .await?;
+
+        Ok(projects)
+    }
+}

src/models/mod.rs

@@ -1,3 +1,4 @@
 pub mod attendance;
 pub mod member;
+pub mod project;
 pub mod status_update_streak;

src/models/project.rs

@@ -0,0 +1,15 @@
+use async_graphql::{InputObject, SimpleObject};
+use sqlx::FromRow;
+
+#[derive(FromRow, SimpleObject)]
+pub struct Project {
+    pub project_id: i32,
+    pub member_id: i32,
+    pub title: Option<String>,
+}
+
+#[derive(InputObject)]
+pub struct SetProjectInput {
+    pub member_id: i32,
+    pub title: String,
+}