101 lines
2.8 KiB
TypeScript
101 lines
2.8 KiB
TypeScript
import { PrismaClient } from '@prisma/client';
|
|
import crypto from 'crypto';
|
|
|
|
const prisma = new PrismaClient();
|
|
|
|
/**
|
|
* 生成 VIP 卡密
|
|
*/
|
|
function generateCardKey(days: number): string {
|
|
const prefix = `VIP${days}`;
|
|
const segments = [];
|
|
for (let i = 0; i < 4; i++) {
|
|
segments.push(crypto.randomBytes(2).toString('hex').toUpperCase());
|
|
}
|
|
return `${prefix}-${segments.join('-')}`;
|
|
}
|
|
|
|
async function main() {
|
|
console.log('🌱 开始种子数据初始化...');
|
|
|
|
// ============================================
|
|
// 创建分类
|
|
// ============================================
|
|
const categories = [
|
|
// 烽火地带
|
|
{ name: '突击步枪', type: 'SCHEME' },
|
|
{ name: '冲锋枪', type: 'SCHEME' },
|
|
{ name: '狙击步枪', type: 'SCHEME' },
|
|
{ name: '轻机枪', type: 'SCHEME' },
|
|
{ name: '霰弹枪', type: 'SCHEME' },
|
|
{ name: '手枪', type: 'SCHEME' },
|
|
|
|
// 全面战场
|
|
{ name: '突击步枪', type: 'SCHEME_AOB' },
|
|
{ name: '冲锋枪', type: 'SCHEME_AOB' },
|
|
{ name: '狙击步枪', type: 'SCHEME_AOB' },
|
|
{ name: '轻机枪', type: 'SCHEME_AOB' },
|
|
{ name: '霰弹枪', type: 'SCHEME_AOB' },
|
|
|
|
// 滤镜
|
|
{ name: '烽火地带', type: 'FILTER' },
|
|
{ name: '全面战场', type: 'FILTER' },
|
|
{ name: '通用', type: 'FILTER' },
|
|
];
|
|
|
|
for (const cat of categories) {
|
|
await prisma.category.create({
|
|
data: cat,
|
|
}).catch(() => {}); // 忽略重复错误
|
|
}
|
|
console.log(`✅ 创建 ${categories.length} 个分类`);
|
|
|
|
// ============================================
|
|
// 生成 VIP 卡密
|
|
// ============================================
|
|
const cardConfigs = [
|
|
{ type: 'MONTH', days: 30, count: 100 },
|
|
{ type: 'QUARTER', days: 90, count: 50 },
|
|
{ type: 'YEAR', days: 365, count: 20 },
|
|
];
|
|
|
|
// 测试卡密 (固定值，方便开发验证)
|
|
await prisma.vipCard.create({
|
|
data: {
|
|
cardKey: 'VIP365-0000-0000-0000-0000',
|
|
cardType: 'YEAR',
|
|
days: 365,
|
|
},
|
|
}).catch(() => {});
|
|
console.log(`✅ 创建测试卡密: VIP365-0000-0000-0000-0000`);
|
|
|
|
let totalCards = 0;
|
|
for (const config of cardConfigs) {
|
|
const cards = [];
|
|
for (let i = 0; i < config.count; i++) {
|
|
cards.push({
|
|
cardKey: generateCardKey(config.days),
|
|
cardType: config.type,
|
|
days: config.days,
|
|
});
|
|
}
|
|
|
|
await prisma.vipCard.createMany({ data: cards, skipDuplicates: true });
|
|
totalCards += config.count;
|
|
console.log(`✅ 生成 ${config.type} 卡密 ${config.count} 张`);
|
|
}
|
|
|
|
console.log(`\n🎉 种子数据初始化完成！`);
|
|
console.log(` - 分类：${categories.length} 个`);
|
|
console.log(` - VIP 卡密：${totalCards} 张`);
|
|
}
|
|
|
|
main()
|
|
.catch((e) => {
|
|
console.error('❌ 种子数据初始化失败:', e);
|
|
process.exit(1);
|
|
})
|
|
.finally(async () => {
|
|
await prisma.$disconnect();
|
|
});
|