package com.gxwebsoft.ai.controller;

import java.io.OutputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.servlet.http.HttpServletResponse;

import org.apache.poi.openxml4j.util.ZipSecureFile;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.apache.poi.xwpf.usermodel.XWPFParagraph;
import org.apache.poi.xwpf.usermodel.XWPFRun;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import com.alibaba.fastjson.JSONObject;
import com.gxwebsoft.ai.config.TemplateConfig;
import com.gxwebsoft.ai.dto.AuditReportRequest;
import com.gxwebsoft.ai.dto.KnowledgeBaseRequest;
import com.gxwebsoft.ai.enums.AuditReportEnum;
import com.gxwebsoft.ai.service.KnowledgeBaseService;
import com.gxwebsoft.ai.util.AuditReportUtil;
import com.gxwebsoft.common.core.web.ApiResult;
import com.gxwebsoft.common.core.web.BaseController;
import com.gxwebsoft.common.system.entity.User;

import cn.afterturn.easypoi.word.WordExportUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.http.HttpUtil;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;

/**
 * 审计报告控制器
 * @author GIIT-YC
 *
 */
@Tag(name = "审计报告")
@RestController
@RequestMapping("/api/ai/auditReport")
public class AuditReportController extends BaseController {
    
    @Autowired
    private TemplateConfig templateConfig;
    
    @Autowired
    private KnowledgeBaseService knowledgeBaseService;
    
    private String invok(String query, String knowledge, String history, String suggestion, String title, String userName) {
        // 构建请求体
        JSONObject requestBody = new JSONObject();
        JSONObject inputs = new JSONObject();
        inputs.put("query", query);
        inputs.put("knowledge", knowledge);
        inputs.put("history", history);
        inputs.put("suggestion", suggestion);
        inputs.put("title", title);
        
        requestBody.put("inputs", inputs);
        requestBody.put("response_mode", "blocking");
        requestBody.put("user", userName);
        
        // 发送 POST 请求
        String result = HttpUtil.createPost("http://1.14.159.185:8180/v1/workflows/run")
                .header("Authorization", "Bearer app-d7Ok9FECVZG2Ybw9wpg7tGu9")
                .header("Content-Type", "application/json")
                .body(requestBody.toString())
                .timeout(600000)
                .execute()
                .body();
        // 解析返回的JSON字符串
        JSONObject jsonResponse = JSONObject.parseObject(result);
        // 获取data字段
        JSONObject data = jsonResponse.getJSONObject("data");
        // 获取outputs字段
        JSONObject outputs = data.getJSONObject("outputs");
        // 获取outputs中的result字符串
        String resultStr = outputs.getString("result");
        return resultStr;
    }
    
    /**
     * 生成审计报告-单一模块
     */
    @Operation(summary = "生成审计报告-单一模块")
    @PostMapping("/generate")
    public ApiResult<?> generateAuditReport(@RequestBody AuditReportRequest req) {
    	final User loginUser = getLoginUser();
    	
    	Set<String> kbIdSet = new HashSet<>();
    	List<String> kbIdList = StrUtil.split(req.getKbIds(), ',');
    	for(String kbId : kbIdList) {
    		KnowledgeBaseRequest knowledgeBaseRequest = new KnowledgeBaseRequest();
    		knowledgeBaseRequest.setKbId(kbId);
    		//召回切片数上限
    		knowledgeBaseRequest.setTopK(10);
    		knowledgeBaseRequest.setFormCommit((req.getFormCommit() >= 10) ? req.getFormCommit() / 10 : req.getFormCommit());
    		kbIdSet.addAll(knowledgeBaseService.queryKnowledgeBase(knowledgeBaseRequest));
    	}
    	String knowledge = kbIdSet.toString();
    	
    	String query = AuditReportEnum.getByCode(req.getFormCommit()).getDesc();
//    	String ret = this.invok(query, knowledge, AuditReportUtil.generateReportContent(req), req.getSuggestion(), loginUser.getUsername());
    	String ret = this.invok(query, knowledge, AuditReportUtil.generateReportContentByFormCommit(req), req.getSuggestion(), req.getFrom00(), loginUser.getUsername());
    	
    	return success(ret);
    }
    
