<!-- Sa-Token 權(quán)限認(rèn)證，在線文檔：https://sa-token.cc -->
<dependency>
    <groupId>cn.dev33</groupId>
    <artifactId>sa-token-spring-boot-starter</artifactId>
    <version>1.44.0</version>
</dependency>

spring:
  application:
    name: ai-codegen-microservice
sa-token: 
    # token 名稱（同時(shí)也是 cookie 名稱）
    token-name: satoken
    # token 有效期（單位：秒） 默認(rèn)30天，-1 代表永久有效
    timeout: 2592000
    # token 最低活躍頻率，如果 token 超過(guò)此時(shí)間沒(méi)有訪問(wèn)系統(tǒng)就會(huì)被凍結(jié)，默認(rèn)-1 代表不限制，永不凍結(jié)
    active-timeout: -1
    # 是否允許同一賬號(hào)多地同時(shí)登錄 （為 true 時(shí)允許一起登錄, 為 false 時(shí)新登錄擠掉舊登錄）
    is-concurrent: true
    # 在多人登錄同一賬號(hào)時(shí)，是否共用一個(gè) token （為 true 時(shí)所有登錄共用一個(gè) token, 為 false 時(shí)每次登錄新建一個(gè) token）
    is-share: false
    # token 風(fēng)格（默認(rèn)可取值：uuid、simple-uuid、random-32、random-64、random-128、tik）
    token-style: uuid
    # 是否輸出操作日志 
    is-log: true

/**
 * 測(cè)試 Sa-Token 框架實(shí)現(xiàn)用戶登錄
 */
@RestController
@RequestMapping("/test/user/")
public class TestSaTokenLoginController {
    // 測(cè)試登錄，瀏覽器訪問(wèn)： http://localhost:8101/api/test/user/doLogin?username=zhang&password=123456
    @RequestMapping("doLogin")
    public String doLogin(String username, String password) {
        // 此處僅作模擬示例，真實(shí)項(xiàng)目需要從數(shù)據(jù)庫(kù)中查詢數(shù)據(jù)進(jìn)行比對(duì)
        if ("zhang".equals(username) && "123456".equals(password)) {
            StpUtil.login(10001);
            return "登錄成功";
        }
        return "登錄失敗";
    }
    // 查詢登錄狀態(tài)，瀏覽器訪問(wèn)： http://localhost:8101/api/test/user/isLogin
    @RequestMapping("isLogin")
    public String isLogin() {
        return "當(dāng)前會(huì)話是否登錄：" + StpUtil.isLogin();
    }
}

@Configuration
public class SaTokenConfigure implements WebMvcConfigurer {
    // 注冊(cè) Sa-Token 攔截器，打開(kāi)注解式鑒權(quán)功能 
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        // 注冊(cè) Sa-Token 攔截器，打開(kāi)注解式鑒權(quán)功能 
        registry.addInterceptor(new SaInterceptor()).addPathPatterns("/**");    
    }
}

// 登錄校驗(yàn)：只有登錄之后才能進(jìn)入該方法 
@SaCheckLogin                        
@RequestMapping("info")
public String info() {
    return "查詢用戶信息";
}
// 角色校驗(yàn)：必須具有指定角色才能進(jìn)入該方法 
@SaCheckRole("super-admin")        
@RequestMapping("add")
public String add() {
    return "用戶增加";
}

@Component // 保證此類被 SpringBoot 掃描，完成 Sa-Token 的自定義權(quán)限驗(yàn)證擴(kuò)展 
public class StpInterfaceImpl implements StpInterface {
    /**
     * 返回一個(gè)賬號(hào)所擁有的角色標(biāo)識(shí)集合 (權(quán)限與角色可分開(kāi)校驗(yàn))
     */
    @Override
    public List<String> getRoleList(Object loginId, String s) {
        // 從當(dāng)前登錄用戶信息中獲取角色
        User user = (User) StpUtil.getSessionByLoginId(loginId).get(USER_LOGIN_STATE);
        return Collections.singletonList(user.getUserRole());
    }
}

