Initial commit
This commit is contained in:
97
scripts/init-db.js
Normal file
97
scripts/init-db.js
Normal file
@@ -0,0 +1,97 @@
|
||||
const Database = require('better-sqlite3');
|
||||
const bcrypt = require('bcryptjs');
|
||||
const path = require('path');
|
||||
|
||||
// 创建数据库连接
|
||||
const dbPath = path.join(__dirname, '../data/database.sqlite');
|
||||
const db = new Database(dbPath, { verbose: console.log });
|
||||
|
||||
// 创建表
|
||||
const createTables = () => {
|
||||
// 用户表
|
||||
db.exec(`
|
||||
CREATE TABLE IF NOT EXISTS users (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
username TEXT UNIQUE NOT NULL,
|
||||
password TEXT NOT NULL,
|
||||
name TEXT NOT NULL,
|
||||
email TEXT,
|
||||
role TEXT DEFAULT 'user',
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
`);
|
||||
|
||||
// 企业表
|
||||
db.exec(`
|
||||
CREATE TABLE IF NOT EXISTS companies (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
company_name TEXT UNIQUE NOT NULL,
|
||||
is_active BOOLEAN DEFAULT 1,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||
)
|
||||
`);
|
||||
|
||||
// 序列号表
|
||||
db.exec(`
|
||||
CREATE TABLE IF NOT EXISTS serials (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
serial_number TEXT UNIQUE NOT NULL,
|
||||
company_name TEXT NOT NULL,
|
||||
valid_until DATETIME,
|
||||
is_active BOOLEAN DEFAULT 1,
|
||||
created_by INTEGER,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
FOREIGN KEY (created_by) REFERENCES users (id),
|
||||
FOREIGN KEY (company_name) REFERENCES companies (company_name)
|
||||
)
|
||||
`);
|
||||
|
||||
// 创建索引
|
||||
db.exec('CREATE INDEX IF NOT EXISTS idx_company_name_companies ON companies (company_name)');
|
||||
db.exec('CREATE INDEX IF NOT EXISTS idx_serial_number ON serials (serial_number)');
|
||||
db.exec('CREATE INDEX IF NOT EXISTS idx_company_name_serials ON serials (company_name)');
|
||||
db.exec('CREATE INDEX IF NOT EXISTS idx_created_by ON serials (created_by)');
|
||||
|
||||
console.log('数据库表创建完成');
|
||||
};
|
||||
|
||||
// 创建默认管理员用户
|
||||
const createDefaultUser = async () => {
|
||||
const username = 'admin';
|
||||
const password = 'Beifan@2026';
|
||||
const name = '系统管理员';
|
||||
|
||||
// 检查用户是否已存在
|
||||
const user = db.prepare('SELECT * FROM users WHERE username = ?').get(username);
|
||||
|
||||
if (!user) {
|
||||
// 哈希密码
|
||||
const hashedPassword = await bcrypt.hash(password, 10);
|
||||
|
||||
// 创建用户
|
||||
db.prepare(
|
||||
'INSERT INTO users (username, password, name, email, role) VALUES (?, ?, ?, ?, ?)'
|
||||
).run(username, hashedPassword, name, 'admin@example.com', 'admin');
|
||||
|
||||
console.log('默认管理员用户创建完成:');
|
||||
console.log('用户名:', username);
|
||||
console.log('密码:', password);
|
||||
} else {
|
||||
console.log('默认管理员用户已存在');
|
||||
}
|
||||
};
|
||||
|
||||
// 初始化数据库
|
||||
const initDatabase = async () => {
|
||||
createTables();
|
||||
await createDefaultUser();
|
||||
|
||||
// 关闭数据库连接
|
||||
db.close();
|
||||
console.log('数据库连接已关闭');
|
||||
};
|
||||
|
||||
initDatabase();
|
||||
Reference in New Issue
Block a user