112 lines
3.4 KiB
JavaScript
112 lines
3.4 KiB
JavaScript
/**
|
||
* 数据库初始化脚本
|
||
* @file init-db.js
|
||
* @description 初始化数据库结构和基础数据
|
||
*/
|
||
const { sequelize, syncModels } = require('../models');
|
||
const { User, Role, UserRole } = require('../models');
|
||
const bcrypt = require('bcrypt');
|
||
const migrationManager = require('./migration-manager');
|
||
const seedManager = require('./seed-manager');
|
||
|
||
async function initDb() {
|
||
try {
|
||
console.log('开始初始化数据库...');
|
||
|
||
// 测试数据库连接
|
||
await sequelize.authenticate();
|
||
console.log('数据库连接成功');
|
||
|
||
// 创建迁移表
|
||
await migrationManager.createMigrationTable();
|
||
console.log('迁移表创建成功');
|
||
|
||
// 创建种子表
|
||
await seedManager.createSeedTable();
|
||
console.log('种子表创建成功');
|
||
|
||
// 运行待处理的迁移
|
||
await migrationManager.runPendingMigrations();
|
||
console.log('迁移完成');
|
||
|
||
// 运行种子数据
|
||
await seedManager.runAllSeeds();
|
||
console.log('种子数据应用完成');
|
||
|
||
// 同步模型（确保所有模型都已同步到数据库）
|
||
await syncModels({ alter: true });
|
||
console.log('模型同步完成');
|
||
|
||
// 检查是否有管理员用户
|
||
const adminUser = await User.findOne({ where: { username: 'admin' } });
|
||
|
||
// 如果有管理员用户，检查密码是否为123456的哈希值
|
||
if (adminUser) {
|
||
// 检查密码是否为123456的哈希值
|
||
const isCorrectPassword = await adminUser.validPassword('123456');
|
||
|
||
// 如果密码不是123456的哈希值，则更新密码
|
||
if (!isCorrectPassword) {
|
||
adminUser.password = await bcrypt.hash('123456', 10);
|
||
await adminUser.save();
|
||
console.log('管理员密码已重置为123456');
|
||
} else {
|
||
console.log('管理员密码已是123456');
|
||
}
|
||
|
||
// 确保管理员有admin角色
|
||
const adminRole = await Role.findOne({ where: { name: 'admin' } });
|
||
if (adminRole) {
|
||
const hasAdminRole = await adminUser.hasRole('admin');
|
||
if (!hasAdminRole) {
|
||
await adminUser.assignRole(adminRole.id);
|
||
console.log('已为管理员分配admin角色');
|
||
} else {
|
||
console.log('管理员已有admin角色');
|
||
}
|
||
}
|
||
} else {
|
||
// 如果没有管理员用户，则创建一个
|
||
const newAdmin = await User.create({
|
||
username: 'admin',
|
||
email: 'admin@example.com',
|
||
password: await bcrypt.hash('123456', 10)
|
||
});
|
||
console.log('管理员用户已创建，用户名: admin，密码: 123456');
|
||
|
||
// 为新管理员分配admin角色
|
||
const adminRole = await Role.findOne({ where: { name: 'admin' } });
|
||
if (adminRole) {
|
||
await newAdmin.assignRole(adminRole.id);
|
||
console.log('已为新管理员分配admin角色');
|
||
}
|
||
}
|
||
|
||
console.log('数据库初始化完成');
|
||
|
||
// 关闭数据库连接
|
||
await sequelize.close();
|
||
console.log('数据库连接已关闭');
|
||
|
||
process.exit(0);
|
||
} catch (error) {
|
||
console.error('数据库初始化失败:', error);
|
||
|
||
// 尝试关闭数据库连接
|
||
try {
|
||
await sequelize.close();
|
||
console.log('数据库连接已关闭');
|
||
} catch (closeError) {
|
||
console.error('关闭数据库连接失败:', closeError);
|
||
}
|
||
|
||
process.exit(1);
|
||
}
|
||
}
|
||
|
||
// 如果直接运行此脚本，则执行初始化
|
||
if (require.main === module) {
|
||
initDb();
|
||
}
|
||
|
||
module.exports = initDb; |