package com.simon.springbootchm.util;

import com.jfinal.kit.StrKit;
import com.jfinal.template.Engine;
import com.jfinal.template.Template;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Collections;
import java.util.Map;
import java.util.StringTokenizer;

/**
 * 执行命令工具类
 *
 * @author Simon
 */
public class CommandUtils {
    private Logger logger = LoggerFactory.getLogger(this.getClass());

    private static final String ROOT_PATH = CommandUtils.class.getClassLoader().getResource("").getPath().substring(1);
    private static final Engine engine = Engine.use().setBaseTemplatePath(ROOT_PATH);

    public static String getPath(String projectName) {
        return ROOT_PATH + projectName;
    }

    public static Template getTemplate(String projectName, String name) {
        return engine.getTemplate(projectName + '/' + name);
    }

    private ProcessBuilder pb;
    private String command;
    private String[] envp;
    private File dir;
    private String[] cmdarray;

    public CommandUtils(String command, String[] envp, String dirPath) {
        this.envp = envp;
        if (dirPath != null) {
            dir = new File(dirPath);
        }
        if (command.length() == 0) {
            throw new IllegalArgumentException("Empty command");
        }
        this.command = command;
        StringTokenizer st = new StringTokenizer(command);
        cmdarray = new String[st.countTokens()];
        for (int i = 0; st.hasMoreTokens(); i++) {
            cmdarray[i] = st.nextToken();
        }
        getPb();
    }

    public CommandUtils(String[] cmdarray, String[] envp, String dirPath) {
        this.envp = envp;
        if (dirPath != null) {
            dir = new File(dirPath);
        }
        this.cmdarray = cmdarray;
        this.command = StrKit.join(cmdarray);
        getPb();
    }

    public ProcessBuilder getPb() {
        if (pb == null) {
            pb = new ProcessBuilder(cmdarray);
            Map<String, String> env = pb.environment();
            if (envp != null) {
                for (String envstring : envp) {
                    if (envstring.indexOf((int) '\u0000') != -1) {
                        envstring = envstring.replaceFirst("\u0000.*", "");
                    }
                    int eqlsign = envstring.indexOf('=', 1);
                    if (eqlsign != -1) {
                        env.put(envstring.substring(0, eqlsign), envstring.substring(eqlsign + 1));
                    }
                }
            }
            pb.redirectErrorStream(true);//重定向错误流到输入流,一起打印,防止缓存满
            pb.directory(dir);
        }
        return pb;
    }

    public void run() {
        logger.info("command:{} dir:{}", command, dir);
        BufferedReader reader = null;
        try {
            Process p = getPb().start();
            reader = new BufferedReader(new InputStreamReader(p.getInputStream()));
            String s = null;
            int a = 0;
            while ((s = reader.readLine()) != null) {
                if (a > 0) {
                    String join = String.join("", Collections.nCopies(a + 80, "\b"));
                    logger.info(join);
                    a = 0;
                }
                if (!"".equals(s)) {
                    if (s.matches("[^\\n]*\\[[=*>\\s]+\\][^\\n]*")) {
                        a = s.length();
                        logger.info(s);
                    } else {
                        logger.info(s);
                    }
                }
            }
            p.waitFor();
            logger.info("execute end");
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (reader != null) {
                try {
                    reader.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    public static void run(String command, String[] envp, String dir) {
        new CommandUtils(command, envp, dir).run();
    }
}