83 lines
2.9 KiB
TypeScript
83 lines
2.9 KiB
TypeScript
import { PrismaClient } from '@prisma/client';
|
|
import crypto from 'crypto';
|
|
|
|
const prisma = new PrismaClient();
|
|
|
|
/**
|
|
* 生成 VIP 卡密 (原始 + 哈希)
|
|
*/
|
|
function generateCardKey(days: number): { raw: string; hash: string } {
|
|
const prefix = `VIP${days}`;
|
|
const segments = [];
|
|
for (let i = 0; i < 4; i++) {
|
|
segments.push(crypto.randomBytes(2).toString('hex').toUpperCase());
|
|
}
|
|
const raw = `${prefix}-${segments.join('-')}`;
|
|
const hash = crypto.createHash('sha256').update(raw).digest('hex');
|
|
return { raw, hash };
|
|
}
|
|
|
|
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' },
|
|
{ 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 testKey = { raw: 'VIP365-0000-0000-0000-0000', hash: crypto.createHash('sha256').update('VIP365-0000-0000-0000-0000').digest('hex') };
|
|
|
|
await prisma.vipCard.create({
|
|
data: { cardKey: testKey.hash, cardType: 'YEAR', days: 365 },
|
|
}).catch(() => {});
|
|
console.log(`\n🔑 测试卡密 (仅显示一次): ${testKey.raw}`);
|
|
|
|
const cardConfigs = [
|
|
{ type: 'MONTH', days: 30, count: 100 },
|
|
{ type: 'QUARTER', days: 90, count: 50 },
|
|
{ type: 'YEAR', days: 365, count: 20 },
|
|
];
|
|
|
|
let totalCards = 0;
|
|
for (const config of cardConfigs) {
|
|
const cards = [];
|
|
console.log(`\n📦 ${config.type} 卡密 (${config.count} 张):`);
|
|
for (let i = 0; i < config.count; i++) {
|
|
const { raw, hash } = generateCardKey(config.days);
|
|
cards.push({ cardKey: hash, cardType: config.type, days: config.days });
|
|
console.log(` ${raw}`);
|
|
}
|
|
await prisma.vipCard.createMany({ data: cards, skipDuplicates: true });
|
|
totalCards += config.count;
|
|
}
|
|
|
|
console.log(`\n🎉 种子数据初始化完成！`);
|
|
console.log(` ⚠️ 以上卡密仅显示一次，请妥善保存`);
|
|
console.log(` - 分类：${categories.length} 个`);
|
|
console.log(` - VIP 卡密：${totalCards + 1} 张`);
|
|
}
|