package st;

import syntaxtree.*;
import visitor.*;
import java.util.*;
import misc.*;

/**
 * Provides default methods which visit each node in the tree in depth-first
 * order.  Your visitors may extend this class.
 */
public class SymTableVis extends GJDepthFirst<String,ArrayList<TypeInstance>> {

    public HashMap<String,AbstractInstance> symt = new HashMap<>();


    /**
     * f0 -> "class"
     * f1 -> Identifier()
     * f2 -> "{"
     * f3 -> "public"
     * f4 -> "static"
     * f5 -> "void"
     * f6 -> "main"
     * f7 -> "("
     * f8 -> "String"
     * f9 -> "["
     * f10 -> "]"
     * f11 -> Identifier()
     * f12 -> ")"
     * f13 -> "{"
     * f14 -> ( VarDeclaration() )*
     * f15 -> ( Statement() )*
     * f16 -> "}"
     * f17 -> "}"
     */
    public String visit(MainClass n, ArrayList<TypeInstance> argu) {

        ArrayList<TypeInstance> attr_list = new ArrayList<TypeInstance>();

        n.f0.accept(this, argu);
        n.f1.accept(this, argu);
        n.f2.accept(this, argu);
        n.f3.accept(this, argu);
        n.f4.accept(this, argu);
        n.f5.accept(this, argu);
        n.f6.accept(this, argu);
        n.f7.accept(this, argu);
        n.f8.accept(this, argu);
        n.f9.accept(this, argu);
        n.f10.accept(this, argu);
        n.f11.accept(this, argu);
        n.f12.accept(this, argu);
        n.f13.accept(this, argu);
        n.f14.accept(this, attr_list);
        n.f15.accept(this, argu);
        n.f16.accept(this, argu);
        n.f17.accept(this, argu);


        PrintFilter.print("Processing main", true);

        String id = n.f1.f0.tokenImage;

        ClassInstance type = new ClassInstance(id);
        type.set_attrs(attr_list);
        PrintFilter.print("Inserting " + id + " => " + type, true);
        symt.put(id, type);

        return null;

    }

    /**
     * f0 -> "class"
     * f1 -> Identifier()
     * f2 -> "{"
     * f3 -> ( VarDeclaration() )*
     * f4 -> ( MethodDeclaration() )*
     * f5 -> "}"
     */
    public String visit(ClassDeclaration n, ArrayList<TypeInstance> argu) {

        ArrayList<TypeInstance> attr_list = new ArrayList<TypeInstance>();
        // ArrayList<TypeInstance> mtd_list = new ArrayList<TypeInstance>();

        n.f0.accept(this, argu);
        n.f1.accept(this, argu);
        n.f2.accept(this, argu);
        n.f3.accept(this, attr_list);
        n.f4.accept(this, argu);
        n.f5.accept(this, argu);


        PrintFilter.print("Processing class", true);

        String id = n.f1.f0.tokenImage;

        ClassInstance type = new ClassInstance(id);
        type.set_attrs(attr_list);
        // type.set_mtds(mtd_list);
        PrintFilter.print("Inserting " + id + " => " + type, true);
        // Safe?

        symt.put(id, type);

        return null;

    }

    /**
     * f0 -> "class"
     * f1 -> Identifier()
     * f2 -> "extends"
     * f3 -> Identifier()
     * f4 -> "{"
     * f5 -> ( VarDeclaration() )*
     * f6 -> ( MethodDeclaration() )*
     * f7 -> "}"
     */
    public String visit(ClassExtendsDeclaration n, ArrayList<TypeInstance> argu) {

        ArrayList<TypeInstance> attr_list = new ArrayList<TypeInstance>();
        // ArrayList<TypeInstance> mtd_list = new ArrayList<TypeInstance>();

        n.f0.accept(this, argu);
        n.f1.accept(this, argu);
        n.f2.accept(this, argu);
        n.f3.accept(this, argu);
        n.f4.accept(this, argu);
        n.f5.accept(this, attr_list);
        n.f6.accept(this, argu);
        n.f7.accept(this, argu);


        PrintFilter.print("Processing extension class", true);

        String id = n.f1.f0.tokenImage;
        String ext = n.f3.f0.tokenImage;

        ClassInstance type = new ClassInstance(id, ext);
        type.set_attrs(attr_list);
        // type.set_mtds(mtd_list);
        PrintFilter.print("Inserting " + id + " => " + type, true);

        symt.put(id, type);

        return null;
    }


