@allbluecn/web-app 0.4.16 → 0.4.18

package/src/server/serverFns/notifications.ts

@@ -0,0 +1,155 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+import { createServerFn } from '@tanstack/react-start'
+import { getRequest } from '@tanstack/react-start/server'
+import { z } from 'zod'
+import { prisma } from '@allbluecn/database'
+import { requireAuth } from '@/server/auth-helpers'
+import {
+  NOTIFICATION_CATEGORIES,
+  type NotificationCategory,
+  type CategoryPreferences,
+  type NotificationItem,
+  type NotificationListResult,
+} from '@allbluecn/shared'
+import {
+  getOrCreatePreferences,
+  DEFAULT_CATEGORIES,
+} from '@/server/notifications/notification-service'
+
+const PAGE_SIZE = 20
+
+const listSchema = z.object({
+  category: z.enum(NOTIFICATION_CATEGORIES).optional(),
+  cursor: z.string().optional(),
+})
+
+export const listNotificationsFn = createServerFn({ method: 'GET' })
+  .validator(listSchema)
+  .handler(async ({ data }): Promise<NotificationListResult> => {
+    const request = getRequest()
+    const session = await requireAuth(request)
+    const userId = session.user!.id!
+
+    const items = await prisma.notification.findMany({
+      where: {
+        userId,
+        ...(data.category ? { category: data.category } : {}),
+      },
+      orderBy: { createdAt: 'desc' },
+      take: PAGE_SIZE + 1,
+      ...(data.cursor ? { cursor: { id: data.cursor }, skip: 1 } : {}),
+    })
+
+    const hasMore = items.length > PAGE_SIZE
+    const page = (hasMore ? items.slice(0, PAGE_SIZE) : items).map(
+      (n): NotificationItem => ({
+        id: n.id,
+        category: n.category as NotificationCategory,
+        type: n.type,
+        title: n.title,
+        body: n.body,
+        link: n.link,
+        readAt: n.readAt ? n.readAt.toISOString() : null,
+        createdAt: n.createdAt.toISOString(),
+      }),
+    )
+    return { items: page, total: page.length, hasMore }
+  })
+
+const markReadSchema = z.object({ id: z.string().min(1) })
+
+export const markNotificationReadFn = createServerFn({ method: 'POST' })
+  .validator(markReadSchema)
+  .handler(async ({ data }) => {
+    const request = getRequest()
+    const session = await requireAuth(request)
+    const userId = session.user!.id!
+    await prisma.notification.updateMany({
+      where: { id: data.id, userId, readAt: null },
+      data: { readAt: new Date() },
+    })
+    return { success: true }
+  })
+
+export const markAllNotificationsReadFn = createServerFn({ method: 'POST' })
+  .validator(z.object({}).optional())
+  .handler(async () => {
+    const request = getRequest()
+    const session = await requireAuth(request)
+    const userId = session.user!.id!
+    const result = await prisma.notification.updateMany({
+      where: { userId, readAt: null },
+      data: { readAt: new Date() },
+    })
+    return { success: true, count: result.count }
+  })
+
+const deleteSchema = z.object({ id: z.string().min(1) })
+
+export const deleteNotificationFn = createServerFn({ method: 'POST' })
+  .validator(deleteSchema)
+  .handler(async ({ data }) => {
+    const request = getRequest()
+    const session = await requireAuth(request)
+    const userId = session.user!.id!
+    await prisma.notification.deleteMany({ where: { id: data.id, userId } })
+    return { success: true }
+  })
+
+export const getUnreadCountFn = createServerFn({ method: 'GET' })
+  .validator(z.object({}).optional())
+  .handler(async () => {
+    const request = getRequest()
+    const session = await requireAuth(request)
+    const userId = session.user!.id!
+    const count = await prisma.notification.count({
+      where: { userId, readAt: null },
+    })
+    return { count }
+  })
+
+export const getNotificationPreferencesFn = createServerFn({ method: 'GET' })
+  .validator(z.object({}).optional())
+  .handler(async () => {
+    const request = getRequest()
+    const session = await requireAuth(request)
+    return getOrCreatePreferences(session.user!.id!)
+  })
+
+const preferencesSchema = z.object({
+  categories: z.record(z.string(), z.boolean()),
+  dndEnabled: z.boolean(),
+  dndStart: z.string().regex(/^\d{2}:\d{2}$/),
+  dndEnd: z.string().regex(/^\d{2}:\d{2}$/),
+})
+
+export const updateNotificationPreferencesFn = createServerFn({
+  method: 'POST',
+})
+  .validator(preferencesSchema)
+  .handler(async ({ data }) => {
+    const request = getRequest()
+    const session = await requireAuth(request)
+    const userId = session.user!.id!
+    const categories = { ...DEFAULT_CATEGORIES } as CategoryPreferences
+    for (const c of NOTIFICATION_CATEGORIES) {
+      if (typeof data.categories[c] === 'boolean') categories[c] = data.categories[c]
+    }
+    await prisma.notificationPreference.upsert({
+      where: { userId },
+      create: {
+        userId,
+        categories,
+        dndEnabled: data.dndEnabled,
+        dndStart: data.dndStart,
+        dndEnd: data.dndEnd,
+      },
+      update: {
+        categories,
+        dndEnabled: data.dndEnabled,
+        dndStart: data.dndStart,
+        dndEnd: data.dndEnd,
+      },
+    })
+    return { success: true }
+  })