/**
 * 設(shè)備工具類
 */
public class DeviceUtils {
    /**
     * 根據(jù)請(qǐng)求獲取設(shè)備信息
     **/
    public static String getRequestDevice(HttpServletRequest request) {
        String userAgentStr = request.getHeader(Header.USER_AGENT.toString());
        // 使用 Hutool 解析 UserAgent
        UserAgent userAgent = UserAgentUtil.parse(userAgentStr);
        ThrowUtils.throwIf(userAgent == null, ErrorCode.OPERATION_ERROR, "非法請(qǐng)求");
        // 默認(rèn)值是 PC
        String device = "pc";
        // 是否為小程序
        if (isMiniProgram(userAgentStr)) {
            device = "miniProgram";
        } else if (isPad(userAgentStr)) {
            // 是否為 Pad
            device = "pad";
        } else if (userAgent.isMobile()) {
            // 是否為手機(jī)
            device = "mobile";
        }
        return device;
    }
    /**
     * 判斷是否是小程序
     * 一般通過(guò) User-Agent 字符串中的 "MicroMessenger" 來(lái)判斷是否是微信小程序
     **/
    private static boolean isMiniProgram(String userAgentStr) {
        // 判斷 User-Agent 是否包含 "MicroMessenger" 表示是微信環(huán)境
        return StrUtil.containsIgnoreCase(userAgentStr, "MicroMessenger")
                && StrUtil.containsIgnoreCase(userAgentStr, "MiniProgram");
    }
    /**
     * 判斷是否為平板設(shè)備
     * 支持 iOS（如 iPad）和 Android 平板的檢測(cè)
     **/
    private static boolean isPad(String userAgentStr) {
        // 檢查 iPad 的 User-Agent 標(biāo)志
        boolean isIpad = StrUtil.containsIgnoreCase(userAgentStr, "iPad");
        // 檢查 Android 平板（包含 "Android" 且不包含 "Mobile"）
        boolean isAndroidTablet = StrUtil.containsIgnoreCase(userAgentStr, "Android")
                && !StrUtil.containsIgnoreCase(userAgentStr, "Mobile");
        // 如果是 iPad 或 Android 平板，則返回 true
        return isIpad || isAndroidTablet;
    }
}

// Sa-Token 登錄，并指定設(shè)備，同端登錄互斥
StpUtil.login(user.getId(), DeviceUtils.getRequestDevice(request));
StpUtil.getSession().set(USER_LOGIN_STATE, user);

@Override
public User getLoginUser(HttpServletRequest request) {
    // 先判斷是否已登錄
    Object loginUserId = StpUtil.getLoginIdDefaultNull();
    if (loginUserId == null) {
        throw new BusinessException(ErrorCode.NOT_LOGIN_ERROR);
    }
    // 從數(shù)據(jù)庫(kù)查詢（追求性能的話可以注釋，直接走緩存）
    User currentUser = this.getById((String) loginUserId);
    if (currentUser == null) {
        throw new BusinessException(ErrorCode.NOT_LOGIN_ERROR);
    }
    return currentUser;
}

@Override
public User getLoginUser(HttpServletRequest request) {
    // 先判斷是否已登錄
    Object loginId = StpUtil.getLoginIdDefaultNull();
    if (Objects.isNull(loginId)) {
        throw new BusinessException(ErrorCode.NOT_LOGIN_ERROR);
    }
    return (User) StpUtil.getSessionByLoginId(loginId).get(USER_LOGIN_STATE);
}

@Override
public boolean userLogout(HttpServletRequest request) {
    StpUtil.checkLogin();
    // 移除登錄態(tài)
    StpUtil.logout();
    return true;
}

@SaCheckRole(UserConstant.ADMIN_ROLE)