feat: 添加PostgreSQL数据库集成和机台管理功能
- 新增PostgreSQL数据库支持，包含内置和外部数据库两种部署模式 - 实现机台管理功能，包括保护状态控制和VHD关键词设置 - 添加用户认证系统，支持会话管理和权限控制 - 更新Docker配置，支持数据库持久化和健康检查 - 扩展API接口，提供完整的RESTful API文档 - 更新客户端代码以支持机台ID和保护检查功能 - 添加数据库初始化脚本和配置指南
This commit is contained in:
Lannamokia
+336
-15
@@ -3,15 +3,37 @@ const cors = require('cors');
|
||||
const bodyParser = require('body-parser');
|
||||
const path = require('path');
|
||||
const fs = require('fs');
|
||||
const session = require('express-session');
|
||||
const bcrypt = require('bcryptjs');
|
||||
require('dotenv').config();
|
||||
|
||||
const database = require('./database');
|
||||
|
||||
const app = express();
|
||||
const PORT = process.env.PORT || 8080;
|
||||
|
||||
// 管理员密码配置 (默认密码: admin123)
|
||||
const ADMIN_PASSWORD_HASH = process.env.ADMIN_PASSWORD_HASH || bcrypt.hashSync('admin123', 10);
|
||||
|
||||
// 会话配置
|
||||
app.use(session({
|
||||
secret: process.env.SESSION_SECRET || 'vhd-select-server-secret-key-2024',
|
||||
resave: false,
|
||||
saveUninitialized: false,
|
||||
cookie: {
|
||||
secure: false, // 在生产环境中应该设置为true (需要HTTPS)
|
||||
httpOnly: true,
|
||||
maxAge: 24 * 60 * 60 * 1000 // 24小时
|
||||
}
|
||||
}));
|
||||
|
||||
// 中间件
|
||||
app.use(cors());
|
||||
app.use(cors({
|
||||
origin: true,
|
||||
credentials: true // 允许发送cookies
|
||||
}));
|
||||
app.use(bodyParser.json());
|
||||
app.use(bodyParser.urlencoded({ extended: true }));
|
||||
app.use(express.static(path.join(__dirname, 'public')));
|
||||
|
||||
// VHD关键词存储
|
||||
let currentVhdKeyword = 'SDEZ'; // 默认值
|
||||
@@ -47,20 +69,129 @@ function saveConfig() {
|
||||
}
|
||||
}
|
||||
|
||||
// API路由
|
||||
// 认证中间件
|
||||
function requireAuth(req, res, next) {
|
||||
if (req.session && req.session.isAuthenticated) {
|
||||
return next();
|
||||
} else {
|
||||
return res.status(401).json({
|
||||
success: false,
|
||||
message: '需要登录',
|
||||
requireAuth: true
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
// 获取当前VHD关键词
|
||||
app.get('/api/boot-image-select', (req, res) => {
|
||||
console.log(`[${new Date().toISOString()}] API请求: GET /api/boot-image-select`);
|
||||
res.json({
|
||||
success: true,
|
||||
BootImageSelected: currentVhdKeyword,
|
||||
timestamp: new Date().toISOString()
|
||||
// 静态文件服务 (登录页面不需要认证)
|
||||
app.use(express.static(path.join(__dirname, 'public')));
|
||||
|
||||
// 认证API路由
|
||||
app.post('/api/auth/login', async (req, res) => {
|
||||
try {
|
||||
const { password } = req.body;
|
||||
|
||||
console.log(`[${new Date().toISOString()}] 登录尝试`);
|
||||
|
||||
if (!password) {
|
||||
return res.status(400).json({
|
||||
success: false,
|
||||
message: '请输入密码'
|
||||
});
|
||||
}
|
||||
|
||||
// 验证密码
|
||||
const isValidPassword = await bcrypt.compare(password, ADMIN_PASSWORD_HASH);
|
||||
|
||||
if (isValidPassword) {
|
||||
req.session.isAuthenticated = true;
|
||||
console.log(`[${new Date().toISOString()}] 登录成功`);
|
||||
res.json({
|
||||
success: true,
|
||||
message: '登录成功'
|
||||
});
|
||||
} else {
|
||||
console.log(`[${new Date().toISOString()}] 登录失败: 密码错误`);
|
||||
res.status(401).json({
|
||||
success: false,
|
||||
message: '密码错误'
|
||||
});
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('登录错误:', error);
|
||||
res.status(500).json({
|
||||
success: false,
|
||||
message: '服务器错误'
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
app.post('/api/auth/logout', (req, res) => {
|
||||
req.session.destroy((err) => {
|
||||
if (err) {
|
||||
console.error('登出错误:', err);
|
||||
return res.status(500).json({
|
||||
success: false,
|
||||
message: '登出失败'
|
||||
});
|
||||
}
|
||||
res.json({
|
||||
success: true,
|
||||
message: '已登出'
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
// 设置VHD关键词
|
||||
app.post('/api/set-vhd', (req, res) => {
|
||||
app.get('/api/auth/check', (req, res) => {
|
||||
res.json({
|
||||
isAuthenticated: !!(req.session && req.session.isAuthenticated)
|
||||
});
|
||||
});
|
||||
|
||||
// API路由
|
||||
|
||||
// 获取当前VHD关键词 - 需要Machine ID验证
|
||||
app.get('/api/boot-image-select', async (req, res) => {
|
||||
const machineId = req.query.machineId;
|
||||
|
||||
console.log(`[${new Date().toISOString()}] API请求: GET /api/boot-image-select, Machine ID: ${machineId}`);
|
||||
|
||||
if (!machineId) {
|
||||
return res.status(400).json({
|
||||
success: false,
|
||||
error: 'Machine ID是必需的'
|
||||
});
|
||||
}
|
||||
|
||||
try {
|
||||
// 从数据库获取机台信息
|
||||
let machine = await database.getMachine(machineId);
|
||||
|
||||
// 如果机台不存在，创建新的机台记录
|
||||
if (!machine) {
|
||||
machine = await database.upsertMachine(machineId, false, currentVhdKeyword);
|
||||
console.log(`[${new Date().toISOString()}] 创建新机台: ${machineId}`);
|
||||
}
|
||||
|
||||
res.json({
|
||||
success: true,
|
||||
BootImageSelected: machine ? machine.vhd_keyword : currentVhdKeyword,
|
||||
machineId: machineId,
|
||||
timestamp: new Date().toISOString()
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('获取VHD关键词失败:', error.message);
|
||||
// 降级到内存存储
|
||||
res.json({
|
||||
success: true,
|
||||
BootImageSelected: currentVhdKeyword,
|
||||
machineId: machineId,
|
||||
timestamp: new Date().toISOString()
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
// 设置VHD关键词 (需要认证)
|
||||
app.post('/api/set-vhd', requireAuth, (req, res) => {
|
||||
const { BootImageSelected } = req.body;
|
||||
|
||||
if (!BootImageSelected || typeof BootImageSelected !== 'string') {
|
||||
@@ -91,6 +222,194 @@ app.post('/api/set-vhd', (req, res) => {
|
||||
});
|
||||
});
|
||||
|
||||
// 保护状态检查端点 - 需要Machine ID验证
|
||||
app.get('/api/protect', async (req, res) => {
|
||||
const machineId = req.query.machineId;
|
||||
|
||||
console.log(`[${new Date().toISOString()}] API请求: GET /api/protect, Machine ID: ${machineId}`);
|
||||
|
||||
if (!machineId) {
|
||||
return res.status(400).json({
|
||||
success: false,
|
||||
error: 'Machine ID是必需的'
|
||||
});
|
||||
}
|
||||
|
||||
try {
|
||||
// 从数据库获取机台信息
|
||||
let machine = await database.getMachine(machineId);
|
||||
|
||||
// 如果机台不存在，创建新的机台记录
|
||||
if (!machine) {
|
||||
machine = await database.upsertMachine(machineId, false, currentVhdKeyword);
|
||||
console.log(`[${new Date().toISOString()}] 创建新机台: ${machineId}`);
|
||||
}
|
||||
|
||||
res.json({
|
||||
success: true,
|
||||
protected: machine ? machine.protected : false,
|
||||
machineId: machineId,
|
||||
timestamp: new Date().toISOString()
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('获取保护状态失败:', error.message);
|
||||
// 降级到默认状态
|
||||
res.json({
|
||||
success: true,
|
||||
protected: false,
|
||||
machineId: machineId,
|
||||
timestamp: new Date().toISOString()
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
// 设置机台保护状态
|
||||
app.post('/api/protect', requireAuth, async (req, res) => {
|
||||
const { machineId, protected } = req.body;
|
||||
|
||||
console.log(`[${new Date().toISOString()}] API请求: POST /api/protect, Machine ID: ${machineId}, Protected: ${protected}`);
|
||||
|
||||
if (!machineId) {
|
||||
return res.status(400).json({
|
||||
success: false,
|
||||
error: 'Machine ID是必需的'
|
||||
});
|
||||
}
|
||||
|
||||
if (typeof protected !== 'boolean') {
|
||||
return res.status(400).json({
|
||||
success: false,
|
||||
error: 'protected状态必须是布尔值'
|
||||
});
|
||||
}
|
||||
|
||||
try {
|
||||
// 更新机台保护状态
|
||||
const machine = await database.updateMachineProtection(machineId, protected);
|
||||
|
||||
if (!machine) {
|
||||
// 如果机台不存在，创建新的机台记录
|
||||
const newMachine = await database.upsertMachine(machineId, protected, currentVhdKeyword);
|
||||
return res.json({
|
||||
success: true,
|
||||
protected: newMachine.protected,
|
||||
machineId: machineId,
|
||||
message: '机台保护状态已设置'
|
||||
});
|
||||
}
|
||||
|
||||
res.json({
|
||||
success: true,
|
||||
protected: machine.protected,
|
||||
machineId: machineId,
|
||||
message: '机台保护状态已更新'
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('设置保护状态失败:', error.message);
|
||||
res.status(500).json({
|
||||
success: false,
|
||||
error: '设置保护状态失败'
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
// 获取所有机台信息
|
||||
app.get('/api/machines', requireAuth, async (req, res) => {
|
||||
console.log(`[${new Date().toISOString()}] API请求: GET /api/machines`);
|
||||
|
||||
try {
|
||||
const machines = await database.getAllMachines();
|
||||
res.json({
|
||||
success: true,
|
||||
machines: machines,
|
||||
count: machines.length,
|
||||
timestamp: new Date().toISOString()
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('获取机台列表失败:', error.message);
|
||||
res.status(500).json({
|
||||
success: false,
|
||||
error: '获取机台列表失败'
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
// 设置特定机台的VHD关键词
|
||||
app.post('/api/machines/:machineId/vhd', requireAuth, async (req, res) => {
|
||||
const { machineId } = req.params;
|
||||
const { vhdKeyword } = req.body;
|
||||
|
||||
console.log(`[${new Date().toISOString()}] API请求: POST /api/machines/${machineId}/vhd, VHD: ${vhdKeyword}`);
|
||||
|
||||
if (!vhdKeyword || typeof vhdKeyword !== 'string') {
|
||||
return res.status(400).json({
|
||||
success: false,
|
||||
error: 'VHD关键词不能为空且必须是字符串'
|
||||
});
|
||||
}
|
||||
|
||||
const trimmedKeyword = vhdKeyword.trim().toUpperCase();
|
||||
if (trimmedKeyword.length === 0) {
|
||||
return res.status(400).json({
|
||||
success: false,
|
||||
error: 'VHD关键词不能为空'
|
||||
});
|
||||
}
|
||||
|
||||
try {
|
||||
// 更新机台VHD关键词
|
||||
let machine = await database.updateMachineVhdKeyword(machineId, trimmedKeyword);
|
||||
|
||||
if (!machine) {
|
||||
// 如果机台不存在，创建新的机台记录
|
||||
machine = await database.upsertMachine(machineId, false, trimmedKeyword);
|
||||
}
|
||||
|
||||
res.json({
|
||||
success: true,
|
||||
machineId: machineId,
|
||||
vhdKeyword: machine.vhd_keyword,
|
||||
message: '机台VHD关键词已更新'
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('设置机台VHD关键词失败:', error.message);
|
||||
res.status(500).json({
|
||||
success: false,
|
||||
error: '设置机台VHD关键词失败'
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
// 删除机台
|
||||
app.delete('/api/machines/:machineId', requireAuth, async (req, res) => {
|
||||
const { machineId } = req.params;
|
||||
|
||||
console.log(`[${new Date().toISOString()}] API请求: DELETE /api/machines/${machineId}`);
|
||||
|
||||
try {
|
||||
const deletedMachine = await database.deleteMachine(machineId);
|
||||
|
||||
if (!deletedMachine) {
|
||||
return res.status(404).json({
|
||||
success: false,
|
||||
error: '机台不存在'
|
||||
});
|
||||
}
|
||||
|
||||
res.json({
|
||||
success: true,
|
||||
machineId: machineId,
|
||||
message: '机台已删除'
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('删除机台失败:', error.message);
|
||||
res.status(500).json({
|
||||
success: false,
|
||||
error: '删除机台失败'
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
// 获取服务器状态
|
||||
app.get('/api/status', (req, res) => {
|
||||
res.json({
|
||||
@@ -99,7 +418,7 @@ app.get('/api/status', (req, res) => {
|
||||
BootImageSelected: currentVhdKeyword,
|
||||
uptime: process.uptime(),
|
||||
timestamp: new Date().toISOString(),
|
||||
version: '1.0.0'
|
||||
version: '1.2.0'
|
||||
});
|
||||
});
|
||||
|
||||
@@ -139,14 +458,16 @@ app.listen(PORT, () => {
|
||||
});
|
||||
|
||||
// 优雅关闭
|
||||
process.on('SIGINT', () => {
|
||||
process.on('SIGINT', async () => {
|
||||
console.log('\n正在关闭服务器...');
|
||||
saveConfig();
|
||||
await database.close();
|
||||
process.exit(0);
|
||||
});
|
||||
|
||||
process.on('SIGTERM', () => {
|
||||
process.on('SIGTERM', async () => {
|
||||
console.log('\n正在关闭服务器...');
|
||||
saveConfig();
|
||||
await database.close();
|
||||
process.exit(0);
|
||||
});
|
||||
Reference in New Issue
Block a user