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 zskId = list.getFirst().getStr("id");

        //在上传文档前，需要删除掉旧的同名文档
        File fi = new File(uploadFile);
        String fileName = fi.getName();
        //获取身份识别
        String authCode = MaxKbCallUtil.getAuthorization();
        //上传文档
        List<Record> listDocuments = mm.getDocumentListByDataSetId(zskId);
        for (Record record : listDocuments) {
            String documentName = record.getStr("name");
            if (documentName.equals(fileName)) {
                String documentId = record.getStr("id");
                //准备删除掉旧的文档
                MaxKbCallUtil.delDocument(authCode, zskId, documentId);
                System.out.println("删除旧的文档成功：" + documentName);
            }
        }
        //上传新的文档
        MaxKbCallUtil.uploadQA(authCode, zskId, uploadFile);
        System.out.println("恭喜，所有操作成功完成！");
    }
}