refactor(api): split notification procedure support

This commit is contained in:
2026-03-31 21:56:15 +02:00
parent bec1b98688
commit 4111b7b661
5 changed files with 1036 additions and 945 deletions
@@ -0,0 +1,222 @@
import { TRPCError } from "@trpc/server";
import { z } from "zod";
import { findUniqueOrThrow } from "../db/helpers.js";
import { createNotification } from "../lib/create-notification.js";
import { resolveRecipients } from "../lib/notification-targeting.js";
import {
emitNotificationCreated,
emitTaskAssigned,
} from "../sse/event-bus.js";
import {
type BroadcastPersistenceDb,
type BroadcastRecipientNotification,
CreateBroadcastInputSchema,
ListBroadcastsInputSchema,
type NotificationProcedureContext,
NotificationIdInputSchema,
requireNotificationDbUser,
rethrowNotificationReferenceError,
SCHEDULED_TASK_BROADCAST_UNSUPPORTED_MESSAGE,
sendNotificationEmail,
} from "./notification-procedure-base.js";
function isFutureScheduledBroadcast(input: z.infer<typeof CreateBroadcastInputSchema>): boolean {
return Boolean(input.scheduledAt && input.scheduledAt > new Date());
}
function hasTaskLikeBroadcastMetadata(input: z.infer<typeof CreateBroadcastInputSchema>): boolean {
return input.category === "TASK"
|| input.category === "APPROVAL"
|| input.taskAction !== undefined
|| input.dueDate !== undefined;
}
function buildBroadcastCreateData(
senderId: string,
input: z.infer<typeof CreateBroadcastInputSchema>,
recipientCount?: number,
) {
return {
senderId,
title: input.title,
...(input.body !== undefined ? { body: input.body } : {}),
...(input.link !== undefined ? { link: input.link } : {}),
category: input.category,
priority: input.priority,
channel: input.channel,
targetType: input.targetType,
...(input.targetValue !== undefined ? { targetValue: input.targetValue } : {}),
...(input.scheduledAt !== undefined ? { scheduledAt: input.scheduledAt } : {}),
...(recipientCount !== undefined ? { recipientCount } : {}),
};
}
async function resolveBroadcastRecipientIds(
ctx: NotificationProcedureContext,
input: z.infer<typeof CreateBroadcastInputSchema>,
senderId: string,
): Promise<string[]> {
const recipientIds = await resolveRecipients(
input.targetType,
input.targetValue,
ctx.db,
senderId,
);
if (recipientIds.length === 0) {
throw new TRPCError({
code: "BAD_REQUEST",
message: "No recipients matched the broadcast target.",
});
}
return recipientIds;
}
async function createScheduledBroadcastRecord(
db: BroadcastPersistenceDb,
senderId: string,
input: z.infer<typeof CreateBroadcastInputSchema>,
recipientIds: string[],
) {
return db.notificationBroadcast.create({
data: buildBroadcastCreateData(senderId, input, recipientIds.length),
});
}
async function persistImmediateBroadcast(
db: BroadcastPersistenceDb,
senderId: string,
input: z.infer<typeof CreateBroadcastInputSchema>,
recipientIds: string[],
) {
const isTask = input.category === "TASK" || input.category === "APPROVAL";
const sentAt = new Date();
const broadcast = await db.notificationBroadcast.create({
data: buildBroadcastCreateData(senderId, input),
});
const notificationIds: BroadcastRecipientNotification[] = [];
for (const recipientUserId of recipientIds) {
const notificationId = await createNotification({
db,
userId: recipientUserId,
type: `BROADCAST_${input.category}`,
title: input.title,
body: input.body,
link: input.link,
category: input.category,
priority: input.priority,
channel: input.channel,
sourceId: broadcast.id,
senderId,
taskStatus: isTask ? "OPEN" : undefined,
taskAction: input.taskAction,
dueDate: input.dueDate,
emit: false,
});
notificationIds.push({ id: notificationId, userId: recipientUserId });
}
const updatedBroadcast = await db.notificationBroadcast.update({
where: { id: broadcast.id },
data: {
sentAt,
recipientCount: notificationIds.length,
},
});
return { broadcast: updatedBroadcast, notificationIds };
}
function emitImmediateBroadcastSideEffects(
db: BroadcastPersistenceDb,
input: z.infer<typeof CreateBroadcastInputSchema>,
notificationIds: BroadcastRecipientNotification[],
) {
const isTask = input.category === "TASK" || input.category === "APPROVAL";
for (const notification of notificationIds) {
emitNotificationCreated(notification.userId, notification.id);
if (isTask) {
emitTaskAssigned(notification.userId, notification.id);
}
}
if (input.channel === "email" || input.channel === "both") {
for (const notification of notificationIds) {
void sendNotificationEmail(db, notification.userId, input.title, input.body);
}
}
}
export async function createBroadcast(
ctx: NotificationProcedureContext,
input: z.infer<typeof CreateBroadcastInputSchema>,
) {
const senderId = requireNotificationDbUser(ctx).id;
const isScheduledFutureBroadcast = isFutureScheduledBroadcast(input);
if (isScheduledFutureBroadcast && hasTaskLikeBroadcastMetadata(input)) {
throw new TRPCError({
code: "BAD_REQUEST",
message: SCHEDULED_TASK_BROADCAST_UNSUPPORTED_MESSAGE,
});
}
const recipientIds = await resolveBroadcastRecipientIds(ctx, input, senderId);
if (isScheduledFutureBroadcast) {
try {
return await createScheduledBroadcastRecord(ctx.db, senderId, input, recipientIds);
} catch (error) {
rethrowNotificationReferenceError(error);
}
}
let persistedBroadcast: Awaited<ReturnType<typeof persistImmediateBroadcast>>["broadcast"];
let notificationIds: BroadcastRecipientNotification[] = [];
try {
const transactionResult = typeof ctx.db.$transaction === "function"
? await ctx.db.$transaction((tx) =>
persistImmediateBroadcast(tx as typeof ctx.db, senderId, input, recipientIds))
: await persistImmediateBroadcast(ctx.db, senderId, input, recipientIds);
persistedBroadcast = transactionResult.broadcast;
notificationIds = transactionResult.notificationIds;
} catch (error) {
rethrowNotificationReferenceError(error);
}
emitImmediateBroadcastSideEffects(ctx.db, input, notificationIds);
return persistedBroadcast;
}
export async function listBroadcasts(
ctx: NotificationProcedureContext,
input: z.infer<typeof ListBroadcastsInputSchema>,
) {
return ctx.db.notificationBroadcast.findMany({
orderBy: { createdAt: "desc" },
take: input.limit,
include: {
sender: { select: { id: true, name: true, email: true } },
},
});
}
export async function getBroadcastById(
ctx: NotificationProcedureContext,
input: z.infer<typeof NotificationIdInputSchema>,
) {
return findUniqueOrThrow(
ctx.db.notificationBroadcast.findUnique({
where: { id: input.id },
include: {
sender: { select: { id: true, name: true, email: true } },
},
}),
"Broadcast",
);
}
@@ -0,0 +1,398 @@
import { TRPCError } from "@trpc/server";
import { z } from "zod";
import { createNotification } from "../lib/create-notification.js";
import { sendEmail } from "../lib/email.js";
import { emitTaskAssigned } from "../sse/event-bus.js";
import type { TRPCContext } from "../trpc.js";
export type NotificationProcedureContext = Pick<TRPCContext, "db" | "dbUser" | "roleDefaults" | "session">;
export function requireNotificationDbUser(ctx: NotificationProcedureContext) {
if (!ctx.dbUser) {
throw new TRPCError({ code: "UNAUTHORIZED", message: "User account not found" });
}
return ctx.dbUser;
}
export async function resolveUserId(ctx: {
db: {
user: {
findUnique: (args: {
where: { email: string };
select: { id: true };
}) => Promise<{ id: string } | null>;
};
};
session: { user?: { email?: string | null } | null } | null;
}): Promise<string> {
const email = ctx.session?.user?.email;
if (!email) {
throw new TRPCError({ code: "UNAUTHORIZED" });
}
const user = await ctx.db.user.findUnique({
where: { email },
select: { id: true },
});
if (!user) {
throw new TRPCError({ code: "UNAUTHORIZED" });
}
return user.id;
}
export function getNotificationErrorCandidates(error: unknown): Array<{
code?: unknown;
message?: unknown;
meta?: { field_name?: unknown; modelName?: unknown };
cause?: unknown;
}> {
const queue: unknown[] = [error];
const seen = new Set<unknown>();
const candidates: Array<{
code?: unknown;
message?: unknown;
meta?: { field_name?: unknown; modelName?: unknown };
cause?: unknown;
}> = [];
while (queue.length > 0) {
const current = queue.shift();
if (!current || seen.has(current) || typeof current !== "object") {
continue;
}
seen.add(current);
const candidate = current as {
code?: unknown;
message?: unknown;
meta?: { field_name?: unknown; modelName?: unknown };
cause?: unknown;
shape?: { message?: unknown; data?: { cause?: unknown } };
};
candidates.push(candidate);
if ("cause" in candidate) {
queue.push(candidate.cause);
}
if (candidate.shape?.data?.cause) {
queue.push(candidate.shape.data.cause);
}
}
return candidates;
}
export function rethrowNotificationReferenceError(error: unknown): never {
for (const candidate of getNotificationErrorCandidates(error)) {
const fieldName = typeof candidate.meta?.field_name === "string"
? candidate.meta.field_name.toLowerCase()
: "";
const modelName = typeof candidate.meta?.modelName === "string"
? candidate.meta.modelName.toLowerCase()
: "";
if (
typeof candidate.code === "string"
&& (candidate.code === "P2003" || candidate.code === "P2025")
&& fieldName.includes("sender")
) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Sender user not found",
cause: error,
});
}
if (
typeof candidate.code === "string"
&& (candidate.code === "P2003" || candidate.code === "P2025")
&& fieldName.includes("userid")
) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Broadcast recipient user not found",
cause: error,
});
}
if (
typeof candidate.code === "string"
&& (candidate.code === "P2003" || candidate.code === "P2025")
&& (
modelName.includes("notificationbroadcast")
|| fieldName.includes("broadcast")
|| fieldName.includes("sourceid")
)
) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Notification broadcast not found",
cause: error,
});
}
}
throw error;
}
export const categoryEnum = z.enum(["NOTIFICATION", "REMINDER", "TASK", "APPROVAL"]);
export const priorityEnum = z.enum(["LOW", "NORMAL", "HIGH", "URGENT"]);
export const taskStatusEnum = z.enum(["OPEN", "IN_PROGRESS", "DONE", "DISMISSED"]);
export const channelEnum = z.enum(["in_app", "email", "both"]);
export const recurrenceEnum = z.enum(["daily", "weekly", "monthly"]);
export const targetTypeEnum = z.enum(["user", "role", "project", "orgUnit", "all"]);
export const SCHEDULED_TASK_BROADCAST_UNSUPPORTED_MESSAGE =
"Scheduled broadcasts with task metadata are not supported yet.";
export const NotificationListInputSchema = z.object({
unreadOnly: z.boolean().optional(),
category: categoryEnum.optional(),
taskStatus: taskStatusEnum.optional(),
priority: priorityEnum.optional(),
limit: z.number().min(1).max(100).default(50),
});
export const MarkNotificationReadInputSchema = z.object({
id: z.string().optional(),
});
export const CreateManagedNotificationInputSchema = z.object({
userId: z.string(),
type: z.string(),
title: z.string(),
body: z.string().optional(),
entityId: z.string().optional(),
entityType: z.string().optional(),
category: categoryEnum.optional(),
priority: priorityEnum.optional(),
link: z.string().optional(),
taskStatus: taskStatusEnum.optional(),
taskAction: z.string().optional(),
assigneeId: z.string().optional(),
dueDate: z.date().optional(),
channel: channelEnum.optional(),
senderId: z.string().optional(),
});
export const ListNotificationTasksInputSchema = z.object({
status: taskStatusEnum.optional(),
includeAssigned: z.boolean().default(true),
limit: z.number().min(1).max(100).default(20),
});
export const NotificationIdInputSchema = z.object({
id: z.string(),
});
export const UpdateNotificationTaskStatusInputSchema = z.object({
id: z.string(),
status: taskStatusEnum,
});
export const CreateReminderInputSchema = z.object({
title: z.string().min(1).max(200),
body: z.string().max(2000).optional(),
remindAt: z.date(),
recurrence: recurrenceEnum.optional(),
entityId: z.string().optional(),
entityType: z.string().optional(),
link: z.string().optional(),
});
export const UpdateReminderInputSchema = z.object({
id: z.string(),
title: z.string().min(1).max(200).optional(),
body: z.string().max(2000).optional(),
remindAt: z.date().optional(),
recurrence: recurrenceEnum.nullish(),
});
export const ListRemindersInputSchema = z.object({
limit: z.number().min(1).max(100).default(20),
});
export const CreateBroadcastInputSchema = z.object({
title: z.string().min(1).max(200),
body: z.string().max(2000).optional(),
link: z.string().optional(),
category: categoryEnum.default("NOTIFICATION"),
priority: priorityEnum.default("NORMAL"),
channel: channelEnum.default("in_app"),
targetType: targetTypeEnum,
targetValue: z.string().optional(),
scheduledAt: z.date().optional(),
taskAction: z.string().optional(),
dueDate: z.date().optional(),
});
export const ListBroadcastsInputSchema = z.object({
limit: z.number().min(1).max(50).default(20),
});
export const CreateTaskInputSchema = z.object({
userId: z.string(),
title: z.string().min(1).max(200),
body: z.string().max(2000).optional(),
priority: priorityEnum.default("NORMAL"),
dueDate: z.date().optional(),
taskAction: z.string().optional(),
entityId: z.string().optional(),
entityType: z.string().optional(),
link: z.string().optional(),
channel: channelEnum.default("in_app"),
});
export const AssignTaskInputSchema = z.object({
id: z.string(),
assigneeId: z.string(),
});
export type BroadcastRecipientNotification = { id: string; userId: string };
export type BroadcastPersistenceDb = NotificationProcedureContext["db"];
export async function sendNotificationEmail(
db: {
user: {
findUnique: (args: {
where: { id: string };
select: { email: true; name: true };
}) => Promise<{ email: string; name: string | null } | null>;
};
},
userId: string,
title: string,
body?: string | null,
): Promise<void> {
try {
const user = await db.user.findUnique({
where: { id: userId },
select: { email: true, name: true },
});
if (!user) {
return;
}
void sendEmail({
to: user.email,
subject: title,
text: body ?? title,
...(body !== undefined && body !== null ? { html: `<p>${body}</p>` } : {}),
});
} catch {
// non-blocking
}
}
export async function listNotifications(
ctx: NotificationProcedureContext,
input: z.infer<typeof NotificationListInputSchema>,
) {
const userId = await resolveUserId(ctx);
return ctx.db.notification.findMany({
where: {
userId,
...(input.unreadOnly ? { readAt: null } : {}),
...(input.category !== undefined ? { category: input.category } : {}),
...(input.taskStatus !== undefined ? { taskStatus: input.taskStatus } : {}),
...(input.priority !== undefined ? { priority: input.priority } : {}),
},
orderBy: { createdAt: "desc" },
take: input.limit,
});
}
export async function countUnreadNotifications(ctx: NotificationProcedureContext) {
const userId = await resolveUserId(ctx);
return ctx.db.notification.count({
where: { userId, readAt: null },
});
}
export async function markNotificationsRead(
ctx: NotificationProcedureContext,
input: z.infer<typeof MarkNotificationReadInputSchema>,
) {
const userId = await resolveUserId(ctx);
const now = new Date();
if (input.id) {
await ctx.db.notification.update({
where: { id: input.id, userId },
data: { readAt: now },
});
return;
}
await ctx.db.notification.updateMany({
where: { userId, readAt: null },
data: { readAt: now },
});
}
export async function createManagedNotification(
ctx: NotificationProcedureContext,
input: z.infer<typeof CreateManagedNotificationInputSchema>,
) {
const currentUserId = requireNotificationDbUser(ctx).id;
const notificationId = await createNotification({
db: ctx.db,
userId: input.userId,
type: input.type,
title: input.title,
body: input.body,
entityId: input.entityId,
entityType: input.entityType,
category: input.category,
priority: input.priority,
link: input.link,
taskStatus: input.taskStatus,
taskAction: input.taskAction,
assigneeId: input.assigneeId,
dueDate: input.dueDate,
channel: input.channel,
senderId: input.senderId ?? currentUserId,
});
if (input.category === "TASK" || input.category === "APPROVAL") {
emitTaskAssigned(input.userId, notificationId);
}
const channel = input.channel ?? "in_app";
if (channel === "email" || channel === "both") {
void sendNotificationEmail(ctx.db, input.userId, input.title, input.body);
}
return ctx.db.notification.findUnique({ where: { id: notificationId } });
}
export async function deleteNotification(
ctx: NotificationProcedureContext,
input: z.infer<typeof NotificationIdInputSchema>,
) {
const userId = await resolveUserId(ctx);
const existing = await ctx.db.notification.findFirst({
where: { id: input.id, userId },
});
if (!existing) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Notification not found",
});
}
if (
(existing.category === "TASK" || existing.category === "APPROVAL")
&& existing.senderId
&& existing.senderId !== userId
) {
throw new TRPCError({
code: "FORBIDDEN",
message: "Cannot delete tasks created by others",
});
}
await ctx.db.notification.delete({ where: { id: input.id } });
}
@@ -1,945 +1,4 @@
import { PermissionKey, parseTaskAction, resolvePermissions } from "@capakraken/shared";
import { TRPCError } from "@trpc/server";
import { z } from "zod";
import { findUniqueOrThrow } from "../db/helpers.js";
import { sendEmail } from "../lib/email.js";
import { createNotification } from "../lib/create-notification.js";
import { resolveRecipients } from "../lib/notification-targeting.js";
import { getTaskAction } from "../lib/task-actions.js";
import {
emitNotificationCreated,
emitTaskAssigned,
emitTaskCompleted,
emitTaskStatusChanged,
} from "../sse/event-bus.js";
import type { TRPCContext } from "../trpc.js";
type NotificationProcedureContext = Pick<TRPCContext, "db" | "dbUser" | "roleDefaults" | "session">;
export async function resolveUserId(ctx: {
db: {
user: {
findUnique: (args: {
where: { email: string };
select: { id: true };
}) => Promise<{ id: string } | null>;
};
};
session: { user?: { email?: string | null } | null };
}): Promise<string> {
const email = ctx.session.user?.email;
if (!email) {
throw new TRPCError({ code: "UNAUTHORIZED" });
}
const user = await ctx.db.user.findUnique({
where: { email },
select: { id: true },
});
if (!user) {
throw new TRPCError({ code: "UNAUTHORIZED" });
}
return user.id;
}
async function sendNotificationEmail(
db: {
user: {
findUnique: (args: {
where: { id: string };
select: { email: true; name: true };
}) => Promise<{ email: string; name: string | null } | null>;
};
},
userId: string,
title: string,
body?: string | null,
): Promise<void> {
try {
const user = await db.user.findUnique({
where: { id: userId },
select: { email: true, name: true },
});
if (!user) {
return;
}
void sendEmail({
to: user.email,
subject: title,
text: body ?? title,
...(body !== undefined && body !== null ? { html: `<p>${body}</p>` } : {}),
});
} catch {
// non-blocking
}
}
export function getNotificationErrorCandidates(error: unknown): Array<{
code?: unknown;
message?: unknown;
meta?: { field_name?: unknown; modelName?: unknown };
cause?: unknown;
}> {
const queue: unknown[] = [error];
const seen = new Set<unknown>();
const candidates: Array<{
code?: unknown;
message?: unknown;
meta?: { field_name?: unknown; modelName?: unknown };
cause?: unknown;
}> = [];
while (queue.length > 0) {
const current = queue.shift();
if (!current || seen.has(current) || typeof current !== "object") {
continue;
}
seen.add(current);
const candidate = current as {
code?: unknown;
message?: unknown;
meta?: { field_name?: unknown; modelName?: unknown };
cause?: unknown;
shape?: { message?: unknown; data?: { cause?: unknown } };
};
candidates.push(candidate);
if ("cause" in candidate) {
queue.push(candidate.cause);
}
if (candidate.shape?.data?.cause) {
queue.push(candidate.shape.data.cause);
}
}
return candidates;
}
export function rethrowNotificationReferenceError(error: unknown): never {
for (const candidate of getNotificationErrorCandidates(error)) {
const fieldName = typeof candidate.meta?.field_name === "string"
? candidate.meta.field_name.toLowerCase()
: "";
const modelName = typeof candidate.meta?.modelName === "string"
? candidate.meta.modelName.toLowerCase()
: "";
if (
typeof candidate.code === "string"
&& (candidate.code === "P2003" || candidate.code === "P2025")
&& fieldName.includes("sender")
) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Sender user not found",
cause: error,
});
}
if (
typeof candidate.code === "string"
&& (candidate.code === "P2003" || candidate.code === "P2025")
&& fieldName.includes("userid")
) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Broadcast recipient user not found",
cause: error,
});
}
if (
typeof candidate.code === "string"
&& (candidate.code === "P2003" || candidate.code === "P2025")
&& (modelName.includes("notificationbroadcast") || fieldName.includes("broadcast"))
) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Notification broadcast not found",
cause: error,
});
}
}
throw error;
}
export const categoryEnum = z.enum(["NOTIFICATION", "REMINDER", "TASK", "APPROVAL"]);
export const priorityEnum = z.enum(["LOW", "NORMAL", "HIGH", "URGENT"]);
export const taskStatusEnum = z.enum(["OPEN", "IN_PROGRESS", "DONE", "DISMISSED"]);
export const channelEnum = z.enum(["in_app", "email", "both"]);
export const recurrenceEnum = z.enum(["daily", "weekly", "monthly"]);
export const targetTypeEnum = z.enum(["user", "role", "project", "orgUnit", "all"]);
export const SCHEDULED_TASK_BROADCAST_UNSUPPORTED_MESSAGE =
"Scheduled broadcasts with task metadata are not supported yet.";
export const NotificationListInputSchema = z.object({
unreadOnly: z.boolean().optional(),
category: categoryEnum.optional(),
taskStatus: taskStatusEnum.optional(),
priority: priorityEnum.optional(),
limit: z.number().min(1).max(100).default(50),
});
export const MarkNotificationReadInputSchema = z.object({
id: z.string().optional(),
});
export const CreateManagedNotificationInputSchema = z.object({
userId: z.string(),
type: z.string(),
title: z.string(),
body: z.string().optional(),
entityId: z.string().optional(),
entityType: z.string().optional(),
category: categoryEnum.optional(),
priority: priorityEnum.optional(),
link: z.string().optional(),
taskStatus: taskStatusEnum.optional(),
taskAction: z.string().optional(),
assigneeId: z.string().optional(),
dueDate: z.date().optional(),
channel: channelEnum.optional(),
senderId: z.string().optional(),
});
export const ListNotificationTasksInputSchema = z.object({
status: taskStatusEnum.optional(),
includeAssigned: z.boolean().default(true),
limit: z.number().min(1).max(100).default(20),
});
export const NotificationIdInputSchema = z.object({
id: z.string(),
});
export const UpdateNotificationTaskStatusInputSchema = z.object({
id: z.string(),
status: taskStatusEnum,
});
export const CreateReminderInputSchema = z.object({
title: z.string().min(1).max(200),
body: z.string().max(2000).optional(),
remindAt: z.date(),
recurrence: recurrenceEnum.optional(),
entityId: z.string().optional(),
entityType: z.string().optional(),
link: z.string().optional(),
});
export const UpdateReminderInputSchema = z.object({
id: z.string(),
title: z.string().min(1).max(200).optional(),
body: z.string().max(2000).optional(),
remindAt: z.date().optional(),
recurrence: recurrenceEnum.nullish(),
});
export const ListRemindersInputSchema = z.object({
limit: z.number().min(1).max(100).default(20),
});
export const CreateBroadcastInputSchema = z.object({
title: z.string().min(1).max(200),
body: z.string().max(2000).optional(),
link: z.string().optional(),
category: categoryEnum.default("NOTIFICATION"),
priority: priorityEnum.default("NORMAL"),
channel: channelEnum.default("in_app"),
targetType: targetTypeEnum,
targetValue: z.string().optional(),
scheduledAt: z.date().optional(),
taskAction: z.string().optional(),
dueDate: z.date().optional(),
});
export const ListBroadcastsInputSchema = z.object({
limit: z.number().min(1).max(50).default(20),
});
export const CreateTaskInputSchema = z.object({
userId: z.string(),
title: z.string().min(1).max(200),
body: z.string().max(2000).optional(),
priority: priorityEnum.default("NORMAL"),
dueDate: z.date().optional(),
taskAction: z.string().optional(),
entityId: z.string().optional(),
entityType: z.string().optional(),
link: z.string().optional(),
channel: channelEnum.default("in_app"),
});
export const AssignTaskInputSchema = z.object({
id: z.string(),
assigneeId: z.string(),
});
type NotificationListInput = z.infer<typeof NotificationListInputSchema>;
type MarkNotificationReadInput = z.infer<typeof MarkNotificationReadInputSchema>;
type CreateManagedNotificationInput = z.infer<typeof CreateManagedNotificationInputSchema>;
type ListNotificationTasksInput = z.infer<typeof ListNotificationTasksInputSchema>;
type NotificationIdInput = z.infer<typeof NotificationIdInputSchema>;
type UpdateNotificationTaskStatusInput = z.infer<typeof UpdateNotificationTaskStatusInputSchema>;
type CreateReminderInput = z.infer<typeof CreateReminderInputSchema>;
type UpdateReminderInput = z.infer<typeof UpdateReminderInputSchema>;
type ListRemindersInput = z.infer<typeof ListRemindersInputSchema>;
type CreateBroadcastInput = z.infer<typeof CreateBroadcastInputSchema>;
type ListBroadcastsInput = z.infer<typeof ListBroadcastsInputSchema>;
type CreateTaskInput = z.infer<typeof CreateTaskInputSchema>;
type AssignTaskInput = z.infer<typeof AssignTaskInputSchema>;
export async function listNotifications(
ctx: NotificationProcedureContext,
input: NotificationListInput,
) {
const userId = await resolveUserId(ctx);
return ctx.db.notification.findMany({
where: {
userId,
...(input.unreadOnly ? { readAt: null } : {}),
...(input.category !== undefined ? { category: input.category } : {}),
...(input.taskStatus !== undefined ? { taskStatus: input.taskStatus } : {}),
...(input.priority !== undefined ? { priority: input.priority } : {}),
},
orderBy: { createdAt: "desc" },
take: input.limit,
});
}
export async function countUnreadNotifications(ctx: NotificationProcedureContext) {
const userId = await resolveUserId(ctx);
return ctx.db.notification.count({
where: { userId, readAt: null },
});
}
export async function markNotificationsRead(
ctx: NotificationProcedureContext,
input: MarkNotificationReadInput,
) {
const userId = await resolveUserId(ctx);
const now = new Date();
if (input.id) {
await ctx.db.notification.update({
where: { id: input.id, userId },
data: { readAt: now },
});
return;
}
await ctx.db.notification.updateMany({
where: { userId, readAt: null },
data: { readAt: now },
});
}
export async function createManagedNotification(
ctx: NotificationProcedureContext,
input: CreateManagedNotificationInput,
) {
const currentUserId = ctx.dbUser.id;
const notificationId = await createNotification({
db: ctx.db,
userId: input.userId,
type: input.type,
title: input.title,
body: input.body,
entityId: input.entityId,
entityType: input.entityType,
category: input.category,
priority: input.priority,
link: input.link,
taskStatus: input.taskStatus,
taskAction: input.taskAction,
assigneeId: input.assigneeId,
dueDate: input.dueDate,
channel: input.channel,
senderId: input.senderId ?? currentUserId,
});
if (input.category === "TASK" || input.category === "APPROVAL") {
emitTaskAssigned(input.userId, notificationId);
}
const channel = input.channel ?? "in_app";
if (channel === "email" || channel === "both") {
void sendNotificationEmail(ctx.db, input.userId, input.title, input.body);
}
return ctx.db.notification.findUnique({ where: { id: notificationId } });
}
export async function listNotificationTasks(
ctx: NotificationProcedureContext,
input: ListNotificationTasksInput,
) {
const userId = await resolveUserId(ctx);
const userFilter = input.includeAssigned
? { OR: [{ userId }, { assigneeId: userId }] }
: { userId };
return ctx.db.notification.findMany({
where: {
...userFilter,
category: { in: ["TASK", "APPROVAL"] },
...(input.status !== undefined ? { taskStatus: input.status } : {}),
},
orderBy: [{ priority: "desc" }, { dueDate: "asc" }, { createdAt: "desc" }],
take: input.limit,
});
}
export async function getNotificationTaskCounts(ctx: NotificationProcedureContext) {
const userId = await resolveUserId(ctx);
const now = new Date();
const where = {
OR: [{ userId }, { assigneeId: userId }],
category: { in: ["TASK" as const, "APPROVAL" as const] },
};
const [grouped, overdue] = await Promise.all([
ctx.db.notification.groupBy({
by: ["taskStatus"],
where,
_count: true,
}),
ctx.db.notification.count({
where: {
...where,
taskStatus: { in: ["OPEN", "IN_PROGRESS"] },
dueDate: { lt: now },
},
}),
]);
const counts: Record<string, number> = {};
for (const group of grouped) {
if (group.taskStatus) {
counts[group.taskStatus] = group._count;
}
}
return {
open: counts.OPEN ?? 0,
inProgress: counts.IN_PROGRESS ?? 0,
done: counts.DONE ?? 0,
dismissed: counts.DISMISSED ?? 0,
overdue,
};
}
export async function getNotificationTaskDetail(
ctx: NotificationProcedureContext,
input: NotificationIdInput,
) {
const userId = await resolveUserId(ctx);
const task = await ctx.db.notification.findFirst({
where: {
id: input.id,
OR: [{ userId }, { assigneeId: userId }],
category: { in: ["TASK", "APPROVAL"] },
},
select: {
id: true,
title: true,
body: true,
type: true,
priority: true,
category: true,
taskStatus: true,
taskAction: true,
dueDate: true,
entityId: true,
entityType: true,
completedAt: true,
completedBy: true,
createdAt: true,
userId: true,
assigneeId: true,
sender: { select: { id: true, name: true, email: true } },
},
});
if (!task) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Task not found or you do not have permission",
});
}
return task;
}
export async function updateNotificationTaskStatus(
ctx: NotificationProcedureContext,
input: UpdateNotificationTaskStatusInput,
) {
const userId = await resolveUserId(ctx);
const existing = await ctx.db.notification.findFirst({
where: {
id: input.id,
OR: [{ userId }, { assigneeId: userId }],
},
});
if (!existing) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Task not found or you do not have permission",
});
}
const isCompleting = input.status === "DONE";
const updated = await ctx.db.notification.update({
where: { id: input.id },
data: {
taskStatus: input.status,
...(isCompleting ? { completedAt: new Date(), completedBy: userId } : {}),
},
});
if (isCompleting) {
emitTaskCompleted(existing.userId, updated.id);
if (existing.assigneeId && existing.assigneeId !== existing.userId) {
emitTaskCompleted(existing.assigneeId, updated.id);
}
} else {
emitTaskStatusChanged(existing.userId, updated.id);
if (existing.assigneeId && existing.assigneeId !== existing.userId) {
emitTaskStatusChanged(existing.assigneeId, updated.id);
}
}
return updated;
}
export async function executeNotificationTaskAction(
ctx: NotificationProcedureContext,
input: NotificationIdInput,
) {
const userId = await resolveUserId(ctx);
const task = await ctx.db.notification.findFirst({
where: {
id: input.id,
OR: [{ userId }, { assigneeId: userId }],
category: { in: ["TASK", "APPROVAL"] },
},
select: {
id: true,
userId: true,
assigneeId: true,
taskAction: true,
taskStatus: true,
},
});
if (!task) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Task not found or you do not have permission",
});
}
if (!task.taskAction) {
throw new TRPCError({
code: "PRECONDITION_FAILED",
message: "This task has no executable action",
});
}
if (task.taskStatus === "DONE") {
throw new TRPCError({
code: "PRECONDITION_FAILED",
message: "This task is already completed",
});
}
const parsed = parseTaskAction(task.taskAction);
if (!parsed) {
throw new TRPCError({
code: "BAD_REQUEST",
message: `Invalid taskAction format: ${task.taskAction}`,
});
}
const handler = getTaskAction(parsed.action);
if (!handler) {
throw new TRPCError({
code: "BAD_REQUEST",
message: `Unknown action: ${parsed.action}`,
});
}
const permissions = resolvePermissions(
ctx.dbUser.systemRole as import("@capakraken/shared").SystemRole,
ctx.dbUser.permissionOverrides as import("@capakraken/shared").PermissionOverrides | null,
ctx.roleDefaults ?? undefined,
);
if (handler.permission && !permissions.has(handler.permission as PermissionKey)) {
throw new TRPCError({
code: "FORBIDDEN",
message: `Permission denied: you need "${handler.permission}" to perform this action`,
});
}
const actionResult = await handler.execute(parsed.entityId, ctx.db, userId);
if (!actionResult.success) {
throw new TRPCError({
code: "BAD_REQUEST",
message: actionResult.message,
});
}
const completedTask = await ctx.db.notification.update({
where: { id: input.id },
data: {
taskStatus: "DONE",
completedAt: new Date(),
completedBy: userId,
},
});
emitTaskCompleted(task.userId, task.id);
if (task.assigneeId && task.assigneeId !== task.userId) {
emitTaskCompleted(task.assigneeId, task.id);
}
return {
task: completedTask,
actionResult,
};
}
export async function createReminder(
ctx: NotificationProcedureContext,
input: CreateReminderInput,
) {
const userId = await resolveUserId(ctx);
return ctx.db.notification.create({
data: {
userId,
type: "REMINDER",
category: "REMINDER",
title: input.title,
...(input.body !== undefined ? { body: input.body } : {}),
remindAt: input.remindAt,
nextRemindAt: input.remindAt,
...(input.recurrence !== undefined ? { recurrence: input.recurrence } : {}),
...(input.entityId !== undefined ? { entityId: input.entityId } : {}),
...(input.entityType !== undefined ? { entityType: input.entityType } : {}),
...(input.link !== undefined ? { link: input.link } : {}),
channel: "in_app",
},
});
}
export async function updateReminder(
ctx: NotificationProcedureContext,
input: UpdateReminderInput,
) {
const userId = await resolveUserId(ctx);
const existing = await ctx.db.notification.findFirst({
where: { id: input.id, userId, category: "REMINDER" },
});
if (!existing) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Reminder not found or you do not have permission",
});
}
return ctx.db.notification.update({
where: { id: input.id },
data: {
...(input.title !== undefined ? { title: input.title } : {}),
...(input.body !== undefined ? { body: input.body } : {}),
...(input.remindAt !== undefined
? { remindAt: input.remindAt, nextRemindAt: input.remindAt }
: {}),
...(input.recurrence !== undefined ? { recurrence: input.recurrence } : {}),
},
});
}
export async function deleteReminder(
ctx: NotificationProcedureContext,
input: NotificationIdInput,
) {
const userId = await resolveUserId(ctx);
const existing = await ctx.db.notification.findFirst({
where: { id: input.id, userId, category: "REMINDER" },
});
if (!existing) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Reminder not found or you do not have permission",
});
}
await ctx.db.notification.delete({ where: { id: input.id } });
}
export async function listReminders(
ctx: NotificationProcedureContext,
input: ListRemindersInput,
) {
const userId = await resolveUserId(ctx);
return ctx.db.notification.findMany({
where: { userId, category: "REMINDER" },
orderBy: { nextRemindAt: "asc" },
take: input.limit,
});
}
export async function createBroadcast(
ctx: NotificationProcedureContext,
input: CreateBroadcastInput,
) {
const senderId = ctx.dbUser.id;
const isScheduledFutureBroadcast = Boolean(input.scheduledAt && input.scheduledAt > new Date());
const hasTaskLikeMetadata =
input.category === "TASK"
|| input.category === "APPROVAL"
|| input.taskAction !== undefined
|| input.dueDate !== undefined;
if (isScheduledFutureBroadcast && hasTaskLikeMetadata) {
throw new TRPCError({
code: "BAD_REQUEST",
message: SCHEDULED_TASK_BROADCAST_UNSUPPORTED_MESSAGE,
});
}
const recipientIds = await resolveRecipients(
input.targetType,
input.targetValue,
ctx.db,
senderId,
);
if (recipientIds.length === 0) {
throw new TRPCError({
code: "BAD_REQUEST",
message: "No recipients matched the broadcast target.",
});
}
if (isScheduledFutureBroadcast) {
return ctx.db.notificationBroadcast.create({
data: {
senderId,
title: input.title,
...(input.body !== undefined ? { body: input.body } : {}),
...(input.link !== undefined ? { link: input.link } : {}),
category: input.category,
priority: input.priority,
channel: input.channel,
targetType: input.targetType,
...(input.targetValue !== undefined ? { targetValue: input.targetValue } : {}),
...(input.scheduledAt !== undefined ? { scheduledAt: input.scheduledAt } : {}),
recipientCount: recipientIds.length,
},
});
}
const isTask = input.category === "TASK" || input.category === "APPROVAL";
const sentAt = new Date();
const runImmediateBroadcast = async (db: typeof ctx.db) => {
const broadcast = await db.notificationBroadcast.create({
data: {
senderId,
title: input.title,
...(input.body !== undefined ? { body: input.body } : {}),
...(input.link !== undefined ? { link: input.link } : {}),
category: input.category,
priority: input.priority,
channel: input.channel,
targetType: input.targetType,
...(input.targetValue !== undefined ? { targetValue: input.targetValue } : {}),
...(input.scheduledAt !== undefined ? { scheduledAt: input.scheduledAt } : {}),
},
});
const createdNotificationIds: Array<{ id: string; userId: string }> = [];
for (const recipientUserId of recipientIds) {
const notificationId = await createNotification({
db,
userId: recipientUserId,
type: `BROADCAST_${input.category}`,
title: input.title,
body: input.body,
link: input.link,
category: input.category,
priority: input.priority,
channel: input.channel,
sourceId: broadcast.id,
senderId,
taskStatus: isTask ? "OPEN" : undefined,
taskAction: input.taskAction,
dueDate: input.dueDate,
emit: false,
});
createdNotificationIds.push({ id: notificationId, userId: recipientUserId });
}
const updatedBroadcast = await db.notificationBroadcast.update({
where: { id: broadcast.id },
data: {
sentAt,
recipientCount: createdNotificationIds.length,
},
});
return { broadcast: updatedBroadcast, notificationIds: createdNotificationIds };
};
let persistedBroadcast: Awaited<ReturnType<typeof runImmediateBroadcast>>["broadcast"];
let notificationIds: Array<{ id: string; userId: string }> = [];
try {
const transactionResult = typeof ctx.db.$transaction === "function"
? await ctx.db.$transaction((tx) => runImmediateBroadcast(tx as typeof ctx.db))
: await runImmediateBroadcast(ctx.db);
persistedBroadcast = transactionResult.broadcast;
notificationIds = transactionResult.notificationIds;
} catch (error) {
rethrowNotificationReferenceError(error);
}
for (const notification of notificationIds) {
emitNotificationCreated(notification.userId, notification.id);
if (isTask) {
emitTaskAssigned(notification.userId, notification.id);
}
}
if (input.channel === "email" || input.channel === "both") {
for (const notification of notificationIds) {
void sendNotificationEmail(ctx.db, notification.userId, input.title, input.body);
}
}
return persistedBroadcast;
}
export async function listBroadcasts(
ctx: NotificationProcedureContext,
input: ListBroadcastsInput,
) {
return ctx.db.notificationBroadcast.findMany({
orderBy: { createdAt: "desc" },
take: input.limit,
include: {
sender: { select: { id: true, name: true, email: true } },
},
});
}
export async function getBroadcastById(
ctx: NotificationProcedureContext,
input: NotificationIdInput,
) {
return findUniqueOrThrow(
ctx.db.notificationBroadcast.findUnique({
where: { id: input.id },
include: {
sender: { select: { id: true, name: true, email: true } },
},
}),
"Broadcast",
);
}
export async function createTask(
ctx: NotificationProcedureContext,
input: CreateTaskInput,
) {
const senderId = ctx.dbUser.id;
const notificationId = await createNotification({
db: ctx.db,
userId: input.userId,
type: "TASK_CREATED",
category: "TASK",
taskStatus: "OPEN",
title: input.title,
priority: input.priority,
senderId,
channel: input.channel,
body: input.body,
dueDate: input.dueDate,
taskAction: input.taskAction,
entityId: input.entityId,
entityType: input.entityType,
link: input.link,
});
emitTaskAssigned(input.userId, notificationId);
if (input.channel === "email" || input.channel === "both") {
void sendNotificationEmail(ctx.db, input.userId, input.title, input.body);
}
return ctx.db.notification.findUnique({ where: { id: notificationId } });
}
export async function assignTask(
ctx: NotificationProcedureContext,
input: AssignTaskInput,
) {
const existing = await findUniqueOrThrow(
ctx.db.notification.findUnique({ where: { id: input.id } }),
"Task",
);
if (existing.category !== "TASK" && existing.category !== "APPROVAL") {
throw new TRPCError({
code: "BAD_REQUEST",
message: "Only tasks and approvals can be assigned",
});
}
const updated = await ctx.db.notification.update({
where: { id: input.id },
data: { assigneeId: input.assigneeId },
});
emitTaskAssigned(input.assigneeId, updated.id);
return updated;
}
export async function deleteNotification(
ctx: NotificationProcedureContext,
input: NotificationIdInput,
) {
const userId = await resolveUserId(ctx);
const existing = await ctx.db.notification.findFirst({
where: { id: input.id, userId },
});
if (!existing) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Notification not found",
});
}
if (
(existing.category === "TASK" || existing.category === "APPROVAL")
&& existing.senderId
&& existing.senderId !== userId
) {
throw new TRPCError({
code: "FORBIDDEN",
message: "Cannot delete tasks created by others",
});
}
await ctx.db.notification.delete({ where: { id: input.id } });
}
export * from "./notification-procedure-base.js";
export * from "./notification-task-procedure-support.js";
export * from "./notification-reminder-procedure-support.js";
export * from "./notification-broadcast-procedure-support.js";
@@ -0,0 +1,93 @@
import { TRPCError } from "@trpc/server";
import { z } from "zod";
import {
CreateReminderInputSchema,
ListRemindersInputSchema,
type NotificationProcedureContext,
NotificationIdInputSchema,
resolveUserId,
UpdateReminderInputSchema,
} from "./notification-procedure-base.js";
export async function createReminder(
ctx: NotificationProcedureContext,
input: z.infer<typeof CreateReminderInputSchema>,
) {
const userId = await resolveUserId(ctx);
return ctx.db.notification.create({
data: {
userId,
type: "REMINDER",
category: "REMINDER",
title: input.title,
...(input.body !== undefined ? { body: input.body } : {}),
remindAt: input.remindAt,
nextRemindAt: input.remindAt,
...(input.recurrence !== undefined ? { recurrence: input.recurrence } : {}),
...(input.entityId !== undefined ? { entityId: input.entityId } : {}),
...(input.entityType !== undefined ? { entityType: input.entityType } : {}),
...(input.link !== undefined ? { link: input.link } : {}),
channel: "in_app",
},
});
}
export async function updateReminder(
ctx: NotificationProcedureContext,
input: z.infer<typeof UpdateReminderInputSchema>,
) {
const userId = await resolveUserId(ctx);
const existing = await ctx.db.notification.findFirst({
where: { id: input.id, userId, category: "REMINDER" },
});
if (!existing) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Reminder not found or you do not have permission",
});
}
return ctx.db.notification.update({
where: { id: input.id },
data: {
...(input.title !== undefined ? { title: input.title } : {}),
...(input.body !== undefined ? { body: input.body } : {}),
...(input.remindAt !== undefined
? { remindAt: input.remindAt, nextRemindAt: input.remindAt }
: {}),
...(input.recurrence !== undefined ? { recurrence: input.recurrence } : {}),
},
});
}
export async function deleteReminder(
ctx: NotificationProcedureContext,
input: z.infer<typeof NotificationIdInputSchema>,
) {
const userId = await resolveUserId(ctx);
const existing = await ctx.db.notification.findFirst({
where: { id: input.id, userId, category: "REMINDER" },
});
if (!existing) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Reminder not found or you do not have permission",
});
}
await ctx.db.notification.delete({ where: { id: input.id } });
}
export async function listReminders(
ctx: NotificationProcedureContext,
input: z.infer<typeof ListRemindersInputSchema>,
) {
const userId = await resolveUserId(ctx);
return ctx.db.notification.findMany({
where: { userId, category: "REMINDER" },
orderBy: { nextRemindAt: "asc" },
take: input.limit,
});
}
@@ -0,0 +1,319 @@
import { PermissionKey, parseTaskAction, resolvePermissions } from "@capakraken/shared";
import { TRPCError } from "@trpc/server";
import { z } from "zod";
import { findUniqueOrThrow } from "../db/helpers.js";
import { createNotification } from "../lib/create-notification.js";
import { getTaskAction } from "../lib/task-actions.js";
import {
emitTaskAssigned,
emitTaskCompleted,
emitTaskStatusChanged,
} from "../sse/event-bus.js";
import {
AssignTaskInputSchema,
CreateTaskInputSchema,
ListNotificationTasksInputSchema,
NotificationIdInputSchema,
type NotificationProcedureContext,
requireNotificationDbUser,
resolveUserId,
sendNotificationEmail,
UpdateNotificationTaskStatusInputSchema,
} from "./notification-procedure-base.js";
export async function listNotificationTasks(
ctx: NotificationProcedureContext,
input: z.infer<typeof ListNotificationTasksInputSchema>,
) {
const userId = await resolveUserId(ctx);
const userFilter = input.includeAssigned
? { OR: [{ userId }, { assigneeId: userId }] }
: { userId };
return ctx.db.notification.findMany({
where: {
...userFilter,
category: { in: ["TASK", "APPROVAL"] },
...(input.status !== undefined ? { taskStatus: input.status } : {}),
},
orderBy: [{ priority: "desc" }, { dueDate: "asc" }, { createdAt: "desc" }],
take: input.limit,
});
}
export async function getNotificationTaskCounts(ctx: NotificationProcedureContext) {
const userId = await resolveUserId(ctx);
const now = new Date();
const where = {
OR: [{ userId }, { assigneeId: userId }],
category: { in: ["TASK" as const, "APPROVAL" as const] },
};
const [grouped, overdue] = await Promise.all([
ctx.db.notification.groupBy({
by: ["taskStatus"],
where,
_count: true,
}),
ctx.db.notification.count({
where: {
...where,
taskStatus: { in: ["OPEN", "IN_PROGRESS"] },
dueDate: { lt: now },
},
}),
]);
const counts: Record<string, number> = {};
for (const group of grouped) {
if (group.taskStatus) {
counts[group.taskStatus] = group._count;
}
}
return {
open: counts.OPEN ?? 0,
inProgress: counts.IN_PROGRESS ?? 0,
done: counts.DONE ?? 0,
dismissed: counts.DISMISSED ?? 0,
overdue,
};
}
export async function getNotificationTaskDetail(
ctx: NotificationProcedureContext,
input: z.infer<typeof NotificationIdInputSchema>,
) {
const userId = await resolveUserId(ctx);
const task = await ctx.db.notification.findFirst({
where: {
id: input.id,
OR: [{ userId }, { assigneeId: userId }],
category: { in: ["TASK", "APPROVAL"] },
},
select: {
id: true,
title: true,
body: true,
type: true,
priority: true,
category: true,
taskStatus: true,
taskAction: true,
dueDate: true,
entityId: true,
entityType: true,
completedAt: true,
completedBy: true,
createdAt: true,
userId: true,
assigneeId: true,
sender: { select: { id: true, name: true, email: true } },
},
});
if (!task) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Task not found or you do not have permission",
});
}
return task;
}
export async function updateNotificationTaskStatus(
ctx: NotificationProcedureContext,
input: z.infer<typeof UpdateNotificationTaskStatusInputSchema>,
) {
const userId = await resolveUserId(ctx);
const existing = await ctx.db.notification.findFirst({
where: {
id: input.id,
OR: [{ userId }, { assigneeId: userId }],
},
});
if (!existing) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Task not found or you do not have permission",
});
}
const isCompleting = input.status === "DONE";
const updated = await ctx.db.notification.update({
where: { id: input.id },
data: {
taskStatus: input.status,
...(isCompleting ? { completedAt: new Date(), completedBy: userId } : {}),
},
});
if (isCompleting) {
emitTaskCompleted(existing.userId, updated.id);
if (existing.assigneeId && existing.assigneeId !== existing.userId) {
emitTaskCompleted(existing.assigneeId, updated.id);
}
} else {
emitTaskStatusChanged(existing.userId, updated.id);
if (existing.assigneeId && existing.assigneeId !== existing.userId) {
emitTaskStatusChanged(existing.assigneeId, updated.id);
}
}
return updated;
}
export async function executeNotificationTaskAction(
ctx: NotificationProcedureContext,
input: z.infer<typeof NotificationIdInputSchema>,
) {
const userId = await resolveUserId(ctx);
const task = await ctx.db.notification.findFirst({
where: {
id: input.id,
OR: [{ userId }, { assigneeId: userId }],
category: { in: ["TASK", "APPROVAL"] },
},
select: {
id: true,
userId: true,
assigneeId: true,
taskAction: true,
taskStatus: true,
},
});
if (!task) {
throw new TRPCError({
code: "NOT_FOUND",
message: "Task not found or you do not have permission",
});
}
if (!task.taskAction) {
throw new TRPCError({
code: "PRECONDITION_FAILED",
message: "This task has no executable action",
});
}
if (task.taskStatus === "DONE") {
throw new TRPCError({
code: "PRECONDITION_FAILED",
message: "This task is already completed",
});
}
const parsed = parseTaskAction(task.taskAction);
if (!parsed) {
throw new TRPCError({
code: "BAD_REQUEST",
message: `Invalid taskAction format: ${task.taskAction}`,
});
}
const handler = getTaskAction(parsed.action);
if (!handler) {
throw new TRPCError({
code: "BAD_REQUEST",
message: `Unknown action: ${parsed.action}`,
});
}
const currentUser = requireNotificationDbUser(ctx);
const permissions = resolvePermissions(
currentUser.systemRole as import("@capakraken/shared").SystemRole,
currentUser.permissionOverrides as import("@capakraken/shared").PermissionOverrides | null,
ctx.roleDefaults ?? undefined,
);
if (handler.permission && !permissions.has(handler.permission as PermissionKey)) {
throw new TRPCError({
code: "FORBIDDEN",
message: `Permission denied: you need "${handler.permission}" to perform this action`,
});
}
const actionResult = await handler.execute(parsed.entityId, ctx.db, userId);
if (!actionResult.success) {
throw new TRPCError({
code: "BAD_REQUEST",
message: actionResult.message,
});
}
const completedTask = await ctx.db.notification.update({
where: { id: input.id },
data: {
taskStatus: "DONE",
completedAt: new Date(),
completedBy: userId,
},
});
emitTaskCompleted(task.userId, task.id);
if (task.assigneeId && task.assigneeId !== task.userId) {
emitTaskCompleted(task.assigneeId, task.id);
}
return {
task: completedTask,
actionResult,
};
}
export async function createTask(
ctx: NotificationProcedureContext,
input: z.infer<typeof CreateTaskInputSchema>,
) {
const senderId = requireNotificationDbUser(ctx).id;
const notificationId = await createNotification({
db: ctx.db,
userId: input.userId,
type: "TASK_CREATED",
category: "TASK",
taskStatus: "OPEN",
title: input.title,
priority: input.priority,
senderId,
channel: input.channel,
body: input.body,
dueDate: input.dueDate,
taskAction: input.taskAction,
entityId: input.entityId,
entityType: input.entityType,
link: input.link,
});
emitTaskAssigned(input.userId, notificationId);
if (input.channel === "email" || input.channel === "both") {
void sendNotificationEmail(ctx.db, input.userId, input.title, input.body);
}
return ctx.db.notification.findUnique({ where: { id: notificationId } });
}
export async function assignTask(
ctx: NotificationProcedureContext,
input: z.infer<typeof AssignTaskInputSchema>,
) {
const existing = await findUniqueOrThrow(
ctx.db.notification.findUnique({ where: { id: input.id } }),
"Task",
);
if (existing.category !== "TASK" && existing.category !== "APPROVAL") {
throw new TRPCError({
code: "BAD_REQUEST",
message: "Only tasks and approvals can be assigned",
});
}
const updated = await ctx.db.notification.update({
where: { id: input.id },
data: { assigneeId: input.assigneeId },
});
emitTaskAssigned(input.assigneeId, updated.id);
return updated;
}