package Tools.MaxKb.Service.Impl;

import cn.hutool.core.util.StrUtil;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import com.alibaba.fastjson.JSONObject;
import com.jfinal.kit.PropKit;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class MaxKbImpl {
    //MaxKB的url
    static final String baseUrl = PropKit.get("baseUrl");
    //获取身份识别码接口
    static final String Authorization_URL = baseUrl + "/api/user/login";

    /**
     * 上传QA文件到知识库
     *
     * @param authCode   身份识别码
     * @param uploadFile 上传文件路径
     */
    public static String uploadQA(String authCode, String zskId, String uploadFile) {
        //知识库url上传地址
        String qaUrl = baseUrl + "/api/dataset/" + zskId + "/document/qa";
        //使用hutool工具箱里面的httpPost功能，向指定的接口发起上传文件的请求，在Header中携带参数authCode
        HttpResponse response = HttpRequest.post(qaUrl)
                .header("Accept", "application/json, text/plain, */*")
                .header("Accept-Encoding", "gzip,deflate")
                .header("Accept-Language", "zh-CN,zh;q=0.9")
                .header("Authorization", authCode)// 添加header
                .header("Connection", "keep-alive")
                .header("Content-Type", "multipart/form-data; boundary=----WebKitFormBoundaryhEiEDpZDka6byweA")
                .header("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36")
                .form("file", new File(uploadFile)).execute();// 添加文件
        // 获取响应内容
        String result = response.body();
        JSONObject jo = JSONObject.parseObject(result);
        int code = jo.getInteger("code");
        if (code != 200) {
            System.out.println("请求失败");
            return null;
        }
        return jo.getJSONArray("data").getJSONObject(0).getString("id");
    }

    /**
     * 上传文档并分段
     *
     * @param authCode   验证码
     * @param uploadFile 上传的文件
     * @return
     */
    public static JSONObject uploadDocumentSplit(String authCode, String uploadFile) {
        //知识库url上传地址
        String qaUrl = baseUrl + "/api/dataset/document/split";
        //使用hutool工具箱里面的httpPost功能，向指定的接口发起上传文件的请求，在Header中携带参数authCode
        HttpResponse response = HttpRequest.post(qaUrl)
                .header("Accept", "application/json, text/plain, */*")
                .header("Accept-Encoding", "gzip,deflate")
                .header("Accept-Language", "zh-CN,zh;q=0.9")
                .header("Authorization", authCode)// 添加header
                .header("Connection", "keep-alive")
                .header("Content-Type", "multipart/form-data; boundary=----WebKitFormBoundaryhEiEDpZDka6byweA")
                .header("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36")
                .form("file", new File(uploadFile)).execute();// 添加文件
        // 获取响应内容
        String result = response.body();
        JSONObject jo = JSONObject.parseObject(result);
        int code = jo.getInteger("code");
        if (code != 200) {
            System.out.println("请求失败");
            return null;
        }
        return jo;
    }


    /**
     * 发送请求获取识别码
     *
     * @throws Exception
     */
    public static String getAuthorization() {
        //管理员账号与密码
        String username = PropKit.get("MaxKBUsername");
        String password = PropKit.get("MaxKBPassword");

        JSONObject requestBody = new JSONObject();
        requestBody.put("username", username);
        requestBody.put("password", password);

        HttpResponse response = HttpRequest.post(Authorization_URL)
                .header("Content-Type", "application/json")
                .header("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36")
                .header("Origin", baseUrl)
                .header("Referer", baseUrl + "/ui/login")
                .body(requestBody.toString())
                .timeout(10000) // 可选：设置超时时间（毫秒）
                .execute();

        //System.out.println(response);
        // 详细的状态处理
        if (response.isOk()) {
            String body = response.body();
            if (StrUtil.isBlank(body)) {
                throw new RuntimeException("响应内容为空");
            }

            try {
                JSONObject responseJson = JSONObject.parseObject(body);
                String data = responseJson.getString("data");
                if (StrUtil.isBlank(data)) {
                    throw new RuntimeException("未获取到data字段");
                }
                return data;
            } catch (Exception e) {
                throw new RuntimeException("解析响应JSON失败: " + e.getMessage());
            }
        } else {
            String errorBody = response.body();
            throw new RuntimeException(String.format("请求失败 [%d]: %s",
                    response.getStatus(),
                    StrUtil.isBlank(errorBody) ? "无错误信息" : errorBody));
        }
    }

    /**
     * 删除知识库文档
     *
     * @param authCode   身份识别码
     * @param zskId      知识库id
     * @param documentId 文档id
     * @return
     */
    public static com.alibaba.fastjson.JSONObject delDocument(String authCode, String zskId, String documentId) {
        String url = baseUrl + "/api/dataset/" + zskId + "/document/" + documentId + "?dep=[object+Object]&__v_isRef=true&__v_isShallow=false&_rawValue=false&_value=false";
        HttpResponse response = HttpRequest.delete(url)
                .header("Accept", "application/json, text/plain, */*")
                .header("Accept-Encoding", "gzip,deflate")
                .header("Accept-Language", "zh-CN,zh;q=0.9")
                .header("Authorization", authCode)// 添加header
                .header("Connection", "keep-alive")
                .header("Content-Type", "application/json")
                .header("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36")
                .execute();
        return com.alibaba.fastjson.JSONObject.parseObject(response.body());
    }

    /**
     * 生成问题
     *
     * @param authCode   身份识别码
     * @param zskId      知识库id
     * @param modelId    模型id
     * @param documentId 文档id
     * @return
     */
    public static com.alibaba.fastjson.JSONObject generateQuestion(String authCode, String zskId, String modelId, String documentId) {
        String url = baseUrl + "/api/dataset/" + zskId + "/document/batch_generate_related";
        com.alibaba.fastjson.JSONObject jo = new com.alibaba.fastjson.JSONObject();
        jo.put("model_id", modelId);
        jo.put("prompt", "内容：{data}\\n\\n请总结上面的内容，并根据内容总结生成 5 个问题。\\n回答要求：\\n- 请只输出问题；\\n- 请将每个问题放置<question></question>标签中。");
        List<String> list = new ArrayList<>();
        list.add(documentId);
        jo.put("document_id_list", list);
        HttpResponse response = HttpRequest.put(url)
                .header("Accept", "application/json, text/plain, */*")
                .header("Accept-Encoding", "gzip,deflate")
                .header("Accept-Language", "zh-CN,zh;q=0.9")
                .header("Authorization", authCode)// 添加header
                .header("Connection", "keep-alive")
                .header("Content-Type", "application/json")
                .body(jo.toJSONString()).execute();
        return com.alibaba.fastjson.JSONObject.parseObject(response.body());

    }

    /**
     * 向知识库批量上传文档
     *
     * @param authCode 授权码
     * @param zskId    知识库ID
     * @param body     分段后的文字内容
     * @return
     */
    public static String _bach(String authCode, String zskId, String body) {
        String url = baseUrl + "/api/dataset/" + zskId + "/document/_bach";
        HttpResponse response = HttpRequest.post(url)
                .header("Accept", "application/json, text/plain, */*")
                .header("Accept-Encoding", "gzip,deflate")
                .header("Accept-Language", "zh-CN,zh;q=0.9")
                .header("Authorization", authCode)// 添加header
                .header("Connection", "keep-alive")
                .header("Content-Type", "application/json")
                .body(body).execute();
        return response.body();
    }
}