更新政府端和银行端

bank-backend/middleware/auth.js

@@ -0,0 +1,226 @@
+/**
+ * 认证中间件
+ * @file auth.js
+ * @description 处理用户认证和授权
+ */
+const jwt = require('jsonwebtoken');
+const { User, Role } = require('../models');
+
+/**
+ * 验证JWT令牌
+ * @param {Object} req 请求对象
+ * @param {Object} res 响应对象
+ * @param {Function} next 下一个中间件
+ */
+const verifyToken = async (req, res, next) => {
+  try {
+    const token = req.header('Authorization')?.replace('Bearer ', '');
+    
+    if (!token) {
+      return res.status(401).json({
+        success: false,
+        message: '访问被拒绝，未提供令牌'
+      });
+    }
+
+    const decoded = jwt.verify(token, process.env.JWT_SECRET);
+    const user = await User.findByPk(decoded.id, {
+      include: [{
+        model: Role,
+        as: 'role'
+      }]
+    });
+
+    if (!user) {
+      return res.status(401).json({
+        success: false,
+        message: '令牌无效，用户不存在'
+      });
+    }
+
+    if (user.status !== 'active') {
+      return res.status(401).json({
+        success: false,
+        message: '账户已被禁用'
+      });
+    }
+
+    req.user = user;
+    next();
+  } catch (error) {
+    if (error.name === 'TokenExpiredError') {
+      return res.status(401).json({
+        success: false,
+        message: '令牌已过期'
+      });
+    } else if (error.name === 'JsonWebTokenError') {
+      return res.status(401).json({
+        success: false,
+        message: '令牌无效'
+      });
+    } else {
+      console.error('认证中间件错误:', error);
+      return res.status(500).json({
+        success: false,
+        message: '服务器内部错误'
+      });
+    }
+  }
+};
+
+/**
+ * 检查用户角色权限
+ * @param {String|Array} roles 允许的角色
+ * @returns {Function} 中间件函数
+ */
+const requireRole = (roles) => {
+  return async (req, res, next) => {
+    try {
+      if (!req.user) {
+        return res.status(401).json({
+          success: false,
+          message: '请先登录'
+        });
+      }
+
+      const userRole = req.user.role;
+      if (!userRole) {
+        return res.status(403).json({
+          success: false,
+          message: '用户角色未分配'
+        });
+      }
+
+      const allowedRoles = Array.isArray(roles) ? roles : [roles];
+      if (!allowedRoles.includes(userRole.name)) {
+        return res.status(403).json({
+          success: false,
+          message: '权限不足'
+        });
+      }
+
+      next();
+    } catch (error) {
+      console.error('角色权限检查错误:', error);
+      return res.status(500).json({
+        success: false,
+        message: '服务器内部错误'
+      });
+    }
+  };
+};
+
+/**
+ * 检查用户权限级别
+ * @param {Number} minLevel 最小权限级别
+ * @returns {Function} 中间件函数
+ */
+const requireLevel = (minLevel) => {
+  return async (req, res, next) => {
+    try {
+      if (!req.user) {
+        return res.status(401).json({
+          success: false,
+          message: '请先登录'
+        });
+      }
+
+      const userRole = req.user.role;
+      if (!userRole || userRole.level < minLevel) {
+        return res.status(403).json({
+          success: false,
+          message: '权限级别不足'
+        });
+      }
+
+      next();
+    } catch (error) {
+      console.error('权限级别检查错误:', error);
+      return res.status(500).json({
+        success: false,
+        message: '服务器内部错误'
+      });
+    }
+  };
+};
+
+/**
+ * 可选认证中间件（不强制要求登录）
+ * @param {Object} req 请求对象
+ * @param {Object} res 响应对象
+ * @param {Function} next 下一个中间件
+ */
+const optionalAuth = async (req, res, next) => {
+  try {
+    const token = req.header('Authorization')?.replace('Bearer ', '');
+    
+    if (token) {
+      const decoded = jwt.verify(token, process.env.JWT_SECRET);
+      const user = await User.findByPk(decoded.id, {
+        include: [{
+          model: Role,
+          as: 'role'
+        }]
+      });
+
+      if (user && user.status === 'active') {
+        req.user = user;
+      }
+    }
+
+    next();
+  } catch (error) {
+    // 可选认证失败时不返回错误，继续执行
+    next();
+  }
+};
+
+/**
+ * 检查账户所有权
+ * @param {Object} req 请求对象
+ * @param {Object} res 响应对象
+ * @param {Function} next 下一个中间件
+ */
+const checkAccountOwnership = async (req, res, next) => {
+  try {
+    const { accountId } = req.params;
+    const userId = req.user.id;
+
+    // 管理员可以访问所有账户
+    if (req.user.role && req.user.role.name === 'admin') {
+      return next();
+    }
+
+    const { Account } = require('../models');
+    const account = await Account.findOne({
+      where: {
+        id: accountId,
+        user_id: userId
+      }
+    });
+
+    if (!account) {
+      return res.status(403).json({
+        success: false,
+        message: '无权访问该账户'
+      });
+    }
+
+    req.account = account;
+    next();
+  } catch (error) {
+    console.error('账户所有权检查错误:', error);
+    return res.status(500).json({
+      success: false,
+      message: '服务器内部错误'
+    });
+  }
+};
+
+module.exports = {
+  verifyToken,
+  requireRole,
+  requireLevel,
+  optionalAuth,
+  checkAccountOwnership
+};