    /**
     * 生成并下载审计报告
     */
    @Operation(summary = "生成并下载审计报告")
    @PostMapping("/download")
    public void downloadAuditReport(@RequestBody AuditReportRequest req, HttpServletResponse response) {
        double originalMinInflateRatio = ZipSecureFile.getMinInflateRatio();

        try {
            ZipSecureFile.setMinInflateRatio(0.001);

            // 准备模板数据
            Map<String, Object> map = new HashMap<>();
            map.put(AuditReportEnum.TITLE.getCode().toString(), req.getFrom00());
            map.put(AuditReportEnum.BASIS.getCode().toString(), req.getFrom10());
            map.put(AuditReportEnum.OBJECTIVE.getCode().toString(), req.getFrom20());
            map.put(AuditReportEnum.SCOPE.getCode().toString(), req.getFrom30());
            map.put(AuditReportEnum.UNIT_OVERVIEW.getCode().toString(), req.getFrom41());
            map.put(AuditReportEnum.ORG_PERSONNEL.getCode().toString(), req.getFrom42());
            map.put(AuditReportEnum.FINANCIAL_ACCOUNTING.getCode().toString(), req.getFrom43());
            map.put(AuditReportEnum.ANNUAL_BUSINESS.getCode().toString(), req.getFrom44());
            map.put(AuditReportEnum.INTERNAL_CONTROL.getCode().toString(), req.getFrom45());
            map.put(AuditReportEnum.ECONOMIC_POLICIES.getCode().toString(), req.getFrom51());
            map.put(AuditReportEnum.DEV_STRATEGY.getCode().toString(), req.getFrom52());
            map.put(AuditReportEnum.MAJOR_ECONOMIC.getCode().toString(), req.getFrom53());
            map.put(AuditReportEnum.CORP_GOVERNANCE.getCode().toString(), req.getFrom54());
            map.put(AuditReportEnum.FINANCIAL_LEGAL.getCode().toString(), req.getFrom55());
            map.put(AuditReportEnum.INTEGRITY_COMPLIANCE.getCode().toString(), req.getFrom56());
            map.put(AuditReportEnum.PREV_AUDIT_ISSUES.getCode().toString(), req.getFrom57());
            map.put(AuditReportEnum.OTHER_MATTERS.getCode().toString(), req.getFrom58());
            map.put(AuditReportEnum.RISK_IDENTIFY.getCode().toString(), req.getFrom61());
            map.put(AuditReportEnum.RISK_RESPONSE.getCode().toString(), req.getFrom62());
            map.put(AuditReportEnum.TECHNIQUES.getCode().toString(), req.getFrom70());
            map.put(AuditReportEnum.SCHEDULE.getCode().toString(), req.getFrom80());
            map.put(AuditReportEnum.ORGANIZATION.getCode().toString(), req.getFrom90());

            // 使用 Easypoi 的 Word 模板功能
            XWPFDocument document = WordExportUtil.exportWord07(templateConfig.getWordTemplatePath(), map);

            // 处理换行，确保 \n 转为硬回车（新段落）
            processParagraphs(document);

            response.setContentType("application/vnd.openxmlformats-officedocument.wordprocessingml.document");
            response.setHeader("Content-Disposition", "attachment; filename=audit_report.docx");

            try (OutputStream out = response.getOutputStream()) {
                document.write(out);
                out.flush();
            }

        } catch (Exception e) {
            throw new RuntimeException("生成审计报告失败", e);
        } finally {
            ZipSecureFile.setMinInflateRatio(originalMinInflateRatio);
        }
    }

    private void processParagraphs(XWPFDocument document) {
        List<XWPFParagraph> originalParas = new ArrayList<>(document.getParagraphs());

        for (XWPFParagraph para : originalParas) {
            String text = para.getText();
            if (text == null || !text.contains("\n")) {
                continue;
            }

            String[] parts = text.replace("\r\n", "\n").replace("\r", "\n").split("\n");
            
            // 在原段落位置之前插入新段落
            int pos = document.getPosOfParagraph(para);
            
            // 修正：按正序插入（从前往后）
            for (int i = 0; i < parts.length; i++) {
                // 创建新段落
                XWPFParagraph newPara = document.insertNewParagraph(para.getCTP().newCursor());
                
                // 复制样式
                newPara.getCTP().setPPr(para.getCTP().getPPr());
                
                XWPFRun newRun = newPara.createRun();
                newRun.setText(parts[i].trim());
                
                if (!para.getRuns().isEmpty()) {
                    newRun.getCTR().setRPr(para.getRuns().get(0).getCTR().getRPr());
                }
            }
            
            // 删除原段落
            document.removeBodyElement(pos + parts.length);
        }
    }

}