package com.inspect.gateway.handler;

import com.inspect.base.core.utils.ServletUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.web.reactive.error.ErrorWebExceptionHandler;
import org.springframework.cloud.gateway.support.NotFoundException;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.annotation.Order;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.web.server.ResponseStatusException;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

@Order(-1)
@Configuration
public class GatewayExceptionHandler implements ErrorWebExceptionHandler {
    private static final Logger log = LoggerFactory.getLogger(GatewayExceptionHandler.class);

    public Mono<Void> handle(ServerWebExchange exchange, Throwable ex) {
        ServerHttpResponse response = exchange.getResponse();
        if (exchange.getResponse().isCommitted()) {
            return Mono.error(ex);
        } else {
            String msg;
            if (ex instanceof NotFoundException) {
                msg = "SERVICE UNFOUNDED!";
            } else if (ex instanceof ResponseStatusException) {
                ResponseStatusException responseStatusException = (ResponseStatusException) ex;
                msg = responseStatusException.getMessage();
            } else {
                msg = "INTERNAL SERVER ERROR!";
            }

            ex.printStackTrace();
            log.error("[GATEWAY EXCEPTION] PATH: {}, INFO: {}", exchange.getRequest().getPath(), ex.getMessage());
            return ServletUtils.webFluxResponseWriter(response, msg);
        }
    }
}
