package com.hx.auto;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.HashMap;

import com.hx.auto.common.UrlData;
import com.hx.auto.util.CommonTool;

import freemarker.template.Configuration;
import freemarker.template.DefaultObjectWrapper;
import freemarker.template.Template;

/**
 * 自动生成DAO
 * @author chenjiahe 2019年09月08日16:57:47
 * 
 */
public class GeneratorDaoUtil {

    private static String templateDir = "/ftl";//获取模板路径
    private static String templateName = "Dao.tpl";//action模板名称

    /**生成Dao
     * @param cl 实体类
     * @param urlData 生成配置信息
     * @throws Exception
     */
    public static void generatorDao(Class<?> cl,UrlData urlData) throws Exception {
        try {
            // 反射start
            // 类名
            String entityName = cl.getSimpleName();
            // 类名首字母小写
            String initial = entityName.substring(0, 1).toLowerCase();
            String entityNameSmall = initial + entityName.substring(1, entityName.length());

			//生成文件路径
			String targetFile = urlData.getDaoUrl()[1].replace(".", "/")+"/";
			//生成文件名称
			targetFile += entityName + "Mapper.java";
			//映射文件的文件夹
			//补全路径
			targetFile = "./"+urlData.getDaoUrl()[0].replace(".", "/")+"/"+targetFile;
			File file = new File(targetFile);
			if(file.exists()){
				//System.out.println("存在333...:");
				//存在就结束
				return;
			}

			//获取实体类包名
            String[] strs = cl.getName().split("\\.");
            String packageName = "";
            //去掉类名
            for(int i=0;i<strs.length-1;i++) {
            	packageName += "."+strs[i];
            }
            packageName = packageName.replaceFirst(".", "");

			//获取模板
			Configuration configuration = new Configuration(Configuration.getVersion());
			configuration.setClassForTemplateLoading(GeneratorDaoUtil.class, templateDir);
			//configuration.setTemplateLoader(new ClassTemplateLoader(this.getClass(), "填你的resource下的路径，比如/ftl"));
			Template template = configuration.getTemplate(templateName);
        	
        	// 创建数据模型
        	HashMap<String, Object> root = new HashMap<String, Object>();
        	
        	//action包名
        	if(!CommonTool.checkNotNull(urlData.getActionUrl())) {
        		System.err.println("没有生成action路径");
        		return;
        	}
        	root.put("packageName",urlData.getActionUrl()[1]);
        	//实体类的类名
        	root.put("entityName", entityName);
        	//实体类的类名（首字母小写）
        	root.put("entityNameSmall", entityNameSmall);
        	
        	//dao的包名
        	if(!CommonTool.checkNotNull(urlData.getDaoUrl())) {
        		System.err.println("没有dao路径");
        		return;
        	}

			if(!CommonTool.checkNotNull(urlData.getTotalUrl())) {
				System.err.println("没有设置总包路径");
				return;
			}
			root.put("TotalPackageName",urlData.getTotalUrl());
        	root.put("DAOPackageName",urlData.getDaoUrl()[1]);
        	//实体类的包名
        	root.put("entityPackageName",packageName);

        	System.out.println("actionUrl:"+targetFile);
        	// 将模板和数据模型合并 输出到Console
        	Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(targetFile), "UTF-8"));
        	template.process(root, out);
        	out.flush();
        	out.close();
        } catch (IOException e) {
            e.printStackTrace();
        } 
    }
}