first commit

2025-07-08 00:52:10 +08:00
commit aa2416c5d6
69 changed files with 16628 additions and 0 deletions
--- a/backend/prisma/seed.ts
+++ b/backend/prisma/seed.ts
@@ -0,0 +1,153 @@
+import { PrismaClient } from '@prisma/client';
+import bcrypt from 'bcryptjs';
+
+const prisma = new PrismaClient();
+
+async function main() {
+  console.log('开始初始化数据库...');
+
+  // 创建管理员用户
+  const adminPassword = await bcrypt.hash('admin123', 12);
+  const admin = await prisma.user.upsert({
+    where: { email: 'admin@pandora.com' },
+    update: {},
+    create: {
+      email: 'admin@pandora.com',
+      username: 'admin',
+      password: adminPassword,
+      firstName: '管理员',
+      lastName: '系统',
+      isAdmin: true,
+      isActive: true,
+      emailVerified: true,
+    },
+  });
+
+  // 创建测试用户
+  const userPassword = await bcrypt.hash('user123', 12);
+  const user = await prisma.user.upsert({
+    where: { email: 'user@pandora.com' },
+    update: {},
+    create: {
+      email: 'user@pandora.com',
+      username: 'testuser',
+      password: userPassword,
+      firstName: '测试',
+      lastName: '用户',
+      isAdmin: false,
+      isActive: true,
+      emailVerified: true,
+    },
+  });
+
+  // 创建网站账号
+  const accounts = [
+    {
+      website: 'claude.ai',
+      accountName: 'claude_pro_1',
+      token: 'sk-ant-api03-xxx-claude-pro-1',
+      maxUsers: 3,
+      currentUsers: 0,
+    },
+    {
+      website: 'openai.com',
+      accountName: 'gpt4_plus_1',
+      token: 'sk-xxx-gpt4-plus-1',
+      maxUsers: 2,
+      currentUsers: 0,
+    },
+    {
+      website: 'gemini.google.com',
+      accountName: 'gemini_pro_1',
+      token: 'AIzaSyCxxx-gemini-pro-1',
+      maxUsers: 1,
+      currentUsers: 0,
+    },
+  ];
+
+  for (const accountData of accounts) {
+    await prisma.websiteAccount.upsert({
+      where: {
+        website_accountName: {
+          website: accountData.website,
+          accountName: accountData.accountName,
+        }
+      },
+      update: {},
+      create: accountData,
+    });
+  }
+
+  // 为管理员分配所有账号
+  const allAccounts = await prisma.websiteAccount.findMany();
+  for (const account of allAccounts) {
+    await prisma.accountAssignment.upsert({
+      where: {
+        userId_accountId: {
+          userId: admin.id,
+          accountId: account.id,
+        }
+      },
+      update: {},
+      create: {
+        userId: admin.id,
+        accountId: account.id,
+        isActive: true,
+      },
+    });
+  }
+
+  // 为用户分配部分账号
+  const userAccounts = await prisma.websiteAccount.findMany({
+    where: {
+      website: {
+        in: ['claude.ai', 'openai.com']
+      }
+    }
+  });
+
+  for (const account of userAccounts) {
+    await prisma.accountAssignment.upsert({
+      where: {
+        userId_accountId: {
+          userId: user.id,
+          accountId: account.id,
+        }
+      },
+      update: {},
+      create: {
+        userId: user.id,
+        accountId: account.id,
+        isActive: true,
+      },
+    });
+  }
+
+  // 更新账号当前用户数
+  for (const account of allAccounts) {
+    const userCount = await prisma.accountAssignment.count({
+      where: {
+        accountId: account.id,
+        isActive: true,
+      }
+    });
+
+    await prisma.websiteAccount.update({
+      where: { id: account.id },
+      data: { currentUsers: userCount }
+    });
+  }
+
+  console.log('数据库初始化完成！');
+  console.log('管理员账户:', admin.email, '密码: admin123');
+  console.log('测试用户账户:', user.email, '密码: user123');
+}
+
+main()
+  .catch((e) => {
+    console.error('数据库初始化失败:', e);
+    process.exit(1);
+  })
+  .finally(async () => {
+    await prisma.$disconnect();
+  });