package com.iailab.module.model.mpk.common.utils;

import cn.hutool.core.io.FileUtil;
import com.iailab.module.model.mpk.common.MdkConstant;
import com.iailab.module.model.mpk.dto.MpkFileDTO;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.Velocity;

import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.io.StringWriter;
import java.nio.file.Files;
import java.util.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

@Slf4j
public class GenUtils {
    public static List<String> getTemplates(){
        List<String> templates = new ArrayList<String>();
        templates.add("abstract.java.vm");
        templates.add("impl.java.vm");
        templates.add("cpp.vm");
        templates.add("Jni.cpp.vm");
        templates.add("h.vm");
        templates.add("Jni.h.vm");
        return templates;
    }
    /**
     * 生成代码
     */
    public static void generatorCode(MpkFileDTO entity, ZipOutputStream zip){
        //设置velocity资源加载器
        Properties prop = new Properties();
        prop.put("file.resource.loader.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader");
        Velocity.init(prop);

        //封装模板数据
        Map<String, Object> map = new HashMap<>();
        map.put("pkgName",entity.getPkgName());
        map.put("modelMethods",entity.getModelMethods());
        map.put("pyName",entity.getPyName());
        map.put("pyModule",entity.getPyModule());

        VelocityContext context = new VelocityContext(map);

        //获取模板列表
        List<String> templates = getTemplates();
        for(String template : templates){
            //渲染模板
            StringWriter sw = drawTemplate(template,context);

            try {
                //添加到zip
                zip.putNextEntry(new ZipEntry(getFileName(template, entity.getPyName(),entity.getPkgName(),entity.getPyModule())));
                IOUtils.write(sw.toString(), zip, "UTF-8");
                IOUtils.closeQuietly(sw);
                zip.closeEntry();

                if (template.equals("cpp.vm")) {
                    File dirPath = new File("C:/DLUT/tmp/");
                    if (!dirPath.exists()) {
                        dirPath.mkdirs();
                    }
                    // 生成临时cpp文件
                    File cppFile = new File(dirPath.getAbsolutePath() + File.separator + entity.getPyName() + ".cpp");
                    cppFile.createNewFile();
                    OutputStream cppOutputStream = Files.newOutputStream(cppFile.toPath());
                    IOUtils.write(sw.toString(), cppOutputStream, "UTF-8");
                    IOUtils.closeQuietly(cppOutputStream);
                }

                if (template.equals("Jni.cpp.vm")) {
                    File dirPath = new File("C:/DLUT/tmp/");
                    if (!dirPath.exists()) {
                        dirPath.mkdirs();
                    }

                    // 生成临时Jni.cpp文件
                    File cppFile = new File(dirPath.getAbsolutePath() + File.separator + entity.getPyName() + ".cpp");
                    File jniCppFile = new File(dirPath.getAbsolutePath() + File.separator + entity.getPyName() + "Jni.cpp");
                    jniCppFile.createNewFile();
                    OutputStream jniCppOutputStream = Files.newOutputStream(jniCppFile.toPath());
                    IOUtils.write(sw.toString(), jniCppOutputStream, "UTF-8");
                    IOUtils.closeQuietly(jniCppOutputStream);
                    try {
                        // 根据file 通过cmd命令 生成dll
                        String dllSavePath = dirPath.getAbsolutePath() + File.separator + entity.getPyName() + ".dll";
                        String cppFilePath = cppFile.getAbsolutePath();
                        String jniCppFilePath = jniCppFile.getAbsolutePath();
//                        String command = "cmd.exe /c cl -o " + dllSavePath + " /LD " + jniCppFilePath + " " + cppFilePath;
                        String command = "cmd.exe /c cl -o " + dllSavePath + " /LD D:\\work\\mdk\\code\\makeDll\\src\\main\\java\\org\\example\\MakeDll.c D:\\work\\mdk\\code\\makeDll\\src\\main\\java\\org\\example\\MakeDll2.c";
                        Process process = Runtime.getRuntime().exec(command);
                        // 等待命令执行完成
                        process.waitFor();

                        File dllFile = new File(dllSavePath);
                        //添加到zip
                        zip.putNextEntry(new ZipEntry("dll" + File.separator + entity.getPyName() + ".dll"));
                        IOUtils.write(FileUtils.readFileToByteArray(dllFile), zip);
                        zip.closeEntry();

                        FileUtils.deleteDirectory(dirPath);
                    } catch (InterruptedException e) {
                        log.error("DLL生成失败" + entity.getPyName(), e);
                        throw new RuntimeException(e);
                    }
                }
            } catch (IOException e) {
                log.error("渲染模板失败，模型名称：" + entity.getPyName(), e);
                throw new RuntimeException("渲染模板失败，模型名称：" + entity.getPyName(), e);
            }
        }
    }

    /**
     * 获取文件名
     */
    public static String getFileName(String template, String moduleName, String pkgName, String pyModule) {
        // java
        if (template.equals("abstract.java.vm")) {
            return pkgName.replace(".", File.separator) + File.separator + moduleName + ".java";
        }
        if (template.equals("impl.java.vm")) {
            return pkgName.replace(".", File.separator) + File.separator + MdkConstant.IMPL + File.separator + moduleName + "Impl.java";
        }
        // c++
        if (template.equals("cpp.vm")) {
            return pyModule.replace(".", File.separator) + File.separator + moduleName + ".cpp";
        }
        if (template.equals("h.vm")) {
            return pyModule.replace(".", File.separator) + File.separator + moduleName + ".h";
        }
        // Jni c++
        if (template.equals("Jni.cpp.vm")) {
            return pyModule.replace(".", File.separator) + File.separator + MdkConstant.JNI + File.separator + moduleName + "Jni.cpp";
        }
        if (template.equals("Jni.h.vm")) {
            return pyModule.replace(".", File.separator) + File.separator + MdkConstant.JNI + File.separator + moduleName + "Jni.h";
        }
        return null;
    }

    /**
     * 渲染模板
     **/
    public static StringWriter drawTemplate(String template,Map<String, Object> map) {
        //模板数据
        VelocityContext context = new VelocityContext(map);
        return drawTemplate(template,context);
    }

    public static StringWriter drawTemplate(String template,VelocityContext context) {
        Properties prop = new Properties();
        prop.put("file.resource.loader.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader");
        Velocity.init(prop);

        StringWriter writer = new StringWriter();
        Template tpl = Velocity.getTemplate("template/" + template, "UTF-8");
        tpl.merge(context, writer);
        return writer;
    }

    public static void drawTemplate(String template,Map<String, Object> map,File toFile) throws IOException {
        VelocityContext context = new VelocityContext(map);
        drawTemplate(template,context,toFile);
    }

    public static void drawTemplate(String template,VelocityContext context,File toFile) throws IOException {
        StringWriter writer = drawTemplate(template,context);

        FileUtil.mkParentDirs(toFile);

        if (!toFile.exists()) {
            toFile.createNewFile();
        }

        FileUtil.writeUtf8String(writer.toString(),toFile);
    }
}