/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill 庄骞 (smallchill@163.com)
 */
package org.springblade.develop.support;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.TemplateType;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.annotations.Mapper;
import org.springblade.core.tool.utils.Func;
import org.springblade.core.tool.utils.StringUtil;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PropertiesLoaderUtils;

import java.io.IOException;
import java.util.*;
import java.util.stream.Collectors;

import static org.springblade.develop.constant.DevelopConstant.*;

/**
 * 代码生成器配置类
 *
 * @author Chill
 */
@Data
@Slf4j
public class BladeCodeGenerator {
	/**
	 * 代码风格
	 */
	private String codeStyle = SABER_NAME;
	/**
	 * 代码模块名称
	 */
	private String codeName;
	/**
	 * 模型编号
	 */
	private String modelCode;
	/**
	 * 模型实体类
	 */
	private String modelClass;
	/**
	 * 代码所在服务名
	 */
	private String serviceName = "blade-service";
	/**
	 * 代码生成的包名
	 */
	private String packageName = "org.springblade.test";
	/**
	 * 模版类型
	 */
	private String templateType;
	/**
	 * 作者信息
	 */
	private String author;
	/**
	 * 子表模型主键
	 */
	private String subModelId;
	/**
	 * 子表绑定外键
	 */
	private String subFkId;
	/**
	 * 树主键字段
	 */
	private String treeId;
	/**
	 * 树父主键字段
	 */
	private String treePid;
	/**
	 * 树名称字段
	 */
	private String treeName;
	/**
	 * 代码后端生成的地址
	 */
	private String packageDir;
	/**
	 * 代码前端生成的地址
	 */
	private String packageWebDir;
	/**
	 * 需要去掉的表前缀
	 */
	private String[] tablePrefix = {"blade_"};
	/**
	 * 需要生成的表名(两者只能取其一)
	 */
	private String[] includeTables = {"blade_test"};
	/**
	 * 需要排除的表名(两者只能取其一)
	 */
	private String[] excludeTables = {};
	/**
	 * 是否包含基础业务字段
	 */
	private Boolean hasSuperEntity = Boolean.TRUE;
	/**
	 * 是否包含包装器
	 */
	private Boolean hasWrapper = Boolean.TRUE;
	/**
	 * 是否包含远程调用
	 */
	private Boolean hasFeign = Boolean.FALSE;
	/**
	 * 是否包含服务名
	 */
	private Boolean hasServiceName = Boolean.FALSE;
	/**
	 * 基础业务字段
	 */
	private String[] superEntityColumns = {"create_time", "create_user", "create_dept", "update_time", "update_user", "status", "is_deleted"};
	/**
	 * 租户字段
	 */
	private String tenantColumn = "tenant_id";
	/**
	 * 数据库驱动名
	 */
	private String driverName;
	/**
	 * 数据库链接地址
	 */
	private String url;
	/**
	 * 数据库用户名
	 */
	private String username;
	/**
	 * 数据库密码
	 */
	private String password;
	/**
	 * 数据模型
	 */
	private Map<String, Object> model;
	/**
	 * 数据原型
	 */
	private List<Map<String, Object>> prototypes;
	/**
	 * 子数据模型
	 */
	private Map<String, Object> subModel;
	/**
	 * 子数据原型
	 */
	private List<Map<String, Object>> subPrototypes;

	/**
	 * 代码生成执行
	 */
	public void run() {
		// 主模块代码生成
		getAutoGenerator(getCustomMap(TEMPLATE_MAIN), getCustomFile(TEMPLATE_MAIN)).templateEngine(new BladeTemplateEngine(getOutputDir(), getOutputWebDir())).execute();
		// 子模块代码生成
		if (Func.equals(templateType, TEMPLATE_SUB) && StringUtil.isNotBlank(subModelId)) {
			getAutoGenerator(getCustomMap(TEMPLATE_SUB), getCustomFile(TEMPLATE_SUB)).templateEngine(new BladeTemplateEngine(getOutputDir(), getOutputWebDir())).execute();
		}
	}

