Baobhan Sith
292 lines
13 KiB
TypeScript
292 lines
13 KiB
TypeScript
import express = require('express');
|
|
import { Request, Response, NextFunction, RequestHandler } from 'express';
|
|
import http from 'http';
|
|
import fs from 'fs';
|
|
import path from 'path';
|
|
import crypto from 'crypto';
|
|
import dotenv from 'dotenv';
|
|
import session from 'express-session';
|
|
import sessionFileStore from 'session-file-store';
|
|
import { getDbInstance } from './database/connection';
|
|
import authRouter from './auth/auth.routes';
|
|
import connectionsRouter from './connections/connections.routes';
|
|
import sftpRouter from './sftp/sftp.routes';
|
|
import proxyRoutes from './proxies/proxies.routes';
|
|
import tagsRouter from './tags/tags.routes';
|
|
import settingsRoutes from './settings/settings.routes';
|
|
import notificationRoutes from './notifications/notification.routes';
|
|
import auditRoutes from './audit/audit.routes';
|
|
import commandHistoryRoutes from './command-history/command-history.routes';
|
|
import quickCommandsRoutes from './quick-commands/quick-commands.routes';
|
|
import terminalThemeRoutes from './terminal-themes/terminal-theme.routes';
|
|
import appearanceRoutes from './appearance/appearance.routes';
|
|
import sshKeysRouter from './ssh_keys/ssh_keys.routes'; // +++ Import SSH Key routes +++
|
|
import quickCommandTagRoutes from './quick-command-tags/quick-command-tag.routes'; // +++ Import Quick Command Tag routes +++
|
|
import { initializeWebSocket } from './websocket';
|
|
import { ipWhitelistMiddleware } from './auth/ipWhitelist.middleware';
|
|
|
|
// --- 初始化通知系统 (导入即初始化单例) ---
|
|
import './services/event.service'; // 确保事件服务被加载
|
|
import './services/notification.processor.service'; // 确保处理器被加载并监听事件
|
|
import './services/notification.dispatcher.service'; // 确保分发器被加载并监听处理器事件
|
|
// --- 结束通知系统初始化 ---
|
|
// --- 环境变量和密钥初始化 ---
|
|
|
|
// --- 全局错误处理 ---
|
|
// 捕获未处理的 Promise Rejection
|
|
process.on('unhandledRejection', (reason: any, promise: Promise<any>) => {
|
|
console.error('---未处理的 Promise Rejection---');
|
|
console.error('原因:', reason);
|
|
// 可以在这里添加更详细的日志记录，例如将错误发送到监控系统
|
|
// 注意：根据 Node.js 官方建议，未来版本的 Node.js 可能会默认在 unhandledRejection 时终止进程。
|
|
// 目前我们选择记录错误并继续运行，但这可能导致应用程序状态不一致。
|
|
});
|
|
|
|
// 捕获未捕获的同步异常
|
|
process.on('uncaughtException', (error: Error) => {
|
|
console.error('---未捕获的异常---');
|
|
console.error('错误:', error);
|
|
// 记录错误，但避免退出进程，尝试让服务器继续运行（有风险）
|
|
// 在生产环境中，更安全的做法可能是记录错误后优雅地关闭服务器并重启。
|
|
// process.exit(1); // 强制退出（更安全，但会中断服务）
|
|
});
|
|
// --- 结束全局错误处理 ---
|
|
|
|
|
|
const initializeEnvironment = async () => {
|
|
// 1. 加载根目录的 .env 文件 (定义部署模式等)
|
|
// 注意: __dirname 在 dist/src 中，所以需要回退三级到项目根目录
|
|
const projectRootEnvPath = path.resolve(__dirname, '../../../.env');
|
|
const rootConfigResult = dotenv.config({ path: projectRootEnvPath });
|
|
// Use type assertion for error code checking
|
|
if (rootConfigResult.error && (rootConfigResult.error as NodeJS.ErrnoException).code !== 'ENOENT') {
|
|
// 只在文件存在但无法加载时发出警告
|
|
console.warn(`[ENV Init] Warning: Could not load root .env file from ${projectRootEnvPath}. Error: ${rootConfigResult.error.message}`);
|
|
} else if (!rootConfigResult.error) {
|
|
console.log(`[ENV Init] Loaded environment variables from root .env file: ${projectRootEnvPath}`);
|
|
} else {
|
|
console.log(`[ENV Init] Root .env file not found at ${projectRootEnvPath}, proceeding without it (expected in non-local deployments where env vars are injected).`);
|
|
}
|
|
|
|
// 2. 加载 data/.env 文件 (定义密钥等)
|
|
// 注意: 这个路径是相对于编译后的 dist/src/index.js
|
|
const dataEnvPath = path.resolve(__dirname, '../data/.env');
|
|
let keysGenerated = false;
|
|
let keysToAppend = '';
|
|
|
|
// dotenv.config 默认不会覆盖已存在的 process.env 变量
|
|
// 这意味着如果根 .env 和 data/.env 定义了相同的变量，先加载的（根 .env）的值会优先
|
|
const dataConfigResult = dotenv.config({ path: dataEnvPath });
|
|
// Use type assertion for error code checking
|
|
if (dataConfigResult.error && (dataConfigResult.error as NodeJS.ErrnoException).code !== 'ENOENT') {
|
|
// 只在文件存在但无法加载时发出警告，文件不存在是正常情况
|
|
console.warn(`[ENV Init] Warning: Could not load data .env file from ${dataEnvPath}. Error: ${dataConfigResult.error.message}`);
|
|
} else if (!dataConfigResult.error) {
|
|
console.log(`[ENV Init] Loaded environment variables from data .env file: ${dataEnvPath}`);
|
|
}
|
|
|
|
// 2. 检查 ENCRYPTION_KEY
|
|
if (!process.env.ENCRYPTION_KEY) {
|
|
console.log('[ENV Init] ENCRYPTION_KEY 未设置，正在生成...');
|
|
const newEncryptionKey = crypto.randomBytes(32).toString('hex');
|
|
process.env.ENCRYPTION_KEY = newEncryptionKey; // 更新当前进程环境
|
|
keysToAppend += `\nENCRYPTION_KEY=${newEncryptionKey}`;
|
|
keysGenerated = true;
|
|
}
|
|
|
|
// 3. 检查 SESSION_SECRET
|
|
if (!process.env.SESSION_SECRET) {
|
|
console.log('[ENV Init] SESSION_SECRET 未设置，正在生成...');
|
|
const newSessionSecret = crypto.randomBytes(64).toString('hex');
|
|
process.env.SESSION_SECRET = newSessionSecret; // 更新当前进程环境
|
|
keysToAppend += `\nSESSION_SECRET=${newSessionSecret}`;
|
|
keysGenerated = true;
|
|
}
|
|
|
|
// 4. 检查 GUACD_HOST 和 GUACD_PORT
|
|
if (!process.env.GUACD_HOST) {
|
|
console.warn('[ENV Init] GUACD_HOST 未设置，将使用默认值 "localhost"');
|
|
process.env.GUACD_HOST = 'localhost';
|
|
// Optionally add to keysToAppend if you want to save the default
|
|
// keysToAppend += `\nGUACD_HOST=localhost`;
|
|
// keysGenerated = true; // Mark if you want to save
|
|
}
|
|
if (!process.env.GUACD_PORT) {
|
|
console.warn('[ENV Init] GUACD_PORT 未设置，将使用默认值 "4822"');
|
|
process.env.GUACD_PORT = '4822';
|
|
// Optionally add to keysToAppend
|
|
// keysToAppend += `\nGUACD_PORT=4822`;
|
|
// keysGenerated = true; // Mark if you want to save
|
|
}
|
|
|
|
|
|
// 5. 如果生成了新密钥或添加了默认值，则追加到 .env 文件
|
|
if (keysGenerated) {
|
|
try {
|
|
// 确保追加前有换行符 (如果文件非空) - Use dataEnvPath here
|
|
let prefix = '';
|
|
if (fs.existsSync(dataEnvPath)) { // Use dataEnvPath
|
|
const content = fs.readFileSync(dataEnvPath, 'utf-8'); // Use dataEnvPath
|
|
if (content.trim().length > 0 && !content.endsWith('\n')) {
|
|
prefix = '\n';
|
|
}
|
|
}
|
|
fs.appendFileSync(dataEnvPath, prefix + keysToAppend.trim()); // Use dataEnvPath, trim() 移除开头的换行符
|
|
console.warn(`[ENV Init] 已自动生成密钥并保存到 ${dataEnvPath}`); // Use dataEnvPath
|
|
console.warn('[ENV Init] !!! 重要：请务必备份此 data/.env 文件，并在生产环境中妥善保管 !!!');
|
|
} catch (error) {
|
|
console.error(`[ENV Init] 无法写入密钥到 ${dataEnvPath}:`, error); // Use dataEnvPath
|
|
console.error('[ENV Init] 请检查文件权限或手动创建 data/.env 文件并添加生成的密钥。');
|
|
// 即使写入失败，密钥已在 process.env 中，程序可以继续运行本次
|
|
}
|
|
}
|
|
|
|
// 5. 生产环境最终检查 (虽然理论上已被覆盖，但作为保险)
|
|
if (process.env.NODE_ENV === 'production') {
|
|
if (!process.env.ENCRYPTION_KEY) {
|
|
console.error('错误：生产环境中 ENCRYPTION_KEY 最终未能设置！');
|
|
process.exit(1);
|
|
}
|
|
if (!process.env.SESSION_SECRET) {
|
|
console.error('错误：生产环境中 SESSION_SECRET 最终未能设置！');
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
// 6. 最终检查 (包括 Guacamole 相关)
|
|
if (process.env.NODE_ENV === 'production') {
|
|
if (!process.env.ENCRYPTION_KEY) {
|
|
console.error('错误：生产环境中 ENCRYPTION_KEY 最终未能设置！');
|
|
process.exit(1);
|
|
}
|
|
if (!process.env.SESSION_SECRET) {
|
|
console.error('错误：生产环境中 SESSION_SECRET 最终未能设置！');
|
|
process.exit(1);
|
|
}
|
|
// Guacd host/port are less critical to halt on, defaults might work
|
|
}
|
|
|
|
};
|
|
// --- 结束环境变量和密钥初始化 ---
|
|
|
|
|
|
// 基础 Express 应用设置
|
|
const app = express();
|
|
const server = http.createServer(app);
|
|
|
|
// --- 信任代理设置 ---
|
|
app.set('trust proxy', true);
|
|
|
|
// --- 中间件 ---
|
|
app.use(ipWhitelistMiddleware as RequestHandler);
|
|
app.use(express.json());
|
|
|
|
// --- 静态文件服务 ---
|
|
const uploadsPath = path.join(__dirname, '../uploads');
|
|
if (!fs.existsSync(uploadsPath)) { // 确保 uploads 目录存在
|
|
fs.mkdirSync(uploadsPath, { recursive: true });
|
|
}
|
|
// app.use('/uploads', express.static(uploadsPath)); // 不再需要，文件通过 API 提供
|
|
|
|
|
|
// 扩展 Express Request 类型
|
|
declare module 'express-session' {
|
|
interface SessionData {
|
|
userId?: number;
|
|
username?: string;
|
|
}
|
|
}
|
|
|
|
const port = process.env.PORT || 3001;
|
|
|
|
// 初始化数据库
|
|
const initializeDatabase = async () => {
|
|
try {
|
|
const db = await getDbInstance();
|
|
console.log('[Index] 正在检查用户数量...');
|
|
const userCount = await new Promise<number>((resolve, reject) => {
|
|
db.get('SELECT COUNT(*) as count FROM users', (err: Error | null, row: { count: number }) => {
|
|
if (err) {
|
|
console.error('检查 users 表时出错:', err.message);
|
|
return reject(err);
|
|
}
|
|
resolve(row.count);
|
|
});
|
|
});
|
|
console.log(`[Index] 用户数量检查完成。找到 ${userCount} 个用户。`);
|
|
} catch (error) {
|
|
console.error('数据库初始化或检查失败:', error);
|
|
process.exit(1);
|
|
}
|
|
};
|
|
|
|
// 启动服务器
|
|
const startServer = () => {
|
|
// --- 会话中间件配置 ---
|
|
const FileStore = sessionFileStore(session);
|
|
// 修改路径以匹配 Docker volume 挂载点 /app/data
|
|
const sessionsPath = path.join('/app/data', 'sessions');
|
|
if (!fs.existsSync(sessionsPath)) {
|
|
fs.mkdirSync(sessionsPath, { recursive: true });
|
|
}
|
|
const sessionMiddleware = session({
|
|
store: new FileStore({
|
|
path: sessionsPath,
|
|
ttl: 31536000, // 1 year
|
|
// logFn: console.log // 可选：启用详细日志
|
|
}),
|
|
// 直接从 process.env 读取，initializeEnvironment 已确保其存在
|
|
secret: process.env.SESSION_SECRET as string,
|
|
resave: false,
|
|
saveUninitialized: false,
|
|
proxy: true, // 信任反向代理设置的 X-Forwarded-Proto 头
|
|
cookie: {
|
|
httpOnly: true,
|
|
}
|
|
});
|
|
app.use(sessionMiddleware);
|
|
// --- 结束会话中间件配置 ---
|
|
|
|
|
|
// --- 应用 API 路由 ---
|
|
app.use('/api/v1/auth', authRouter);
|
|
app.use('/api/v1/connections', connectionsRouter);
|
|
app.use('/api/v1/sftp', sftpRouter);
|
|
app.use('/api/v1/proxies', proxyRoutes);
|
|
app.use('/api/v1/tags', tagsRouter);
|
|
app.use('/api/v1/settings', settingsRoutes);
|
|
app.use('/api/v1/notifications', notificationRoutes);
|
|
app.use('/api/v1/audit-logs', auditRoutes);
|
|
app.use('/api/v1/command-history', commandHistoryRoutes);
|
|
app.use('/api/v1/quick-commands', quickCommandsRoutes);
|
|
app.use('/api/v1/terminal-themes', terminalThemeRoutes);
|
|
app.use('/api/v1/appearance', appearanceRoutes);
|
|
app.use('/api/v1/ssh-keys', sshKeysRouter); // +++ Register SSH Key routes +++
|
|
app.use('/api/v1/quick-command-tags', quickCommandTagRoutes); // +++ Register Quick Command Tag routes +++
|
|
|
|
// 状态检查接口
|
|
app.get('/api/v1/status', (req: Request, res: Response) => {
|
|
res.json({ status: '后端服务运行中！' });
|
|
});
|
|
// --- 结束 API 路由 ---
|
|
|
|
|
|
server.listen(port, () => {
|
|
console.log(`后端服务器正在监听 http://localhost:${port}`);
|
|
initializeWebSocket(server, sessionMiddleware as RequestHandler); // Initialize existing WebSocket
|
|
|
|
});
|
|
};
|
|
|
|
// --- 主程序启动流程 ---
|
|
const main = async () => {
|
|
await initializeEnvironment(); // 首先初始化环境和密钥
|
|
await initializeDatabase(); // 然后初始化数据库
|
|
startServer(); // 最后启动服务器
|
|
};
|
|
|
|
main().catch(error => {
|
|
console.error("启动过程中发生未处理的错误:", error);
|
|
process.exit(1);
|
|
});
|