fix build

Author: gkorland

app/api/repo/route.ts

@@ -5,11 +5,11 @@ import http from 'isomorphic-git/http/node';
 import Parser from 'web-tree-sitter';
 
 import { promises as fs } from 'fs';
-import { Language, SyntaxNode } from 'web-tree-sitter';
+import { SyntaxNode } from 'web-tree-sitter';
 import { NextRequest, NextResponse } from "next/server";
 import { FalkorDB, Graph } from 'falkordb';
 import { RESPOSITORIES } from './repositories';
-import { Python } from '@/lib/languages/python';
+import { Language } from '@/lib/languages/language';
 
 const GraphOps = require('./graph_ops');
 const LIMITED_MODE = process.env.NEXT_PUBLIC_MODE?.toLowerCase() === 'limited';
@@ -18,7 +18,7 @@ const LIMITED_MODE = process.env.NEXT_PUBLIC_MODE?.toLowerCase() === 'limited';
 // Tree-Sitter queries
 //-----------------------------------------------------------------------------
 let parser: Parser;
-let language: Python = new Python();
+let language: Language;
 
 // Process Class declaration
 async function processClassDeclaration
@@ -254,6 +254,10 @@ async function InitializeTreeSitter() {
 	});
 
 	parser = new Parser();
+
+	const {Python} = await import('@/lib/languages/python');
+
+	language = new Python();
 	parser.setLanguage(language.language);
 }
 

lib/languages/language.ts

@@ -0,0 +1,49 @@
+import path from 'path';
+import Parser from 'web-tree-sitter';
+
+const PYTHON_LANG = await Parser.Language.load(path.join(process.cwd(), 'app/parsers/tree-sitter-python.wasm'));
+
+//-----------------------------------------------------------------------------
+// Tree-Sitter queries
+//-----------------------------------------------------------------------------
+export class Language {
+
+    public language: Parser.Language;
+
+    // class definition tree-sitter query
+    // responsible for matching class definition, in addition to extracting the class name
+    public class_definition_query: Parser.Query;
+
+    // function definition tree-sitter query
+    // responsible for matching function definition, in addition to extracting the function name
+    public function_definition_query: Parser.Query;
+
+    // function call tree-sitter query
+    // responsible for matching function calls, in addition to extracting the callee function name
+    public function_call_query: Parser.Query;
+
+    // function call tree-sitter query
+    // responsible for matching function calls of type self.f()
+    // in addition to extracting the callee function name
+    public function_attr_call_query: Parser.Query;
+
+    // identifier tree-sitter query
+    // responsible for matching Identifier nodes
+    public identifier_query: Parser.Query;
+
+    constructor(language: Parser.Language,
+        class_definition_query: Parser.Query,
+        function_definition_query: Parser.Query,
+        function_call_query: Parser.Query,
+        function_attr_call_query: Parser.Query,
+        identifier_query: Parser.Query) {
+
+        this.language = language;
+        this.class_definition_query = class_definition_query;
+        this.function_definition_query = function_definition_query;
+        this.function_call_query = function_call_query;
+        this.function_attr_call_query = function_attr_call_query;
+        this.identifier_query = identifier_query;
+    }
+
+}

lib/languages/python.ts

@@ -1,45 +1,21 @@
 import path from 'path';
 import Parser from 'web-tree-sitter';
-import { Language } from 'web-tree-sitter';
+import { Language } from './language';
 
 const PYTHON_LANG = await Parser.Language.load(path.join(process.cwd(), 'app/parsers/tree-sitter-python.wasm'));
 
 //-----------------------------------------------------------------------------
 // Tree-Sitter queries
 //-----------------------------------------------------------------------------
-export class Python {
-
-    public language: Language;
-
-    // class definition tree-sitter query
-    // responsible for matching class definition, in addition to extracting the class name
-    public class_definition_query: Parser.Query;
-
-    // function definition tree-sitter query
-    // responsible for matching function definition, in addition to extracting the function name
-    public function_definition_query: Parser.Query;
-
-    // function call tree-sitter query
-    // responsible for matching function calls, in addition to extracting the callee function name
-    public function_call_query: Parser.Query;
-
-    // function call tree-sitter query
-    // responsible for matching function calls of type self.f()
-    // in addition to extracting the callee function name
-    public function_attr_call_query: Parser.Query;
-
-    // identifier tree-sitter query
-    // responsible for matching Identifier nodes
-    public identifier_query: Parser.Query;
+export class Python extends Language {
 
     constructor() {
-        this.language = PYTHON_LANG;
-        this.class_definition_query = PYTHON_LANG.query(`(class_definition name: (identifier) @class-name) @class-definition`);
-        this.function_definition_query = PYTHON_LANG.query(`((function_definition name: (identifier) @function-name parameters: (parameters) @parameters) @function-definition)`);
-        this.function_call_query = PYTHON_LANG.query(`((call function: (identifier) @function-name) @function-call)`);
-        this.function_attr_call_query = PYTHON_LANG.query(`((call function: (attribute object: (identifier) attribute: (identifier) @function-name )) @function-call)`);
-        this.identifier_query = PYTHON_LANG.query(`((identifier) @identifier)`);
-
+        super(
+            PYTHON_LANG,
+            PYTHON_LANG.query(`(class_definition name: (identifier) @class-name) @class-definition`),
+            PYTHON_LANG.query(`((function_definition name: (identifier) @function-name parameters: (parameters) @parameters) @function-definition)`),
+            PYTHON_LANG.query(`((call function: (identifier) @function-name) @function-call)`),
+            PYTHON_LANG.query(`((call function: (attribute object: (identifier) attribute: (identifier) @function-name )) @function-call)`),
+            PYTHON_LANG.query(`((identifier) @identifier)`))
     }
-
 }