update

2025-04-14 22:51:05 +08:00
parent 286492fc63
commit a974b8b1d9
49 changed files with 13954 additions and 0 deletions
@@ -0,0 +1,157 @@
+import express = require('express');
+// import express = require('express'); // 移除重复导入
+import { Request, Response, NextFunction, RequestHandler } from 'express'; // 添加 RequestHandler
+import http from 'http'; // 引入 http 模块
+import session from 'express-session';
+import connectSqlite3 from 'connect-sqlite3';
+import path from 'path'; // 需要 path 模块
+import bcrypt from 'bcrypt'; // 引入 bcrypt 用于哈希密码
+import { getDb } from './database';
+import { runMigrations } from './migrations';
+import authRouter from './auth/auth.routes'; // 导入认证路由
+import connectionsRouter from './connections/connections.routes'; // 导入连接路由
+import sftpRouter from './sftp/sftp.routes'; // 导入 SFTP 路由
+import { initializeWebSocket } from './websocket'; // 导入 WebSocket 初始化函数
+
+// 基础 Express 应用设置 (后续会扩展)
+const app = express();
+const server = http.createServer(app); // 创建 HTTP 服务器实例
+
+// --- 会话存储设置 ---
+const SQLiteStore = connectSqlite3(session);
+const dbPath = path.resolve(__dirname, '../../data'); // 数据库目录路径
+
+// --- 中间件 ---
+app.use(express.json()); // 添加此行以解析 JSON 请求体
+
+// 会话中间件配置
+// TODO: 将 secret 移到环境变量中，不要硬编码在代码里！
+const sessionSecret = process.env.SESSION_SECRET || 'a-very-insecure-secret-for-dev';
+if (sessionSecret === 'a-very-insecure-secret-for-dev') {
+    console.warn('警告：正在使用默认的不安全会话密钥，请在生产环境中设置 SESSION_SECRET 环境变量！');
+}
+
+app.use(session({
+    // 使用类型断言 (as any) 来解决 @types/connect-sqlite3 和 @types/express-session 的类型冲突
+    store: new SQLiteStore({
+        db: 'nexus-terminal.db', // 数据库文件名
+        dir: dbPath,          // 数据库文件所在目录
+        table: 'sessions'     // 存储会话的表名 (会自动创建)
+    }) as any,
+    secret: sessionSecret,
+    resave: false,             // 强制保存 session 即使它没有变化 (通常为 false)
+    saveUninitialized: false,  // 强制将未初始化的 session 存储 (通常为 false)
+    cookie: {
+        maxAge: 1000 * 60 * 60 * 24 * 7, // Cookie 有效期：7天 (毫秒)
+        httpOnly: true,                 // 防止客户端脚本访问 cookie
+        secure: process.env.NODE_ENV === 'production' // 仅在 HTTPS 下发送 cookie (生产环境)
+    }
+}));
+// 将 session 中间件保存到一个变量，以便传递给 WebSocket 初始化函数
+const sessionMiddleware = session({
+    // 使用类型断言 (as any) 来解决 @types/connect-sqlite3 和 @types/express-session 的类型冲突
+    store: new SQLiteStore({
+        db: 'nexus-terminal.db', // 数据库文件名
+        dir: dbPath,          // 数据库文件所在目录
+        table: 'sessions'     // 存储会话的表名 (会自动创建)
+    }) as any,
+    secret: sessionSecret,
+    resave: false,             // 强制保存 session 即使它没有变化 (通常为 false)
+    saveUninitialized: false,  // 强制将未初始化的 session 存储 (通常为 false)
+    cookie: {
+        maxAge: 1000 * 60 * 60 * 24 * 7, // Cookie 有效期：7天 (毫秒)
+        httpOnly: true,                 // 防止客户端脚本访问 cookie
+        secure: process.env.NODE_ENV === 'production' // 仅在 HTTPS 下发送 cookie (生产环境)
+    }
+});
+app.use(sessionMiddleware); // 应用会话中间件
+
+
+// 扩展 Express Request 类型以包含 session 数据 (如果需要更明确的类型提示)
+declare module 'express-session' {
+    interface SessionData {
+        userId?: number; // 存储登录用户的 ID
+        username?: string;
+    }
+}
+
+
+const port = process.env.PORT || 3001; // 示例端口，可配置
+
+// --- API 路由 ---
+app.use('/api/v1/auth', authRouter); // 挂载认证相关的路由
+app.use('/api/v1/connections', connectionsRouter); // 挂载连接相关的路由
+app.use('/api/v1/sftp', sftpRouter); // 挂载 SFTP 相关的路由
+
+// 状态检查接口
+app.get('/api/v1/status', (req: Request, res: Response) => {
+  res.json({ status: '后端服务运行中！' }); // 响应也改为中文
+});
+
+// 在服务器启动前初始化数据库并执行迁移
+const initializeDatabase = async () => {
+  try {
+    const db = getDb(); // 获取数据库实例 (同时会建立连接)
+    await runMigrations(db); // 执行数据库迁移 (创建表)
+    // console.log('数据库迁移执行成功。'); // 日志已移至 migrations.ts
+
+    // 检查管理员用户是否存在
+    const userCount = await new Promise<number>((resolve, reject) => {
+      db.get('SELECT COUNT(*) as count FROM users', (err, row: { count: number }) => { // 查询用户数量
+        if (err) {
+          console.error('检查 users 表时出错:', err.message);
+          return reject(err);
+        }
+        resolve(row.count);
+      });
+    });
+
+    if (userCount === 0) {
+      console.warn('------------------------------------------------------');
+      console.warn('警告: 数据库中未找到任何用户。正在创建默认管理员...');
+      // 创建默认管理员
+      const defaultAdminUsername = 'admin';
+      const defaultAdminPassword = 'adminpassword'; // 仅用于首次创建
+      const saltRounds = 10;
+      const hashedPassword = await bcrypt.hash(defaultAdminPassword, saltRounds);
+      const now = Math.floor(Date.now() / 1000);
+
+      await new Promise<void>((resolveInsert, rejectInsert) => {
+          const stmt = db.prepare(
+              `INSERT INTO users (username, hashed_password, created_at, updated_at)
+               VALUES (?, ?, ?, ?)`
+          );
+          stmt.run(defaultAdminUsername, hashedPassword, now, now, function (err: Error | null) {
+              if (err) {
+                  console.error('创建默认管理员时出错:', err.message);
+                  return rejectInsert(new Error('创建默认管理员失败'));
+              }
+              console.log(`默认管理员 '${defaultAdminUsername}' (密码: '${defaultAdminPassword}') 已创建。请尽快修改密码！`);
+              resolveInsert();
+          });
+          stmt.finalize();
+      });
+      console.warn('------------------------------------------------------');
+
+    } else {
+      console.log(`数据库中找到 ${userCount} 个用户。`);
+    }
+
+    console.log('数据库初始化检查完成。');
+  } catch (error) {
+    console.error('数据库初始化失败:', error);
+    process.exit(1); // 如果数据库初始化失败，则退出进程
+  }
+};
+
+// 启动 HTTP 服务器 (而不是直接 app.listen)
+const startServer = () => {
+  server.listen(port, () => { // 使用 server.listen
+    console.log(`后端服务器正在监听 http://localhost:${port}`);
+    // 初始化 WebSocket 服务器，并传入 HTTP 服务器实例和会话解析器
+    initializeWebSocket(server, sessionMiddleware as RequestHandler);
+  });
+};
+
+// 先执行数据库初始化，成功后再启动服务器
+initializeDatabase().then(startServer);