youfool-course/src/main/java/com/chinaweal/youfool/course/controller/LoginController.java

package com.chinaweal.youfool.course.controller;

import cn.dev33.satoken.session.SaSession;
import cn.dev33.satoken.stp.StpUtil;
import com.chinaweal.youfool.framework.springboot.common.base.BaseController;
import com.chinaweal.youfool.framework.springboot.common.util.AssertUtils;
import com.chinaweal.youfool.framework.springboot.common.util.RSAUtil;
import com.chinaweal.youfool.framework.springboot.rest.RestResult;
import com.chinaweal.youfool.framework.springboot.rest.ResultCode;
import com.chinaweal.youfool.framework.springboot.user.entity.UserBase;
import com.chinaweal.youfool.course.common.constants.SessionConstants;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * 登录相关接口
 *
 * @author lroyia
 * @since 2022/4/20 15:02
 **/
@Slf4j
@RestController
@RequestMapping("/user/auth")
public class LoginController extends BaseController {

    /**
     * 登录接口
     *
     * @param username 用户名
     * @param password 密码
     * @param encrypt  密码是否已经加密
     * @return 登录结果
     * @author lroyia
     * @since 2022年4月20日 15:47:35
     */
    @PostMapping("login")
    public RestResult<?> doLogin(String username, String password, Boolean encrypt) {
        AssertUtils.isNotBlank(username, password);
        if (encrypt != null && encrypt) {
            String privateKey = getPrivateKey();
            try {
                password = RSAUtil.decrypt(password, privateKey);
            } catch (Exception e) {
                log.error("密码解密失败：{}", password, e);
                return RestResult.error(ResultCode.BUSINESS_LOGIC_ERROR, "密码解密失败");
            }
        }

        // TODO：与数据库匹配校验，具体按用户信息表

        // 匹配成功的话，登记登录信息 TODO:这里的userId是用户唯一号，应根据实际的数据库信息进行替换
        StpUtil.login("userId");
        SaSession session = StpUtil.getSession();
        // 将用户信息存储至session TODO:登录信息存到这里
        UserBase userBase = new UserBase();
        Set<String> permissionSet = new HashSet<>();
        permissionSet.add("admin");
        userBase.setPermission(permissionSet);
        session.set("user", userBase);

        return RestResult.ok();
    }

    /**
     * 获取当前持有的公钥
     *
     * @return 获取结果
     * @author lroyia
     * @since 2022年4月20日 15:55:13
     */
    @GetMapping("publicKey")
    public RestResult<String> getCurrentPublicKey() {
        return RestResult.ok(getPublicKey());
    }

    /**
     * 登录信息
     *
     * @return 状态判断
     * @author lroyia
     * @since 2022年4月20日 15:58:26
     */
    @GetMapping("login/info")
    public RestResult<UserBase> loginInfo() {
        boolean login = StpUtil.isLogin();
        if (login) {
            SaSession session = getSession();
            return RestResult.ok((UserBase) session.get(SessionConstants.USER_KEY));
        }
        return RestResult.ok(null);
    }

    /**
     * 登出
     *
     * @return 操作结果
     * @author lroyia
     * @since 2022年4月20日 15:59:55
     */
    @GetMapping("logout")
    public RestResult<?> logout() {
        StpUtil.logout();
        return RestResult.ok();
    }

    /**
     * 列出用户权限信息
     *
     * @return 权限列表
     * @author lroyia
     * @since 2022年4月20日 16:12:09
     */
    @GetMapping("perm/list")
    public RestResult<List<String>> getPermList() {
        return RestResult.ok(StpUtil.getPermissionList());
    }
}