feat: 首次推送到Gitea - 完整项目代码 + 安全加固 + 知识库

2026-10-02 18:52:49 +08:00
parent b44601bb66
commit 8205ae709c
1185 changed files with 49841 additions and 12802 deletions
@@ -0,0 +1,185 @@
+// 每日 09:00 运行：为活跃用户生成通知摘要
+// Daily notification digest for active users
+import { PrismaClient } from "@prisma/client";
+import { PrismaMariaDb } from "@prisma/adapter-mariadb";
+import "dotenv/config";
+
+const base = (process.env.DATABASE_URL || "mysql://localhost:3306/zhuiguang_ai?charset=utf8mb4")
+  .replace("mysql://", "mariadb://");
+const sep = base.includes("?") ? "&" : "?";
+const connectionString = `${base}${sep}connection_limit=5&pool_timeout=15`;
+const adapter = new PrismaMariaDb(connectionString);
+const prisma = new PrismaClient({ adapter });
+
+// Map NotifType enum values to human-readable Chinese labels
+const TYPE_LABEL = {
+  COMMENT_REPLY: "新回复",
+  COMMENT_LIKE: "评论点赞",
+  TOPIC_REPLY: "话题回复",
+  MENTION: "提及",
+  BEST_ANSWER: "最佳答案",
+  POST_LIKE: "点赞",
+  TOPIC_LIKE: "话题点赞",
+  SYSTEM: "系统通知",
+  DIGEST: "摘要",
+};
+
+async function main() {
+  const taskKey = "notification-digest";
+  const taskName = "通知摘要生成";
+
+  const startedAt = new Date();
+
+  // Step 1: Find active users - users with activity in the last 7 days
+  const sevenDaysAgo = new Date(Date.now() - 7 * 24 * 3600 * 1000);
+
+  // Get users who have notifications or recent profile updates (proxy for active)
+  const activeUserIds = await prisma.notification.findMany({
+    where: {
+      createdAt: { gte: sevenDaysAgo },
+      type: { notIn: ["DIGEST"] },
+    },
+    distinct: ["userId"],
+    select: { userId: true },
+  });
+
+  console.log(
+    `[notification-digest] 找到 ${activeUserIds.length} 位活跃用户（近7天有通知）`
+  );
+
+  // Step 2: For each active user, check their notification prefs and compile digest
+  const past24h = new Date(Date.now() - 24 * 3600 * 1000);
+  let digestCreated = 0;
+  let usersSkipped = 0;
+  let totalDigested = 0;
+
+  for (const { userId } of activeUserIds) {
+    // Check preferences
+    const prefKey = `notif_prefs_${userId}`;
+    const prefConfig = await prisma.systemConfig.findUnique({
+      where: { key: prefKey },
+    });
+
+    let prefs = {
+      digestMode: "realtime",
+      channels: {},
+      quietHours: { enabled: false, start: "22:00", end: "08:00" },
+    };
+
+    if (prefConfig) {
+      try {
+        prefs = { ...prefs, ...(prefConfig.value || {}) };
+      } catch {
+        // fall back to defaults
+      }
+    }
+
+    // Skip users who want real-time notifications
+    if (prefs.digestMode === "realtime") {
+      usersSkipped++;
+      continue;
+    }
+
+    // Step 3: Compile their unread notifications from the past 24h (or past 7 days for weekly)
+    const digestWindow =
+      prefs.digestMode === "weekly"
+        ? new Date(Date.now() - 7 * 24 * 3600 * 1000)
+        : past24h;
+
+    const recentNotifs = await prisma.notification.findMany({
+      where: {
+        userId,
+        type: { notIn: ["DIGEST"] },
+        isRead: false,
+        createdAt: { gte: digestWindow },
+      },
+      orderBy: { createdAt: "desc" },
+    });
+
+    if (recentNotifs.length === 0) {
+      usersSkipped++;
+      continue;
+    }
+
+    // Group by type
+    const typeCounts = {};
+    for (const n of recentNotifs) {
+      const label = TYPE_LABEL[n.type] || n.type;
+      typeCounts[label] = (typeCounts[label] || 0) + 1;
+    }
+
+    // Build summary message
+    const parts = Object.entries(typeCounts).map(
+      ([label, count]) => `${count} 条${label}`
+    );
+
+    const modeLabel =
+      prefs.digestMode === "weekly" ? "本周" : "今天";
+    const title = `${modeLabel}你有 ${recentNotifs.length} 条新通知`;
+    const content = parts.join("、");
+
+    // Create the digest notification
+    try {
+      await prisma.notification.create({
+        data: {
+          userId,
+          type: "DIGEST",
+          title,
+          content,
+          link: "/notifications",
+          isRead: false,
+        },
+      });
+      digestCreated++;
+      totalDigested += recentNotifs.length;
+    } catch (err) {
+      console.error(`[notification-digest] 为用户 ${userId} 创建摘要失败:`, err.message);
+    }
+  }
+
+  const finishedAt = new Date();
+  const duration = Math.round((finishedAt - startedAt) / 1000);
+
+  console.log(
+    `[notification-digest] 完成: 创建 ${digestCreated} 条摘要，覆盖 ${totalDigested} 条通知，跳过 ${usersSkipped} 位用户，耗时 ${duration}s`
+  );
+
+  // Step 4: Log to TaskLog
+  await prisma.taskLog.create({
+    data: {
+      taskKey,
+      taskName,
+      status: "completed",
+      startedAt,
+      finishedAt,
+      duration,
+      result: {
+        activeUsers: activeUserIds.length,
+        digestCreated,
+        totalDigested,
+        usersSkipped,
+      },
+    },
+  });
+
+  await prisma.$disconnect();
+}
+
+main().catch(async (e) => {
+  console.error("[notification-digest] 执行失败:", e.message);
+  try {
+    await prisma.taskLog.create({
+      data: {
+        taskKey: "notification-digest",
+        taskName: "通知摘要生成",
+        status: "failed",
+        startedAt: new Date(),
+        finishedAt: new Date(),
+        duration: 0,
+        error: e.message?.slice(0, 500),
+      },
+    });
+  } catch {}
+  if (prisma) await prisma.$disconnect();
+  process.exit(1);
+});