package com.product.file.util;

import cn.hutool.core.lang.UUID;
import com.product.core.config.Global;
import com.spire.doc.*;

import java.io.*;

public class MergeWordDocuments {

	public static File mergeFile(File source, File target) throws Exception {
		if ("doc".equals(FileUtil.checkDocType(source))) {
			source = FileUtil.toDocx(source);
		}
		boolean delTarget = false;
		if ("doc".equals(FileUtil.checkDocType(target))) {
			target = FileUtil.toDocx(target);
			delTarget = true;
		}
		try (InputStream sourceIn = new FileInputStream(source); InputStream targetIn = new FileInputStream(target)) {
			Document sourceDoc = new Document(sourceIn);
			Document targetDoc = new Document(targetIn);
			for (Object section : targetDoc.getSections()) {
				Section sec = (Section) section;
				for (Object docObj : sec.getBody().getChildObjects()) {
					DocumentObject obj = (DocumentObject) docObj;
					Section lastSection = sourceDoc.getLastSection();
					Body body = lastSection.getBody();
					body.getChildObjects().add(obj.deepClone());
				}

			}
			String fileType=".docx";
			FileFormat saveType = FileFormat.Docx;
			//判断来源文件是否为wps
			if (source.getName().endsWith(".wps")) {
				saveType = FileFormat.WPS;
				fileType=".wps";
			}


//			doc.insertTextFromStream(targetIn, com.spire.doc.FileFormat.Docx);
			String tempFile = Global.getSystemConfig("temp.dir", "./attachment/temp") + File.separator + UUID.randomUUID() + fileType;
			try (FileOutputStream out = new FileOutputStream(tempFile)) {
				sourceDoc.saveToStream(out, FileFormat.Docx);
				return new File(tempFile);
			} catch (IOException e) {
				throw e;
			} finally {
				if (delTarget) {
					target.delete();
				}
//				doc.close();
			}
		} catch (IOException e) {
			throw e;
		} catch (Exception e) {
			throw e;
		}
	}
}