package com.hx.util;

import freemarker.template.Configuration;
import freemarker.template.DefaultObjectWrapper;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import org.springframework.core.io.ClassPathResource;

import java.io.*;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

public class TempltUtil {  

    public static final String PREVIEW_DOC = "/temp.docx";
    public static final String TEMP_URL = "temp";
    public static final String TEMP_NAME = "contract.tpl";

    /**
     *
     * @param tempUrl 模板所在文件夹
     * @param tempName 模板名称
     * @return
     * @throws IOException
     */
    public static Template configTemplate(String tempUrl,String tempName) throws IOException {
        Configuration config = new Configuration();
        ClassPathResource classPathResource = new ClassPathResource(tempUrl);
        config.setDirectoryForTemplateLoading(classPathResource.getFile());
        config.setObjectWrapper(new DefaultObjectWrapper());
        Template template = config.getTemplate(tempName, "UTF-8");
        return template;
    }

    /**
     *
     * @param tempUrl 模板所在文件夹
     * @param temp 模板名称
     * @param copyUrl 复制文件路径
     * @param data 模板数据map
     * return 生成路径
     */
    public static String toPreview(String tempUrl,String temp,String copyUrl, Map<?, ?> data){
        try {
        Template template = configTemplate(tempUrl,temp);
        ClassPathResource classPathResource = new ClassPathResource(tempUrl+"/"+PREVIEW_DOC);
        //生成新的模板
        File tempFile =  classPathResource.getFile();
        copyUrl = copyUrl+"/"+DateUtil.dateFormat(new Date(),"yyyyMMddHHmmss")+tempFile.getName();
        File file = new File(copyUrl);
        FileUtils.copyFile(tempFile,file);

        File file2 = new File(copyUrl);

        FileOutputStream fos = new FileOutputStream(file2);
        Writer out = new OutputStreamWriter(fos, "UTF-8");
            template.process(data, out);
            out.flush();
            out.close();
        } catch (Exception e) {
        e.printStackTrace();
        }
        return copyUrl;
    }

    /**
     *
     * @param templateUrl 模板路径目录，不包括模板名称,如 ./src/main/resources/jsonFile
     * @param templateName 模板名称，如：data.tpl
     * @param savaPath 生成文件存放目录，如 ./src/main/resources/jsonFile
     * @param name 生成的文件名称，包括后缀名称
     * @param data 模板里面内容
     * @throws Exception
     */
    public static String createDataFile(String templateUrl,String templateName,
                                      String savaPath, String name,Map<String, Object> data) throws IOException {
        String targetFile = null;
        Writer out = null;
        if(!SimpleTool.checkNotNull(templateUrl)) {
            throw new RuntimeException("templateUrl 为空");
        }
        if(!SimpleTool.checkNotNull(templateName)) {
            throw new RuntimeException("templateName 为空");
        }
        if(!SimpleTool.checkNotNull(savaPath)) {
            throw new RuntimeException("savaPath 为空");
        }
        try {
            if(!savaPath.endsWith("/")){
                savaPath = savaPath+"/";
            }
            if(!templateUrl.endsWith("/")){
                templateUrl = templateUrl +"/";
            }

            //为了解决spingBoot打包成jar获取不了文件，把文件保存到本地
            //拿到jar里面的文件流
            ClassPathResource classPathResource = new ClassPathResource(templateUrl+templateName);
            //保存模板图片路径
            String saveTempUrl = savaPath+"temp/";
            //判断是否存在了
            File temFile = new File(saveTempUrl+templateName);
            if(temFile.exists()){
                temFile = temFile.getParentFile();
            }else{
                //不存在就创建一个
                temFile = FileUtils.inputStreamToFile(classPathResource.getInputStream(),saveTempUrl,templateName);
                temFile = temFile.getParentFile();
            }

            // 反射end
            Configuration cfg = new Configuration();
            // 指定模板文件从何处加载的数据源，这里设置成一个文件目录
            cfg.setDirectoryForTemplateLoading(temFile);
            cfg.setObjectWrapper(new DefaultObjectWrapper());
            cfg.setDefaultEncoding("UTF-8");
            // 获取或创建模板
            Template template = cfg.getTemplate(templateName);
            template.setEncoding("UTF-8");

            // String targetFile = "./src/template/test.html";
            //生成路径
            targetFile = savaPath;
            File file = new File(targetFile);
            //没有找到就新建一个
            if (!file.isDirectory() && !file.exists()) {
                file.mkdirs();
            }
            targetFile += name;
            System.out.println("targetFile:"+targetFile);
            // 将模板和数据模型合并 输出到Console
            out = new BufferedWriter(new OutputStreamWriter(
                    new FileOutputStream(targetFile), "UTF-8"));
            template.process(data, out);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (TemplateException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }finally {
            if(SimpleTool.checkNotNull(out)){
                out.flush();
                out.close();
            }
        }
        return targetFile;
    }

}