import { Request, Response } from 'express'; import bcrypt from 'bcrypt'; import { getDbInstance, runDb, getDb, allDb } from '../database/connection'; import speakeasy from 'speakeasy'; import qrcode from 'qrcode'; import { NotificationService } from '../services/notification.service'; import { AuditLogService } from '../services/audit.service'; import { ipBlacklistService } from '../services/ip-blacklist.service'; import { captchaService } from '../services/captcha.service'; import { settingsService } from '../services/settings.service'; const notificationService = new NotificationService(); const auditLogService = new AuditLogService(); export interface User { // Add export keyword id: number; username: string; hashed_password: string; two_factor_secret?: string | null; } declare module 'express-session' { interface SessionData { userId?: number; username?: string; tempTwoFactorSecret?: string; requiresTwoFactor?: boolean; // currentChallenge?: string; // Removed Passkey challenge storage rememberMe?: boolean; } } /** * 处理用户登录请求 (POST /api/v1/auth/login) */ export const login = async (req: Request, res: Response): Promise => { // 从请求体中解构 username, password 和可选的 rememberMe const { username, password, rememberMe } = req.body; if (!username || !password) { res.status(400).json({ message: '用户名和密码不能为空。' }); return; } try { // --- CAPTCHA Verification Step --- const captchaConfig = await settingsService.getCaptchaConfig(); if (captchaConfig.enabled) { const { captchaToken } = req.body; if (!captchaToken) { res.status(400).json({ message: '需要提供 CAPTCHA 令牌。' }); return; } try { const isCaptchaValid = await captchaService.verifyToken(captchaToken); if (!isCaptchaValid) { console.log(`[AuthController] 登录尝试失败: CAPTCHA 验证失败 - ${username}`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; ipBlacklistService.recordFailedAttempt(clientIp); auditLogService.logAction('LOGIN_FAILURE', { username, reason: 'Invalid CAPTCHA token', ip: clientIp }); notificationService.sendNotification('LOGIN_FAILURE', { username, reason: 'Invalid CAPTCHA token', ip: clientIp }); // 取消注释 res.status(401).json({ message: 'CAPTCHA 验证失败。' }); return; } console.log(`[AuthController] CAPTCHA 验证成功 - ${username}`); } catch (captchaError: any) { console.error(`[AuthController] CAPTCHA 验证过程中出错 (${username}):`, captchaError.message); res.status(500).json({ message: 'CAPTCHA 验证服务出错,请稍后重试或检查配置。' }); return; } } else { console.log(`[AuthController] CAPTCHA 未启用,跳过验证 - ${username}`); } const db = await getDbInstance(); const user = await getDb(db, 'SELECT id, username, hashed_password, two_factor_secret FROM users WHERE username = ?', [username]); if (!user) { console.log(`登录尝试失败: 用户未找到 - ${username}`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; // 记录失败尝试 ipBlacklistService.recordFailedAttempt(clientIp); // 记录审计日志 (添加 IP) auditLogService.logAction('LOGIN_FAILURE', { username, reason: 'User not found', ip: clientIp }); // 发送登录失败通知 notificationService.sendNotification('LOGIN_FAILURE', { username, reason: 'User not found', ip: clientIp }); // 取消注释 res.status(401).json({ message: '无效的凭据。' }); return; } const isMatch = await bcrypt.compare(password, user.hashed_password); if (!isMatch) { console.log(`登录尝试失败: 密码错误 - ${username}`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; // 获取客户端 IP // 记录失败尝试 ipBlacklistService.recordFailedAttempt(clientIp); // 记录审计日志 (添加 IP) auditLogService.logAction('LOGIN_FAILURE', { username, reason: 'Invalid password', ip: clientIp }); // 发送登录失败通知 notificationService.sendNotification('LOGIN_FAILURE', { username, reason: 'Invalid password', ip: clientIp }); // 取消注释 res.status(401).json({ message: '无效的凭据。' }); return; } // 检查是否启用了 2FA if (user.two_factor_secret) { console.log(`用户 ${username} 已启用 2FA,需要进行二次验证。`); // 不设置完整 session,只标记需要 2FA req.session.userId = user.id; // 临时存储 userId 以便 2FA 验证 req.session.requiresTwoFactor = true; req.session.rememberMe = rememberMe; // 临时存储 rememberMe 状态 res.status(200).json({ message: '需要进行两步验证。', requiresTwoFactor: true }); } else { // --- 认证成功 (未启用 2FA) --- console.log(`登录成功 (无 2FA): ${username}`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; // 获取客户端 IP // 重置失败尝试次数 ipBlacklistService.resetAttempts(clientIp); // 记录审计日志 (添加 IP) auditLogService.logAction('LOGIN_SUCCESS', { userId: user.id, username, ip: clientIp }); notificationService.sendNotification('LOGIN_SUCCESS', { userId: user.id, username, ip: clientIp }); // 添加通知调用 req.session.userId = user.id; req.session.username = user.username; req.session.requiresTwoFactor = false; // 明确标记不需要 2FA // 根据 rememberMe 设置 cookie maxAge if (rememberMe) { // 如果勾选了“记住我”,设置 cookie 有效期为 10 年 (毫秒) req.session.cookie.maxAge = 315360000000; } else { // 如果未勾选,则不设置 maxAge,使其成为会话 cookie req.session.cookie.maxAge = undefined; } res.status(200).json({ message: '登录成功。', user: { id: user.id, username: user.username } }); } } catch (error) { console.error('登录时出错:', error); res.status(500).json({ message: '登录过程中发生内部服务器错误。' }); } }; /** * 获取当前用户的认证状态 (GET /api/v1/auth/status) */ export const getAuthStatus = async (req: Request, res: Response): Promise => { const userId = req.session.userId; const username = req.session.username; if (!userId || !username || req.session.requiresTwoFactor) { // 如果 session 无效或 2FA 未完成,视为未认证 res.status(401).json({ isAuthenticated: false }); return; } try { const db = await getDbInstance(); // Get DB instance // 查询用户的 2FA 状态 using promisified getDb const user = await getDb<{ two_factor_secret: string | null }>(db, 'SELECT two_factor_secret FROM users WHERE id = ?', [userId]); // 如果找不到用户,也视为未认证 if (!user) { res.status(401).json({ isAuthenticated: false }); return; } res.status(200).json({ isAuthenticated: true, user: { id: userId, username: username, isTwoFactorEnabled: !!user.two_factor_secret // 返回 2FA 是否启用 } }); } catch (error) { console.error(`获取用户 ${userId} 状态时发生内部错误:`, error); res.status(500).json({ message: '获取认证状态时发生内部服务器错误。' }); } }; /** * 处理登录时的 2FA 验证 (POST /api/v1/auth/login/2fa) */ export const verifyLogin2FA = async (req: Request, res: Response): Promise => { const { token } = req.body; const userId = req.session.userId; // 获取之前临时存储的 userId // 检查 session 状态 if (!userId || !req.session.requiresTwoFactor) { res.status(400).json({ message: '无效的请求或会话状态。' }); return; } if (!token) { res.status(400).json({ message: '验证码不能为空。' }); return; } try { const db = await getDbInstance(); // 获取用户的 2FA 密钥 using promisified getDb const user = await getDb(db, 'SELECT id, username, two_factor_secret FROM users WHERE id = ?', [userId]); if (!user || !user.two_factor_secret) { console.error(`2FA 验证错误: 未找到用户 ${userId} 或未设置密钥。`); res.status(400).json({ message: '无法验证,请重新登录。' }); return; } // 验证 TOTP 令牌 const verified = speakeasy.totp.verify({ secret: user.two_factor_secret, encoding: 'base32', token: token, window: 1 // 允许前后一个时间窗口 (30秒) 的容错 }); if (verified) { console.log(`用户 ${user.username} 2FA 验证成功。`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; // 获取客户端 IP // 重置失败尝试次数 ipBlacklistService.resetAttempts(clientIp); // 记录审计日志 (2FA 成功也算登录成功) (添加 IP) auditLogService.logAction('LOGIN_SUCCESS', { userId: user.id, username: user.username, ip: clientIp, twoFactor: true }); notificationService.sendNotification('LOGIN_SUCCESS', { userId: user.id, username: user.username, ip: clientIp, twoFactor: true }); // 添加通知调用 // 验证成功,建立完整会话 req.session.username = user.username; req.session.requiresTwoFactor = false; // 标记 2FA 已完成 // 根据之前存储在 session 中的 rememberMe 设置 cookie maxAge if (req.session.rememberMe) { // 如果勾选了“记住我”,设置 cookie 有效期为 1 年 (毫秒) req.session.cookie.maxAge = 315360000000; // 10 years (Effectively permanent) } else { // 如果未勾选,则不设置 maxAge,使其成为会话 cookie req.session.cookie.maxAge = undefined; // 或者 null } // 清除临时的 rememberMe 状态 delete req.session.rememberMe; res.status(200).json({ message: '登录成功。', user: { id: user.id, username: user.username } }); } else { console.log(`用户 ${user.username} 2FA 验证失败: 验证码错误。`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; // 获取客户端 IP // 记录失败尝试 ipBlacklistService.recordFailedAttempt(clientIp); // 记录审计日志 (添加 IP) auditLogService.logAction('LOGIN_FAILURE', { userId: user.id, username: user.username, reason: 'Invalid 2FA token', ip: clientIp }); notificationService.sendNotification('LOGIN_FAILURE', { userId: user.id, username: user.username, reason: 'Invalid 2FA token', ip: clientIp }); // 添加通知调用 res.status(401).json({ message: '验证码无效。' }); } } catch (error) { console.error(`用户 ${userId} 2FA 验证时发生内部错误:`, error); res.status(500).json({ message: '两步验证过程中发生内部服务器错误。' }); } }; /** * 处理修改密码请求 (PUT /api/v1/auth/password) */ export const changePassword = async (req: Request, res: Response): Promise => { const { currentPassword, newPassword } = req.body; const userId = req.session.userId; // 从会话中获取用户 ID // 检查用户是否登录且 2FA 已完成 (如果需要) if (!userId || req.session.requiresTwoFactor) { res.status(401).json({ message: '用户未认证或认证未完成,请先登录。' }); return; } // 基础输入验证 if (!currentPassword || !newPassword) { res.status(400).json({ message: '当前密码和新密码不能为空。' }); return; } if (newPassword.length < 8) { res.status(400).json({ message: '新密码长度至少需要 8 位。' }); return; } if (currentPassword === newPassword) { res.status(400).json({ message: '新密码不能与当前密码相同。' }); return; } try { const db = await getDbInstance(); const user = await getDb(db, 'SELECT id, hashed_password FROM users WHERE id = ?', [userId]); if (!user) { console.error(`修改密码错误: 未找到 ID 为 ${userId} 的用户。`); res.status(404).json({ message: '用户不存在。' }); return; } const isMatch = await bcrypt.compare(currentPassword, user.hashed_password); if (!isMatch) { console.log(`修改密码尝试失败: 当前密码错误 - 用户 ID ${userId}`); res.status(400).json({ message: '当前密码不正确。' }); return; } const saltRounds = 10; const newHashedPassword = await bcrypt.hash(newPassword, saltRounds); const now = Math.floor(Date.now() / 1000); const result = await runDb(db, 'UPDATE users SET hashed_password = ?, updated_at = ? WHERE id = ?', [newHashedPassword, now, userId] ); if (result.changes === 0) { console.error(`修改密码错误: 更新影响行数为 0 - 用户 ID ${userId}`); throw new Error('未找到要更新的用户'); } console.log(`用户 ${userId} 密码已成功修改。`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; // 获取客户端 IP // 记录审计日志 (添加 IP) auditLogService.logAction('PASSWORD_CHANGED', { userId, ip: clientIp }); notificationService.sendNotification('PASSWORD_CHANGED', { userId, ip: clientIp }); // 添加通知调用 res.status(200).json({ message: '密码已成功修改。' }); } catch (error) { console.error(`修改用户 ${userId} 密码时发生内部错误:`, error); res.status(500).json({ message: '修改密码过程中发生内部服务器错误。' }); } }; /** * 开始 2FA 设置流程 (POST /api/v1/auth/2fa/setup) * 生成临时密钥和二维码 */ export const setup2FA = async (req: Request, res: Response): Promise => { const userId = req.session.userId; const username = req.session.username; // 获取用户名用于 OTP URL if (!userId || !username || req.session.requiresTwoFactor) { res.status(401).json({ message: '用户未认证或认证未完成。' }); return; } try { const db = await getDbInstance(); // 检查用户是否已启用 2FA using promisified getDb const user = await getDb<{ two_factor_secret: string | null }>(db, 'SELECT two_factor_secret FROM users WHERE id = ?', [userId]); const existingSecret = user ? user.two_factor_secret : null; if (existingSecret) { res.status(400).json({ message: '两步验证已启用。如需重置,请先禁用。' }); return; } // 生成新的 2FA 密钥 const secret = speakeasy.generateSecret({ length: 20, name: `NexusTerminal (${username})` // 应用名称和用户名,显示在 Authenticator 应用中 }); // 将临时密钥存储在 session 中,等待验证 req.session.tempTwoFactorSecret = secret.base32; // 生成 OTP Auth URL (用于生成二维码) if (!secret.otpauth_url) { throw new Error('无法生成 OTP Auth URL'); } // 生成二维码 Data URL qrcode.toDataURL(secret.otpauth_url, (err, data_url) => { if (err) { console.error('生成二维码时出错:', err); throw new Error('生成二维码失败'); } // 返回密钥 (base32) 和二维码数据 URL 给前端 res.json({ secret: secret.base32, // 供用户手动输入 qrCodeUrl: data_url // 用于显示二维码图片 }); }); } catch (error: any) { console.error(`用户 ${userId} 设置 2FA 时出错:`, error); res.status(500).json({ message: '设置两步验证时发生错误。', error: error.message }); } }; /** * 验证并激活 2FA (POST /api/v1/auth/2fa/verify) */ export const verifyAndActivate2FA = async (req: Request, res: Response): Promise => { const { token } = req.body; const userId = req.session.userId; const tempSecret = req.session.tempTwoFactorSecret; // 获取存储在 session 中的临时密钥 if (!userId || req.session.requiresTwoFactor) { res.status(401).json({ message: '用户未认证或认证未完成。' }); return; } if (!tempSecret) { res.status(400).json({ message: '未找到临时密钥,请重新开始设置流程。' }); return; } if (!token) { res.status(400).json({ message: '验证码不能为空。' }); return; } try { const db = await getDbInstance(); // 使用临时密钥验证用户提交的令牌 const verified = speakeasy.totp.verify({ secret: tempSecret, encoding: 'base32', token: token, window: 1 // 允许一定的时钟漂移 }); if (verified) { // 验证成功,将密钥永久存储到数据库 using promisified runDb const now = Math.floor(Date.now() / 1000); const result = await runDb(db, 'UPDATE users SET two_factor_secret = ?, updated_at = ? WHERE id = ?', [tempSecret, now, userId] ); if (result.changes === 0) { console.error(`激活 2FA 错误: 更新影响行数为 0 - 用户 ID ${userId}`); throw new Error('未找到要更新的用户'); } console.log(`用户 ${userId} 已成功激活两步验证。`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; // 记录审计日志 (添加 IP) auditLogService.logAction('2FA_ENABLED', { userId, ip: clientIp }); notificationService.sendNotification('2FA_ENABLED', { userId, ip: clientIp }); // 添加通知调用 // 清除 session 中的临时密钥 delete req.session.tempTwoFactorSecret; res.status(200).json({ message: '两步验证已成功激活!' }); } else { // 验证失败 console.log(`用户 ${userId} 2FA 激活失败: 验证码错误。`); res.status(400).json({ message: '验证码无效。' }); } } catch (error: any) { console.error(`用户 ${userId} 验证并激活 2FA 时出错:`, error); res.status(500).json({ message: '验证两步验证码时发生错误。', error: error.message }); } }; /** * 禁用 2FA (DELETE /api/v1/auth/2fa) */ export const disable2FA = async (req: Request, res: Response): Promise => { const userId = req.session.userId; const { password } = req.body; // 需要验证当前密码以禁用 if (!userId || req.session.requiresTwoFactor) { res.status(401).json({ message: '用户未认证或认证未完成。' }); return; } if (!password) { res.status(400).json({ message: '需要提供当前密码才能禁用两步验证。' }); return; } try { const db = await getDbInstance(); // 验证当前密码 const user = await getDb(db, 'SELECT id, hashed_password FROM users WHERE id = ?', [userId]); if (!user) { res.status(404).json({ message: '用户不存在。' }); return; } const isMatch = await bcrypt.compare(password, user.hashed_password); if (!isMatch) { res.status(400).json({ message: '当前密码不正确。' }); return; } // 清除数据库中的 2FA 密钥 const now = Math.floor(Date.now() / 1000); const result = await runDb(db, 'UPDATE users SET two_factor_secret = NULL, updated_at = ? WHERE id = ?', [now, userId] ); if (result.changes === 0) { console.error(`禁用 2FA 错误: 更新影响行数为 0 - 用户 ID ${userId}`); throw new Error('未找到要更新的用户'); } console.log(`用户 ${userId} 已成功禁用两步验证。`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; // 记录审计日志 (添加 IP) auditLogService.logAction('2FA_DISABLED', { userId, ip: clientIp }); notificationService.sendNotification('2FA_DISABLED', { userId, ip: clientIp }); // 添加通知调用 res.status(200).json({ message: '两步验证已成功禁用。' }); } catch (error: any) { console.error(`用户 ${userId} 禁用 2FA 时出错:`, error); res.status(500).json({ message: '禁用两步验证时发生错误。', error: error.message }); } }; /** * 检查是否需要进行初始设置 (GET /api/v1/auth/needs-setup) * 如果数据库中没有用户,则需要设置。 */ export const needsSetup = async (req: Request, res: Response): Promise => { try { const db = await getDbInstance(); // Get DB instance // Use promisified getDb const row = await getDb<{ count: number }>(db, 'SELECT COUNT(*) as count FROM users'); const userCount = row ? row.count : 0; res.status(200).json({ needsSetup: userCount === 0 }); } catch (error) { console.error('检查设置状态时发生内部错误:', error); // 如果检查失败,保守起见返回 false,避免用户卡在设置页面 res.status(500).json({ message: '检查设置状态时发生错误。', needsSetup: false }); } }; /** * 处理初始管理员账号设置请求 (POST /api/v1/auth/setup) */ export const setupAdmin = async (req: Request, res: Response): Promise => { const { username, password, confirmPassword } = req.body; // 基本输入验证 if (!username || !password || !confirmPassword) { res.status(400).json({ message: '用户名、密码和确认密码不能为空。' }); return; } if (password !== confirmPassword) { res.status(400).json({ message: '两次输入的密码不匹配。' }); return; } if (password.length < 8) { res.status(400).json({ message: '密码长度至少需要 8 位。' }); return; } try { const db = await getDbInstance(); // 检查数据库中是否已存在用户 const row = await getDb<{ count: number }>(db, 'SELECT COUNT(*) as count FROM users'); const userCount = row ? row.count : 0; if (userCount > 0) { console.warn('尝试在已有用户的情况下执行初始设置。'); res.status(403).json({ message: '设置已完成,无法重复执行。' }); return; } // 哈希密码 const saltRounds = 10; const hashedPassword = await bcrypt.hash(password, saltRounds); const now = Math.floor(Date.now() / 1000); // 插入新用户 const result = await runDb(db, `INSERT INTO users (username, hashed_password, created_at, updated_at) VALUES (?, ?, ?, ?)`, [username, hashedPassword, now, now] ); if (typeof result.lastID !== 'number' || result.lastID <= 0) { console.error('创建初始管理员后未能获取有效的 lastID。可能原因:用户名已存在或其他数据库错误。'); throw new Error('创建初始管理员失败,可能用户名已存在。'); } const newUser = { id: result.lastID }; console.log(`初始管理员账号 '${username}' (ID: ${newUser.id}) 已成功创建。`); const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; // 记录审计日志 (添加 IP) auditLogService.logAction('ADMIN_SETUP_COMPLETE', { userId: newUser.id, username, ip: clientIp }); notificationService.sendNotification('ADMIN_SETUP_COMPLETE', { userId: newUser.id, username, ip: clientIp }); // 添加通知调用 res.status(201).json({ message: '初始管理员账号创建成功!' }); } catch (error: any) { console.error('初始设置过程中发生内部错误:', error); res.status(500).json({ message: error.message || '初始设置过程中发生内部服务器错误。' }); } }; /** * 处理用户登出请求 (POST /api/v1/auth/logout) */ export const logout = (req: Request, res: Response): void => { const userId = req.session.userId; const username = req.session.username; req.session.destroy((err) => { if (err) { console.error(`销毁用户 ${userId} (${username}) 的会话时出错:`, err); // 即使销毁失败,也尝试让前端认为已登出 res.status(500).json({ message: '登出时发生服务器内部错误。' }); } else { console.log(`用户 ${userId} (${username}) 已成功登出。`); // 清除客户端的 session cookie (通常 connect-sqlite3 会处理,但显式设置更保险) res.clearCookie('connect.sid'); // 'connect.sid' 是 express-session 的默认 cookie 名称 // 记录审计日志 if (userId) { // 仅在能获取到 userId 时记录 const clientIp = req.ip || req.socket?.remoteAddress || 'unknown'; auditLogService.logAction('LOGOUT', { userId, username, ip: clientIp }); notificationService.sendNotification('LOGOUT', { userId, username, ip: clientIp }); // 添加通知调用 } res.status(200).json({ message: '已成功登出。' }); } }); }; /** * 获取公共 CAPTCHA 配置 (GET /api/v1/auth/captcha/config) * 返回给前端用于显示 CAPTCHA 小部件所需的信息 (不含密钥)。 */ export const getPublicCaptchaConfig = async (req: Request, res: Response): Promise => { try { console.log('[AuthController] Received request for public CAPTCHA config.'); const fullConfig = await settingsService.getCaptchaConfig(); const publicConfig = { enabled: fullConfig.enabled, provider: fullConfig.provider, hcaptchaSiteKey: fullConfig.hcaptchaSiteKey, recaptchaSiteKey: fullConfig.recaptchaSiteKey, }; console.log('[AuthController] Sending public CAPTCHA config to client:', publicConfig); res.status(200).json(publicConfig); } catch (error: any) { console.error('[AuthController] 获取公共 CAPTCHA 配置时出错:', error); // 即使出错,也返回一个“禁用”状态,避免前端出错 res.status(500).json({ enabled: false, provider: 'none', hcaptchaSiteKey: '', recaptchaSiteKey: '', error: '获取 CAPTCHA 配置失败' }); } };