	/**
	 * 设置 customMap
	 */
	private Map<String, Object> getCustomMap(String generateType) {
		List<Map<String, Object>> prototypeList;
		String[] split = packageName.split("\\.");
		String serviceCode = split[split.length - 1];
		Map<String, Object> customMap = new HashMap<>(11);
		customMap.put("generateType", generateType);
		customMap.put("codeName", codeName);
		customMap.put("serviceName", serviceName);
		customMap.put("serviceCode", serviceCode);
		customMap.put("packageName", packageName);
		customMap.put("tenantColumn", tenantColumn);
		customMap.put("hasWrapper", hasWrapper);
		customMap.put("hasServiceName", hasServiceName);
		customMap.put("templateType", templateType);
		customMap.put("author", author);
		customMap.put("subModelId", subModelId);
		customMap.put("subFkId", subFkId);
		customMap.put("treeId", treeId);
		customMap.put("treePid", treePid);
		customMap.put("treeName", treeName);
		customMap.put("subFkIdHump", StringUtil.underlineToHump(subFkId));
		customMap.put("treeIdHump", StringUtil.underlineToHump(treeId));
		customMap.put("treePidHump", StringUtil.underlineToHump(treePid));
		if (Func.equals(generateType, TEMPLATE_SUB)) {
			prototypeList = subPrototypes;
			customMap.put("model", subModel);
			customMap.put("prototypes", subPrototypes);
			customMap.put("modelCode", subModel.get("modelCode"));
			customMap.put("modelClass", subModel.get("modelClass"));
			customMap.put("modelTable", subModel.get("modelTable"));
		} else {
			prototypeList = prototypes;
			customMap.put("model", model);
			customMap.put("prototypes", prototypes);
			customMap.put("subModel", subModel);
			customMap.put("subPrototypes", subPrototypes);
			customMap.put("modelCode", model.get("modelCode"));
			customMap.put("modelClass", model.get("modelClass"));
			customMap.put("modelTable", model.get("modelTable"));
		}
		List<String> propertyImport = prototypeList.stream().filter(prototype -> {
			String propertyType = String.valueOf(prototype.get("propertyType"));
			return !"String".equals(propertyType) && !"Integer".equals(propertyType) && !"Long".equals(propertyType);
		}).map(prototype -> String.valueOf(prototype.get("propertyEntity"))).distinct().collect(Collectors.toList());
		customMap.put("propertyImport", propertyImport);
		return customMap;
	}

	/**
	 * 设置 customFile
	 */
	private Map<String, String> getCustomFile(String type) {
		Map<String, String> customFile = new HashMap<>(15);
		if (!Func.equals(type, TEMPLATE_SUB)) {
			customFile.put("menu.sql", "/templates/sql/menu.sql.btl");
		}
		customFile.put("entityVO.java", "/templates/api/entityVO.java.btl");
		customFile.put("entityDTO.java", "/templates/api/entityDTO.java.btl");
		if (hasWrapper) {
			customFile.put("wrapper.java", "/templates/api/wrapper.java.btl");
		}
		if (hasFeign) {
			customFile.put("feign.java", "/templates/api/feign.java.btl");
			customFile.put("feignclient.java", "/templates/api/feignclient.java.btl");
		}
		if (Func.isNotBlank(packageWebDir)) {
			if (Func.equals(codeStyle, SWORD_NAME)) {
				customFile.put("action.js", "/templates/sword/action.js.btl");
				customFile.put("model.js", "/templates/sword/model.js.btl");
				customFile.put("service.js", "/templates/sword/service.js.btl");
				customFile.put("list.js", "/templates/sword/list.js.btl");
				customFile.put("add.js", "/templates/sword/add.js.btl");
				customFile.put("edit.js", "/templates/sword/edit.js.btl");
				customFile.put("view.js", "/templates/sword/view.js.btl");
			} else if (Func.equals(codeStyle, SABER_NAME)) {
				customFile.put("api.js", "/templates/saber/" + templateType + "/api.js.btl");
				customFile.put("const.js", "/templates/saber/" + templateType + "/const.js.btl");
				if (!Func.equals(type, TEMPLATE_SUB)) {
					customFile.put("crud.vue", "/templates/saber/" + templateType + "/crud.vue.btl");
				}
			} else if (Func.equals(codeStyle, ELEMENT_NAME)) {
				customFile.put("api.js", "/templates/element/" + templateType + "/api.js.btl");
				customFile.put("const.js", "/templates/element/" + templateType + "/const.js.btl");
				if (!Func.equals(type, TEMPLATE_SUB)) {
					customFile.put("crud.vue", "/templates/element/" + templateType + "/crud.vue.btl");
				} else {
					customFile.put("sub.vue", "/templates/element/" + templateType + "/sub.vue.btl");
				}
			} else if (Func.equals(codeStyle, LEMON_NAME)) {
				customFile.put("data.ts", "/templates/lemon/" + templateType + "/data.ts.btl");
				customFile.put("Modal.vue", "/templates/lemon/" + templateType + "/Modal.vue.btl");
				customFile.put("data.data.ts", "/templates/lemon/" + templateType + "/data.data.ts.btl");
				if (!Func.equals(type, TEMPLATE_SUB)) {
					customFile.put("index.vue", "/templates/lemon/" + templateType + "/index.vue.btl");
				} else {
					customFile.put("lemonSub.vue", "/templates/lemon/" + templateType + "/sub.vue.btl");
				}

			}
		}
		return customFile;
	}

