46 lines
1.2 KiB
JavaScript
46 lines
1.2 KiB
JavaScript
const jwt = require('jsonwebtoken');
|
|
const db = require('../utils/database');
|
|
|
|
// 验证JWT令牌
|
|
const authenticateToken = async (req, res, next) => {
|
|
// 从请求头获取令牌
|
|
const authHeader = req.headers['authorization'];
|
|
const token = authHeader && authHeader.split(' ')[1];
|
|
|
|
if (!token) {
|
|
return res.status(401).json({ error: '访问令牌缺失' });
|
|
}
|
|
|
|
try {
|
|
// 验证令牌
|
|
const decoded = jwt.verify(token, process.env.JWT_SECRET);
|
|
|
|
// 获取用户信息
|
|
const user = await db.get('SELECT id, username, name, role FROM users WHERE id = ?', [decoded.userId]);
|
|
|
|
if (!user) {
|
|
return res.status(401).json({ error: '用户不存在' });
|
|
}
|
|
|
|
req.user = user;
|
|
next();
|
|
} catch (error) {
|
|
if (error.name === 'TokenExpiredError') {
|
|
return res.status(401).json({ error: '令牌已过期' });
|
|
}
|
|
return res.status(403).json({ error: '无效的令牌' });
|
|
}
|
|
};
|
|
|
|
// 验证管理员权限
|
|
const requireAdmin = (req, res, next) => {
|
|
if (req.user.role !== 'admin') {
|
|
return res.status(403).json({ error: '需要管理员权限' });
|
|
}
|
|
next();
|
|
};
|
|
|
|
module.exports = {
|
|
authenticateToken,
|
|
requireAdmin
|
|
}; |