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

# 端口
server.port=8081
############## Sa-Token 配置 (文檔: https://sa-token.cc) ##############
# token名稱 (同時(shí)也是cookie名稱)
sa-token.token-name=satoken
# token有效期，單位s 默認(rèn)30天, -1代表永不過期 
sa-token.timeout=2592000
# token臨時(shí)有效期 (指定時(shí)間內(nèi)無(wú)操作就視為token過期) 單位: 秒
sa-token.activity-timeout=-1
# 是否允許同一賬號(hào)并發(fā)登錄 (為true時(shí)允許一起登錄, 為false時(shí)新登錄擠掉舊登錄) 
sa-token.is-concurrent=true
# 在多人登錄同一賬號(hào)時(shí)，是否共用一個(gè)token (為true時(shí)所有登錄共用一個(gè)token, 為false時(shí)每次登錄新建一個(gè)token) 
sa-token.is-share=true
# token風(fēng)格
sa-token.token-style=uuid
# 是否輸出操作日志 
sa-token.is-log=false

@SpringBootApplication
public class SaTokenDemoApplication {
    public static void main(String[] args) throws JsonProcessingException {
        SpringApplication.run(SaTokenDemoApplication.class, args);
        System.out.println("啟動(dòng)成功：Sa-Token配置如下：" + SaManager.getConfig());
    }
}

@PostMapping("/login")
public Boolean login(@RequestBody String phone) {
  //從數(shù)據(jù)庫(kù)查詢?cè)撌謾C(jī)號(hào)的用戶
  User user = userInfoDao.selectUser(phone);
  //若不存在該用戶返回false，存在返回true
  return user == null ? false : true;
}

/**
 * 發(fā)送驗(yàn)證碼：
 * 在redis中用hash存儲(chǔ)用戶的相關(guān)信息，用PHONE_NUM+手機(jī)號(hào)作為用戶hash的key，
 * “code”作為用戶信息hash中驗(yàn)證碼的小key，查詢r(jià)edis中用戶的驗(yàn)證碼信息，
 * "num"是驗(yàn)證次數(shù)的小key
 */
@PostMapping("/sendCaptcha")
public String sendCaptcha(String phone){
    //驗(yàn)證碼verCode
    String verCode;
    String key = "PHONE_NUM"+phone;
    //如果redis中有緩存的驗(yàn)證碼
    Object object = redisTemplate.opsForHash().get(key, "code");
    if(null != object){
        throw Error("該用戶驗(yàn)證碼已發(fā)送，且未過期，請(qǐng)輸入驗(yàn)證碼登錄或注冊(cè)！");
    }else {
        Random r = new Random(System.currentTimeMillis());
        int low = 100000;
        int high = 999999;
        //根據(jù)時(shí)間隨機(jī)生成驗(yàn)證碼verCode，將其放入redis中
        int code = (r.nextInt(high - low) + low);
        verCode = String.valueOf(code);
        redisTemplate.opsForHash().put(key,"code",verCode);
        //放入檢驗(yàn)次數(shù)num=5
        redisTemplate.opsForHash().put(key,"num",5);
        //設(shè)置過期時(shí)間
        redisTemplate.expire(key,60*5,TimeUnit.SECONDS);
    }
    try {
        //調(diào)用發(fā)送驗(yàn)證碼的接口發(fā)送驗(yàn)證碼
        String smsResult= sendMsg(phone, verCode);
    }catch (Throwable throwable){
        redisTemplate.delete(key);
        throw Error("短信發(fā)送失??！");
    }
    return "發(fā)送成功";
}

/**
 * 校驗(yàn)驗(yàn)證碼：
 * 驗(yàn)證成功就StpUtil.login(user.getId())進(jìn)行登錄，自動(dòng)生成token并寫入cookie中
 */
@PostMapping("/checkCaptcha")
public String checkCaptcha(String phone,String verCode){
    if(phone==null||verCode==null||phone==""||verCode==""){
        throw Errorr("請(qǐng)輸入手機(jī)及驗(yàn)證碼！");
    }
    //從redis中獲取該手機(jī)號(hào)用戶的信息
    String key = "PHONE_NUM"+phone;
    Object object = redisTemplate.opsForHash().get(key, "code");
    if(null == object){
        throw Error("未請(qǐng)求驗(yàn)證碼或驗(yàn)證碼已失效，請(qǐng)重新登錄！");
    }
    String code= object.toString();
    if(code.equals(verCode)){
        //驗(yàn)證碼比對(duì)正確，刪除redis中驗(yàn)證碼記錄
        redisTemplate.delete(key);
        //從數(shù)據(jù)庫(kù)查詢出該用戶的信息，并調(diào)用stputil進(jìn)行登錄
        User user = userService.findUser(phone);
        StpUtil.login(user.getId());
        return "登陸成功";
    }else{
        //驗(yàn)證碼比對(duì)錯(cuò)誤，校驗(yàn)次數(shù)減1
        double num = redisTemplate.opsForHash().increment(key,"num",-1);
        //若校驗(yàn)次數(shù)小于0則驗(yàn)證碼失效，不小于0則拋出驗(yàn)證碼錯(cuò)誤
        if(num < 0 ){
            redisTemplate.delete(key);
            return "未請(qǐng)求驗(yàn)證碼或驗(yàn)證碼已失效，請(qǐng)重新登錄！";
        } else {
            return "驗(yàn)證碼錯(cuò)誤！";
        }
    }
}

