package com.ishchuk.antlr.json;

import com.ishchuk.antlr.json.model.*;
import com.ishchuk.antlr4.JSONgramBaseListener;
import com.ishchuk.antlr4.JSONgramLexer;
import com.ishchuk.antlr4.JSONgramParser;
import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.tree.ParseTreeWalker;

import java.util.*;


public class JsonListener extends JSONgramBaseListener {

    private JsonElement json;
    private boolean read;

    @Override
    public void enterJson(JSONgramParser.JsonContext ctx) {
        this.read = true;


    }

    public JsonElement getJsonFile() {
        return this.json;
    }

    private JsonElement treatContext(JSONgramParser.ContentContext ctx) {

        JsonElement elem;
        if (Objects.nonNull(ctx.object())) {
            elem = new JsonObject();
            for (JSONgramParser.KeyValueContext kv : ctx.object().keyValue()) {
                ((JsonObject)elem).getMap().put(kv.CHAINE().getSymbol().getText(), treatContext(kv.content()));
            }

        } else if (Objects.nonNull(ctx.table())) {
            elem = new JsonArray();
            for (JSONgramParser.ContentContext ct : ctx.table().content()) {
                ((JsonArray)elem).getArray().add(treatContext(ct));
            }
        } else {
            elem = new JsonPrimitive();
            ((JsonPrimitive)elem).setValue(ctx.getText());
        }

        return elem;
    }

    @Override
    public void enterContent(JSONgramParser.ContentContext ctx) {

        if (read) {
            if (Objects.nonNull(ctx.object())) {
                this.json = new JsonObject();
                ctx.object().keyValue().forEach(kv -> {
                    ((JsonObject)this.json).getMap().put(kv.CHAINE().getSymbol().getText(),
                            treatContext(kv.content()));
                });
            } else if (Objects.nonNull(ctx.table())) {
                this.json = new JsonArray();
                ctx.table().content().forEach(ct -> {
                    ((JsonArray)this.json).getArray().add(treatContext(ct));
                });
            }

            read = false;

        }
    }

    public String convertToYml() {
        StringBuilder builder = new StringBuilder();
        builder.append("---\n");
        if (Objects.nonNull(this.json)) {
            builder.append(this.json.toYml(""));
        }
        return builder.toString();
    }

    public static void main(String[] args) {
        System.out.println("Hello! With this application you can convert JSON file to YML file");
        System.out.println("------------------------------------------------------------------");

        Scanner in = new Scanner(System.in);
        System.out.println("Please, copy your json file here");
        String jsonString = "";
        while(in.hasNextLine()){
            jsonString += in.nextLine();
        }

        JSONgramLexer serverJSONgramLexer = new JSONgramLexer(CharStreams.fromString(jsonString));
        CommonTokenStream tokens = new CommonTokenStream( serverJSONgramLexer );
        JSONgramParser JSONgramParser = new JSONgramParser(tokens);
        ParseTreeWalker walker = new ParseTreeWalker();
        JsonListener jsonWalker = new JsonListener();
        walker.walk(jsonWalker, JSONgramParser.json());

        System.out.println("\nYour file:\n");
        System.out.println(jsonWalker.getJsonFile().toString());

        System.out.println("\nYml file:\n");
        System.out.println(jsonWalker.convertToYml());

    }

}