notification system implemented

need refinements

apps/admin/src/lib/database/drizzle.ts

@@ -6,6 +6,7 @@ import {
   Attendance,
   Recommendation,
   FitnessGoal,
+  Notification,
   DatabaseConfig,
 } from "./types";
 import {
@@ -16,6 +17,7 @@ import {
   attendance,
   recommendations,
   fitnessGoals,
+  notifications,
   eq,
   and,
   desc,
@@ -1419,4 +1421,94 @@ export class DrizzleDatabase implements IDatabase {
       updatedAt: new Date(row.updatedAt as number | Date),
     };
   }
+
+  // Notification operations
+  async createNotification(
+    notification: Omit<Notification, "createdAt">,
+  ): Promise<Notification> {
+    const newNotification = {
+      ...notification,
+      createdAt: new Date(),
+    };
+
+    const result = await this.db.insert(notifications).values(newNotification);
+
+    log.debug("Notification created", { id: notification.id });
+    return newNotification;
+  }
+
+  async getNotificationsByUserId(userId: string): Promise<Notification[]> {
+    const rows = await this.db
+      .select()
+      .from(notifications)
+      .where(eq(notifications.userId, userId))
+      .orderBy(desc(notifications.createdAt))
+      .all();
+
+    return rows.map((row) => this.mapNotification(row));
+  }
+
+  async getUnreadNotificationCount(userId: string): Promise<number> {
+    const result = await this.db
+      .select({ count: sql<number>`count(*)` })
+      .from(notifications)
+      .where(
+        and(eq(notifications.userId, userId), eq(notifications.read, false)),
+      );
+
+    return Number(result[0]?.count || 0);
+  }
+
+  async markNotificationAsRead(id: string): Promise<Notification | null> {
+    await this.db
+      .update(notifications)
+      .set({ read: true })
+      .where(eq(notifications.id, id));
+
+    const rows = await this.db
+      .select()
+      .from(notifications)
+      .where(eq(notifications.id, id));
+
+    if (rows.length === 0) return null;
+
+    return this.mapNotification(rows[0]);
+  }
+
+  async markAllNotificationsAsRead(userId: string): Promise<void> {
+    await this.db
+      .update(notifications)
+      .set({ read: true })
+      .where(
+        and(eq(notifications.userId, userId), eq(notifications.read, false)),
+      );
+
+    log.debug("All notifications marked as read", { userId });
+  }
+
+  async deleteNotification(id: string): Promise<boolean> {
+    const result = await this.db
+      .delete(notifications)
+      .where(eq(notifications.id, id));
+
+    return result.changes > 0;
+  }
+
+  private mapNotification(row: Record<string, unknown>): Notification {
+    const createdAtValue = row.createdAt as number | Date;
+    const createdAt =
+      typeof createdAtValue === "number"
+        ? new Date(createdAtValue * 1000) // SQLite stores in seconds, convert to milliseconds
+        : createdAtValue;
+
+    return {
+      id: String(row.id),
+      userId: String(row.userId || row.user_id),
+      title: String(row.title),
+      message: String(row.message),
+      type: String(row.type) as Notification["type"],
+      read: Boolean(row.read),
+      createdAt,
+    };
+  }
 }