Feat add symbol table visitor

app/build.gradle.kts

@@ -26,10 +26,15 @@ dependencies {
     implementation(kotlin("stdlib-jdk8"))
     implementation("org.jetbrains.kotlinx:kotlinx-coroutines-core:1.8.1-Beta")
     testImplementation("org.junit.jupiter:junit-jupiter-params")
+    implementation("io.arrow-kt:arrow-core:1.2.1")
+
+    testImplementation("io.mockk:mockk:1.13.10")
+    testImplementation("org.assertj:assertj-core:3.25.1")
 }
 
 tasks {
     compileJavacc {
+        dependsOn(compileKotlin)
         inputDirectory = file("src/main/javacc")
         outputDirectory = file(layout.buildDirectory.dir("generated/javacc"))
     }
@@ -45,6 +50,9 @@ sourceSets {
         java {
             srcDir(file(layout.buildDirectory.dir("generated/javacc")))
         }
+        kotlin {
+            srcDir("src/main/kotlin")
+        }
     }
 }
 

app/src/main/java/org/example/ast/ClassDeclExtends.java

@@ -12,12 +12,12 @@
 @Builder
 @AllArgsConstructor
 public class ClassDeclExtends extends ClassDecl {
-    private Identifier className;
-    private Identifier parent;
+    public Identifier className;
+    public Identifier parent;
     @Builder.Default
-    private VarDeclList fields = new VarDeclList();
+    public VarDeclList fields = new VarDeclList();
     @Builder.Default
-    private MethodDeclList methods = new MethodDeclList();
+    public MethodDeclList methods = new MethodDeclList();
 
     @Override
     public void accept(ASTVisitor v) {

app/src/main/java/org/example/ast/ClassDeclList.java

@@ -11,7 +11,7 @@
 @EqualsAndHashCode
 public class ClassDeclList {
     @Builder.Default
-    private ArrayList<ClassDecl> classDecls =  new ArrayList<>();
+    public ArrayList<ClassDecl> classDecls =  new ArrayList<>();
 
     public void addClassDecl(ClassDecl classDecl) {
         classDecls.add(classDecl);

app/src/main/java/org/example/ast/ClassDeclSimple.java

@@ -12,11 +12,11 @@
 @Builder
 @AllArgsConstructor
 public class ClassDeclSimple extends ClassDecl {
-    private Identifier className;
+    public Identifier className;
     @Builder.Default
-    private VarDeclList fields = new VarDeclList();
+    public VarDeclList fields = new VarDeclList();
     @Builder.Default
-    private MethodDeclList methods = new MethodDeclList();
+    public MethodDeclList methods = new MethodDeclList();
 
     @Override
     public void accept(ASTVisitor v) {

app/src/main/java/org/example/ast/ExpressionList.java

@@ -11,7 +11,7 @@
 @AllArgsConstructor
 @NoArgsConstructor
 public class ExpressionList {
-    private ArrayList<Expression> list = new ArrayList<>();
+    public ArrayList<Expression> list = new ArrayList<>();
 
     public void addExpression(Expression expression) {
         list.add(expression);

app/src/main/java/org/example/ast/Formal.java

@@ -12,8 +12,8 @@
 @Builder
 @AllArgsConstructor
 public class Formal extends Node {
-    private Type type;
-    private String name;
+    public Type type;
+    public String name;
 
     @Override
     public void accept(ASTVisitor v) {

app/src/main/java/org/example/ast/FormalList.java

@@ -11,7 +11,7 @@
 @EqualsAndHashCode
 public class FormalList {
     @Builder.Default
-    private ArrayList<Formal> formals = new ArrayList<>();
+    public ArrayList<Formal> formals = new ArrayList<>();
 
     public void addFormal(Formal formal) {
         formals.add(formal);

app/src/main/java/org/example/ast/Identifier.java

@@ -1,18 +1,16 @@
 package org.example.ast;
 
-import lombok.AllArgsConstructor;
-import lombok.Data;
-import lombok.EqualsAndHashCode;
-import lombok.ToString;
+import lombok.*;
 import org.example.visitor.ASTVisitor;
 import org.example.visitor.TypeVisitor;
 
 @EqualsAndHashCode(callSuper = false)
 @ToString
 @Data
 @AllArgsConstructor
+@Builder
 public class Identifier extends Expression {
-    private String s;
+    public String s;
     @Override
     public void accept(ASTVisitor v) {
         v.visit(this);

app/src/main/java/org/example/ast/IdentifierExpression.java

@@ -12,7 +12,7 @@
 @Builder
 @AllArgsConstructor
 public class IdentifierExpression extends Expression {
-    private String id;
+    public String id;
 
     @Override
     public void accept(ASTVisitor v) {

app/src/main/java/org/example/ast/IdentifierType.java

@@ -10,7 +10,7 @@
 @Data
 @AllArgsConstructor
 public class IdentifierType extends Type {
-    private String s;
+    public String s;
 
     @Override
     public void accept(ASTVisitor v) {

app/src/main/java/org/example/ast/IntegerType.java

@@ -1,5 +1,6 @@
 package org.example.ast;
 
+import lombok.Builder;
 import lombok.EqualsAndHashCode;
 import org.example.visitor.ASTVisitor;
 import org.example.visitor.TypeVisitor;

app/src/main/java/org/example/ast/MainClass.java

@@ -12,9 +12,9 @@
 @Builder
 @AllArgsConstructor
 public class MainClass extends Node {
-    private Identifier className;
-    private Identifier argsName;
-    private StatementList statements;
+    public Identifier className;
+    public Identifier argsName;
+    public StatementList statements;
 
     @Override
     public void accept(ASTVisitor v) {

app/src/main/java/org/example/ast/MethodDecl.java

@@ -12,12 +12,12 @@
 @Builder
 @AllArgsConstructor
 public class MethodDecl extends Node {
-    private Type type;
-    private String identifier;
-    private FormalList formals;
-    private VarDeclList varDecls;
-    private StatementList statements;
-    private Expression returnExpression;
+    public Type type;
+    public String identifier;
+    public FormalList formals;
+    public VarDeclList varDecls;
+    public StatementList statements;
+    public Expression returnExpression;
 
     @Override
     public void accept(ASTVisitor v) {

app/src/main/java/org/example/ast/MethodDeclList.java

@@ -11,7 +11,7 @@
 @EqualsAndHashCode
 public class MethodDeclList {
     @Builder.Default
-    private ArrayList<MethodDecl> methodDecls = new ArrayList<>();
+    public ArrayList<MethodDecl> methodDecls = new ArrayList<>();
 
     public void addMethodDecl(MethodDecl methodDecl) {
         methodDecls.add(methodDecl);

app/src/main/java/org/example/ast/Program.java

@@ -12,8 +12,8 @@
 @Builder
 @AllArgsConstructor
 public class Program extends Node {
-    private MainClass mainClass;
-    private ClassDeclList classes;
+    public MainClass mainClass;
+    public ClassDeclList classes;
 
     @Override
     public void accept(ASTVisitor v) {

app/src/main/java/org/example/ast/Type.java

@@ -1,4 +1,6 @@
 package org.example.ast;
 
+import lombok.Builder;
+
 public abstract class Type extends Node {
 }

app/src/main/java/org/example/ast/VarDecl.java

@@ -12,8 +12,8 @@
 @Builder
 @AllArgsConstructor
 public class VarDecl extends Node {
-    private Type type;
-    private String name;
+    public Type type;
+    public String name;
 
     @Override
     public void accept(ASTVisitor v) {

app/src/main/java/org/example/ast/VarDeclList.java

@@ -13,7 +13,7 @@
 @NoArgsConstructor
 public class VarDeclList {
     @Builder.Default
-    private ArrayList<VarDecl> varDecls = new ArrayList<>();
+    public ArrayList<VarDecl> varDecls = new ArrayList<>();
 
     public void addVarDecl(VarDecl varDecl) {
         varDecls.add(varDecl);

app/src/main/kotlin/org/example/visitor/ClassDeclVisitor.kt

@@ -0,0 +1,62 @@
+package org.example.visitor
+
+import arrow.core.Either
+import arrow.core.raise.either
+import arrow.core.raise.ensure
+import org.example.ast.*
+import org.example.ast2.org.example.visitor.MethodDeclListVisitor
+import org.example.visitor.SymbolVisitor.Companion.dispatch
+
+object ClassDeclListVisitor : SymbolVisitor<ClassDeclList> {
+    override fun Table.visit(entity: ClassDeclList): Either<Error, Table> =
+        ClassDeclVisitor.fold(
+            entity.classDecls.toList()
+        ) { dispatch(it) }
+}
+
+object ClassDeclVisitor : SymbolVisitor<ClassDecl> {
+    private fun extractName(entity: ClassDecl): String =
+        when (entity) {
+            is ClassDeclSimple -> entity.className.s
+            is ClassDeclExtends -> entity.className.s
+            else -> throw IllegalArgumentException(
+                "ClassDeclVisitor: ClassDecl must be either ClassDeclSimple or ClassDeclExtends"
+            )
+        }
+
+    object ClassDeclSimpleVisitor : SymbolVisitor<ClassDeclSimple> {
+        override fun Table.visit(entity: ClassDeclSimple): Either<Error, Table> = either {
+            this@visit + Table(
+                ClassData(
+                    name = extractName(entity),
+                    fields = dispatch(entity.fields).bind(),
+                    methods = dispatch(entity.methods).bind()
+                )
+            )
+        }
+    }
+
+    object ClassDeclExtendsVisitor : SymbolVisitor<ClassDeclExtends> {
+        override fun Table.visit(entity: ClassDeclExtends): Either<Error, Table> = either {
+            this@visit + Table(
+                ClassData(
+                    name = extractName(entity),
+                    fields = dispatch(entity.fields).bind(),
+                    methods = dispatch(entity.methods).bind()
+                )
+            )
+        }
+    }
+
+    override fun Table.visit(entity: ClassDecl): Either<Error, Table> = either {
+        val name = extractName(entity)
+
+        ensure(!this@visit.contains(name)) {
+            Error("ClassDeclVisitor: ClassDecl must have a unique name")
+        }
+
+        val newTable = dispatch(entity, table = this@visit).bind()
+
+        this@visit + newTable
+    }
+}

app/src/main/kotlin/org/example/visitor/MainClassVisitor.kt

@@ -0,0 +1,42 @@
+package org.example.visitor
+
+import arrow.core.Either
+import arrow.core.raise.either
+import arrow.core.raise.ensure
+import org.example.ast.IntegerLiteral
+import org.example.ast.IntegerType
+import org.example.ast.MainClass
+
+object MainClassVisitor : SymbolVisitor<MainClass> {
+    override fun Table.visit(entity: MainClass): Either<Error, Table> = either {
+        ensure(!contains(entity.className.s)) {
+            Error("MainClassVisitor: MainClass must have a unique name")
+        }
+        ensure(entity.argsName.s == "args") {
+            Error("MainClassVisitor: MainClass args must be named 'args'")
+        }
+
+        Table(
+            ClassData(
+                name = entity.className.s,
+                fields = Table(),
+                methods = Table(
+                    MethodData(
+                        name = entity.className.s,
+                        args = Table(
+                            ParamData(
+                                name = entity.argsName.s,
+                                type = IntegerType()
+                            )
+                        ),
+                        varDeclList = Table(
+
+                        ),
+                    )
+                )
+            )
+        )
+    }
+
+
+}

app/src/main/kotlin/org/example/visitor/MethodDeclVisitor.kt

@@ -0,0 +1,63 @@
+package org.example.ast2.org.example.visitor
+
+import arrow.core.Either
+import arrow.core.raise.either
+import arrow.core.raise.ensure
+import org.example.ast.Formal
+import org.example.ast.FormalList
+import org.example.ast.MethodDecl
+import org.example.ast.MethodDeclList
+import org.example.visitor.*
+import org.example.visitor.SymbolVisitor.Companion.dispatch
+
+
+object FormalsListVisitor : SymbolVisitor<FormalList> {
+    override fun Table.visit(entity: FormalList): Either<Error, Table> =
+        with(FormalsVisitor) {
+            fold(
+                entity.formals.toList(),
+                ::dispatch
+            )
+        }
+}
+
+object FormalsVisitor : SymbolVisitor<Formal> {
+    override fun Table.visit(entity: Formal): Either<Error, Table> = either {
+        ensure(!contains(entity.name)) {
+            Error("FormalsVisitor: Formals must have unique names")
+        }
+
+        Table(
+            FormalData(
+                name = entity.name,
+                type = entity.type
+            )
+        )
+    }
+}
+
+object MethodDeclListVisitor : SymbolVisitor<MethodDeclList> {
+    override fun Table.visit(entity: MethodDeclList): Either<Error, Table> =
+        with(MethodDeclVisitor) {
+            fold(
+                entity.methodDecls.toList(),
+                ::dispatch
+            )
+        }
+}
+
+object MethodDeclVisitor : SymbolVisitor<MethodDecl> {
+    override fun Table.visit(entity: MethodDecl): Either<Error, Table> = either {
+        ensure(!contains(entity.identifier)) {
+            Error("MethodDeclVisitor: MethodDecl must have a unique name")
+        }
+
+        Table(
+            MethodData(
+                name = entity.identifier,
+                args = dispatch(entity.formals).bind(),
+                varDeclList = dispatch(entity.varDecls).bind()
+            )
+        )
+    }
+}