web/src/main/java/com/chenxuan/controller/SysLoginController.java

package com.chenxuan.controller;

import com.chenxuan.base.entity.AjaxResult;
import com.chenxuan.constants.Constants;
import com.chenxuan.entity.model.SysMenu;
import com.chenxuan.entity.model.SysUser;
import com.chenxuan.security.LoginUser;
import com.chenxuan.security.service.CustomPermissionService;
import com.chenxuan.security.service.JwtTokenService;
import com.chenxuan.security.service.LoginService;
import com.chenxuan.service.SysMenuService;
import com.chenxuan.utils.ServletUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;

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

/**
 * @ClassName：SysLoginController
 * @Description： 登录验证
 * @Author： Arno_Fu
 * @CreatTime：11/26/2019 - 1:47 PM
 * @Version V1.0
 */

@RestController
public class SysLoginController {


    @Autowired
    private LoginService loginService;

    @Autowired
    private SysMenuService menuService;

    @Autowired
    private CustomPermissionService customPermissionService;

    @Autowired
    private JwtTokenService jwtTokenService;


    /**
     * @Description 登录方法
     *
     * @param username 用户名
     * @param password 密码
     * @param code 验证码
     * @param uuid 唯一标识
     * @return com.chenxuan.domain.AjaxResult
     */
    @PostMapping("/login")
    public AjaxResult login(String username, String password, String code, String uuid) throws Exception {
        AjaxResult ajax = AjaxResult.success();
        String token = loginService.login(username, password, code, uuid);
        ajax.put(Constants.TOKEN, token);
        return ajax;
    }


    /**
     * @Description 获取用户信息
     *
     * @return com.chenxuan.base.entity.AjaxResult
     */
    @GetMapping("getInfo")
    public AjaxResult getInfo(){
        SysUser user = getSysUser();
        // 权限集合
        Set<String> permissions = customPermissionService.getMenuPermission(user);
        // 角色集合
        Set<String> roles = customPermissionService.getRolePermission(user);
        AjaxResult ajax = AjaxResult.success();
        ajax.put("user", user);
        ajax.put("roles", roles);
        ajax.put("permissions", permissions);
        return ajax;
    }




    /**
     * @Description 获取路由信息
     *
     * @return com.chenxuan.base.entity.AjaxResult
     */
    @GetMapping("getRouters")
    public AjaxResult getRouters(){
        SysUser user = getSysUser();
        List<SysMenu> menus = menuService.getMenuTreeByUserId(user);
        return AjaxResult.success(menuService.buildMenus(menus));
    }

    private SysUser getSysUser() {
        LoginUser loginUser = jwtTokenService.getLoginUser( ServletUtils.getRequest());
        return loginUser.getUser();
    }




}