	private FastAutoGenerator getAutoGenerator(Map<String, Object> customMap, Map<String, String> customFile) {
		Properties props = getProperties();
		String url = Func.toStr(this.url, props.getProperty("spring.datasource.url"));
		String username = Func.toStr(this.username, props.getProperty("spring.datasource.username"));
		String password = Func.toStr(this.password, props.getProperty("spring.datasource.password"));
		return FastAutoGenerator.create(url, username, password)
			.globalConfig(builder -> builder.author(StringUtil.isBlank(author) ? props.getProperty("author") : author).dateType(DateType.TIME_PACK).enableSwagger().outputDir(getOutputDir()).disableOpenDir())
			.packageConfig(builder -> builder.parent(packageName).controller("controller").entity("entity").service("service").serviceImpl("service.impl").mapper("mapper").xml("mapper"))
			.strategyConfig(builder -> builder.addTablePrefix(tablePrefix).addInclude(Func.toStrArray(String.valueOf(customMap.get("modelTable")))).addExclude(excludeTables)
				.entityBuilder().naming(NamingStrategy.underline_to_camel).columnNaming(NamingStrategy.underline_to_camel).enableLombok().superClass("org.springblade.core.mp.base.BaseEntity").formatFileName("%sEntity").addSuperEntityColumns(superEntityColumns).enableFileOverride()
				.serviceBuilder().superServiceClass("org.springblade.core.mp.base.BaseService").superServiceImplClass("org.springblade.core.mp.base.BaseServiceImpl").formatServiceFileName("I%sService").formatServiceImplFileName("%sServiceImpl").enableFileOverride()
				.mapperBuilder().mapperAnnotation(Mapper.class).enableBaseResultMap().enableBaseColumnList().formatMapperFileName("%sMapper").formatXmlFileName("%sMapper").enableFileOverride()
				.controllerBuilder().superClass("org.springblade.core.boot.ctrl.BladeController").formatFileName("%sController").enableRestStyle().enableHyphenStyle().enableFileOverride()
			)
			.templateConfig(builder -> builder.disable(TemplateType.ENTITY)
				.entity("/templates/api/entity.java")
				.service("/templates/api/service.java")
				.serviceImpl("/templates/api/serviceImpl.java")
				.mapper("/templates/api/mapper.java")
				.xml("/templates/api/mapper.xml")
				.controller("/templates/api/controller.java"))
			.injectionConfig(builder -> builder.beforeOutputFile(
					(tableInfo, objectMap) -> System.out.println("tableInfo: " + tableInfo.getEntityName() + " objectMap: " + objectMap.size())
				).customMap(customMap).customFile(customFile)
			);
	}

	/**
	 * 获取配置文件
	 *
	 * @return 配置Props
	 */
	private Properties getProperties() {
		// 读取配置文件
		Resource resource = new ClassPathResource("/templates/code.properties");
		Properties props = new Properties();
		try {
			props = PropertiesLoaderUtils.loadProperties(resource);
		} catch (IOException e) {
			e.printStackTrace();
		}
		return props;
	}

	/**
	 * 生成到项目中
	 *
	 * @return outputDir
	 */
	public String getOutputDir() {
		return (Func.isBlank(packageDir) ? System.getProperty("user.dir") + "/blade-ops/blade-develop" : packageDir) + "/src/main/java";
	}


	/**
	 * 生成到Web项目中
	 *
	 * @return outputDir
	 */
	public String getOutputWebDir() {
		return (Func.isBlank(packageWebDir) ? System.getProperty("user.dir") : packageWebDir) + "/src";
	}

}