fix(grammar): super invocation

Author: devanshj

src/index.spec.ts

@@ -1,4 +1,7 @@
+import { ParseTreeWalker } from 'antlr4ts/tree/ParseTreeWalker';
 import { parse } from './index';
+import { ExpressionContext } from './parser/JavaParser';
+import { JavaParserListener } from './parser/JavaParserListener';
 
 describe('Java AST parser', () => {
   it('should parse the given Java code and return the AST', () => {
@@ -8,4 +11,26 @@ describe('Java AST parser', () => {
     `);
     expect(tree.children[0].getChild(0).getChild(1).text).toEqual('TestClass');
   });
+
+  it('should handle super invocation with arguments', () => {
+    const tree = parse(`
+      class B extends A {
+        public B() {
+          super(1);
+        }
+      }
+    `);
+
+    const expressions = [];
+    ParseTreeWalker.DEFAULT.walk(
+      {
+        enterExpression(context: ExpressionContext) {
+          expressions.push(context.text);
+        },
+      } as JavaParserListener,
+      tree,
+    );
+
+    expect(expressions).toContain('super(1)');
+  });
 });

src/parser/JavaParser.g4

@@ -524,7 +524,7 @@ lambdaBody
 primary
     : '(' expression ')'
     | THIS
-    | SUPER
+    | SUPER superSuffix
     | literal
     | IDENTIFIER
     | typeTypeOrVoid '.' CLASS