package Tools.MaxKb;

import Tools.MaxKb.Model.MaxKbModel;
import Tools.MaxKb.Util.MaxKbCallUtil;
import Tools.MaxKb.Util.MaxKbDb;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.Record;

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

public class UploadMaxKBFile {

    public static MaxKbModel mm = new MaxKbModel();


    //要上传的QA知识内容
    static String uploadFile = "D:\\dsWork\\QingLong\\Doc\\MaxKB\\各学校人员和班级统计【结果】.xlsx";

    public static void main(String[] args) throws Exception {
        //加载配置文件
        PropKit.use("MaxKb.properties");

        //初始化数据库
        MaxKbDb.Init();

        //获取数据集名称
        String dataSetName = PropKit.get("dataSetName");
        List<Record> list = mm.getDataSetList(dataSetName);
        if (list.size() > 1) {
            System.out.println("数据集:【" + dataSetName + "】名称重复,请检查后重新运行！");
            return;
        }
        if (list.isEmpty()) {
            System.out.println("数据集:【" + dataSetName + "】不存在,请检查后重新运行！");
            return;
        }
        //数据集ID
        String dataSetId = list.getFirst().getStr("id");

        //在上传文档前，需要删除掉旧的同名文档
        File fi = new File(uploadFile);
        String fileName = fi.getName();

        List<Record> listDocuments = mm.getDocumentListByDataSetId(dataSetId);
        for (Record record : listDocuments) {
            String documentName = record.getStr("name");
            if (documentName.equals(fileName)) {
                String documentId = record.getStr("id");
                //准备删除掉旧的文档

                //上传新的文档
            }
        }

        //获取身份识别
//        String authCode = MaxKbCallUtil.getAuthorization();
//        //测试上传文件到知识库
//        String id = MaxKbCallUtil.uploadQA(authCode, zskId, uploadFile);
//        System.out.println("生成的文档ID=" + id);
    }
}