package/src/server/serverFns/story/__tests__/story-groups.test.ts

@@ -50,12 +50,14 @@ const mocks = vi.hoisted(() => ({
   groupBy: vi.fn(),
   storyLabelGroupBy: vi.fn(),
   currentSession: vi.fn(),
+  projectFindFirst: vi.fn(),
 }));
 
 vi.mock("@allbluecn/database", () => ({
   prisma: {
     story: { findMany: mocks.findMany, count: mocks.count, groupBy: mocks.groupBy },
     storyLabel: { groupBy: mocks.storyLabelGroupBy },
+    project: { findFirst: mocks.projectFindFirst },
   },
   Prisma: {},
 }));
@@ -70,6 +72,7 @@ describe("listStoriesInGroupFn", () => {
   beforeEach(() => {
     vi.clearAllMocks();
     mocks.currentSession.mockResolvedValue({ user: { id: "u1" } });
+    mocks.projectFindFirst.mockResolvedValue({ id: "proj-independent-1" });
   });
 
   it("构建含 group 过滤的 where 并返回 total/hasMore", async () => {
@@ -170,6 +173,7 @@ describe("listStoryGroupsFn", () => {
   beforeEach(() => {
     vi.clearAllMocks();
     mocks.currentSession.mockResolvedValue({ user: { id: "u1" } });
+    mocks.projectFindFirst.mockResolvedValue({ id: "proj-independent-1" });
   });
 
   it("按 groupBy 聚合返回组 key 列表", async () => {

package/src/server/serverFns/story/story-comments.ts

@@ -22,6 +22,7 @@ import { prisma } from "@allbluecn/database";
 import { requireAuth } from "@/server/auth-helpers";
 import type { CommentWithReplies, ReactionGroup } from "@/components/story/types";
 import { USER_BRIEF_SELECT, castUserBrief, aggregateReactions } from "./types";
+import { createNotification } from '@/server/notifications/notification-service'
 
 export const listCommentsFn = createServerFn({ method: "GET" })
   .validator(z.object({ storyId: z.string() }))
@@ -101,6 +102,47 @@ export const createCommentFn = createServerFn({ method: "POST" })
       },
     });
 
+    const commenter = await prisma.user.findUnique({
+      where: { id: userId },
+      select: { username: true },
+    })
+    const story = await prisma.story.findUnique({
+      where: { id: data.storyId },
+      select: { createdById: true },
+    })
+    if (story?.createdById && story.createdById !== userId) {
+      void createNotification({
+        userId: story.createdById,
+        category: 'collaboration',
+        type: 'collaboration.comment',
+        title: `${commenter?.username ?? '有人'} 评论了你的 Story`,
+        link: `/stories/${data.storyId}`,
+      })
+    }
+
+    const mentionMatches = data.content.match(/@([a-zA-Z0-9_\u4e00-\u9fa5]+)/g) ?? []
+    if (mentionMatches.length > 0) {
+      const mentionedNames = [...new Set(mentionMatches.map((m) => m.slice(1)))].filter(
+        (name) => name.toLowerCase() !== commenter?.username?.toLowerCase(),
+      )
+      if (mentionedNames.length > 0) {
+        const mentionedUsers = await prisma.user.findMany({
+          where: { username: { in: mentionedNames } },
+          select: { id: true },
+        })
+        for (const u of mentionedUsers) {
+          if (u.id === userId) continue
+          void createNotification({
+            userId: u.id,
+            category: 'collaboration',
+            type: 'collaboration.mention',
+            title: `${commenter?.username ?? '有人'} 在评论中提到了你`,
+            link: `/stories/${data.storyId}`,
+          })
+        }
+      }
+    }
+
     await prisma.storyActivity.create({
       data: {
         storyId: data.storyId,

package/src/server/serverFns/story/story-crud/labels.ts

@@ -27,12 +27,52 @@ const labelCreateSchema = z.object({
   color: z.string().regex(/^#[0-9a-fA-F]{6}$/).default("#6b7280"),
 });
 
-export const listUsersBriefFn = createServerFn({ method: "GET" }).handler(
-  async () => {
+export const listUsersBriefFn = createServerFn({ method: "GET" })
+  .validator((data?: { projectId?: string }) => data)
+  .handler(async ({ data }) => {
     const request = getRequest();
-    await requireAuth(request);
+    const session = await requireAuth(request);
+    const currentUserId = session.user!.id!;
+
+    if (data?.projectId) {
+      // 项目级别：只返回项目成员
+      const project = await prisma.project.findUnique({
+        where: { id: data.projectId },
+        select: { ownerId: true, workspaceId: true },
+      });
+      if (!project) return [];
+
+      const members = await prisma.projectMember.findMany({
+        where: { projectId: data.projectId },
+        select: { userId: true },
+      });
+      const memberIds = new Set(members.map((m) => m.userId));
+      memberIds.add(project.ownerId);
+
+      const users = await prisma.user.findMany({
+        where: { id: { in: Array.from(memberIds) }, status: "ACTIVE" },
+        orderBy: { username: "asc" },
+        select: USER_BRIEF_SELECT,
+      });
+      return users.map((u) => ({
+        ...u,
+        avatarConfig: (u.avatarConfig as import("@allbluecn/shared").AvatarConfig | null) ?? null,
+      }));
+    }
+
+    // 团队级别：只返回当前用户团队（team_memberships.ownerId = 当前用户）的成员
+    const memberships = await prisma.teamMembership.findMany({
+      where: { ownerId: currentUserId },
+      select: { memberId: true },
+    });
+    if (memberships.length === 0) {
+      return [];
+    }
+    const memberIds = new Set(memberships.map((m) => m.memberId));
+    memberIds.add(currentUserId); // 包含自己
+
     const users = await prisma.user.findMany({
-      where: { status: "ACTIVE" },
+      where: { id: { in: Array.from(memberIds) }, status: "ACTIVE" },
       orderBy: { username: "asc" },
       select: USER_BRIEF_SELECT,
     });
@@ -40,8 +80,7 @@ export const listUsersBriefFn = createServerFn({ method: "GET" }).handler(
       ...u,
       avatarConfig: (u.avatarConfig as import("@allbluecn/shared").AvatarConfig | null) ?? null,
     }));
-  },
-);
+  });
 
 export const listLabelsFn = createServerFn({ method: "GET" }).handler(
   async () => {

package/src/server/serverFns/story/story-crud.ts

@@ -56,9 +56,15 @@ export const listStoriesFn = createServerFn({ method: "GET" })
   .validator(listSchema)
   .handler(async ({ data }) => {
     const request = getRequest();
-    await requireAuth(request);
+    const session = await requireAuth(request);
+    const userId = session.user!.id!;
+
+    let projectId = data.projectId;
+    if (!projectId) {
+      projectId = await ensureBuiltinIndependentProject(userId);
+    }
 
-    const where = buildListWhere(data);
+    const where = buildListWhere({ ...data, projectId });
 
     const [items, total] = await Promise.all([
       prisma.story.findMany({

package/src/server/serverFns/story/story-groups.ts

@@ -20,6 +20,7 @@ import { getRequest } from "@tanstack/react-start/server";
 import { z } from "zod";
 import { prisma, Prisma } from "@allbluecn/database";
 import { requireAuth } from "@/server/auth-helpers";
+import { ensureBuiltinIndependentProject } from "@/server/builtin-project";
 import { GROUP_KEYS, ORDER_KEYS, ORDER_DIRECTIONS, type GroupKey, type OrderKey } from "@/components/story/display/types";
 import { STORY_STATES, STORY_PRIORITIES } from "@/components/story/constants";
 import { STORY_LIST_SELECT_LIGHT, castStoryUsersLight, type StoryListRowLight } from "./types";
@@ -57,9 +58,15 @@ export const listStoriesInGroupFn = createServerFn({ method: "GET" })
   .validator(schema)
   .handler(async ({ data }) => {
     const request = getRequest();
-    await requireAuth(request);
+    const session = await requireAuth(request);
+    const userId = session.user!.id!;
 
-    const base = buildListWhere(data);
+    let projectId = data.projectId;
+    if (!projectId) {
+      projectId = await ensureBuiltinIndependentProject(userId);
+    }
+
+    const base = buildListWhere({ ...data, projectId });
     const baseAnd = Array.isArray(base.AND) ? base.AND : base.AND ? [base.AND] : [];
     const extras: Prisma.StoryWhereInput[] = [];
     const groupWhere = buildGroupFilter(data.groupBy, data.groupKey);
@@ -114,9 +121,15 @@ export const listAllGroupsItemsFn = createServerFn({ method: "GET" })
   .validator(allGroupsItemsSchema)
   .handler(async ({ data }) => {
     const request = getRequest();
-    await requireAuth(request);
+    const session = await requireAuth(request);
+    const userId = session.user!.id!;
+
+    let projectId = data.projectId;
+    if (!projectId) {
+      projectId = await ensureBuiltinIndependentProject(userId);
+    }
 
-    const base = buildListWhere(data);
+    const base = buildListWhere({ ...data, projectId });
     const field = ORDER_FIELD[data.orderBy];
     const dir: "asc" | "desc" = data.orderDir === "asc" ? "asc" : "desc";
     const orderBy: Prisma.StoryOrderByWithRelationInput = field
@@ -207,10 +220,16 @@ export const listStoryGroupsFn = createServerFn({ method: "GET" })
   .validator(listGroupsSchema)
   .handler(async ({ data }) => {
     const request = getRequest();
-    await requireAuth(request);
-    if (data.groupBy === "none") return { groups: [] };
+    const session = await requireAuth(request);
+    const userId = session.user!.id!;
+
+    let projectId = data.projectId;
+    if (!projectId) {
+      projectId = await ensureBuiltinIndependentProject(userId);
+    }
 
-    const where = buildListWhere(data);
+    const where = buildListWhere({ ...data, projectId });
+    if (data.groupBy === "none") return { groups: [] };
     if (data.parentGroupBy !== "none" && data.parentGroupKey) {
       const parentWhere = buildGroupFilter(data.parentGroupBy, data.parentGroupKey);
       if (Object.keys(parentWhere).length) {

package/src/server/serverFns/story/story-search.ts

@@ -20,16 +20,20 @@ import { getRequest } from "@tanstack/react-start/server";
 import { z } from "zod";
 import { prisma } from "@allbluecn/database";
 import { requireAuth } from "@/server/auth-helpers";
+import { ensureBuiltinIndependentProject } from "@/server/builtin-project";
 
 export const searchStoriesFn = createServerFn({ method: "GET" })
   .validator(z.object({ query: z.string(), excludeIds: z.array(z.string()).optional() }))
   .handler(async ({ data }) => {
     const request = getRequest();
     const session = await requireAuth(request);
+    const userId = session.user!.id!;
+    const projectId = await ensureBuiltinIndependentProject(userId);
 
     const stories = await prisma.story.findMany({
       where: {
         id: data.excludeIds?.length ? { notIn: data.excludeIds } : undefined,
+        projectId,
         OR: [
           { name: { contains: data.query, mode: "insensitive" } },
           { identifier: { contains: data.query, mode: "insensitive" } },

package/src/server/serverFns/story-display-options.ts

@@ -22,6 +22,7 @@ import { z } from "zod"
 import { prisma, type Prisma } from "@allbluecn/database"
 import { requireAuth } from "@/server/auth-helpers"
 import { requireFeature } from "@/server/entitlements"
+import { isProEdition } from "@/server/pro-services"
 import {
   DEFAULT_DISPLAY_OPTIONS,
   GROUP_KEYS,
@@ -123,11 +124,16 @@ export const getStoryDisplayOptionsFn = createServerFn({ method: "GET" }).handle
     const options = parseStoredOptions(user?.storyDisplayOptions)
 
     // 订阅过期/降级：已持久化的 500/1000 无权限时降级为 200
+    // OSS 环境直接 cap 到 200；Pro 环境检查 feature
     if (options.pageSize > 200) {
-      try {
-        await requireFeature(request, "largePageSize")
-      } catch {
+      if (!isProEdition()) {
         options.pageSize = 200
+      } else {
+        try {
+          await requireFeature(request, "largePageSize")
+        } catch {
+          options.pageSize = 200
+        }
       }
     }
     return options
@@ -143,8 +149,13 @@ export const updateStoryDisplayOptionsFn = createServerFn({ method: "POST" })
     const userId = session.user!.id!
 
     // 服务端强制校验：pageSize > 200（500/1000）为 Pro 专属，前端锁定只是 UX
+    // OSS 环境直接 cap 到 200；Pro 环境走 feature gate
     if (data.pageSize > 200) {
-      await requireFeature(request, "largePageSize")
+      if (!isProEdition()) {
+        data.pageSize = 200
+      } else {
+        await requireFeature(request, "largePageSize")
+      }
     }
 
     const options: DisplayOptions = data