Skip to content

Commit d2ac1c0

Browse files
committed
fix treesitter loading
1 parent fc01df4 commit d2ac1c0

File tree

3 files changed

+11
-15
lines changed

3 files changed

+11
-15
lines changed

app/api/repo/route.ts

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@ import { SyntaxNode } from 'web-tree-sitter';
99
import { NextRequest, NextResponse } from "next/server";
1010
import { FalkorDB, Graph } from 'falkordb';
1111
import { RESPOSITORIES } from './repositories';
12-
import { Language } from '@/lib/languages/language';
12+
import Language from '@/lib/languages/language';
13+
import Python from '@/lib/languages/python'
1314

1415
const GraphOps = require('./graph_ops');
1516
const LIMITED_MODE = process.env.NEXT_PUBLIC_MODE?.toLowerCase() === 'limited';
@@ -246,17 +247,7 @@ async function BuildGraph
246247
}
247248

248249
async function InitializeTreeSitter() {
249-
// Initialize Tree-Sitter parser
250-
await Parser.init({
251-
locateFile(scriptName: string, scriptDirectory: string) {
252-
return path.join(process.cwd(), 'app/parsers', scriptName);
253-
},
254-
});
255-
256250
parser = new Parser();
257-
258-
const {Python} = await import('@/lib/languages/python');
259-
260251
language = new Python();
261252
parser.setLanguage(language.language);
262253
}

lib/languages/language.ts

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,17 @@
11
import path from 'path';
22
import Parser from 'web-tree-sitter';
33

4-
const PYTHON_LANG = await Parser.Language.load(path.join(process.cwd(), 'app/parsers/tree-sitter-python.wasm'));
4+
// Initialize Tree-Sitter parser
5+
await Parser.init({
6+
locateFile(scriptName: string, scriptDirectory: string) {
7+
return path.join(process.cwd(), 'app/parsers', scriptName);
8+
},
9+
});
510

611
//-----------------------------------------------------------------------------
712
// Tree-Sitter queries
813
//-----------------------------------------------------------------------------
9-
export class Language {
14+
export default class Language {
1015

1116
public language: Parser.Language;
1217

lib/languages/python.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
import path from 'path';
22
import Parser from 'web-tree-sitter';
3-
import { Language } from './language';
3+
import Language from './language';
44

55
const PYTHON_LANG = await Parser.Language.load(path.join(process.cwd(), 'app/parsers/tree-sitter-python.wasm'));
66

77
//-----------------------------------------------------------------------------
88
// Tree-Sitter queries
99
//-----------------------------------------------------------------------------
10-
export class Python extends Language {
10+
export default class Python extends Language {
1111

1212
constructor() {
1313
super(

0 commit comments

Comments
 (0)