package tech.aiflowy.ai.node;

import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import com.agentsflex.core.chain.Chain;
import com.agentsflex.core.chain.node.BaseNode;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.apache.poi.xwpf.usermodel.XWPFParagraph;
import org.apache.poi.xwpf.usermodel.XWPFRun;
import org.apache.poi.xwpf.usermodel.XWPFStyle;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTInd;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTPPrGeneral;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTStyle;
import tech.aiflowy.common.filestorage.FileStorageService;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class MakeFileNode extends BaseNode {

    private final FileStorageService fileStorageService;
    private final String suffix;

    public MakeFileNode(FileStorageService fileStorageService, String suffix) {
        this.fileStorageService = fileStorageService;
        this.suffix = suffix;
    }

    @Override
    protected Map<String, Object> execute(Chain chain) {
        Map<String, Object> map = chain.getParameterValues(this);

        Map<String, Object> res = new HashMap<>();

        String content = map.get("content").toString();
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        createFile(suffix, content, os);

        String fileName = IdUtil.fastSimpleUUID() + "." + suffix;
        CustomFile file = new CustomFile(fileName, os.toByteArray());
        String url = fileStorageService.save(file);
        res.put("url", url);
        return res;
    }

    private void createFile(String suffix, String content, ByteArrayOutputStream os) {
        if ("docx".equals(suffix)) {
            docx(content, os);
        }
    }

    private void docx(String content, ByteArrayOutputStream os) {
        String separator = "\n";
        List<String> split = StrUtil.split(content, separator);
        // 创建一个新的Word文档
        XWPFDocument doc = new XWPFDocument();
        // 创建样式
        CTStyle ctStyle = CTStyle.Factory.newInstance();
        ctStyle.setStyleId("IndentStyle");
        CTPPrGeneral pPr = ctStyle.addNewPPr();
        CTInd ind = pPr.addNewInd();
        ind.setFirstLine(400);
        doc.createStyles().addStyle(new XWPFStyle(ctStyle));

        for (String str : split) {
            // 创建段落
            XWPFParagraph paragraph = doc.createParagraph();
            paragraph.setStyle("IndentStyle");
            XWPFRun run = paragraph.createRun();
            run.setText(str);
        }
        try {
            doc.write(os);
        } catch (IOException e) {
            throw new RuntimeException(e);
        } finally {
            try {
                os.close();
                doc.close();
            } catch (IOException e) {
                System.out.println("关闭流异常" + e.getMessage());
            }
        }
    }
}