@jpoly1219/context-extractor 0.2.7 → 0.2.9

package/dist/src/vscode-ide.js

@@ -0,0 +1,161 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VsCode = void 0;
+const vscode = __importStar(require("vscode"));
+var VsCode;
+(function (VsCode) {
+    const MAX_CACHE_SIZE = 50;
+    const lspCache = new Map();
+    function gotoInputKey(input) {
+        return `${input.name}${input.uri.toString()}${input.line}${input.character}`;
+    }
+    async function hover(location) {
+        const result = await executeHoverProvider({
+            uri: vscode.Uri.parse(location.filepath),
+            line: location.position.line,
+            character: location.position.character,
+            name: "vscode.executeHoverProvider",
+        });
+        return result;
+    }
+    VsCode.hover = hover;
+    async function gotoTypeDefinition(location) {
+        const result = await executeGotoProvider({
+            uri: vscode.Uri.parse(location.filepath),
+            line: location.position.line,
+            character: location.position.character,
+            name: "vscode.executeTypeDefinitionProvider",
+        });
+        return result;
+    }
+    VsCode.gotoTypeDefinition = gotoTypeDefinition;
+    async function getDocumentSymbols(location) {
+        const result = await executeDocumentSymbolProvider({
+            uri: vscode.Uri.parse(location.filepath),
+            name: "vscode.executeDocumentSymbolProvider",
+        });
+        return result;
+    }
+    VsCode.getDocumentSymbols = getDocumentSymbols;
+    // Core logic for executing vscode providers.
+    async function executeGotoProvider(input) {
+        // Check the cache before executing vscode command.
+        const cacheKey = gotoInputKey(input);
+        const cached = lspCache.get(cacheKey);
+        if (cached) {
+            return cached;
+        }
+        try {
+            const definitions = (await vscode.commands.executeCommand(input.name, input.uri, new vscode.Position(input.line, input.character)));
+            const results = definitions
+                .filter((d) => (d.targetUri || d.uri) && (d.targetRange || d.range))
+                .map((d) => ({
+                filepath: (d.targetUri || d.uri).toString(),
+                range: d.targetRange || d.range,
+            }));
+            // Update the cache via LRU.
+            if (lspCache.size >= MAX_CACHE_SIZE) {
+                const oldestKey = lspCache.keys().next().value;
+                if (oldestKey) {
+                    lspCache.delete(oldestKey);
+                }
+            }
+            lspCache.set(cacheKey, results);
+            return results;
+        }
+        catch (e) {
+            console.warn(`Error executing ${input.name}:`, e);
+            return [];
+        }
+    }
+    async function executeHoverProvider(input) {
+        // Check the cache before executing vscode command.
+        const cacheKey = gotoInputKey(input);
+        const cached = lspCache.get(cacheKey);
+        if (cached) {
+            return cached;
+        }
+        try {
+            const definitions = (await vscode.commands.executeCommand(input.name, input.uri, new vscode.Position(input.line, input.character)));
+            const result = definitions[0].contents[0].value;
+            // Update the cache via LRU.
+            if (lspCache.size >= MAX_CACHE_SIZE) {
+                const oldestKey = lspCache.keys().next().value;
+                if (oldestKey) {
+                    lspCache.delete(oldestKey);
+                }
+            }
+            lspCache.set(cacheKey, result);
+            return result;
+        }
+        catch (e) {
+            console.warn(`Error executing ${input.name}:`, e);
+            return { text: "" };
+        }
+    }
+    async function executeDocumentSymbolProvider(input) {
+        // Check the cache before executing vscode command.
+        const cacheKey = gotoInputKey(input);
+        const cached = lspCache.get(cacheKey);
+        if (cached) {
+            return cached;
+        }
+        try {
+            const definitions = (await vscode.commands.executeCommand(input.name, input.uri));
+            const results = definitions
+                .filter((d) => d.location && d.location.uri && d.range && d.selectionRange && d.name)
+                .map((d) => ({
+                filepath: (d.location.uri).toString(),
+                range: d.range,
+                selectionRange: d.selectionRange,
+                name: d.name
+            }));
+            // Update the cache via LRU.
+            if (lspCache.size >= MAX_CACHE_SIZE) {
+                const oldestKey = lspCache.keys().next().value;
+                if (oldestKey) {
+                    lspCache.delete(oldestKey);
+                }
+            }
+            lspCache.set(cacheKey, results);
+            return results;
+        }
+        catch (e) {
+            console.warn(`Error executing ${input.name}:`, e);
+            return [];
+        }
+    }
+})(VsCode || (exports.VsCode = VsCode = {}));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jpoly1219/context-extractor",
-  "version": "0.2.7",
+  "version": "0.2.9",
   "description": "Extract relevant context from an incomplete program sketch.",
   "repository": {
     "type": "git",
@@ -13,22 +13,28 @@
   ],
   "scripts": {
     "build": "rm -rf dist/ && npx tsc && npm run copy-deps",
-    "copy-deps": "npm run copy-lsp-client && npm run copy-ocaml",
+    "copy-deps": "npm run copy-lsp-client && npm run copy-ocaml && npm run copy-tree-sitter",
     "copy-lsp-client": "shx cp -r ts-lsp-client-dist dist/",
     "copy-ocaml": "shx mkdir -p dist/src/ocaml-utils/ && shx cp -r src/ocaml-utils/_build dist/src/ocaml-utils/_build/",
+    "copy-tree-sitter": "shx cp -r src/tree-sitter-files/ dist/src/",
     "test": "echo \"Error: no test specified\" && exit 1"
   },
   "author": "",
   "license": "ISC",
   "dependencies": {
     "json-rpc-2.0": "^1.7.0",
-    "openai": "^4.30.0",
-    "pino": "^9.3.1",
+    "openai": "^4.86.2",
+    "pino": "^9.6.0",
+    "pprof": "^4.0.0",
     "shx": "^0.3.4",
+    "tree-sitter-wasms": "^0.1.11",
     "ts-node": "^10.9.2",
-    "tslib": "^2.6.3"
+    "tslib": "^2.8.1",
+    "uri-js": "^4.4.1",
+    "web-tree-sitter": "^0.21.0"
   },
   "devDependencies": {
-    "typescript": "^5.4.5"
+    "@types/vscode": "^1.99.1",
+    "typescript": "^5.8.2"
   }
 }