(feat): add java parser

package.json

@@ -18,6 +18,9 @@
     "gpt-tokenizer": "^2.1.2",
     "groq-sdk": "^0.8.0",
     "octokit": "^3.1.1",
+    "tree-sitter": "^0.21.1",
+    "tree-sitter-java": "^0.23.4",
+    "tree-sitter-python": "^0.23.4",
     "xml2js": "^0.6.2"
   },
   "devDependencies": {

src/constants.ts

@@ -1,6 +1,8 @@
 import { Node } from "@babel/traverse";
 import { JavascriptParser } from "./context/language/javascript-parser";
 import { ChatCompletionMessageParam } from "groq-sdk/resources/chat/completions";
+import { PythonParser } from "./context/language/python-parser";
+import { JavaParser } from "./context/language/java-parser";
 
 export interface PRFile {
   sha: string;
@@ -107,6 +109,8 @@ const EXTENSIONS_TO_PARSERS: Map<string, AbstractParser> = new Map([
   ["tsx", new JavascriptParser()],
   ["js", new JavascriptParser()],
   ["jsx", new JavascriptParser()],
+  ["py", new PythonParser()],
+  ["java", new JavaParser()],
 ]);
 
 export const getParserForExtension = (filename: string) => {

src/context/language/java-parser.ts

@@ -0,0 +1,78 @@
+import { AbstractParser, EnclosingContext } from "../../constants";
+import traverse, { NodePath, Node } from "@babel/traverse";
+import { SyntaxNode } from "tree-sitter";
+
+const Parser = require("tree-sitter");
+const Java = require("tree-sitter-java");
+
+const parser = new Parser();
+parser.setLanguage(Java);
+
+const processNode = (
+    node: SyntaxNode,
+    lineStart: number,
+    lineEnd: number,
+    largestSize: number,
+    largestEnclosingContext: SyntaxNode | null
+) => {
+    const start = node.startPosition;
+    const end = node.endPosition;
+
+    if (start.row <= lineStart && lineEnd <= end.row) {
+        const size = end.row - start.row;
+        if (size > largestSize) {
+            largestSize = size;
+            largestEnclosingContext = node;
+        }
+    }
+
+    return { largestSize, largestEnclosingContext };
+};
+
+export class JavaParser implements AbstractParser {
+    findEnclosingContext(
+        file: string,
+        lineStart: number,
+        lineEnd: number
+    ): EnclosingContext {
+        const tree = parser.parse(file);
+        let largestEnclosingContext: SyntaxNode | null = null;
+        let largestSize = 0;
+
+        const cursor = tree.walk();
+        let first = true
+        while (first || cursor.gotoNextSibling() || cursor.gotoParent())
+        {
+            const node = cursor.currentNode;
+            const node_types = ["method_declaration", "class_declaration", "compilation_unit"];
+            if (
+                node_types.includes(node.type)
+            ) {
+                ({ largestSize, largestEnclosingContext } = processNode(
+                    node,
+                    lineStart,
+                    lineEnd,
+                    largestSize,
+                    largestEnclosingContext,
+                ));
+            }
+            first = false
+        }
+        return { enclosingContext: largestEnclosingContext } as EnclosingContext;
+    }
+
+    dryRun(file: string): { valid: boolean; error: string } {
+      try {
+        const tree = parser.parse(file);
+        return {
+          valid: true,
+          error: "",
+        };
+      } catch (exc) {
+        return {
+          valid: false,
+          error: exc,
+        };
+      }
+    }
+}