package com.ruoyi.framework.interceptor;

import com.ruoyi.common.config.RuoYiConfig;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.nio.charset.StandardCharsets;
import java.util.Arrays;
import java.util.List;

/**
 * 集群版对接中台时的写操作兜底拦截。
 *
 * <p>目的：前端隐藏按钮只能减少误触，不能阻止绕过页面直接调用接口。</p>
 * <p>影响范围：访客后台非本地管理模块的新增、编辑、删除、状态变更、注销、导入等写接口。</p>
 */
@Component
public class MiddlePlatformReadonlyInterceptor implements HandlerInterceptor {

    private static final String MESSAGE = "集群版对接中台时，当前模块仅支持查询和导出";
    private static final String RESET_PASSWORD_MESSAGE = "集群版对接中台时，不支持重置密码";

    /** 受中台只读规则约束的访客后台接口前缀。 */
    private static final List<String> CONTROLLED_PREFIXES = Arrays.asList(
            "/system/visit/area",
            "/system/dept",
            "/system/employee",
            "/system/visitor/record",
            "/system/visitor/machine",
            "/system/door/device",
            "/system/version/update",
            "/system/protocol/config",
            "/system/platform/admin",
            "/system/role/manage"
    );

    /** 本地仍可维护的白名单模块：账号管理、系统配置、入园承诺、访客机版本、首页、操作日志。 */
    private static final List<String> LOCAL_MODULE_PREFIXES = Arrays.asList(
            "/system/platform/admin",
            "/system/role/manage",
            "/system/visitor/config",
            "/system/protocol/config",
            "/system/version/update",
            "/system/dashboard",
            "/system/operation/log"
    );

    /** 非白名单模块在中台模式下只允许这些查询、导出和模板下载接口通过。 */
    private static final List<String> READONLY_SUFFIXES = Arrays.asList(
            "/page",
            "/detail",
            "/tree",
            "/overview",
            "/roleOptions",
            "/permissionTree",
            "/export",
            "/importTemplate"
    );

    @Autowired
    private RuoYiConfig ruoyiConfig;

    /**
     * 目的：在请求进入 Controller 前判断中台只读规则，阻止非白名单模块写操作。
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        if (!ruoyiConfig.isConnectMiddlePlatform()) {
            return true;
        }

        String path = request.getRequestURI();
        String contextPath = request.getContextPath();
        if (contextPath != null && contextPath.length() > 0 && path.startsWith(contextPath)) {
            path = path.substring(contextPath.length());
        }

        if (!isControlledPath(path)) {
            return true;
        }
        String method = request.getMethod();
        // 放行预检请求，避免 CORS OPTIONS 被业务只读规则误拦截。
        if (isPreflight(method)) {
            return true;
        }
        // 账号管理属于本地白名单，但业务要求中台模式下不允许展示或调用重置密码。
        if (path.startsWith("/system/platform/admin/resetPassword")) {
            return reject(response, RESET_PASSWORD_MESSAGE);
        }
        // GET/HEAD 和明确的查询导出接口为只读请求；白名单模块按本地管理逻辑继续放行。
        if (isLocalModule(path) || isReadonlyEndpoint(path) || isReadMethod(method)) {
            return true;
        }
        return reject(response);
    }

    /**
     * 目的：判断当前请求是否属于访客后台需要受控的接口范围。
     */
    private boolean isControlledPath(String path) {
        for (String prefix : CONTROLLED_PREFIXES) {
            if (path.startsWith(prefix)) {
                return true;
            }
        }
        return false;
    }

    /**
     * 目的：判断当前请求是否属于中台模式仍允许本地维护的白名单模块。
     */
    private boolean isLocalModule(String path) {
        for (String prefix : LOCAL_MODULE_PREFIXES) {
            if (path.startsWith(prefix)) {
                return true;
            }
        }
        return false;
    }

    /**
     * 目的：判断 POST 查询、导出和模板下载这类只读接口是否可放行。
     */
    private boolean isReadonlyEndpoint(String path) {
        for (String suffix : READONLY_SUFFIXES) {
            if (path.endsWith(suffix)) {
                return true;
            }
        }
        return false;
    }

    /**
     * 目的：识别 HTTP 层面的只读方法，支持原生 GET 查询接口。
     */
    private boolean isReadMethod(String method) {
        return "GET".equalsIgnoreCase(method) || "HEAD".equalsIgnoreCase(method);
    }

    /**
     * 目的：识别 CORS 预检请求，避免 OPTIONS 被业务规则误拦截。
     */
    private boolean isPreflight(String method) {
        return "OPTIONS".equalsIgnoreCase(method);
    }

    /**
     * 目的：使用默认只读提示拒绝当前写操作请求。
     */
    private boolean reject(HttpServletResponse response) throws Exception {
        return reject(response, MESSAGE);
    }

    /**
     * 目的：按指定提示返回统一 Ajax JSON 错误。
     */
    private boolean reject(HttpServletResponse response, String message) throws Exception {
        response.setCharacterEncoding(StandardCharsets.UTF_8.name());
        response.setContentType("application/json;charset=UTF-8");
        response.getWriter().write("{\"code\":500,\"msg\":\"" + message + "\"}");
        return false;
    }
}