package TP2.asd;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import TP2.asd.Interface.*;
import TP2.asd.Eval.*;

public class Program{
    public static record ProgramImp(ArrayList<Function> instructions) implements ProgramI{
        public <H, S> S accept(ProgramVisitor<H, S> v, H h) {
            return v.visitProgram(this, h);
        }
    }

    public static record FunctionImp(Type type, String nom, ArrayList<Instruction> instructions)implements Function {
        public FunctionImp(Type type, String name, Instruction instruction) {
            this(type, name, new ArrayList<>() {{ add(instruction); }});
        }
    
        public <H, S> S accept(FunctionVisitor<H, S> v, H h) {
            return v.visitFunction(this, h);
        }
    }
    
    public static record ConstImp(int c) implements Expression{
        public <H, S> S accept(ExprVisitor<H, S> v, H h) {
            return v.visitConst(this, h);
        }   
    }
    
    public static record BinopExpressionImp(Op op,Expression e1, Expression e2) implements Expression{
        public <H, S> S accept(ExprVisitor<H, S> v, H h) {
            return v.visitBinOp(this, h);
        }
    
    }
    
    public static record Return_instrImp(Expression e) implements Instruction{
        public <H, S> S accept(InstrVisitor<H, S> v, H h) {
            return v.visitReturn(this,h);
        }
    }
    
    public static record Type_voidImp() implements Type{
        public String prettyprinter() {
            return "VOID";
        }
    }
    
    public static record Type_intImp() implements Type{
        public String prettyprinter() {
            return "INT";
        }
    }

    //Eval
    public static class ProgramEval implements ProgramVisitor<Map<String, Integer>, Integer> {
        @Override
        public Integer visitProgram(ProgramImp e, Map<String, Integer> h) {
            Integer result = null;
            FunctionEval functionEval = new FunctionEval();
            for (Function function : e.instructions()) {
                result = function.accept(functionEval, h);
            }
            return result;
        }    
    }

    public static class FunctionEval implements FunctionVisitor<Map<String, Integer>, Integer> {
        
        @Override
        public Integer visitFunction(FunctionImp e, Map<String, Integer> h) {
            InstructionEval instructionEval = new InstructionEval();
            Integer result = null;
            for (Instruction instr : e.instructions()) {
                result = instr.accept(instructionEval, h); 
            }
            return result;
        }
    }

    public static class InstructionEval implements InstrVisitor<Map<String,Integer>,Integer>{

        @Override
        public Integer visitReturn(Return_instrImp e, Map<String, Integer> h) {
           ExprEval exprEval = new ExprEval();
           return e.e().accept(exprEval, h);
        }
    }

    public static class ExprEval implements ExprVisitor<Map<String,Integer>,Integer>{
        public Integer visitConst(ConstImp c, Map<String,Integer> h){
            return c.c();
        }

        public Integer visitBinOp(BinopExpressionImp e, Map<String, Integer> h) {
            switch(e.op()) {
                case Op.PLUS: return e.e1().accept(this, h)+e.e2().accept(this, h);
                case Op.MINUS: return e.e1().accept(this, h)-e.e2().accept(this, h);
                case Op.TIMES: return e.e1().accept(this, h)*e.e2().accept(this, h);
                case Op.DIV: return e.e1().accept(this, h)/e.e2().accept(this, h);
                case Op.MOD: return e.e1().accept(this, h)%e.e2().accept(this, h);
                default: throw new IllegalArgumentException();
            }
        }
    }
}