feat: add query for leave count

Author: naveensrinivas282

src/graphql/mutations/attendance_mutations.rs

@@ -7,7 +7,9 @@ use sha2::Sha256;
 use sqlx::PgPool;
 
 use crate::auth::guards::AdminOrBotGuard;
-use crate::models::attendance::{AttendanceRecord, MarkAttendanceInput, MarkLeaveInput, MarkLeaveOutput};
+use crate::models::attendance::{
+    AttendanceRecord, MarkAttendanceInput, MarkLeaveInput, MarkLeaveOutput,
+};
 
 type HmacSha256 = Hmac<Sha256>;
 
@@ -71,7 +73,7 @@ impl AttendanceMutations {
             .data::<Arc<PgPool>>()
             .expect("Pool not found in context");
         let now = chrono::Utc::now().with_timezone(&Kolkata);
-        
+
         let leave: MarkLeaveOutput = sqlx::query_as::<_, MarkLeaveOutput>(
             "INSERT INTO Leave
             (discord_id, date, duration, reason, approved_by) 

src/graphql/queries/member_queries.rs

@@ -1,6 +1,9 @@
 use crate::auth::guards::AuthGuard;
 use crate::auth::AuthContext;
-use crate::models::{attendance::AttendanceRecord, status_update::StatusUpdateRecord};
+use crate::models::{
+    attendance::{AttendanceRecord, LeaveCountOutput},
+    status_update::StatusUpdateRecord,
+};
 use async_graphql::{ComplexObject, Context, Object, Result};
 use chrono::NaiveDate;
 use sqlx::PgPool;
@@ -397,4 +400,37 @@ impl Member {
             member_id: self.member_id,
         }
     }
+
+    async fn leave_count(
+        &self,
+        ctx: &Context<'_>,
+        start_date: NaiveDate,
+        end_date: NaiveDate,
+    ) -> Result<LeaveCountOutput> {
+        let pool = ctx.data::<Arc<PgPool>>().expect("Pool must be in context.");
+
+        if end_date < start_date {
+            return Err("end_date must be >= start_date".into());
+        }
+        let leave = sqlx::query_as::<_, LeaveCountOutput>(
+            r#"
+                SELECT discord_id, SUM(duration) AS count
+                FROM "Leave"
+                WHERE date > $1
+                  AND (date + duration) < $2
+                  AND discord_id = $3
+                GROUP BY discord_id
+                "#,
+        )
+        .bind(start_date)
+        .bind(end_date)
+        .bind(
+            self.discord_id
+                .as_ref()
+                .expect("Leave count needs discord_id"),
+        )
+        .fetch_one(pool.as_ref())
+        .await?;
+        Ok(leave)
+    }
 }

src/models/attendance.rs

@@ -37,4 +37,10 @@ pub struct MarkLeaveOutput {
     pub reason: String,
     pub duration: i32,
     pub approved_by: Option<String>,
-}
+}
+
+#[derive(SimpleObject, FromRow)]
+pub struct LeaveCountOutput {
+    pub discord_id: String,
+    pub count: i32,
+}