package com.why.main;

import java.io.bufferedwriter;
import java.io.file;
import java.io.fileinputstream;
import java.io.filenotfoundexception;
import java.io.fileoutputstream;
import java.io.ioexception;
import java.io.inputstream;
import java.io.outputstreamwriter;
import java.io.unsupportedencodingexception;
import java.io.writer;
import java.util.hashmap;
import java.util.map;

import freemarker.template.configuration;
import freemarker.template.template;
import freemarker.template.templateexception;
import sun.misc.base64encoder;

/**
 * 生成doc文档
 */
public class documenthandler {

    //测试
    public static void main(string[] args) {
        documenthandler documenthandler = new documenthandler();
        documenthandler.createdoc();
    }

    // 配置实例：只需要一个实例（单例模式）
    private configuration configuration = null;

    private string tempdirpath = "d:/temp";

    public documenthandler() {
        // 通过freemaker的configuration读取相应的ftl
        configuration = new configuration(configuration.version_2_3_28);
        configuration.setdefaultencoding("utf-8");// 设置默认编码方式
    }

    /**
     * 生成doc文档
     */
    public void createdoc() {
        // 要填入模本的数据文件
        map<string,object> datamap = new hashmap<string,object>();
        getdata(datamap);
        // 设置模本装置方法和路径,freemarker支持多种模板装载方法。可以重servlet，classpath，数据库装载，
        // 如果模板是放在程序代码的包下面
        //configuration.setclassfortemplateloading(this.getclass(),"../");
        //如果放到服务器目录中，则使用下面的代码
        try {
            configuration.setdirectoryfortemplateloading(new file(tempdirpath));
        } catch (ioexception e2) {
            e2.printstacktrace();
        }
        //这里要设置取消使用local语言环境
        configuration.setlocalizedlookup(false);
        template template = null;
        try {
            // leavetemplet.ftl为要装载的模板
            template = configuration.gettemplate("leavetemplet.ftl","utf-8");
        } catch (ioexception e) {
            e.printstacktrace();
        }
        // 输出文档路径及名称
        file dir = new file(tempdirpath);
        file outfile = new file(tempdirpath + "/请假条.doc");
        if (!dir.isdirectory()) {
            dir.mkdir();
            if (!outfile.exists()) {
                try {
                    outfile.createnewfile();
                } catch (ioexception e) {
                    system.out.println("创建文件失败");
                    e.printstacktrace();
                }
            }
        }
        writer out = null;
        try {
            try {
                out = new bufferedwriter(new outputstreamwriter(new fileoutputstream(outfile), "utf-8"));
            } catch (unsupportedencodingexception e) {
                e.printstacktrace();
            }

        } catch (filenotfoundexception e1) {
            system.out.println("输出文件失败");
            e1.printstacktrace();
        }
        try {
            template.process(datamap, out);
            system.out.println("it's success!");
        } catch (templateexception e) {
            system.out.println("生成失败");
            e.printstacktrace();
        } catch (ioexception e) {
            e.printstacktrace();
        }
    }

    /**
     * 注意datamap里存放的数据key值要与模板中的参数相对应
     */
    private void getdata(map<string,object> datamap) {
        string imgbase64str = getimagestr(tempdirpath + "/leaderopinion_img.png");

        // 使用string的有参构造方法
        datamap.put("writedate","2018年10月13日");//填写日期
        datamap.put("name","haiyuking");//姓名
        datamap.put("dept","移动组");//部门
        datamap.put("leavetype","☑倒休 √年假 ✔事假 ☐病假 ☐婚假 ☐产假 ☐其他");//请假类型
        datamap.put("leavereason","倒休休息两天");//请假理由
        datamap.put("leavestartdate","2018年10月13日上午");//请假开始日期
        datamap.put("leaveenddate","2018年10月14日下午");//请假结束日期
        datamap.put("leaveday","2");//请假天数
        datamap.put("leaveleader","同意");//直属领导意见
        datamap.put("leavedeptleaderimg",imgbase64str);//部门领导意见

    }

    /**
     * 获取图片的base64值*/
    private string getimagestr(string imgfile) {
        inputstream in = null;
        byte[] data = null;
        try {
            in = new fileinputstream(imgfile);
            data = new byte[in.available()];
            in.read(data);
            in.close();
        } catch (ioexception e) {
            e.printstacktrace();
        }
        base64encoder encoder = new base64encoder();
        return encoder.encode(data);
    }
}