83 lines
2.1 KiB
JavaScript
83 lines
2.1 KiB
JavaScript
|
const jwt = require("jsonwebtoken");
|
|||
|
|
const logger = require("../utils/logger");
|
|||
|
|
const { sendError } = require("../utils/response");
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* JWT 认证中间件
|
|||
|
|
* 验证请求头中的 token，如果有效则允许访问，否则返回 401
|
|||
|
|
* 支持从 Authorization header 或 URL 查询参数（token）中获取 token
|
|||
|
|
*/
|
|||
|
|
const jwtAuth = (req, res, next) => {
|
|||
|
|
// 优先从请求头中获取 token
|
|||
|
|
let token = null;
|
|||
|
|
const authHeader = req.headers.authorization;
|
|||
|
|
|
|||
|
|
if (authHeader) {
|
|||
|
|
// 支持两种格式: "Bearer token" 或 "token"
|
|||
|
|
token = authHeader.startsWith("Bearer ")
|
|||
|
|
? authHeader.substring(7)
|
|||
|
|
: authHeader;
|
|||
|
|
} else if (req.query && req.query.token) {
|
|||
|
|
// 如果请求头中没有，尝试从 URL 查询参数中获取（用于文件下载等场景）
|
|||
|
|
token = req.query.token;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!token) {
|
|||
|
|
return sendError(res, "未提供认证令牌", "No token provided", 401);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const secret = process.env.JWT_SECRET || "your-secret-key";
|
|||
|
|
|
|||
|
|
jwt.verify(token, secret, (err, decoded) => {
|
|||
|
|
if (err) {
|
|||
|
|
if (err.name === "TokenExpiredError") {
|
|||
|
|
return sendError(res, "令牌已过期", "Token expired", 401);
|
|||
|
|
}
|
|||
|
|
if (err.name === "JsonWebTokenError") {
|
|||
|
|
return sendError(res, "无效的令牌", "Invalid token", 401);
|
|||
|
|
}
|
|||
|
|
logger.error("Token验证失败:", err);
|
|||
|
|
return sendError(res, "令牌验证失败", "Token verification failed", 401);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 将解码后的用户信息添加到请求对象
|
|||
|
|
req.user = decoded;
|
|||
|
|
next();
|
|||
|
|
});
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 可选的 JWT 认证中间件
|
|||
|
|
* 如果提供了有效 token 则解析用户信息，否则继续执行（不阻止请求）
|
|||
|
|
*/
|
|||
|
|
const optionalJwtAuth = (req, res, next) => {
|
|||
|
|
const authHeader = req.headers.authorization;
|
|||
|
|
|
|||
|
|
if (!authHeader) {
|
|||
|
|
return next();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const token = authHeader.startsWith("Bearer ")
|
|||
|
|
? authHeader.substring(7)
|
|||
|
|
: authHeader;
|
|||
|
|
|
|||
|
|
if (!token) {
|
|||
|
|
return next();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const secret = process.env.JWT_SECRET || "your-secret-key";
|
|||
|
|
|
|||
|
|
jwt.verify(token, secret, (err, decoded) => {
|
|||
|
|
if (!err) {
|
|||
|
|
req.user = decoded;
|
|||
|
|
}
|
|||
|
|
next();
|
|||
|
|
});
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
module.exports = {
|
|||
|
|
jwtAuth,
|
|||
|
|
optionalJwtAuth,
|
|||
|
|
};
|
|||
|
|
|