Thanks java...

Author: NoSharp

src/main/java/org/luapp/language/handlers/ClassBodyHandler.java

@@ -2,6 +2,7 @@
 
 import org.antlr.v4.runtime.ParserRuleContext;
 import org.antlr.v4.runtime.tree.ParseTree;
+import org.antlr.v4.runtime.tree.TerminalNodeImpl;
 import org.luapp.language.generator.luappParser;
 import org.luapp.language.listeners.LuaPPListener;
 
@@ -12,15 +13,15 @@ public ClassBodyHandler() {
     }
 
     public boolean isClassFunction(ParseTree context){
-        return context instanceof luappParser.ClassfunctionContext;
+        return ((ParserRuleContext)context).getRuleIndex() == luappParser.RULE_classfunction;
     }
 
     public boolean isClassGetSet(ParseTree context){
-        return context instanceof luappParser.ClassgetsetContext;
+        return ((ParserRuleContext)context).getRuleIndex() == luappParser.RULE_classgetset;
     }
 
     public boolean isClassConstructor(ParseTree context){
-        return context instanceof luappParser.ConstructorContext;
+        return ((ParserRuleContext)context).getRuleIndex() == luappParser.RULE_constructor;
     }
 
 
@@ -38,10 +39,14 @@ public void onEnterContext(ParserRuleContext enterContext) {
             return;
         }
         for (ParseTree child : enterContext.children) {
+            if(child instanceof TerminalNodeImpl) continue;
             if(isClassFunction(child)){
                 System.out.println("Class function");
             }else if(isClassConstructor(child)){
-                System.out.println("Constructor");
+                System.out.println("Constructor!");
+                this.listenerManager
+                        .GetInstangeByTarget(luappParser.RULE_constructor)
+                        .handleEnterContext((ParserRuleContext)child);
             }else if(isClassGetSet(child)){
                 System.out.println("Get/Set");
             }