ajy/code/backend/src/main/java/com/aijiuyi/admin/controller/AuthController.java

package com.aijiuyi.admin.controller;

import com.aijiuyi.admin.common.annotation.Log;
import com.aijiuyi.admin.common.annotation.NoAuth;
import com.aijiuyi.admin.common.context.RequestContext;
import com.aijiuyi.admin.common.entity.Result;
import com.aijiuyi.admin.common.enums.OperationType;
import com.aijiuyi.admin.entity.AppUser;
import com.aijiuyi.admin.controller.dto.LoginDTO;
import com.aijiuyi.admin.service.AdminService;
import com.aijiuyi.admin.service.AppUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import javax.validation.Valid;
import java.util.HashMap;
import java.util.Map;

/**
 * 认证管理 Controller
 * 提供登录、退出、获取当前用户信息接口
 * 统一使用应用用户表（app_user）进行身份验证
 */
@RestController
@RequestMapping("/auth")
public class AuthController {

    @Autowired
    private AppUserService appUserService;

    @Autowired
    private AdminService adminService;

    /**
     * 用户登录接口
     * 免鉴权（使用 @NoAuth 注解跳过 Token 验证）
     * 仅 userType=3（管理员）账号可登录后台
     *
     * @param loginDTO 登录参数（用户名、密码）
     * @return 包含 token 和用户信息的 Result
     */
    @PostMapping("/login")
    @NoAuth
    @Log(value = "用户登录", module = "认证管理", operationType = OperationType.OTHER)
    public Result<Map<String, Object>> login(@RequestBody @Valid LoginDTO loginDTO) {
        return appUserService.login(loginDTO);
    }

    /**
     * 用户退出登录
     * 删除 Redis 中的 Token，使当前 Token 立即失效
     *
     * @param request HTTP 请求（用于获取 Authorization 头中的 Token）
     * @return 操作结果
     */
    @PostMapping("/logout")
    @Log(value = "用户退出", module = "认证管理", operationType = OperationType.OTHER)
    public Result<Void> logout(HttpServletRequest request) {
        String token = request.getHeader("Authorization");
        appUserService.logout(token);
        return Result.success();
    }

    /**
     * 获取当前登录用户信息（含权限路由列表）
     *
     * @return 应用用户信息 + permissions 权限路由列表
     */
    @GetMapping("/info")
    @Log(value = "获取当前用户信息", module = "认证管理", operationType = OperationType.QUERY)
    public Result<Map<String, Object>> getUserInfo() {
        // 从 ThreadLocal 获取当前用户ID（由 TokenInterceptor 解析 Token 后设置）
        Long userId = RequestContext.getUserId();
        AppUser user = appUserService.getById(userId);
        if (user != null) {
            user.setPassword(null);
        }
        // 查询权限路由列表（超级管理员返回 null，前端视为全部权限）
        String permissions = adminService.getPermissions(userId);
        Map<String, Object> data = new HashMap<>(8);
        data.put("userId", user != null ? user.getId() : null);
        data.put("username", user != null ? user.getUsername() : null);
        data.put("nickname", user != null ? user.getNickname() : null);
        data.put("avatar", user != null ? user.getAvatar() : null);
        data.put("userType", user != null ? user.getUserType() : null);
        data.put("adminRole", user != null ? user.getAdminRole() : null);
        data.put("permissions", permissions);
        return Result.success(data);
    }
}