package com.hx.auto.manage.xml.scan.util;

import com.gitee.sunchenbin.mybatis.actable.annotation.Column;
import com.gitee.sunchenbin.mybatis.actable.annotation.Table;
import com.gitee.sunchenbin.mybatis.actable.constants.MySqlTypeConstant;
import com.hx.auto.GeneratorMapperUtil;
import com.hx.auto.GeneratorReadXmlUtil;
import com.hx.auto.common.JdbcType;
import com.hx.auto.common.ReadEntityData;
import com.hx.auto.util.CommonTool;
import com.hx.auto.util.GeneratorClassParentUtil;
import freemarker.template.Configuration;
import freemarker.template.DefaultObjectWrapper;
import freemarker.template.Template;

import java.io.*;
import java.lang.reflect.Field;
import java.util.HashMap;

/**
 * 自动生成mapper.xml
 * 
 * @author chenjiahe 2019年09月08日23:57:47
 * 
 */
public class CreateMapperUtil {

    private static String templateDir = "/ftl";//获取模板路径
    private static String templateName = "Mapper.tpl";//action模板名称
    
    /**生成mapper.xml
     * @param cl 实体类
     * @param configUtil 生成配置信息
     * @throws Exception
     */
    public static void generatorMapper(Class<?> cl, ConfigUtil configUtil) throws Exception {
        try {

			// 反射start
			// 类名
			String entityName = cl.getSimpleName();

			//生成文件路径
			String targetFile = configUtil.getXmlUrl().replace(".", "/")+"/";
			//生成文件名称
			targetFile += entityName + "Mapper.xml";
			//补全路径
			targetFile = "./"+configUtil.getPackFileName().replace(".", "/")+"/"+targetFile;
			//获取是否已经生成过文件，拿取自定义的内容
			String xmlData = GeneratorReadXmlUtil.readMapperXml(targetFile);

			//如果不存在就直接跳过
			if("false".equals(xmlData)){
				return;
			}

            // 类名首字母小写
            String initial = entityName.substring(0, 1).toLowerCase();
            String entityNameSmall = initial + entityName.substring(1, entityName.length());
            
            //获取实体类包名
            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(GeneratorMapperUtil.class, templateDir);
        	// 获取或创建模板
        	Template template = configuration.getTemplate(templateName);
        	
        	//实体类所有的字段
        	//用来存储数据
			ReadEntityData readEntityData = new ReadEntityData();
			//表名称
			Table table = cl.getAnnotation(Table.class);
			readEntityData.setTableName(table.name());
        	// 取得本类的全部属性
			Field[] fields = cl.getDeclaredFields();
			fields = GeneratorClassParentUtil.getPatentFields(fields,cl);
			for (Field field:fields) {
				// 判断方法中是否有指定注解类型的注解
				boolean hasAnnotation = field.isAnnotationPresent(Column.class);
				if (hasAnnotation) {
					// 根据注解类型返回方法的指定类型注解
					Column column = field.getAnnotation(Column.class);

					//判断是不是
					boolean isBol = false;
					Integer isBlob = 0;
					if(column.type().equals(MySqlTypeConstant.TEXT)||column.type().equals(MySqlTypeConstant.LONGTEXT)
					||column.type().equals(MySqlTypeConstant.LONGBLOB)){
						isBol = true;
						isBlob = 1;
					}

					//类型
					//String type = column.type().toUpperCase();
					String type = JdbcType.jdbcTypeData(field.getType().getTypeName().toUpperCase(),isBol);
					//主键
					if (column.isKey()) {
						readEntityData.setEntityIdName(field.getName());
						readEntityData.setEntityIdData("#{"+field.getName()+"}");
						if(!CommonTool.checkNotNull(column.name())) {
							readEntityData.setTableIdName(field.getName());
						}else {
							readEntityData.setTableIdName(column.name());
						}
						readEntityData.setKeyType(type);
						continue;
					}
					//存储数据
					if(!CommonTool.checkNotNull(column.name())) {
						readEntityData.fielData(field.getName(),field.getName(),type,isBlob,"#{"+field.getName()+"}");
					}else {
						readEntityData.fielData(field.getName(),column.name(),type,isBlob,"#{"+field.getName()+"}");
					}
				}
			}
        	// 创建数据模型
        	HashMap<String, Object> root = new HashMap<String, Object>();

        	root.put("packageEntityName",packageName);
        	//实体类的类名
        	root.put("entityName", entityName);
        	//实体类的类名（首字母小写）
        	root.put("entityNameSmall", entityNameSmall);
        	//表字段 数据
        	root.put("fieldData",readEntityData);
        	
        	root.put("sqlSentence","${sqlSentence}");

        	root.put("DAOPackageName",configUtil.getDaoUrl());
        	//实体类的包名
        	root.put("entityPackageName",packageName);

			//获取是否已经生成过文件，拿取自定义的内容
			root.put("customData",xmlData);

        	System.out.println("mapperUrl:"+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();
        } 
    }
}