From a59ac5ef23aec582c700bb32f15fe60c264eb985 Mon Sep 17 00:00:00 2001
From: 404Simon <simsicompi@mail.de>
Date: Thu, 30 May 2024 19:49:45 +0200
Subject: [PATCH] add mainmethod to typedAst; there are Problems with TypeCheck
 though

---
 src/main/java/de/maishai/ASTGenerator.java           |  2 +-
 src/main/java/de/maishai/typedast/CodeGenUtils.java  | 12 ++++++++++++
 .../de/maishai/typedast/typedclass/TypedClass.java   |  9 +++++++++
 .../resources/JavaTestfiles/ClassCanBeBytecoded.java | 11 +++++++++--
 4 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/src/main/java/de/maishai/ASTGenerator.java b/src/main/java/de/maishai/ASTGenerator.java
index 5635814..8cb6a11 100644
--- a/src/main/java/de/maishai/ASTGenerator.java
+++ b/src/main/java/de/maishai/ASTGenerator.java
@@ -28,7 +28,7 @@ public class ASTGenerator {
             constructors.add(new Constructor(ctx.id().IDENTIFIER().getText(), List.of(), new Block(List.of())));
         }
         Block mainmeth = null;
-        if (ctx.mainmeth() != null) {
+        if (!ctx.mainmeth().isEmpty()) {
             if (ctx.mainmeth().size() > 1) {
                 throw new RuntimeException("Only one main method allowed!");
             }
diff --git a/src/main/java/de/maishai/typedast/CodeGenUtils.java b/src/main/java/de/maishai/typedast/CodeGenUtils.java
index dc5ed5f..1509653 100644
--- a/src/main/java/de/maishai/typedast/CodeGenUtils.java
+++ b/src/main/java/de/maishai/typedast/CodeGenUtils.java
@@ -1,7 +1,10 @@
 package de.maishai.typedast;
 
+import de.maishai.typedast.typedclass.TypedBlock;
 import de.maishai.typedast.typedclass.TypedParameter;
 import org.objectweb.asm.ClassWriter;
+import org.objectweb.asm.MethodVisitor;
+import org.objectweb.asm.Opcodes;
 
 import java.io.FileOutputStream;
 import java.util.List;
@@ -30,4 +33,13 @@ public class CodeGenUtils {
             System.out.println("Error writing classfile: " + e);
         }
     }
+
+    public static void generateMainMethod(ClassContext ctx, TypedBlock mainMethodBlock) {
+        System.out.println("Generating main method");
+        MethodVisitor mv = ctx.getCw().visitMethod(Opcodes.ACC_PUBLIC + Opcodes.ACC_STATIC, "main", "([Ljava/lang/String;)V", null, null);
+        MethodContext context = new MethodContext(ctx, mv);
+        context.registerVariable("args", Type.REFERENCE("[Ljava/lang/String;"));
+        mainMethodBlock.codeGen(context);
+        context.wrapUp();
+    }
 }
diff --git a/src/main/java/de/maishai/typedast/typedclass/TypedClass.java b/src/main/java/de/maishai/typedast/typedclass/TypedClass.java
index 3ce8d81..eb1d8a9 100644
--- a/src/main/java/de/maishai/typedast/typedclass/TypedClass.java
+++ b/src/main/java/de/maishai/typedast/typedclass/TypedClass.java
@@ -23,6 +23,7 @@ public class TypedClass implements TypedNode {
     private String className;
     private List<TypedDeclaration> typedDeclarations = new ArrayList<>();
     private List<TypedMethod> typedMethods = new ArrayList<>();
+    private TypedBlock mainMethodBlock;
     private List<TypedConstructor> typedConstructors = new ArrayList<>();
     private TypedMethod currentMethod;
     private TypedConstructor currentConstructor;
@@ -74,6 +75,10 @@ public class TypedClass implements TypedNode {
         for (Method method : c.methods()) {
             enterCurrentMethod(typedMethods.get(j));
             typedMethods.get(j).convertToTypedBlock(typedProgram, method);
+            // Hier wird der Block des main-Methode ausgeführt
+            if (c.mainmeth() != null) {
+                mainMethodBlock = new TypedBlock(typedProgram, c.mainmeth());
+            }
             exitCurrentMethod();
             j++;
         }
@@ -203,6 +208,10 @@ public class TypedClass implements TypedNode {
             m.codeGen(ctx);
         }
 
+        if (mainMethodBlock != null) {
+            CodeGenUtils.generateMainMethod(ctx, mainMethodBlock);
+        }
+
         return cw.toByteArray();
     }
 }
diff --git a/src/main/resources/JavaTestfiles/ClassCanBeBytecoded.java b/src/main/resources/JavaTestfiles/ClassCanBeBytecoded.java
index 3d34626..590cbb9 100644
--- a/src/main/resources/JavaTestfiles/ClassCanBeBytecoded.java
+++ b/src/main/resources/JavaTestfiles/ClassCanBeBytecoded.java
@@ -1,12 +1,19 @@
 public class ClassCanBeBytecoded {
-    public ClassCanBeBytecoded c;
     public int x;
     public ClassCanBeBytecoded() {
     }
 
 
     public int test(int var1) {
-        print(var1);
+        print(5+5);
+        int i = 8;
+        var1 = i + 5;
         return var1;
     }
+
+    public static void main(String[] args) {
+        ClassCanBeBytecoded c = new ClassCanBeBytecoded();
+        c.test(5);
+    }
+
 }
\ No newline at end of file