package com.dy.common.aop;

import com.dy.common.contant.Constant;
import com.dy.common.multiDataSource.DataSourceContext;
import com.dy.common.webFilter.UserTokenContext;
import com.dy.common.webUtil.BaseResponseUtils;
import com.mysql.cj.util.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;
import java.util.Objects;

@Slf4j
@Aspect
@Order(Constant.AspectOrderSsoAutho)
@Component
public class SsoAspect {

    @Value("${pipIrr.global.dev}")
    public String isDevStage ;//是否为开发阶段

    @Autowired
    private SsoCheck ssoCheck ;

    @Pointcut("@annotation(com.dy.common.aop.SsoAop)")
    public void ssoPointCut() {
    }

    @Around("ssoPointCut()")
    public Object execute(ProceedingJoinPoint point) throws Throwable {
        if(isDevStage != null && !isDevStage.trim().equals("") && isDevStage.trim().equalsIgnoreCase("true")){
            return point.proceed();
        }else{
            MethodSignature signature = (MethodSignature) point.getSignature();
            Method method = signature.getMethod();
            SsoAop aop = method.getAnnotation(SsoAop.class) ;
            if (Objects.nonNull(aop)){
                String token = UserTokenContext.get() ;
                Object rObj = this.ssoCheck.check(token);
                if(rObj != null){
                    if(rObj instanceof SsoVo ssoVo){
                        if(ssoVo.logined){
                            if(ssoVo.hasPower){
                                if(!StringUtils.isNullOrEmpty(ssoVo.dataSourceName)){
                                    DataSourceContext.set(ssoVo.dataSourceName);
                                    Object obj = point.proceed();
                                    DataSourceContext.remove();
                                    return obj ;
                                }else{
                                    //无数据源
                                    return BaseResponseUtils.buildError("后端系统出错，未得到当前登录用户所属机构标签（数据源名）") ;
                                }
                            }else{
                                //无权限
                                return BaseResponseUtils.buildNoPower() ;
                            }
                        }else{
                            //未登录
                            return BaseResponseUtils.buildToLogin() ;
                        }
                    }else{
                        return rObj ;
                    }
                }else{
                    return BaseResponseUtils.buildError("后端系统出错，check方法返回null") ;
                }
            }else{
                //已经进入注解处理了，还得不到注解，这种情况是不可能的。
                return BaseResponseUtils.buildError("后端系统出错，DyAop注解为null") ;
            }
        }
    }


}