package cn.huge.module.buildcode.service;

import cn.huge.module.buildcode.domain.bo.UserTabColsBO;
import cn.huge.module.buildcode.dao.mapper.BuildCodeMapper;
import com.google.common.collect.Maps;
import freemarker.core.ParseException;
import freemarker.template.*;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import javax.annotation.PostConstruct;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.net.MalformedURLException;
import java.net.URISyntaxException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @title: 代码生成器服务层基类
 * @description: 使用单元测试直接对buildAll执行测试完成代码生成
 * @company: hugeinfo
 * @author: liyj
 * @time: 2021-11-05 16:51:48
 * @version: 1.0.0
 */
@Slf4j
@Service
public class BuildCodeService {

    public static void main(String[] args) throws MalformedURLException, URISyntaxException {

    }

    @Autowired
    private BuildCodeMapper buildCodeMapper;

    private Configuration cfg;
    private static StringBuffer ROOT_MAIN_PATH = new StringBuffer(System.getProperty("user.dir")).append(("/src/main/java/"));
    private static StringBuffer ROOT_TEST_PATH = new StringBuffer(System.getProperty("user.dir")).append(("/src/test/java/"));

    @PostConstruct
    private void init() throws IOException {
        try {
            Configuration configuration = new Configuration(Configuration.getVersion());
            configuration.setClassForTemplateLoading(BuildCodeService.class, "/template");
            configuration.setDefaultEncoding("UTF-8");
            this.cfg = configuration;
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取表数据属性
     * @param packagePath 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     * @return Map
     */
    private Map getTplData(String packagePath, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = Maps.newHashMap();
        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        tplData.put("packageName", packageName);
        tplData.put("className", className);
        tplData.put("author", author);
        tplData.put("version", version);
        tplData.put("createTime", format.format(new Date()));
        tplData.put("basePackage", packagePath);
        tplData.put("basePackagePath", packagePath.replace(".", "/") + "/");
        List<UserTabColsBO> list = buildCodeMapper.getTabInfo(tableSchema, tableName);
        if (!CollectionUtils.isEmpty(list)) {
            UserTabColsBO vo = list.get(0);
            tplData.put("tableName", vo.getTableName());
            tplData.put("tableComments", vo.getTableComments());
            tplData.put("cols", list);
            tplData.put("hasDate", 0);
            tplData.put("hasClob", 0);
            tplData.put("hasBlob", 0);
            for (UserTabColsBO bo : list) {
                if (bo.getHasDate() > 0) {
                    tplData.put("hasDate", 1);
                }
            }
        }
        return tplData;
    }

    /**
     * 生成所有层的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildAll(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            // 必须
            this.buildPo(tplData);
            this.buildBo(tplData);
            this.buildDaoMapper(tplData);
            this.buildDaoMapperXml(tplData);
            this.buildService(tplData);
            this.buildWebController(tplData);
            // 可选
//            this.buildWechatController(tplData);
//            this.buildJson(tplData);
//            this.buildJsForm(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成PO层的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildPo(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildPo(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成BO层的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildBo(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildBo(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成Mapper的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildDaoMapper(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildDaoMapper(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成MapperXml的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildDaoMapperXml(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildDaoMapperXml(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成service层的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildService(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildService(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成webController层的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildWebController(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildWebController(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成wechatController层的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildWechatController(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildWechatController(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成Controller层测试的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildControllerTest(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildControllerTest(tplData);

        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成service层测试的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildServiceTest(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildServiceTest(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成实体json的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildJson(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildJson(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }

    /**
     * 生成实体js的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名

     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildJsForm(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildJsForm(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }


    private void buildFile(Template template, Map tplData, String path) throws IOException, TemplateException {
        File file = new File(path);
        File dir = new File(file.getParent());
        if (!dir.exists()) {
            dir.mkdirs();
        }
        if (!file.exists()) {
            file.createNewFile();
        } else {  //文件存在就做备份
            int i = 1;
            File bakFile = new File(path + i + ".bak");
            while (bakFile.exists()) {
                i++;
                bakFile = new File(path + i + ".bak");
            }
            file.renameTo(bakFile);
        }

        FileOutputStream fos = new FileOutputStream(new File(path));
        template.process(tplData, new OutputStreamWriter(fos, "utf-8"));
        fos.flush();
        fos.close();
    }

    /**
     * 创建持久化对象
     *
     * @param tplData
     */
    private void buildPo(Map tplData) throws TemplateException, IOException {
        Template template = cfg.getTemplate("po.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/domain/po/")
                .append(tplData.get("className"))
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建业务对象
     *
     * @param tplData
     */
    private void buildBo(Map tplData) throws TemplateException, IOException {
        Template template = cfg.getTemplate("bo.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/domain/bo/")
                .append(tplData.get("className")).append("BO")
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建Mybatis持久层接口
     *
     * @param tplData
     */
    private void buildDaoMapper(Map tplData) throws TemplateException, IOException {
        Template template = cfg.getTemplate("dao.mapper.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/dao/mapper/")
                .append(tplData.get("className")).append("Mapper")
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建Mybatis Mapper.xml文件
     *
     * @param tplData
     */
    private void buildDaoMapperXml(Map tplData) throws TemplateException, IOException {
        Template template = cfg.getTemplate("dao.mapperXml.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/dao/mapper/xml/")
                .append(tplData.get("className")).append("Mapper")
                .append(".xml");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建服务层
     *
     * @param tplData
     */
    private void buildService(Map tplData) throws TemplateException, IOException {
        String className = (String) tplData.get("className");
        if (Character.isLowerCase(className.charAt(0))){
            tplData.put("lowerName", className);
        }else{
            String lowerName = (new StringBuilder()).append(Character.toLowerCase(className.charAt(0))).append(className.substring(1)).toString();
            tplData.put("lowerName", lowerName);
        }

        Template template = cfg.getTemplate("service.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/service/")
                .append(tplData.get("className")).append("Service")
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建web控制层
     *
     * @param tplData
     */
    private void buildWebController(Map tplData) throws TemplateException, IOException {
        tplData.put("packagePath", ((String) tplData.get("packageName")).toLowerCase());
        String className = (String) tplData.get("className");
        if (Character.isLowerCase(className.charAt(0))){
            tplData.put("classPath", className);
            tplData.put("lowerName", className);
        }else{
            String lowerName = (new StringBuilder()).append(Character.toLowerCase(className.charAt(0))).append(className.substring(1)).toString();
            tplData.put("classPath", lowerName);
            tplData.put("lowerName", lowerName);
        }

        Template template = cfg.getTemplate("webController.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/controller/web/")
                .append(tplData.get("className")).append("WebController")
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建wechat控制层
     *
     * @param tplData
     */
    private void buildWechatController(Map tplData) throws TemplateException, IOException {
        tplData.put("packagePath", ((String) tplData.get("packageName")).toLowerCase());
        String className = (String) tplData.get("className");
        if (Character.isLowerCase(className.charAt(0))){
            tplData.put("classPath", className);
            tplData.put("lowerName", className);
        }else{
            String lowerName = (new StringBuilder()).append(Character.toLowerCase(className.charAt(0))).append(className.substring(1)).toString();
            tplData.put("classPath", lowerName);
            tplData.put("lowerName", lowerName);
        }

        Template template = cfg.getTemplate("wechatController.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/controller/wechat/")
                .append(tplData.get("className")).append("WechatController")
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建服务层单元测试
     *
     * @param tplData
     */
    private void buildServiceTest(Map tplData) throws TemplateException, IOException {
        String className = (String) tplData.get("className");
        if (Character.isLowerCase(className.charAt(0))){
            tplData.put("lowerName", className);
        }else{
            String lowerName = (new StringBuilder()).append(Character.toLowerCase(className.charAt(0))).append(className.substring(1)).toString();
            tplData.put("lowerName", lowerName);
        }

        Template template = cfg.getTemplate("serviceTest.ftl");
        StringBuffer path = new StringBuffer(ROOT_TEST_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/")
                .append(tplData.get("className")).append("ServiceTest")
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建控制层单元测试
     *
     * @param tplData
     */
    private void buildControllerTest(Map tplData) throws TemplateException, IOException {
        tplData.put("packagePath", ((String) tplData.get("packageName")).toLowerCase());
        String className = (String) tplData.get("className");
        if (Character.isLowerCase(className.charAt(0))){
            tplData.put("classPath", className);
            tplData.put("lowerName", className);
        }else{
            String lowerName = (new StringBuilder()).append(Character.toLowerCase(className.charAt(0))).append(className.substring(1)).toString();
            tplData.put("classPath", lowerName);
            tplData.put("lowerName", lowerName);
        }

        Template template = cfg.getTemplate("controllerTest.ftl");
        StringBuffer path = new StringBuffer(ROOT_TEST_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/")
                .append(tplData.get("className")).append("ControllerTest")
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建Json代码
     *
     * @param tplData
     */
    private void buildJson(Map tplData) throws TemplateException, IOException {
        Template template = cfg.getTemplate("json.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/domain/json/")
                .append(tplData.get("className"))
                .append(".json");
        this.buildFile(template, tplData, path.toString());
    }

    /**
     * 创建界面React JS代码
     *
     * @param tplData
     */
    private void buildJsForm(Map tplData) throws TemplateException, IOException {
        Template template = cfg.getTemplate("jsform.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/domain/js/")
                .append(tplData.get("className"))
                .append(".js");
        this.buildFile(template, tplData, path.toString());
    }
}
/**
 * -------------------_ooOoo_-------------------
 * ------------------o8888888o------------------
 * ------------------88" . "88------------------
 * ------------------(| -_- |)------------------
 * ------------------O\  =  /O------------------
 * ---------------____/`---'\____---------------
 * -------------.'  \\|     |//  `.-------------
 * ------------/  \\|||  :  |||//  \------------
 * -----------/  _||||| -:- |||||-  \-----------
 * -----------|   | \\\  -  /// |   |-----------
 * -----------| \_|  ''\---/''  |   |-----------
 * -----------\  .-\__  `-`  ___/-. /-----------
 * ---------___`. .'  /--.--\  `. . __----------
 * ------."" '<  `.___\_<|>_/___.'  >'"".-------
 * -----| | :  `- \`.;`\ _ /`;.`/ - ` : | |-----
 * -----\  \ `-.   \_ __\ /__ _/   .-` /  /-----
 * ======`-.____`-.___\_____/___.-`____.-'======
 * -------------------`=---='
 * ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 * ---------佛祖保佑---hugeinfo---永无BUG----------
 */