feat: add member-specific status breaks

Author: hrideshmg

migrations/20260121190322_add_member_id_to_status_breaks.sql

@@ -0,0 +1,14 @@
+ALTER TABLE statusbreaks ADD COLUMN member_id INT;
+
+ALTER TABLE statusbreaks 
+ADD CONSTRAINT fk_statusbreaks_member 
+FOREIGN KEY (member_id) 
+REFERENCES member(member_id) 
+ON DELETE CASCADE;
+
+-- Make year nullable since member-specific breaks don't need it
+ALTER TABLE statusbreaks ALTER COLUMN year DROP NOT NULL;
+
+-- Add index for member_id lookups
+CREATE INDEX idx_statusbreaks_member_id ON statusbreaks(member_id);
+

src/graphql/mutations/status_mutations.rs

@@ -43,19 +43,35 @@ impl StatusMutations {
         input: CreateStatusBreakInput,
     ) -> Result<StatusBreakRecord> {
         let pool = ctx.data::<Arc<PgPool>>().expect("Pool must be in context");
-        let status = sqlx::query_as::<_, StatusBreakRecord>(
-            "INSERT INTO StatusBreaks (start_date, end_date, year, reason)
-             VALUES ($1, $2, $3, $4)
+
+        match (&input.year, &input.member_id) {
+            (Some(_), Some(_)) => {
+                return Err("Cannot specify both year and member_id. A status break must apply to either a year or a member, not both.".into());
+            }
+            (None, None) => {
+                return Err("Must specify either year or member_id. A status break must apply to either a year or a member.".into());
+            }
+            _ => {}
+        }
+
+        if input.start_date >= input.end_date {
+            return Err("start_date must be before end_date".into());
+        }
+
+        let status_break = sqlx::query_as::<_, StatusBreakRecord>(
+            "INSERT INTO StatusBreaks (start_date, end_date, year, member_id, reason)
+             VALUES ($1, $2, $3, $4, $5)
              RETURNING *
             ",
         )
         .bind(input.start_date)
         .bind(input.end_date)
         .bind(input.year)
+        .bind(input.member_id)
         .bind(&input.reason)
         .fetch_one(pool.as_ref())
         .await?;
 
-        Ok(status)
+        Ok(status_break)
     }
 }

src/graphql/queries/member_queries.rs

@@ -191,9 +191,15 @@ impl StatusInfo {
               AND (
                 is_sent = TRUE
                 OR NOT EXISTS (
-                    SELECT * FROM StatusBreaks sb
-                    WHERE year = (SELECT year from Member where member_id=$1)
-                    AND suh.date BETWEEN sb.start_date AND sb.end_date
+                    SELECT 1 FROM StatusBreaks sb
+                    WHERE suh.date BETWEEN sb.start_date AND sb.end_date
+                    AND (
+                        -- Year-wide break
+                        sb.year = (SELECT year FROM Member WHERE member_id = $1)
+                        OR
+                        -- Member-specific break
+                        sb.member_id = $1
+                    )
                 )
               )
             )
@@ -235,13 +241,19 @@ impl StatusUpdateRecord {
 
         let is_on_break = sqlx::query_scalar(
             "SELECT EXISTS (
-                SELECT 1 from StatusBreaks
-                WHERE year = (SELECT year FROM Member WHERE member_id = $1)
-                AND $2 BETWEEN start_date AND end_date
+                SELECT 1 FROM StatusBreaks
+                WHERE $1 BETWEEN start_date AND end_date
+                AND (
+                    -- Year-wide break
+                    year = (SELECT year FROM Member WHERE member_id = $2)
+                    OR
+                    -- Member-specific break
+                    member_id = $2
+                )
             )",
         )
-        .bind(self.member_id)
         .bind(self.date)
+        .bind(self.member_id)
         .fetch_optional(pool.as_ref())
         .await?;
 

src/models/status_update.rs

@@ -22,14 +22,16 @@ pub struct StatusBreakRecord {
     pub id: i32,
     pub start_date: NaiveDate,
     pub end_date: NaiveDate,
-    pub year: i32,
+    pub year: Option<i32>,
+    pub member_id: Option<i32>,
     pub reason: Option<String>,
 }
 
 #[derive(InputObject)]
 pub struct CreateStatusBreakInput {
     pub start_date: NaiveDate,
     pub end_date: NaiveDate,
-    pub year: i32,
+    pub year: Option<i32>,
+    pub member_id: Option<i32>,
     pub reason: Option<String>,
 }