Skip to content

Commit 0492092

Browse files
authored
Merge pull request #10 from jpoly1219/jacob/enhancement/fasterextraction/vscodeapi
attempt file resolution fix
2 parents 94ba973 + ffdb703 commit 0492092

File tree

4 files changed

+743
-362
lines changed

4 files changed

+743
-362
lines changed

package-lock.json

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

package.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
{
22
"name": "@jpoly1219/context-extractor",
3-
"version": "0.2.10",
3+
"version": "0.2.11",
44
"description": "Extract relevant context from an incomplete program sketch.",
55
"repository": {
66
"type": "git",

src/tree-sitter.ts

Lines changed: 28 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,25 @@ export async function getParserForFile(filepath: string) {
2929
await Parser.init();
3030
const parser = new Parser();
3131

32+
// const packageRoot = path.dirname(
33+
// require.resolve("./package.json", { paths: [__dirname] })
34+
// );
35+
// console.log(packageRoot);
36+
// const wasmPath2 = require.resolve(
37+
// path.join(
38+
// __dirname,
39+
// `tree-sitter-files/wasms/tree-sitter-${supportedLanguages["ts"]}.wasm`
40+
// )
41+
// );
42+
// console.log(wasmPath2);
43+
// const wasmPath3 = require.resolve(
44+
// path.resolve(
45+
// __dirname,
46+
// `tree-sitter-files/wasms/tree-sitter-${supportedLanguages["ts"]}.wasm`
47+
// )
48+
// );
49+
// console.log(wasmPath3);
50+
3251
const language = await getLanguageForFile(filepath);
3352
if (!language) {
3453
return undefined;
@@ -99,7 +118,9 @@ export async function getQueryForFile(
99118
async function loadLanguageForFileExt(
100119
fileExtension: string
101120
): Promise<Parser.Language> {
102-
const wasmPath = require.resolve(`./tree-sitter-files/wasms/tree-sitter-${supportedLanguages[fileExtension]}.wasm`)
121+
const wasmPath = require.resolve(
122+
`@jpoly1219/context-extractor/src/tree-sitter-files/wasms/tree-sitter-${supportedLanguages[fileExtension]}.wasm`
123+
);
103124
// const wasmPath = path.join(
104125
// __dirname,
105126
// "tree-sitter-files",
@@ -267,7 +288,9 @@ export async function extractTopLevelDecls(currentFile: string) {
267288
throw new Error(`failed to get ast for file ${currentFile}`);
268289
}
269290
const language = getFullLanguageName(currentFile);
270-
const queryPath = require.resolve(`./tree-sitter-files/queries/relevant-headers-queries/${language}-get-toplevel-headers.scm`)
291+
const queryPath = require.resolve(
292+
`@jpoly1219/context-extractor/src/tree-sitter-files/queries/relevant-headers-queries/${language}-get-toplevel-headers.scm`
293+
);
271294
const query = await getQueryForFile(
272295
currentFile,
273296
queryPath
@@ -293,7 +316,9 @@ export async function extractTopLevelDeclsWithFormatting(currentFile: string) {
293316
throw new Error(`failed to get ast for file ${currentFile}`);
294317
}
295318
const language = getFullLanguageName(currentFile);
296-
const queryPath = require.resolve(`./tree-sitter-files/queries/relevant-headers-queries/${language}-get-toplevel-headers.scm`)
319+
const queryPath = require.resolve(
320+
`@jpoly1219/context-extractor/src/tree-sitter-files/queries/relevant-headers-queries/${language}-get-toplevel-headers.scm`
321+
);
297322
const query = await getQueryForFile(
298323
currentFile,
299324
queryPath

0 commit comments

Comments
 (0)