package com.thhy.gateway.config;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.nacos.shaded.io.grpc.netty.shaded.io.netty.util.internal.StringUtil;
import com.thhy.general.common.BasicResult;
import com.thhy.general.common.BasicStatus;
import org.redisson.api.RBucket;
import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.HttpStatus;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

import java.nio.charset.StandardCharsets;

public class GlobalFilters implements GlobalFilter, Ordered {

    @Autowired
    private ExcludeRoute excludeRoute;

    @Autowired
    private RedissonClient redissonClient;

    @Value("${global.keyPrefix}")
    private String keyPrefix;

    private static final String ALL = "*";
    private static final String MAX_AGE = "18000L";

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        exchange.getLogPrefix();
        System.out.println("这是全局");
        ServerHttpRequest request = exchange.getRequest();
        String path = request.getPath().toString();
        System.out.println("访问路径"+path);
        ServerHttpResponse response = exchange.getResponse();
        HttpHeaders headers = response.getHeaders();
        headers.add(HttpHeaders.ACCESS_CONTROL_ALLOW_ORIGIN, "*");
        headers.add(HttpHeaders.ACCESS_CONTROL_ALLOW_METHODS, "POST, GET, PUT, OPTIONS, DELETE, PATCH");
        headers.add(HttpHeaders.ACCESS_CONTROL_ALLOW_CREDENTIALS, "true");
        headers.add(HttpHeaders.ACCESS_CONTROL_ALLOW_HEADERS, "*");
        headers.add("MultipleAllowOriginValues","*");
        headers.add(HttpHeaders.ACCESS_CONTROL_EXPOSE_HEADERS, ALL);
        headers.add(HttpHeaders.ACCESS_CONTROL_MAX_AGE, MAX_AGE);
        if (request.getMethod() == HttpMethod.OPTIONS) {
            response.setStatusCode(HttpStatus.OK);
            return Mono.empty();
        }


        String[] excludeurls = excludeRoute.getExcludeurls();

        for(String excludeurl : excludeurls){
            if(path.equals(excludeurl)||path.contains(excludeurl)){
                return chain.filter(exchange);
            }
        }

        if(!request.getHeaders().containsKey("usertoken")){
            BasicResult basicResult = new BasicResult(false, BasicStatus.TOKEN_IS_NULL);  //请登录
            return buildReturnMono(JSON.toJSONString(basicResult),exchange);
        }

        String userToken = request.getHeaders().get("usertoken").get(0);

        Object obj = getUserInfo(userToken,exchange);
        if(obj==null){
            BasicResult basicResult = new BasicResult(false, BasicStatus.LOGIN_INFO_IS_ERROR);  //登录信息异常
            return buildReturnMono(JSON.toJSONString(basicResult),exchange);
        }
        JSONObject jsonObject = JSON.parseObject(JSON.toJSONString(obj));
        if(!jsonObject.containsKey("userId")||StringUtil.isNullOrEmpty(jsonObject.getString("userId"))){
            BasicResult basicResult = new BasicResult(false, BasicStatus.LOGIN_INFO_IS_ERROR);  //登录信息异常
            return buildReturnMono(JSON.toJSONString(basicResult),exchange);
            //throw new BasicException(BasicStatus.LOGIN_INFO_IS_ERROR);
        }
        //BodyHandlerServerHttpResponseDecorator responseDecorator = new BodyHandlerServerHttpResponseDecorator(bodyHandler, exchange.getResponse());
        return chain.filter(exchange);

    }

    public static Mono<Void> buildReturnMono(String jsonStr, ServerWebExchange exchange) {
        ServerHttpResponse response = exchange.getResponse();
        byte[] bits = jsonStr.getBytes(StandardCharsets.UTF_8);
        DataBuffer buffer = response.bufferFactory().wrap(bits);
        response.setStatusCode(HttpStatus.OK);
        HttpHeaders headers = response.getHeaders();
        headers.add(HttpHeaders.ACCESS_CONTROL_ALLOW_ORIGIN, "*");
        headers.add(HttpHeaders.ACCESS_CONTROL_ALLOW_METHODS, "POST, GET, PUT, OPTIONS, DELETE, PATCH");
        headers.add(HttpHeaders.ACCESS_CONTROL_ALLOW_CREDENTIALS, "true");
        headers.add(HttpHeaders.ACCESS_CONTROL_ALLOW_HEADERS, "*");
        headers.add(HttpHeaders.ACCESS_CONTROL_EXPOSE_HEADERS, ALL);
        headers.add(HttpHeaders.ACCESS_CONTROL_MAX_AGE, MAX_AGE);
        //指定编码，否则在浏览器中会中文乱码
        response.getHeaders().add("Content-Type", "text/plain;charset=UTF-8");
        return response.writeWith(Mono.just(buffer));
    }

    public Object getUserInfo(String token,ServerWebExchange exchange){
        RBucket<Object> rBucket = redissonClient.getBucket(keyPrefix+":usertoken:"+ token);
        Object obj = rBucket.get();
        return obj;
    }

    @Override
    public int getOrder() {
        return 0;
    }
}