| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122 |
- const express = require('express');
- const bcrypt = require('bcryptjs');
- const jwt = require('jsonwebtoken');
- const router = express.Router();
- const authMiddleware = require('../middleware/authMiddleware');
- const User = require('../models/User');
- const Logs = require('../libs/logs');
- // 注册
- router.post('/add',authMiddleware, async (req, res) => {
- const { username, password } = req.body;
- User.add({ username, password })
- .then(() => {
- res.sendSuccess('添加成功');
- })
- .catch(err => {
- Logs.errDev('添加失败:', err);
- if (err.message === 'USER_EXISTS') {
- return res.badRequest('用户已存在');
- }
- res.badRequest(err.message);
- });
- });
- // 登录 - 支持原路径 /login 和 mock 服务路径 /login
- router.post('/login', async (req, res) => {
- const { username, password } = req.body;
- User.login({ username, password })
- .then(({ info, accessToken, refreshToken }) => {
- // 设置 refresh token 到 cookie
- res.cookie('jwt', refreshToken, {
- httpOnly: true,
- secure: true,
- sameSite: 'none',
- maxAge: 7 * 24 * 60 * 60 * 1000 // 7天
- });
- // 返回格式与前端期望一致
- res.sendSuccess({ accessToken, ...info });
- })
- .catch(err => {
- Logs.errDev('登录失败:', err);
- if (err.message === 'USER_NOT_FOUND') {
- return res.badRequest('用户不存在');
- }
- if (err.message === 'PASSWORD_ERROR') {
- return res.badRequest('密码错误');
- }
- res.badRequest(err.message);
- });
- });
- router.post('/logout', authMiddleware, (req, res) => {
- res.clearCookie('jwt');
- res.sendSuccess('退出成功');
- });
- // 刷新 Token - 支持 Vben Admin 的 /refresh 路径
- router.post('/refresh', async (req, res) => {
- const refreshToken = req.cookies.jwt;
- if (!refreshToken) {
- return res.unauthorized('无效的刷新token');
- }
- User.refresh(refreshToken)
- .then(accessToken => {
- res.send(accessToken);
- })
- .catch(err => {
- Logs.errDev('刷新Token失败:', err);
- res.clearCookie('jwt');
- if (err.message === 'USER_NOT_FOUND') {
- return res.unauthorized('用户不存在');
- }
- res.unauthorized(err.message);
- });
- });
- // 用户信息 - 支持 mock 服务的 /info 路径
- router.get('/info', authMiddleware, async (req, res) => {
- User.info(req.userId)
- .then(userInfo => {
- res.sendSuccess(userInfo);
- })
- .catch(err => {
- Logs.errDev('获取用户信息错误:', err);
- if (err.message === 'USER_NOT_FOUND') {
- return res.notFound('用户不存在');
- }
- res.serverError(err.message);
- });
- });
- // 权限码 - 支持 mock 服务的 /codes 路径
- router.get('/codes', authMiddleware, async (req, res) => {
- try {
- // 返回所有功能的权限码
- const codes = [
- 'dashboard',
- 'dashboard:analysis',
- 'dashboard:workbench',
- 'system',
- 'system:account',
- 'system:account:settings',
- 'system:role',
- 'system:menu',
- 'system:dept',
- ];
- res.sendSuccess(codes);
- }
- catch (err) {
- Logs.errDev('获取权限码错误:', err);
- res.serverError(err.message);
- }
- });
- module.exports = router;
|