package com.walker.jdbc.generator.util;

import com.walker.jdbc.generator.Constants;
import com.walker.jdbc.generator.GenerateException;
import org.apache.commons.configuration.Configuration;
import org.apache.commons.io.IOUtils;
import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.Velocity;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.IOException;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class GenPoUtils {

    private static final Logger logger = LoggerFactory.getLogger(GenPoUtils.class);

    public static final String PACKAGE_MODULE = "iplatform";

    public static List<String> getTemplates(String mbbs, boolean platformTable) {
        List<String> templates = new ArrayList<>();
//        if (mbbs.contains("po-old")) {
//            templates.add("potemplate/po.vm");
//        }
        if (mbbs.contains("to")) {
            templates.add("potemplate/vo.vm");
        }
        if (mbbs.contains("mapper")) {
            if(platformTable){
                templates.add("potemplate/mapper.vm");
            } else {
                templates.add("potemplate/mapper-camel.vm");
            }
        }
        if (mbbs.contains("po")) {
            if(platformTable){
                templates.add("potemplate/po-new.vm");
            } else {
                templates.add("potemplate/po-camel.vm");
            }
        }
        return templates;
    }

    public static void generatorCode(String mbbs
            , String table_name, String pk_name, String pk_type, List<Map<String, Object>> columns, ZipOutputStream zip) {
        Properties prop = new Properties();
        prop.put("file.resource.loader.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader");
        Velocity.init(prop);

        // 是否平台表名称：'s_'开头的表？
        table_name = table_name.toLowerCase();
        boolean platformTable = table_name.startsWith(Constants.TABLE_PREFIX_PLATFORM);

        VelocityContext context = new VelocityContext();
        Configuration config = GenUtils.getConfig();
        context.put("columns", columns);
        context.put("table_name", table_name);
        context.put("pk_name", pk_name);
//        if(platformTable){
//        } else {
//            context.put("pk_name", StringUtils.transferUnderlineName2Camel(pk_name, false));
//        }
        context.put("pk_type", pk_type);
        context.put("stringUtils", new StringUtils());

        List<String> templates = getTemplates(mbbs, platformTable);
        try {
            for (String template : templates) {
                StringWriter sw = new StringWriter();
                Template tpl = Velocity.getTemplate(template, "UTF-8");
                tpl.merge(context, sw);

                zip.putNextEntry(new ZipEntry(getFileName(template, table_name)));
                IOUtils.write(sw.toString(), zip, "UTF-8");
                IOUtils.closeQuietly(sw);
            }
        } catch (IOException e) {
            throw new GenerateException("渲染模板失败，表名：" + table_name, e);
        }
    }

    public static String getFileName(String template, String table_name) {
        logger.debug("template = {},", template);
//        String table_name_u = GenUtils.toUpperFirst(table_name);
        String table_name_u = null;
        if(table_name.startsWith(Constants.TABLE_PREFIX_PLATFORM)){
            // 2023-03-23 平台表不转驼峰，仅首字母大写
            table_name_u = StringUtils.toUpperCaseFirst(table_name);
        } else {
            // 业务表，转为驼峰方式
            table_name_u = StringUtils.transferUnderlineName2Camel(table_name, true);
        }
        logger.debug("table_name_u = {}", table_name_u);

        String packagePath = "main" + File.separator + "java" + File.separator + "com" + File.separator + PACKAGE_MODULE + File.separator + "model";
        String[] ts = template.split("/");
//        String template_type = ts[0];
        String template_name = ts[1];
        if (template_name.equals("po.vm")) {
            return packagePath + File.separator + "po" + File.separator + table_name_u + ".java";
        }
        if (template_name.equals("po-new.vm") || template_name.equals("po-camel.vm")) {
//            return packagePath + File.separator + "newpo" + File.separator + table_name_u + ".java";
            return packagePath + File.separator + "po" + File.separator + table_name_u + ".java";
        }
        if (template_name.equals("vo.vm")) {
            return packagePath + File.separator + "to" + File.separator + table_name_u + "_t.java";
        }
        if (template_name.equals("mapper.vm") || template_name.equals("mapper-camel.vm")) {
            return packagePath + File.separator + "mapper" + File.separator + table_name_u + "_mapper.java";
        }
        return null;
    }
}