fix: 定时任务修复+管理后台Task7支持 (Task4外键修复, Task7管理员邮箱动态查询, Task5限流防护, Task7加入SCRIPT_MAP)
This commit is contained in:
ZhuiGuangAI Dev
@@ -0,0 +1,287 @@
|
||||
import { PrismaClient } from "@prisma/client";
|
||||
import { PrismaMariaDb } from "@prisma/adapter-mariadb";
|
||||
import { readFileSync } from "fs";
|
||||
import { resolve, dirname } from "path";
|
||||
import { fileURLToPath } from "url";
|
||||
import "dotenv/config";
|
||||
|
||||
const __dirname = dirname(fileURLToPath(import.meta.url));
|
||||
const base = process.env.DATABASE_URL.replace("mysql://", "mariadb://");
|
||||
const sep = base.includes("?") ? "&" : "?";
|
||||
const connectionString = `${base}${sep}connection_limit=5&pool_timeout=10`;
|
||||
const adapter = new PrismaMariaDb(connectionString);
|
||||
const prisma = new PrismaClient({ adapter });
|
||||
|
||||
const BOT_DATA_PATH = resolve(__dirname, "..", "data", "bot-characters.json");
|
||||
const INDUSTRY_CATEGORIES = [
|
||||
{
|
||||
name: "电商零售", slug: "ecommerce", icon: "🛒", sortOrder: 1,
|
||||
description: "电商平台运营、选品策略、流量玩法与变现",
|
||||
children: [
|
||||
{ name: "平台电商", slug: "ec-platform", icon: "🏪", description: "淘宝/天猫/京东/拼多多/唯品会" },
|
||||
{ name: "直播带货", slug: "ec-livestream", icon: "📱", description: "抖音/快手/视频号直播运营" },
|
||||
{ name: "私域电商", slug: "ec-private", icon: "🔗", description: "微信生态/社群/小程序电商" },
|
||||
{ name: "社区团购", slug: "ec-group", icon: "🏘️", description: "社区团购/买菜/本地配送" },
|
||||
],
|
||||
},
|
||||
{
|
||||
name: "AI智能科技", slug: "ai-tech", icon: "🤖", sortOrder: 2,
|
||||
description: "AI技术应用、创业机会与产品出海",
|
||||
children: [
|
||||
{ name: "AI工具应用", slug: "ai-tools", icon: "🔧", description: "AI工具推荐、评测与变现" },
|
||||
{ name: "AI创业", slug: "ai-startup", icon: "💡", description: "AI创业方向与商业模式" },
|
||||
{ name: "SaaS出海", slug: "ai-saas", icon: "🚀", description: "SaaS产品开发与全球化" },
|
||||
{ name: "智能硬件", slug: "ai-hardware", icon: "🔌", description: "AI硬件产品与IoT创业" },
|
||||
],
|
||||
},
|
||||
{
|
||||
name: "房地产", slug: "real-estate", icon: "🏠", sortOrder: 3,
|
||||
description: "房产买卖、投资分析、装修与物业管理",
|
||||
children: [
|
||||
{ name: "住宅买卖", slug: "re-residential", icon: "🏡", description: "刚需/改善/学区房交易" },
|
||||
{ name: "商业地产", slug: "re-commercial", icon: "🏢", description: "商铺/写字楼/产业园投资" },
|
||||
{ name: "装修家居", slug: "re-renovation", icon: "🛋️", description: "装修设计/建材/智能家居" },
|
||||
{ name: "物业运营", slug: "re-property", icon: "🏗️", description: "物业管理/租赁运营/资产托管" },
|
||||
],
|
||||
},
|
||||
{
|
||||
name: "金融投资", slug: "finance", icon: "📈", sortOrder: 4,
|
||||
description: "股票基金、加密货币、保险理财与股权投资",
|
||||
children: [
|
||||
{ name: "股票基金", slug: "fi-stock", icon: "📊", description: "A股/港股/美股/基金定投" },
|
||||
{ name: "加密货币", slug: "fi-crypto", icon: "₿", description: "比特币/以太坊/Web3" },
|
||||
{ name: "保险理财", slug: "fi-insurance", icon: "🛡️", description: "保险配置/理财规划" },
|
||||
{ name: "私募股权", slug: "fi-pe", icon: "💼", description: "PE/VC/天使投资" },
|
||||
],
|
||||
},
|
||||
{
|
||||
name: "餐饮食品", slug: "food", icon: "🍜", sortOrder: 5,
|
||||
description: "餐饮创业、茶饮咖啡、预制菜与食品供应链",
|
||||
children: [
|
||||
{ name: "餐饮开店", slug: "fd-restaurant", icon: "🍽️", description: "餐厅选址/运营/连锁加盟" },
|
||||
{ name: "茶饮咖啡", slug: "fd-tea", icon: "🧋", description: "新茶饮/咖啡/饮品创业" },
|
||||
{ name: "预制菜", slug: "fd-prepared", icon: "📦", description: "预制菜/半成品/食品加工" },
|
||||
{ name: "食品供应链", slug: "fd-supply", icon: "🚚", description: "食材采购/冷链/中央厨房" },
|
||||
],
|
||||
},
|
||||
{
|
||||
name: "自媒体内容", slug: "content", icon: "🎬", sortOrder: 6,
|
||||
description: "短视频、直播、图文创作与MCN机构运营",
|
||||
children: [
|
||||
{ name: "短视频创作", slug: "ct-short-video", icon: "📱", description: "抖音/快手/B站内容创作" },
|
||||
{ name: "直播运营", slug: "ct-livestream", icon: "🎥", description: "带货直播/娱乐直播/知识直播" },
|
||||
{ name: "图文内容", slug: "ct-writing", icon: "✍️", description: "公众号/小红书/知乎/头条" },
|
||||
{ name: "MCN机构", slug: "ct-mcn", icon: "🏢", description: "MCN运营/达人孵化/商务变现" },
|
||||
],
|
||||
},
|
||||
{
|
||||
name: "本地生活", slug: "local-life", icon: "🏪", sortOrder: 7,
|
||||
description: "美业、健身、宠物、家政等本地服务创业",
|
||||
children: [
|
||||
{ name: "美容美发", slug: "ll-beauty", icon: "💇", description: "美发/美甲/美容/医美" },
|
||||
{ name: "健身运动", slug: "ll-fitness", icon: "🏋️", description: "健身房/瑜伽/私教/运动康复" },
|
||||
{ name: "宠物经济", slug: "ll-pet", icon: "🐾", description: "宠物店/宠物食品/宠物医疗" },
|
||||
{ name: "家政服务", slug: "ll-housekeeping", icon: "🏠", description: "保洁/月嫂/搬家/维修" },
|
||||
],
|
||||
},
|
||||
{
|
||||
name: "大健康", slug: "health", icon: "💊", sortOrder: 8,
|
||||
description: "医疗美容、健康管理、养老服务与运动康复",
|
||||
children: [
|
||||
{ name: "医疗美容", slug: "hl-cosmetic", icon: "💉", description: "轻医美/微整形/皮肤管理" },
|
||||
{ name: "健康管理", slug: "hl-wellness", icon: "🫀", description: "体检/慢病管理/保健品" },
|
||||
{ name: "养老服务", slug: "hl-elderly", icon: "👴", description: "养老院/居家养老/适老化改造" },
|
||||
{ name: "运动康复", slug: "hl-rehab", icon: "🩹", description: "运动损伤/产后康复/中医理疗" },
|
||||
],
|
||||
},
|
||||
{
|
||||
name: "教育培训", slug: "education", icon: "📚", sortOrder: 9,
|
||||
description: "技能培训、知识付费、留学与企业培训",
|
||||
children: [
|
||||
{ name: "技能培训", slug: "edu-skills", icon: "🛠️", description: "编程/设计/运营等职业技能" },
|
||||
{ name: "知识付费", slug: "edu-knowledge", icon: "💎", description: "课程制作/社群运营/个人IP" },
|
||||
{ name: "留学服务", slug: "edu-abroad", icon: "✈️", description: "留学申请/语言培训/海外游学" },
|
||||
{ name: "企业培训", slug: "edu-corporate", icon: "🏢", description: "企业内训/管理咨询/团队建设" },
|
||||
],
|
||||
},
|
||||
{
|
||||
name: "外贸跨境", slug: "cross-border", icon: "🌏", sortOrder: 10,
|
||||
description: "跨境电商、外贸工厂、海外仓与跨境支付",
|
||||
children: [
|
||||
{ name: "跨境电商", slug: "cb-ecommerce", icon: "🛒", description: "Amazon/Temu/TikTok Shop/Shopee" },
|
||||
{ name: "外贸工厂", slug: "cb-factory", icon: "🏭", description: "代工转品牌/工贸一体/B2B" },
|
||||
{ name: "海外仓物流", slug: "cb-logistics", icon: "📦", description: "海外仓/FBA/跨境物流" },
|
||||
{ name: "跨境支付", slug: "cb-payment", icon: "💳", description: "收付款/汇率/资金合规" },
|
||||
],
|
||||
},
|
||||
];
|
||||
|
||||
async function createIndustryCategories() {
|
||||
console.log("Creating industry forum categories...");
|
||||
|
||||
const parent = await prisma.forumCategory.upsert({
|
||||
where: { slug: "industry" },
|
||||
update: {},
|
||||
create: {
|
||||
name: "行业论坛",
|
||||
slug: "industry",
|
||||
icon: "🏭",
|
||||
sortOrder: 10,
|
||||
description: "各行业交流讨论，AI驱动的行业智囊团",
|
||||
},
|
||||
});
|
||||
|
||||
for (const cat of INDUSTRY_CATEGORIES) {
|
||||
let childCat = await prisma.forumCategory.findUnique({
|
||||
where: { slug: cat.slug },
|
||||
include: { children: true },
|
||||
});
|
||||
|
||||
if (!childCat) {
|
||||
childCat = await prisma.forumCategory.create({
|
||||
data: {
|
||||
name: cat.name,
|
||||
slug: cat.slug,
|
||||
icon: cat.icon,
|
||||
description: cat.description,
|
||||
sortOrder: cat.sortOrder,
|
||||
parentId: parent.id,
|
||||
},
|
||||
include: { children: true },
|
||||
});
|
||||
console.log(` + ${cat.name}`);
|
||||
} else {
|
||||
if (!childCat.parentId) {
|
||||
await prisma.forumCategory.update({
|
||||
where: { id: childCat.id },
|
||||
data: { parentId: parent.id },
|
||||
});
|
||||
}
|
||||
console.log(` ✓ ${cat.name} (exists)`);
|
||||
}
|
||||
|
||||
const existingSlugs = new Set(childCat.children.map(c => c.slug));
|
||||
for (const sub of cat.children) {
|
||||
if (existingSlugs.has(sub.slug)) {
|
||||
console.log(` - ${sub.name} (exists)`);
|
||||
} else {
|
||||
const conflictSlug = await prisma.forumCategory.findUnique({
|
||||
where: { slug: sub.slug },
|
||||
});
|
||||
if (conflictSlug) {
|
||||
await prisma.forumCategory.update({
|
||||
where: { slug: sub.slug },
|
||||
data: {
|
||||
name: sub.name,
|
||||
icon: sub.icon,
|
||||
description: sub.description,
|
||||
parentId: childCat.id,
|
||||
},
|
||||
});
|
||||
console.log(` ↻ ${sub.name} (moved under ${cat.name})`);
|
||||
} else {
|
||||
await prisma.forumCategory.create({
|
||||
data: { ...sub, parentId: childCat.id },
|
||||
});
|
||||
console.log(` + ${sub.name}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
console.log("Industry categories created.");
|
||||
}
|
||||
|
||||
async function createBotUsers() {
|
||||
console.log("Creating/updating bot users...");
|
||||
|
||||
const raw = readFileSync(BOT_DATA_PATH, "utf-8");
|
||||
const { characters } = JSON.parse(raw);
|
||||
|
||||
let created = 0;
|
||||
let updated = 0;
|
||||
let skipped = 0;
|
||||
|
||||
for (const char of characters) {
|
||||
const existingUser = await prisma.user.findUnique({
|
||||
where: { email: char.email },
|
||||
include: { botConfig: true },
|
||||
});
|
||||
|
||||
if (existingUser && existingUser.botConfig) {
|
||||
await prisma.botConfig.update({
|
||||
where: { userId: existingUser.id },
|
||||
data: {
|
||||
personality: char.personality,
|
||||
primaryForums: char.primaryForums,
|
||||
activeHours: char.activeHours,
|
||||
},
|
||||
});
|
||||
console.log(` ↻ ${char.displayName} (updated primaryForums)`);
|
||||
updated++;
|
||||
continue;
|
||||
}
|
||||
|
||||
if (existingUser && !existingUser.botConfig) {
|
||||
await prisma.botConfig.create({
|
||||
data: {
|
||||
userId: existingUser.id,
|
||||
displayName: char.displayName,
|
||||
personality: char.personality,
|
||||
primaryForums: char.primaryForums,
|
||||
activeHours: char.activeHours,
|
||||
},
|
||||
});
|
||||
console.log(` + ${char.displayName} (botConfig created)`);
|
||||
updated++;
|
||||
continue;
|
||||
}
|
||||
|
||||
const user = await prisma.user.create({
|
||||
data: {
|
||||
email: char.email,
|
||||
name: char.displayName,
|
||||
oauthProvider: "bot",
|
||||
oauthId: `bot_${char.key}`,
|
||||
role: "user",
|
||||
isBot: true,
|
||||
},
|
||||
});
|
||||
|
||||
await prisma.botConfig.create({
|
||||
data: {
|
||||
userId: user.id,
|
||||
displayName: char.displayName,
|
||||
personality: char.personality,
|
||||
primaryForums: char.primaryForums,
|
||||
activeHours: char.activeHours,
|
||||
},
|
||||
});
|
||||
|
||||
console.log(` + ${char.displayName} (${char.key})`);
|
||||
created++;
|
||||
}
|
||||
|
||||
console.log(`Bot users: ${created} created, ${updated} updated, ${skipped} skipped.`);
|
||||
}
|
||||
|
||||
async function main() {
|
||||
console.log("=== Seed Bot Forum System ===\n");
|
||||
|
||||
await createIndustryCategories();
|
||||
console.log("");
|
||||
|
||||
await createBotUsers();
|
||||
console.log("");
|
||||
|
||||
console.log("=== Done ===");
|
||||
}
|
||||
|
||||
main()
|
||||
.catch((e) => {
|
||||
console.error(e);
|
||||
process.exit(1);
|
||||
})
|
||||
.finally(async () => {
|
||||
await prisma.$disconnect();
|
||||
});
|
||||
Reference in New Issue
Block a user