package com.xzx.gc.common.request;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.util.ReflectUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.xzx.gc.common.utils.BusinessUtil;
import com.xzx.gc.common.utils.ExceptionUtils;
import com.xzx.gc.common.utils.SecurityUtil;
import com.xzx.gc.common.utils.SpringUtil;
import com.xzx.gc.entity.VersionInfo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

import java.util.List;

/**
 * @author
 * @desc 返回数据加密
 */
@ControllerAdvice
public class MyResponseBodyAdvice implements ResponseBodyAdvice {


    @Autowired
    private BusinessUtil businessUtil;

    @Override
    public boolean supports(MethodParameter methodParameter, Class aClass) {
        return true;
    }

    @Override
    public Object beforeBodyWrite(Object body, MethodParameter methodParameter, MediaType mediaType, Class aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
        String url = serverHttpRequest.getURI().toString();
        if (!businessUtil.isAuthUrl(url)||(SpringUtil.isDev()&&CollUtil.isNotEmpty(serverHttpRequest.getHeaders().get("swagger"))&&"true".equals(serverHttpRequest.getHeaders().get("swagger").get(0)))) {
            return body;
        }
        try {
            //添加版本头用于更新
            String authKey = serverHttpRequest.getHeaders().get("authKey").get(0);
            String s = JSONUtil.toJsonStr(body);
            if (!url.contains("/admin")) {
                Object versionService = SpringUtil.getBean("versionService");
                VersionInfo versionInfo = ReflectUtil.invoke(versionService, "find");
                JSONObject jsonObject = JSONUtil.parseObj(s);
                if (versionInfo != null) {
                    jsonObject.put("updateFlag", Convert.toStr(versionInfo.getUpdateFlag()));
                    jsonObject.put("forceUpdateFlag", Convert.toStr(versionInfo.getForceUpdateFlag()));
                    jsonObject.put("apkVersion", Convert.toStr(versionInfo.getApkMd5()));
                }
                s = JSONUtil.toJsonStr(jsonObject);
            }
            String encrypt = SecurityUtil.encryptRsaAndAes(authKey, s);
            return encrypt;

        } catch (Exception e) {
            ExceptionUtils.err("返回数据加密失败", e);
        }
        return body;
    }


}