homeio

1import "server-only";
2
3import { randomUUID } from "node:crypto";
4import { db } from "@/lib/server/db/drizzle";
5import { notifications } from "@/lib/server/db/schema";
6import type { NotificationKind, NotificationRecord } from "@/lib/shared/contracts/notifications";
7import { desc, eq } from "drizzle-orm";
8import { emitNotification, emitNotificationCleared, emitNotificationReadAll } from "./emitter";
9
10const MAX_NOTIFICATIONS = 50;
11
12function toRecord(row: typeof notifications.$inferSelect): NotificationRecord {
13  return {
14    id: row.id,
15    title: row.title,
16    body: row.body,
17    kind: row.kind as NotificationKind,
18    read: row.read,
19    createdAt: row.createdAt.toISOString(),
20  };
21}
22
23export async function listNotifications(): Promise<NotificationRecord[]> {
24  const rows = await db
25    .select()
26    .from(notifications)
27    .orderBy(desc(notifications.createdAt))
28    .limit(MAX_NOTIFICATIONS);
29  return rows.map(toRecord);
30}
31
32export async function createNotification(input: {
33  title: string;
34  body: string;
35  kind?: NotificationKind;
36}): Promise<NotificationRecord> {
37  const id = randomUUID();
38  const [row] = await db
39    .insert(notifications)
40    .values({
41      id,
42      title: input.title,
43      body: input.body,
44      kind: input.kind ?? "info",
45      read: false,
46    })
47    .returning();
48
49  const record = toRecord(row);
50  emitNotification(record);
51
52  // Prune old rows non-blocking
53  void pruneOldNotifications().catch(() => undefined);
54
55  return record;
56}
57
58export async function markAllNotificationsRead(): Promise<void> {
59  await db.update(notifications).set({ read: true }).where(eq(notifications.read, false));
60  emitNotificationReadAll();
61}
62
63export async function clearAllNotifications(): Promise<void> {
64  await db.delete(notifications);
65  emitNotificationCleared();
66}
67
68async function pruneOldNotifications() {
69  const rows = await db
70    .select({ id: notifications.id })
71    .from(notifications)
72    .orderBy(desc(notifications.createdAt))
73    .offset(MAX_NOTIFICATIONS);
74  if (rows.length === 0) return;
75  for (const row of rows) {
76    await db.delete(notifications).where(eq(notifications.id, row.id));
77  }
78}
79