package org.fh.util;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.OutputStreamWriter;

import java.io.PrintWriter;

import java.io.Writer;

import java.util.Locale;

import java.util.Map;

import freemarker.template.Configuration;

import freemarker.template.Template;

import freemarker.template.TemplateException;

/**

 * 说明：Freemarker 模版引擎类

 * 作者：FH Admin

 * from：fhadmin.cn

 */

public class Freemarker {

    /**

     * 打印到控制台(测试用)

     *  @param ftlName

     */

    public static void print(String ftlName, Map<String,Object> root, String ftlPath) throws Exception{

        try {

            Template temp = getTemplate(ftlName, ftlPath);        //通过Template可以将模板文件输出到相应的流

            temp.process(root, new PrintWriter(System.out));

        } catch (TemplateException e) {

            e.printStackTrace();

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

    /**

     * 输出到输出到文件

     * @param ftlName   ftl文件名

     * @param root        传入的map

     * @param outFile    输出后的文件全部路径

     * @param filePath    输出前的文件上部路径

     */

    public static void printFile(String ftlName, Map<String,Object> root, String outFile, String filePath, String ftlPath) throws Exception{

        try {

            File file = new File(PathUtil.getProjectpath() + filePath + outFile);

            if(!file.getParentFile().exists()){                //判断有没有父路径，就是判断文件整个路径是否存在

                file.getParentFile().mkdirs();                //不存在就全部创建

            }

            Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "utf-8"));

            Template template = getTemplate(ftlName, ftlPath);

            template.process(root, out);                    //模版输出

            out.flush();

            out.close();

        } catch (TemplateException e) {

            e.printStackTrace();

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

    /**

     * 通过文件名加载模版

     * @param ftlName

     */

    public static Template getTemplate(String ftlName, String ftlPath) throws Exception{

        try {

            Configuration cfg = new Configuration(Configuration.VERSION_2_3_23);                                  //通过Freemaker的Configuration读取相应的ftl

            cfg.setEncoding(Locale.CHINA, "utf-8");

            cfg.setDirectoryForTemplateLoading(new File(PathUtil.getProjectpath()+"/admin/template/ftl/"+ftlPath));        //设定去哪里读取相应的ftl模板文件

            Template temp = cfg.getTemplate(ftlName);                                                            //在模板文件目录中找到名称为name的文件

            return temp;

        } catch (IOException e) {

            e.printStackTrace();

        }

        return null;

    }

}