package com.vincent.rsf.server.system.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.vincent.rsf.framework.common.Cools;
import com.vincent.rsf.framework.common.R;
import com.vincent.rsf.framework.common.SnowflakeIdWorker;
import com.vincent.rsf.server.ai.constant.AiMcpConstants;
import com.vincent.rsf.server.ai.constant.AiSceneCode;
import com.vincent.rsf.server.ai.service.mcp.AiMcpPayloadMapper;
import com.vincent.rsf.server.ai.service.mcp.AiMcpRegistryService;
import com.vincent.rsf.server.common.annotation.OperationLog;
import com.vincent.rsf.server.common.domain.BaseParam;
import com.vincent.rsf.server.common.domain.PageParam;
import com.vincent.rsf.server.common.utils.ExcelUtil;
import com.vincent.rsf.server.system.entity.AiDiagnosticToolConfig;
import com.vincent.rsf.server.system.entity.AiMcpMount;
import com.vincent.rsf.server.system.service.AiDiagnosticToolConfigService;
import com.vincent.rsf.server.system.service.AiMcpMountService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.util.Arrays;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

@RestController
public class AiMcpMountController extends BaseController {

    @Autowired
    private AiMcpMountService aiMcpMountService;
    @Autowired
    private AiMcpRegistryService aiMcpRegistryService;
    @Autowired
    private AiDiagnosticToolConfigService aiDiagnosticToolConfigService;
    @Autowired
    private SnowflakeIdWorker snowflakeIdWorker;
    @Autowired
    private AiMcpPayloadMapper aiMcpPayloadMapper;

