package burp.yaml;

import burp.yaml.template.Rule;
import burp.yaml.template.Rules;
import org.yaml.snakeyaml.DumperOptions;
import org.yaml.snakeyaml.Yaml;
import org.yaml.snakeyaml.constructor.Constructor;
import org.yaml.snakeyaml.nodes.Tag;
import org.yaml.snakeyaml.representer.Representer;

import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.nio.charset.StandardCharsets;
import java.util.*;

public class SetConfig {

    private Map<String, Object[][]> ruleConfig = LoadConfig.getRules();

    public void format() {
        DumperOptions dop = new DumperOptions();
        dop.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
        Representer representer = new Representer();
        representer.addClassTag(RulesConfig.class, Tag.MAP);
        Yaml yaml = new Yaml(new Constructor(), representer, dop);
        RulesConfig con = new RulesConfig();
        List<Rules> rls = new ArrayList<>();

        ruleConfig.keySet().forEach(i->
        {
            Rules rlsTmp = new Rules();
            rlsTmp.setType(i);
            List<Rule> rl = new ArrayList<>();
            for (Object[] objects : ruleConfig.get(i)) {
                Rule rlTmp = new Rule();
                rlTmp.setName((String) objects[1]);
                rlTmp.setLoaded((Boolean) objects[0]);
                rlTmp.setRegex((String) objects[2]);
                rlTmp.setColor((String) objects[3]);
                rlTmp.setScope((String) objects[4]);
                rlTmp.setEngine((String) objects[5]);
                rl.add(rlTmp);
            }
            rlsTmp.setRule(rl);
            rls.add(rlsTmp);
        });
        con.setRules(rls);
        File f = new File(LoadConfig.getConfigPath());
        try{
            Writer ws = new OutputStreamWriter(new FileOutputStream(f), StandardCharsets.UTF_8);
            yaml.dump(con,ws);
        }catch (Exception ex){
            ex.printStackTrace();
        }
    }

    public void edit(Vector data, int select, String type) {
        ruleConfig = LoadConfig.getRules();
        ruleConfig.get(type)[select] = data.toArray();
        this.format();
    }

    public void add(Vector data, String type) {
        ruleConfig = LoadConfig.getRules();
        ArrayList<Object[]> x = new ArrayList<>(Arrays.asList(ruleConfig.get(type)));
        x.add(data.toArray());
        ruleConfig.put(type,x.toArray(new Object[x.size()][]));
        this.format();
    }
    public void remove(int select,String type) {
        ruleConfig = LoadConfig.getRules();
        ArrayList<Object[]> x = new ArrayList<>(Arrays.asList(ruleConfig.get(type)));
        x.remove(select);
        ruleConfig.put(type,x.toArray(new Object[x.size()][]));
        this.format();
    }

    public void rename(String oldName, String newName) {
        ruleConfig = LoadConfig.getRules();
        ruleConfig.put(newName, ruleConfig.remove(oldName));
        this.format();
    }

    public void deleteRules(String Rules) {
        ruleConfig = LoadConfig.getRules();
        ruleConfig.remove(Rules);
        this.format();
    }
    public String newRules() {
        int i = 0;
        ruleConfig = LoadConfig.getRules();
        String name = "New ";
        Object[][] data = new Object[][]{{false, "New Name", "(New Regex)", "gray", "any", "nfa"}};
        while (ruleConfig.containsKey(name + i)) {
            i++;
        }
        ruleConfig.put(name+i, data);
        this.format();
        return name+i;
    }
}