package com.shangma.cn.common.permission;

import com.cn.common.http.EnumStatus;
import com.cn.common.utils.ServletUtils;
import com.cn.common.utils.TokenService;
import com.cn.domin.entity.Menu;
import com.cn.exception.PermException;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.lang.reflect.Method;
import java.util.List;

/**
 * 创建一个切面
 */
@Aspect //他是一个切面
@Component //放到容器中
public class MyAspect {
    //使用token,获取登录用户信息,获取权限
    @Autowired
    private TokenService tokenService;

    /**
     * 1.前置通知
     *
     * @param joinPoint 里面需要连接点JoinPoint
     */
    @Before("pointCut()")
    public void checkPerm(JoinPoint joinPoint) {
        //Signature 他是一个接口 找到他的被继承的接口(MethodSignature) ,转换一下
        //joinPoint.getSignature()作用: 获取封装了署名信息的对象,在该对象中可以获取到目标方法名,所属类的Class等信息
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();

        //拿到方法,拿到这个被注解定义的方法
        Method method = signature.getMethod();
        HasPerm annotation = method.getAnnotation(HasPerm.class);

        //拿到之后,就拿到方法的注解内容,如果注解不为空
        if (annotation != null) {
            String perm = annotation.perm();

            //判断用户有没有这个权限 拿到权限信息
            List<Menu> perms = tokenService.getLoginAdmin().getPerms();

            //stream().anyMatch() 有任何一个匹配都行
            //menu.getPermission(); 可能为空 不是按钮权限 没有权限标识 的判断
            boolean b = perms.stream().anyMatch(menu -> perm.equals(menu.getPermission()));
            if (!b) {
                System.out.println("打印了: 进入权限判断了!");
                //抛出异常 防止继续执行
                throw new PermException(EnumStatus.NO_PREM);

            }

           /* //返回前端一个json信息
            String str = "{\"status\":000000,\"message\":\"无权限\"}";
            HttpServletResponse response = ServletUtils.getResponse();
            response.setContentType("application/json;charset=utf-8");
            response.getWriter().write(str);*/
        }
    }

    /**
     * 2.需要一个 切入点
     */
    @Pointcut("@annotation(com.shangma.cn.common.permission.HasPerm)")
    public void pointCut() {

    }

}

1.6.权限 运行时异常

package com.cn.exception;

import com.cn.common.http.EnumStatus;

/**
 * 处理 权限验证 运行时异常
 */

public class PermException extends RuntimeException {

    private EnumStatus enumStatus;

    public PermException(EnumStatus enumStatus) {
        this.enumStatus = enumStatus;
    }

    public EnumStatus getEnumStatus() {
        return enumStatus;
    }

    public void setEnumStatus(EnumStatus enumStatus) {
        this.enumStatus = enumStatus;
    }
}

package com.cn.exception;

import com.cn.common.http.AxiosResult;
import com.cn.common.http.EnumStatus;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import java.util.Map;

/**
 * 专门做处理异常的
 */
@RestControllerAdvice
public class MyExceptionHandler {

    /**
     * 处理权限异常
     */
    @ExceptionHandler(PermException.class)
    public AxiosResult<Void> handler(PermException e){
        EnumStatus enumStatus = e.getEnumStatus();

        return AxiosResult.error(enumStatus);

    }
}

package com.cn.common.http;

public enum EnumStatus {
    OK(20000,"操作成功"),
    ERROR(40000,"操作失败"),
    NO_LOGIN(40005,"未登录"),
    ACCOUNT_ERROR(40006,"用户名错误"),
    CODE_ERROR(40007,"验证码错误"),
    PWD_ERROR(40007,"密码错误"),
    CODE_LOSE(40008,"验证码失效"),
    NO_ACTIVE(40009,"用户邮箱未激活"),
    NO_PREM(40010,"无权限");

    private int status;
    private String message;

    EnumStatus(int status, String message) {
        this.status = status;
        this.message = message;
    }

    public int getStatus() {
        return status;
    }

    public void setStatus(int status) {
        this.status = status;
    }

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }
}

package com.cn.common.http;

import com.fasterxml.jackson.annotation.JsonInclude;

/*表示这个类 在转json格式字符串是,如果属性有null,则json字符串中不会有这个属性 */
@JsonInclude(JsonInclude.Include.NON_NULL)
public class AxiosResult<T> {
    private int status;
    private String message;
    private T data;


    private static <T> AxiosResult<T> getAxiosResult(EnumStatus enumStatus, T data){
        return new AxiosResult<T>(enumStatus,data);
    }
    public static <T> AxiosResult<T> error(EnumStatus enumStatus) {
        return getAxiosResult(enumStatus, null);
    }
    public AxiosResult(EnumStatus enumStatus, T data){
        this.message=enumStatus.getMessage();
        this.status=enumStatus.getStatus();
        this.data=data;
    }

    public int getStatus() {
        return status;
    }

    public void setStatus(int status) {
        this.status = status;
    }

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }

    public T getData() {
        return data;
    }

    public void setData(T data) {
        this.data = data;
    }

    /**
     * 返回成功的方法
     */
    public static <T> AxiosResult<T> success(){
       return getAxiosResult(EnumStatus.OK,null);
    }

    public static <T> AxiosResult<T> success(T data){
        return getAxiosResult(EnumStatus.OK,data);
    }

    /**
     * 返回失败的方法
     */
    public static <T> AxiosResult<T> error(T data){
        return getAxiosResult(EnumStatus.ERROR,data);
    }

    public static <T> AxiosResult<T> error(){
        return getAxiosResult(EnumStatus.ERROR,null);
    }
}