    @PreAuthorize("hasAuthority('system:aiMcpMount:list')")
    @PostMapping("/aiMcpMount/page")
    public R page(@RequestBody Map<String, Object> map) {
        BaseParam baseParam = buildParam(map, BaseParam.class);
        PageParam<AiMcpMount, BaseParam> pageParam = new PageParam<>(baseParam, AiMcpMount.class);
        com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<AiMcpMount> wrapper = pageParam.buildWrapper(true);
        wrapper.eq("tenant_id", getTenantId());
        return R.ok().add(aiMcpMountService.page(pageParam, wrapper));
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:list')")
    @GetMapping("/aiMcpMount/{id}")
    public R get(@PathVariable("id") Long id) {
        AiMcpMount mount = aiMcpMountService.getTenantMount(getTenantId(), id);
        if (mount == null) {
            return R.error("mount not found");
        }
        return R.ok().add(mount);
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:save')")
    @OperationLog("Create AiMcpMount")
    @PostMapping("/aiMcpMount/save")
    public R save(@RequestBody AiMcpMount mount) {
        if (Cools.isEmpty(mount.getName()) || Cools.isEmpty(mount.getMountCode()) || Cools.isEmpty(mount.getTransportType())) {
            return R.error("名称、挂载编码和传输类型不能为空");
        }
        Date now = new Date();
        mount.setUuid(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
        mount.setEnabledFlag(mount.getEnabledFlag() == null ? 1 : mount.getEnabledFlag());
        mount.setTimeoutMs(mount.getTimeoutMs() == null ? 10000 : mount.getTimeoutMs());
        mount.setStatus(mount.getStatus() == null ? 1 : mount.getStatus());
        mount.setTransportType(mount.getTransportType().toUpperCase());
        if (AiMcpConstants.TRANSPORT_INTERNAL.equals(mount.getTransportType())) {
            mount.setUrl("/ai/mcp");
        }
        mount.setTenantId(getTenantId());
        mount.setCreateBy(getLoginUserId());
        mount.setCreateTime(now);
        mount.setUpdateBy(getLoginUserId());
        mount.setUpdateTime(now);
        if (!aiMcpMountService.save(mount)) {
            return R.error("Save Fail");
        }
        return R.ok("Save Success").add(mount);
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:update')")
    @OperationLog("Update AiMcpMount")
    @PostMapping("/aiMcpMount/update")
    public R update(@RequestBody AiMcpMount mount) {
        AiMcpMount existed = aiMcpMountService.getTenantMount(getTenantId(), mount.getId());
        if (existed == null) {
            return R.error("mount not found");
        }
        if (AiMcpConstants.TRANSPORT_INTERNAL.equalsIgnoreCase(existed.getTransportType())) {
            return R.error("内置挂载由系统托管，不支持直接编辑");
        }
        if (Cools.isEmpty(mount.getName()) || Cools.isEmpty(mount.getMountCode()) || Cools.isEmpty(mount.getTransportType())) {
            return R.error("名称、挂载编码和传输类型不能为空");
        }
        mount.setTransportType(mount.getTransportType().toUpperCase());
        if (AiMcpConstants.TRANSPORT_INTERNAL.equals(mount.getTransportType())) {
            mount.setUrl("/ai/mcp");
        }
        mount.setTenantId(getTenantId());
        mount.setCreateBy(existed.getCreateBy());
        mount.setCreateTime(existed.getCreateTime());
        mount.setLastTestResult(existed.getLastTestResult());
        mount.setLastTestTime(existed.getLastTestTime());
        mount.setLastTestMessage(existed.getLastTestMessage());
        mount.setLastToolCount(existed.getLastToolCount());
        mount.setUpdateBy(getLoginUserId());
        mount.setUpdateTime(new Date());
        if (!aiMcpMountService.updateById(mount)) {
            return R.error("Update Fail");
        }
        return R.ok("Update Success").add(mount);
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:remove')")
    @OperationLog("Delete AiMcpMount")
    @PostMapping("/aiMcpMount/remove/{ids}")
    public R remove(@PathVariable Long[] ids) {
        List<Long> idList = Arrays.asList(ids);
        List<AiMcpMount> mounts = aiMcpMountService.list(new LambdaQueryWrapper<AiMcpMount>()
                .eq(AiMcpMount::getTenantId, getTenantId())
                .in(AiMcpMount::getId, idList));
        for (AiMcpMount mount : mounts) {
            if (AiMcpConstants.TRANSPORT_INTERNAL.equalsIgnoreCase(mount.getTransportType())) {
                return R.error("内置挂载由系统托管，不支持删除");
            }
        }
        if (mounts.size() != idList.size() || !aiMcpMountService.removeByIds(idList)) {
            return R.error("Delete Fail");
        }
        return R.ok("Delete Success").add(ids);
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:list')")
    @PostMapping("/aiMcpMount/export")
    public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
        ExcelUtil.build(ExcelUtil.create(aiMcpMountService.list(new LambdaQueryWrapper<AiMcpMount>()
                .eq(AiMcpMount::getTenantId, getTenantId())), AiMcpMount.class), response);
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:list')")
    @GetMapping("/ai/mcp/mount/list")
    public R list() {
        aiMcpRegistryService.ensureDefaultMount(getTenantId(), getLoginUserId());
        return R.ok().add(aiMcpMountService.listTenantMounts(getTenantId()));
    }

    @PreAuthorize("hasAnyAuthority('system:aiMcpMount:save','system:aiMcpMount:update')")
    @PostMapping("/ai/mcp/mount/save")
    public R customSave(@RequestBody AiMcpMount mount) {
        if (mount.getId() == null) {
            return save(mount);
        }
        return update(mount);
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:update')")
    @PostMapping("/ai/mcp/mount/initDefaults")
    public R initDefaults() {
        aiMcpRegistryService.ensureDefaultMount(getTenantId(), getLoginUserId());
        return R.ok();
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:list')")
    @GetMapping("/ai/mcp/mount/toolList")
    public R toolList(@RequestParam(required = false) Long mountId) {
        return R.ok().add(aiMcpRegistryService.listTools(getTenantId(), mountId));
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:update')")
    @PostMapping("/ai/mcp/mount/test")
    public R test(@RequestBody Map<String, Object> map) {
        Long id = Long.valueOf(String.valueOf(map.get("id")));
        return R.ok().add(aiMcpRegistryService.testMount(getTenantId(), id));
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:list')")
    @PostMapping("/ai/mcp/mount/toolPreview")
    public R toolPreview(@RequestBody Map<String, Object> map) {
        return R.ok().add(aiMcpRegistryService.previewTool(
                getTenantId(),
                String.valueOf(map.get("mountCode")),
                String.valueOf(map.get("toolCode")),
                map.get("sceneCode") == null ? null : String.valueOf(map.get("sceneCode")),
                map.get("question") == null ? null : String.valueOf(map.get("question"))
        ));
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:list')")
    @GetMapping("/ai/mcp/console/overview")
    public R consoleOverview() {
        aiMcpRegistryService.ensureDefaultMount(getTenantId(), getLoginUserId());
        Map<String, Object> payload = new LinkedHashMap<>();
        payload.put("builtInMount", null);
        payload.put("builtInTools", new java.util.ArrayList<>());
        payload.put("externalServices", new java.util.ArrayList<>());
        for (AiMcpMount mount : aiMcpMountService.listTenantMounts(getTenantId())) {
            if (AiMcpConstants.TRANSPORT_INTERNAL.equalsIgnoreCase(mount.getTransportType())) {
                payload.put("builtInMount", buildMountView(mount, true));
                payload.put("builtInTools", buildBuiltInToolViews());
            } else {
                ((List<Object>) payload.get("externalServices")).add(buildMountView(mount, false));
            }
        }
        return R.ok().add(payload);
    }

    @PreAuthorize("hasAnyAuthority('system:aiMcpMount:save','system:aiMcpMount:update')")
    @PostMapping("/ai/mcp/console/service/save")
    public R consoleSaveService(@RequestBody AiMcpMount mount) {
        if (Cools.isEmpty(mount.getName()) || Cools.isEmpty(mount.getUrl())) {
            return R.error("名称和地址不能为空");
        }
        Date now = new Date();
        AiMcpMount existed = mount.getId() == null ? null : aiMcpMountService.getTenantMount(getTenantId(), mount.getId());
        if (existed != null && AiMcpConstants.TRANSPORT_INTERNAL.equalsIgnoreCase(existed.getTransportType())) {
            return R.error("内置挂载由系统托管，不支持编辑");
        }
        String transportType = Cools.isEmpty(mount.getTransportType()) ? AiMcpConstants.TRANSPORT_AUTO : mount.getTransportType().toUpperCase();
        String usageScope = normalizeUsageScope(mount.getUsageScope());
        AiMcpMount target = existed == null ? new AiMcpMount() : existed;
        target.setName(mount.getName());
        target.setUrl(mount.getUrl());
        target.setTransportType(transportType);
        target.setAuthType(Cools.isEmpty(mount.getAuthType()) ? AiMcpConstants.AUTH_TYPE_NONE : mount.getAuthType().toUpperCase());
        target.setAuthValue(mount.getAuthValue());
        target.setUsageScope(usageScope);
        target.setEnabledFlag(mount.getEnabledFlag() == null ? 1 : mount.getEnabledFlag());
        target.setTimeoutMs(mount.getTimeoutMs() == null ? 10000 : mount.getTimeoutMs());
        target.setStatus(mount.getStatus() == null ? 1 : mount.getStatus());
        target.setMemo(mount.getMemo());
        target.setTenantId(getTenantId());
        target.setUpdateBy(getLoginUserId());
        target.setUpdateTime(now);
        if (existed == null) {
            target.setUuid(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
            target.setMountCode(generateMountCode(target.getName()));
            target.setCreateBy(getLoginUserId());
            target.setCreateTime(now);
            if (!aiMcpMountService.save(target)) {
                return R.error("Save Fail");
            }
        } else if (!aiMcpMountService.updateById(target)) {
            return R.error("Update Fail");
        }
        return R.ok().add(buildMountView(target, false));
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:update')")
    @PostMapping("/ai/mcp/console/service/test")
    public R consoleTestService(@RequestBody Map<String, Object> map) {
        Long id = Long.valueOf(String.valueOf(map.get("id")));
        Map<String, Object> result = (Map<String, Object>) aiMcpRegistryService.testMount(getTenantId(), id);
        AiMcpMount mount = aiMcpMountService.getTenantMount(getTenantId(), id);
        if (mount == null) {
            return R.error("mount not found");
        }
        result.put("service", buildMountView(mount, false));
        return R.ok().add(result);
    }

    @PreAuthorize("hasAuthority('system:aiMcpMount:remove')")
    @PostMapping("/ai/mcp/console/service/remove/{id}")
    public R consoleRemoveService(@PathVariable("id") Long id) {
        AiMcpMount mount = aiMcpMountService.getTenantMount(getTenantId(), id);
        if (mount == null) {
            return R.error("mount not found");
        }
        if (AiMcpConstants.TRANSPORT_INTERNAL.equalsIgnoreCase(mount.getTransportType())) {
            return R.error("内置挂载由系统托管，不支持删除");
        }
        return aiMcpMountService.removeById(id) ? R.ok() : R.error("Delete Fail");
    }

    @PreAuthorize("hasAnyAuthority('system:aiToolConfig:save','system:aiToolConfig:update')")
    @PostMapping("/ai/mcp/console/builtin-tool/save")
    public R consoleSaveBuiltInTool(@RequestBody AiDiagnosticToolConfig config) {
        if (Cools.isEmpty(config.getToolCode())) {
            return R.error("工具编码不能为空");
        }
        String usageScope = aiMcpPayloadMapper.normalizeUsageScope(config.getUsageScope());
        String sceneCode = resolveSceneCodeByUsageScope(usageScope);
        LambdaQueryWrapper<AiDiagnosticToolConfig> wrapper = new LambdaQueryWrapper<AiDiagnosticToolConfig>()
                .eq(AiDiagnosticToolConfig::getTenantId, getTenantId())
                .eq(AiDiagnosticToolConfig::getToolCode, config.getToolCode());
        if (Cools.isEmpty(sceneCode)) {
            wrapper.and(w -> w.isNull(AiDiagnosticToolConfig::getSceneCode).or().eq(AiDiagnosticToolConfig::getSceneCode, ""));
        } else {
            wrapper.eq(AiDiagnosticToolConfig::getSceneCode, sceneCode);
        }
        AiDiagnosticToolConfig existed = aiDiagnosticToolConfigService.getOne(wrapper.last("limit 1"));
        Date now = new Date();
        AiDiagnosticToolConfig target = existed == null ? new AiDiagnosticToolConfig() : existed;
        if (existed == null) {
            target.setUuid(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
            target.setCreateBy(getLoginUserId());
            target.setCreateTime(now);
        }
        target
                .setToolCode(config.getToolCode())
                .setToolName(config.getToolName())
                .setSceneCode(sceneCode)
                .setUsageScope(usageScope)
                .setEnabledFlag(AiMcpConstants.USAGE_SCOPE_DISABLED.equals(usageScope) ? 0 : (config.getEnabledFlag() == null ? 1 : config.getEnabledFlag()))
                .setPriority(config.getPriority() == null ? 10 : config.getPriority())
                .setToolPrompt(config.getToolPrompt())
                .setStatus(config.getStatus() == null ? 1 : config.getStatus())
                .setTenantId(getTenantId())
                .setUpdateBy(getLoginUserId())
                .setUpdateTime(now)
                .setMemo(config.getMemo());
        if (existed == null) {
            aiDiagnosticToolConfigService.save(target);
        } else {
            aiDiagnosticToolConfigService.updateById(target);
        }
        return R.ok().add(buildBuiltInToolViews());
    }

    private List<Map<String, Object>> buildBuiltInToolViews() {
        List<Map<String, Object>> output = new java.util.ArrayList<>();
        for (com.vincent.rsf.server.ai.model.AiMcpToolDescriptor descriptor : aiMcpRegistryService.listInternalTools(getTenantId())) {
            Map<String, Object> item = new LinkedHashMap<>();
            item.put("toolCode", descriptor.getToolCode());
            item.put("toolName", descriptor.getToolName());
            item.put("description", descriptor.getDescription());
            item.put("enabledFlag", descriptor.getEnabledFlag());
            item.put("priority", descriptor.getPriority());
            item.put("toolPrompt", descriptor.getToolPrompt());
            item.put("usageScope", aiMcpPayloadMapper.resolveUsageScope(descriptor.getSceneCode(), descriptor.getEnabledFlag(), descriptor.getUsageScope()));
            item.put("advanced", buildAdvancedFlag(descriptor));
            output.add(item);
        }
        return output;
    }

    private Map<String, Object> buildMountView(AiMcpMount mount, boolean internalManaged) {
        Map<String, Object> item = new LinkedHashMap<>();
        item.put("id", mount.getId());
        item.put("name", mount.getName());
        item.put("mountCode", mount.getMountCode());
        item.put("transportType", mount.getTransportType());
        item.put("displayTransportType", AiMcpConstants.TRANSPORT_INTERNAL.equalsIgnoreCase(mount.getTransportType()) ? "内置" : mount.getTransportType());
        item.put("url", mount.getUrl());
        item.put("enabledFlag", mount.getEnabledFlag());
        item.put("timeoutMs", mount.getTimeoutMs());
        item.put("lastTestResult", mount.getLastTestResult());
        item.put("lastTestResult$", mount.getLastTestResult$());
        item.put("lastTestTime", mount.getLastTestTime$());
        item.put("lastTestMessage", mount.getLastTestMessage());
        item.put("lastToolCount", mount.getLastToolCount());
        item.put("authType", mount.getAuthType());
        item.put("hasAuth", !Cools.isEmpty(mount.getAuthValue()));
        item.put("usageScope", normalizeUsageScope(mount.getUsageScope()));
        item.put("internalManaged", internalManaged);
        return item;
    }

    private boolean buildAdvancedFlag(com.vincent.rsf.server.ai.model.AiMcpToolDescriptor descriptor) {
        return descriptor.getPriority() != null && descriptor.getPriority() != 10
                || (descriptor.getToolPrompt() != null && !descriptor.getToolPrompt().trim().isEmpty());
    }

    private String normalizeUsageScope(String usageScope) {
        return aiMcpPayloadMapper.normalizeUsageScope(usageScope);
    }

    private String resolveUsageScope(String sceneCode, Integer enabledFlag) {
        return aiMcpPayloadMapper.resolveUsageScope(sceneCode, enabledFlag, null);
    }

    private String resolveSceneCodeByUsageScope(String usageScope) {
        return AiMcpConstants.USAGE_SCOPE_DIAGNOSE_ONLY.equals(usageScope) ? AiSceneCode.SYSTEM_DIAGNOSE : "";
    }

    private String generateMountCode(String name) {
        String normalized = name == null ? "remote_mcp" : name.toLowerCase().replaceAll("[^a-z0-9]+", "_");
        normalized = normalized.replaceAll("^_+|_+$", "");
        if (normalized.isEmpty()) {
            normalized = "remote_mcp";
        }
        return normalized + "_" + String.valueOf(System.currentTimeMillis()).substring(7);
    }
}