@PostMapping("/logout")
public Boolean logout() {
    if (!StpUtil.isLogin()) {
    throw Error("未檢測(cè)到登錄信息，請(qǐng)先登錄！");
    }
    StpUtil.logout();
    return true;
}

@PostMapping("/regist)
public Boolean regist(String userName，String verCode) {
    if(userName==null||verCode==null||userName==""||verCode==""){
        throw Errorr("請(qǐng)輸入用戶名或驗(yàn)證碼！");
    }
    查詢數(shù)據(jù)庫(kù)檢查該用戶名是否已經(jīng)被用；
    調(diào)用上面登錄里的校驗(yàn)用戶驗(yàn)證碼的方法；
    用戶名未被占用且驗(yàn)證碼正確，則向數(shù)據(jù)庫(kù)用戶表插入該用戶信息，得到用戶id；
    //調(diào)用stputil進(jìn)行登錄
    StpUtil.login(userId());
    return true；
}

/**
 * 自定義權(quán)限驗(yàn)證接口擴(kuò)展
 */
@Component    // 保證此類被SpringBoot掃描，完成Sa-Token的自定義權(quán)限驗(yàn)證擴(kuò)展
public class StpInterfaceImpl implements StpInterface {
    /**
     * 返回一個(gè)賬號(hào)所擁有的權(quán)限碼集合
     */
    @Override
    public List<String> getPermissionList(Object loginId, String loginType) {
        List<String> list = new ArrayList<String>();
        //1.先從redis中根據(jù)loginId取該用戶的權(quán)限，有則直接返回
        //2.若是redis中沒有，則從數(shù)據(jù)庫(kù)中查詢，再把結(jié)果添加到redis中
        return list;
        // 比如：
        // list.add("101");
        // list.add("user.add");
        // list.add("user.update");
        // list.add("user.get");
        // list.add("user.delete");
        // list.add("art.*");
    }
    /**
     * 返回一個(gè)賬號(hào)所擁有的角色標(biāo)識(shí)集合 (權(quán)限與角色可分開校驗(yàn))
     */
    @Override
    public List<String> getRoleList(Object loginId, String loginType) {
         List<String> list = new ArrayList<String>();
        //1.先從redis中根據(jù)loginId取該用戶的角色集合，有則直接返回
        //2.若是redis中沒有，則從數(shù)據(jù)庫(kù)中查詢，再把結(jié)果添加到redis中
        return list;
        // 比如：
        // list.add("admin");
        // list.add("super-admin");
    }
}

@Configuration
public class SaTokenConfigure implements WebMvcConfigurer {
    // 注冊(cè) Sa-Token 的攔截器
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        // 注冊(cè)路由攔截器，自定義認(rèn)證規(guī)則
        registry.addInterceptor(new SaInterceptor(handler -> {
            //SaRouter.match（參數(shù)一：需要攔截的路由，
                              參數(shù)二：可排除的路由，
                              參數(shù)三：用來(lái)檢驗(yàn)是否通過攔截的方法）
            // 登錄校驗(yàn) -- 用是否登錄攔截所有路由，
            // 在最下面的.excludePathPatterns中并排除登錄等接口的攔截
            SaRouter.match("/**", r -> StpUtil.checkLogin());
            // 角色校驗(yàn) -- 攔截以 admin 開頭的路由，必須具備 admin 角色或者 super-admin 角色才可以通過認(rèn)證
            SaRouter.match("/admin/**", r -> StpUtil.checkRoleOr("admin", "super-admin"));
            // 權(quán)限校驗(yàn) -- 不同模塊校驗(yàn)不同權(quán)限
            SaRouter.match("/user/**", r -> StpUtil.checkPermission("user"));
            SaRouter.match("/admin/**", r -> StpUtil.checkPermission("admin"));
            SaRouter.match("/goods/**", r -> StpUtil.checkPermission("goods"));
            SaRouter.match("/orders/**", r -> StpUtil.checkPermission("orders"));
            SaRouter.match("/notice/**", r -> StpUtil.checkPermission("notice"));
            SaRouter.match("/comment/**", r -> StpUtil.checkPermission("comment"));
            // 甚至你可以隨意的寫一個(gè)打印語(yǔ)句
            SaRouter.match("/**", r -> System.out.println("----啦啦啦----"));
            // 連綴寫法
            SaRouter.match("/**").check(r -> System.out.println("----啦啦啦----"));
        })).addPathPatterns("/**")
           // 排除登錄、注冊(cè)、發(fā)送驗(yàn)證碼等接口 的攔截
           .excludePathPatterns("/user/login")
           .excludePathPatterns("/user/sendCaptcha")
           .excludePathPatterns("/user/checkCaptcha")
           .excludePathPatterns("/user/regist")
        ;
    }
}