57 lines
4.5 KiB
TypeScript
57 lines
4.5 KiB
TypeScript
import 'server-only';
|
|
import { and, desc, eq, inArray, isNull, or, sql } from 'drizzle-orm';
|
|
import { alias } from 'drizzle-orm/pg-core';
|
|
import { getDb } from '@/db';
|
|
import { notifications, comments, commentThreads, guides, users, media } from '@/db/schema';
|
|
import { publicMediaUrl } from '@/lib/media/storage';
|
|
import { notifyNotificationChange } from './events';
|
|
|
|
type Recipient = { userId: string; adminOnly?: boolean; url: string };
|
|
type Message = { eventKey: string; kind: string; title: string; body: string; commentId?: string; messageId?: string };
|
|
export async function createNotifications(recipients: Recipient[], message: Message) {
|
|
for (let offset = 0; offset < recipients.length; offset += 100) {
|
|
const inserted = await getDb().insert(notifications).values(recipients.slice(offset, offset + 100).map(recipient => ({ ...recipient, ...message })))
|
|
.onConflictDoNothing({ target: [notifications.userId, notifications.eventKey] }).returning({ userId: notifications.userId });
|
|
for (let start = 0; start < inserted.length; start += 10) await Promise.all(inserted.slice(start, start + 10).map(row => notifyNotificationChange(row.userId)));
|
|
}
|
|
}
|
|
const root = alias(comments, 'notification_root');
|
|
export async function listNotifications(user: { id: string; role: string | null; emailVerified: boolean }, before?: { time: string; id: string }) {
|
|
const admin = user.role === 'admin' && user.emailVerified;
|
|
const visible = and(eq(notifications.userId, user.id), admin ? undefined : and(eq(notifications.adminOnly, false), or(isNull(notifications.commentId), and(
|
|
eq(comments.hidden, false), sql`coalesce(${root.hidden}, false) = false`,
|
|
or(isNull(commentThreads.guideId), and(eq(guides.isPublic, true), isNull(guides.trashedAt))),
|
|
))));
|
|
const query = () => getDb().select({ notification: notifications, authorName: users.name, authorImage: users.image, cursorTime: sql<string>`to_char(${notifications.createdAt} at time zone 'UTC', 'YYYY-MM-DD"T"HH24:MI:SS.US"Z"')` }).from(notifications)
|
|
.leftJoin(comments, eq(comments.id, notifications.commentId)).leftJoin(root, eq(root.id, comments.rootId))
|
|
.leftJoin(users, and(eq(users.id, sql`case when ${notifications.kind} = 'comment_heart' then ${comments.heartedById} else ${comments.authorId} end`),
|
|
inArray(notifications.kind, ['comment_new', 'comment_reply', 'comment_edit', 'comment_heart'])))
|
|
.leftJoin(commentThreads, eq(commentThreads.id, comments.threadId)).leftJoin(guides, eq(guides.id, commentThreads.guideId));
|
|
const [rows, count] = await Promise.all([
|
|
query().where(and(visible, before ? sql`(${notifications.createdAt}, ${notifications.id}) < (${before.time}::timestamptz, ${before.id}::uuid)` : undefined)).orderBy(desc(notifications.createdAt), desc(notifications.id)).limit(21),
|
|
getDb().select({ count: sql<number>`count(*)::int` }).from(notifications)
|
|
.leftJoin(comments, eq(comments.id, notifications.commentId)).leftJoin(root, eq(root.id, comments.rootId))
|
|
.leftJoin(commentThreads, eq(commentThreads.id, comments.threadId)).leftJoin(guides, eq(guides.id, commentThreads.guideId))
|
|
.where(and(visible, isNull(notifications.readAt))),
|
|
]);
|
|
const page = rows.slice(0, 20);
|
|
const guideIds = [...new Set(page.flatMap(({ notification }) => {
|
|
if (notification.kind !== 'guide_published') return [];
|
|
const match = /^guide:([0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}):published$/i.exec(notification.eventKey);
|
|
return match ? [match[1]] : [];
|
|
}))];
|
|
const covers = guideIds.length ? await getDb().select({ id: guides.id, objectKey: media.objectKey }).from(guides)
|
|
.innerJoin(media, eq(media.id, guides.coverMediaId))
|
|
.where(and(inArray(guides.id, guideIds), eq(guides.isPublic, true), isNull(guides.trashedAt))) : [];
|
|
const guideImages = new Map(covers.map(cover => [`guide:${cover.id}:published`, publicMediaUrl(cover.objectKey)]));
|
|
const items = page.map(row => ({ ...row.notification, authorName: row.authorName, authorImage: row.authorImage,
|
|
imageUrl: row.notification.kind === 'guide_published' ? guideImages.get(row.notification.eventKey) ?? null : null,
|
|
}));
|
|
const last = rows[19];
|
|
return { items, unread: count[0]?.count ?? 0, nextCursor: rows.length > 20 ? `${last.cursorTime}|${last.notification.id}` : null };
|
|
}
|
|
export async function readNotifications(userId: string, id?: string) {
|
|
await getDb().update(notifications).set({ readAt: sql`now()` }).where(and(eq(notifications.userId, userId), isNull(notifications.readAt), id ? eq(notifications.id, id) : undefined));
|
|
await notifyNotificationChange(userId);
|
|
}
|