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;
import freemarker.template.TemplateException;

/**
 * 自动生成Service and ServiceImpl
 * 
 * @author chenjiahe 2019年09月08日16:57:47
 * 
 */
public class GeneratorServiceUtil {

    private static String templateDir = "/ftl";//获取模板路径

    private static String templateServiceName = "Service.tpl";//service名称
    private static String templateServiceImplName = "ServiceImpl.tpl";//serviceImpl名称
    /**
     * @param cl 实体类
     * @param urlData 配置信息
     * @throws Exception
     */
    @SuppressWarnings("deprecation")
    public static void generatorService(Class<?> cl, UrlData urlData) throws Exception {
        try {
        	//获取类的包名
        	String className = cl.getSimpleName();
        	String[] strs = cl.getName().split("\\.");
        	String packageName = "";
        	for(int i=0;i<strs.length-1;i++) {
        		packageName += "."+strs[i];
        	}
        	packageName = packageName.replaceFirst(".","");
        	
        	// 类名首字母小写
        	String initial = className.substring(0, 1).toLowerCase();
        	String classNamex = initial + className.substring(1, className.length());
        	// 取得本类的全部属性

			String targetFile = urlData.getServiceUrl()[1].replace(".", "/")+"/";
			targetFile += className + "Service.java";
			//补全路径
			targetFile = "./"+urlData.getServiceUrl()[0].replace(".", "/")+"/"+targetFile;
			File fileService = new File(targetFile);
			if(fileService.exists()){
				//存在就结束
				return;
			}


			String targetFile2 = urlData.getServiceImplUrl()[1].replace(".", "/")+"/";
			targetFile2 += className + "ServiceImpl.java";
			//补全路径
			targetFile2 = "./"+urlData.getServiceImplUrl()[0].replace(".", "/")+"/"+targetFile2;
			File fileServiceImple = new File(targetFile2);
			if(fileServiceImple.exists()){
				System.out.println("fileServiceImple:"+targetFile2);
				//存在就结束
				return;
			}

			//获取模板
			Configuration configuration = new Configuration(Configuration.getVersion());
			configuration.setClassForTemplateLoading(GeneratorServiceUtil.class, templateDir);
			//configuration.setTemplateLoader(new ClassTemplateLoader(this.getClass(), "填你的resource下的路径，比如/ftl"));

        	// 获取或创建模板
        	Template template = configuration.getTemplate(templateServiceName);
        	
        	// 创建数据模型
        	HashMap<String, Object> root = new HashMap<String, Object>();
        	root.put("className", classNamex);
        	root.put("classNameUP", className);
        	root.put("packageName", packageName);
        	root.put("servicePack", urlData.getServiceUrl()[1]);
        	root.put("actionPack", urlData.getActionUrl()[1]);
        	

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

			if(!CommonTool.checkNotNull(urlData.getTotalUrl())) {
				System.err.println("没有设置总包路径");
				return;
			}
			root.put("TotalPackageName",urlData.getTotalUrl());
        	root.put("serviceImpPack", urlData.getServiceImplUrl()[1]);
        	root.put("daoPack", urlData.getDaoUrl()[1]);
        	//serviceImp文件
        	// 获取或创建模板
        	Template template2 = configuration.getTemplate(templateServiceImplName);

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