@allbluecn/web-app 0.4.17 → 0.4.18

package/src/server/serverFns/__tests__/team-members.test.ts

@@ -33,10 +33,21 @@ vi.mock('@allbluecn/database', () => ({
     teamMembership: {
       findUnique: vi.fn(),
       findMany: vi.fn(),
+      findFirst: vi.fn(),
       delete: vi.fn(),
+      create: vi.fn(),
       count: vi.fn(),
     },
-    teamInvitation: { count: vi.fn() },
+    teamInvitation: {
+      create: vi.fn(),
+      findFirst: vi.fn(),
+      findUnique: vi.fn(),
+      updateMany: vi.fn(),
+      findMany: vi.fn(),
+      update: vi.fn(),
+      count: vi.fn(),
+      deleteMany: vi.fn(),
+    },
     projectMember: { deleteMany: vi.fn() },
     $transaction: vi.fn(),
   },
@@ -55,17 +66,13 @@ vi.mock('@/server/audit', () => ({
 }))
 
 import { prisma } from '@allbluecn/database'
-import {
-  listTeamMembersFn,
-  removeTeamMemberFn,
-  leaveTeamFn,
-  listTeamUsersFn,
-} from '../team'
+import { listTeamMembersFn } from '../team'
 
 describe('listTeamMembersFn', () => {
   beforeEach(() => vi.clearAllMocks())
 
-  it('owner 列出成员', async () => {
+  it('已有 membership → 正常列出成员', async () => {
+    vi.mocked(prisma.teamMembership.findUnique).mockResolvedValue({ id: "m1", ownerId: "owner1" } as never)
     vi.mocked(prisma.teamMembership.findMany).mockResolvedValue([
       {
         id: 'm1',
@@ -78,124 +85,38 @@ describe('listTeamMembersFn', () => {
     ] as never)
     const result = await listTeamMembersFn()
     expect(result).toHaveLength(1)
-    expect(result[0].seatType).toBe('member')
+    expect((result as any)[0].seatType).toBe('member')
+    expect((result as any)[0].memberId).toBe('u2')
   })
-})
-
-describe('removeTeamMemberFn', () => {
-  beforeEach(() => vi.clearAllMocks())
-
-  it('owner 移除成员 → 事务删 membership + 级联删 ProjectMember', async () => {
-    vi.mocked(prisma.teamMembership.findUnique).mockResolvedValue({
-      id: 'm1',
-      ownerId: 'owner1',
-      memberId: 'u2',
-      seatType: 'member',
-    } as never)
-    let capturedTx: any
-    vi.mocked(prisma.$transaction).mockImplementation(async (fn: any) => {
-      capturedTx = {
-        teamMembership: { delete: vi.fn().mockResolvedValue({}) },
-        projectMember: { deleteMany: vi.fn().mockResolvedValue({ count: 3 }) },
-      }
-      return fn(capturedTx)
-    })
-    const result = await removeTeamMemberFn({ data: { membershipId: 'm1' } })
-    expect(result.success).toBe(true)
-    expect(capturedTx.projectMember.deleteMany).toHaveBeenCalledWith(
-      expect.objectContaining({ where: { userId: 'u2' } }),
-    )
-  })
-
-  it('非 owner 移除 → 拒绝', async () => {
-    vi.mocked(prisma.teamMembership.findUnique).mockResolvedValue({
-      id: 'm1',
-      ownerId: 'other',
-      memberId: 'u2',
-    } as never)
-    await expect(removeTeamMemberFn({ data: { membershipId: 'm1' } }))
-      .rejects.toMatchObject({ message: expect.stringContaining('权限') })
-  })
-
-  it('成员不存在 → 拒绝', async () => {
-    vi.mocked(prisma.teamMembership.findUnique).mockResolvedValue(null)
-    await expect(removeTeamMemberFn({ data: { membershipId: 'm999' } }))
-      .rejects.toMatchObject({ message: expect.stringContaining('不存在') })
-  })
-})
-
-describe('leaveTeamFn', () => {
-  beforeEach(() => vi.clearAllMocks())
 
-  it('未加入团队 → 拒绝', async () => {
+  it('无 membership → 自动创建 owner 记录', async () => {
     vi.mocked(prisma.teamMembership.findUnique).mockResolvedValue(null)
-    await expect(leaveTeamFn()).rejects.toMatchObject({ message: expect.stringContaining('未加入') })
-  })
-
-  it('已加入团队 → 退出并级联删 ProjectMember', async () => {
-    vi.mocked(prisma.teamMembership.findUnique).mockResolvedValue({
-      id: 'm1',
+    vi.mocked(prisma.teamMembership.create).mockResolvedValue({
+      id: 'm-new',
       ownerId: 'owner1',
       memberId: 'owner1',
+      seatType: 'owner',
+      joinedAt: new Date(),
+      member: { id: 'owner1', username: 'A', email: 'a@x.com', avatarConfig: null, bgShape: null, bgColor: null },
     } as never)
-    let capturedTx: any
-    vi.mocked(prisma.$transaction).mockImplementation(async (fn: any) => {
-      capturedTx = {
-        teamMembership: { delete: vi.fn().mockResolvedValue({}) },
-        projectMember: { deleteMany: vi.fn().mockResolvedValue({ count: 2 }) },
-      }
-      return fn(capturedTx)
-    })
-    const result = await leaveTeamFn()
-    expect(result.success).toBe(true)
-    expect(capturedTx.projectMember.deleteMany).toHaveBeenCalledWith(
-      expect.objectContaining({ where: { userId: 'owner1' } }),
+    const result = await listTeamMembersFn()
+    expect(prisma.teamMembership.create).toHaveBeenCalledWith(
+      expect.objectContaining({
+        data: expect.objectContaining({ ownerId: 'owner1', memberId: 'owner1', seatType: 'owner' }),
+      }),
     )
-  })
-})
-
-describe('listTeamUsersFn', () => {
-  beforeEach(() => vi.clearAllMocks())
-
-  it('返回 owner + active 成员（含 seatType）', async () => {
-    vi.mocked(prisma.user.findUnique).mockResolvedValue({
-      id: 'owner1',
-      username: 'A',
-      email: 'a@x.com',
-      avatarConfig: null,
-      bgShape: null,
-      bgColor: null,
-    } as never)
-    vi.mocked(prisma.teamMembership.findUnique).mockResolvedValue({
-      ownerId: 'owner1',
-      memberId: 'owner1',
-    } as never)
-    vi.mocked(prisma.teamMembership.findMany).mockResolvedValue([
-      { seatType: 'member', member: { id: 'u2', username: 'B', email: 'b@x.com', avatarConfig: null, bgShape: null, bgColor: null } },
-      { seatType: 'guest', member: { id: 'u3', username: 'C', email: 'c@x.com', avatarConfig: null, bgShape: null, bgColor: null } },
-    ] as never)
-    const result = await listTeamUsersFn()
-    expect(result).toHaveLength(3)
+    expect(result).toHaveLength(1)
     expect(result[0].seatType).toBe('owner')
-    expect(result[0].username).toBe('A')
-    expect(result[1].seatType).toBe('member')
-    expect(result[2].seatType).toBe('guest')
+    expect(result[0].memberId).toBe('owner1')
   })
 
-  it('未加入团队的用户 → 返回自身作为 owner', async () => {
-    vi.mocked(prisma.user.findUnique).mockResolvedValue({
-      id: 'owner1',
-      username: 'Solo',
-      email: 'solo@x.com',
-      avatarConfig: null,
-      bgShape: null,
-      bgColor: null,
-    } as never)
-    vi.mocked(prisma.teamMembership.findUnique).mockResolvedValue(null)
-    vi.mocked(prisma.teamMembership.findMany).mockResolvedValue([] as never)
-    const result = await listTeamUsersFn()
-    expect(result).toHaveLength(1)
-    expect(result[0].seatType).toBe('owner')
-    expect(result[0].username).toBe('Solo')
+  it('已有 membership → 不创建，直接查询', async () => {
+    vi.mocked(prisma.teamMembership.findUnique).mockResolvedValue({ id: "m1", ownerId: "owner1" } as never)
+    vi.mocked(prisma.teamMembership.findMany).mockResolvedValue([])
+    await listTeamMembersFn()
+    expect(prisma.teamMembership.create).not.toHaveBeenCalled()
+    expect(prisma.teamMembership.findMany).toHaveBeenCalledWith(
+      expect.objectContaining({ where: { ownerId: 'owner1' } }),
+    )
   })
-})
+})

package/src/{lib/collections/ai/ai-providers-collection.ts → server/serverFns/auth/setup.ts}

@@ -15,17 +15,12 @@
 // You should have received a copy of the GNU Affero General Public License
 // along with this program. If not, see <https://www.gnu.org/licenses/>.
 
-import { createCollection } from '@tanstack/db'
-import { queryCollectionOptions } from '@tanstack/query-db-collection'
-import { listAiProvidersFn } from '@/server/serverFns/ai/providers'
-import { getQueryClient } from '@allbluecn/shared/lib/query-client-seam'
+import { createServerFn } from '@tanstack/react-start'
+import { prisma } from '@allbluecn/database'
 
-export const aiProvidersCollection = createCollection(
-  queryCollectionOptions({
-    id: 'ai-providers',
-    queryKey: ['ai-providers'],
-    queryFn: () => listAiProvidersFn(),
-    getKey: (item) => item.meta.providerId,
-    queryClient: getQueryClient(),
+export const checkAdminExistsFn = createServerFn({ method: 'GET' }).handler(async () => {
+  const count = await prisma.user.count({
+    where: { role: 'SUPER_ADMIN' },
   })
-)
+  return { hasAdmin: count > 0 }
+})

package/src/server/serverFns/billing.ts

@@ -32,6 +32,17 @@ const requireBilling = () => {
   return billing
 }
 
+async function emitBillingNotification(userId: string, type: string, title: string, body: string, link?: string) {
+  void (async () => {
+    try {
+      const { createNotification } = await import('@/server/notifications/notification-service')
+      await createNotification({ userId, category: 'billing', type, title, body, link })
+    } catch (e) {
+      console.error('[billing] 通知发送失败:', e instanceof Error ? e.message : 'unknown')
+    }
+  })()
+}
+
 export const getEffectivePlanFn = createServerFn({ method: 'GET' }).handler(async () => {
   const request = getRequest()
   const session = await requireAuth(request)
@@ -92,10 +103,12 @@ export const createCustomerPortalSessionFn = createServerFn({ method: 'POST' }).
     const userId = session.user!.id!
 
     const baseUrl = process.env.VITE_APP_URL ?? 'http://localhost:3000'
-    return requireBilling().createCustomerPortalSession({
+    const result = await requireBilling().createCustomerPortalSession({
       userId,
       returnUrl: `${baseUrl}/settings/billing`,
     })
+    await emitBillingNotification(userId, 'billing.portal_access', '客户门户已打开', '您已打开管理订阅的客户门户', '/settings/billing')
+    return result
   },
 )
 
@@ -159,9 +172,17 @@ export const updateSeatsFn = createServerFn({ method: 'POST' })
       )
     }
 
-    return requireBilling().updateSeats({
+    const result = await requireBilling().updateSeats({
       userId,
       addonSeats: data.addonSeats,
       successUrl: data.successUrl ?? `${baseUrl}/settings/billing?checkout=success`,
     })
+    await emitBillingNotification(
+      userId,
+      'billing.seats_changed',
+      '席位变更',
+      `席位已调整为 ${subscription.plan.maxSeats + data.addonSeats}`,
+      '/settings/billing',
+    )
+    return result
   })

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/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,