    /**
     * f0 -> Type()
     * f1 -> Identifier()
     * f2 -> ";"
     */
    public String visit(VarDeclaration n, ArrayList<TypeInstance> argu) {

        PrintFilter.print("Processing declaration", true);

        String id = n.f1.f0.tokenImage;
        TypeEnum rtrn = TypeEnum.ERROR;
        switch (n.f0.f0.which) {
        case 0:
            rtrn = TypeEnum.intarray; break;
        case 1:
            rtrn = TypeEnum.bool; break;
        case 2:
            rtrn = TypeEnum.integer; break;
        case 3:
            rtrn = TypeEnum.classname; break;
        default:
            PrintFilter.print("Unsupported case", true);
        }

        TypeInstance type = new TypeInstance(id, rtrn);
        PrintFilter.print("Inserting " + type, true);
        symt.put(id, type);
        argu.add(type);

        n.f0.accept(this, argu);
        n.f1.accept(this, argu);
        n.f2.accept(this, argu);
        return null;
    }

    /**
     * f0 -> "public"
     * f1 -> Type()
     * f2 -> Identifier()
     * f3 -> "("
     * f4 -> ( FormalParameterList() )?
     * f5 -> ")"
     * f6 -> "{"
     * f7 -> ( VarDeclaration() )*
     * f8 -> ( Statement() )*
     * f9 -> "return"
     * f10 -> Expression()
     * f11 -> ";"
     * f12 -> "}"
     */
    public String visit(MethodDeclaration n, ArrayList<TypeInstance> argu) {

        ArrayList<TypeInstance> argu_list = new ArrayList<TypeInstance>();
        ArrayList<TypeInstance> var_list = new ArrayList<TypeInstance>();

        n.f0.accept(this, argu);
        n.f1.accept(this, argu);
        n.f2.accept(this, argu);
        n.f3.accept(this, argu);
        n.f4.accept(this, argu_list);
        n.f5.accept(this, argu);
        n.f6.accept(this, argu);
        n.f7.accept(this, var_list);
        n.f8.accept(this, argu);
        n.f9.accept(this, argu);
        n.f10.accept(this, argu);
        n.f11.accept(this, argu);
        n.f12.accept(this, argu);


        PrintFilter.print("Processing method", true);

        String id = n.f2.f0.tokenImage;

        TypeEnum rtrn = TypeEnum.ERROR;
        switch (n.f1.f0.which) {
        case 0:
            rtrn = TypeEnum.intarray; break;
        case 1:
            rtrn = TypeEnum.bool; break;
        case 2:
            rtrn = TypeEnum.integer; break;
        default:
            PrintFilter.print("Unsupported case", true);
        }

        MethodInstance type = new MethodInstance(id, rtrn);
        type.set_args(argu_list);
        type.set_locals(var_list);
        PrintFilter.print("Inserting " + type, true);
        symt.put(id, type);

        return null;
    }

    /**
     * f0 -> Type()
     * f1 -> Identifier()
     */
    public String visit(FormalParameter n, ArrayList<TypeInstance> argu) {
        // PrintFilter.print("Processing parameter", true);

        String id = n.f1.f0.tokenImage;
        TypeEnum rtrn = TypeEnum.ERROR;
        switch (n.f0.f0.which) {
        case 0:
            rtrn = TypeEnum.intarray; break;
        case 1:
            rtrn = TypeEnum.bool; break;
        case 2:
            rtrn = TypeEnum.integer; break;
        default:
            // PrintFilter.print("Unsupported case", true);
            ;
        }

        TypeInstance type = new TypeInstance(id, rtrn);
        // PrintFilter.print("Adding Argument " + type, true);
        argu.add(type);

        n.f0.accept(this, argu);
        n.f1.accept(this, argu);
        return null;
    }

}