@generaltranslation/python-extractor 0.2.15 → 0.2.17

package/dist/index.js

@@ -3,7 +3,19 @@ import { extractImports } from './extractImports.js';
 import { extractCalls } from './extractCalls.js';
 export { PYTHON_GT_PACKAGES, PYTHON_GT_DEPENDENCIES, PYTHON_T_FUNCTION, PYTHON_MSG_FUNCTION, PYTHON_DERIVE, PYTHON_DECLARE_STATIC, PYTHON_DECLARE_VAR, PYTHON_TRANSLATION_FUNCTIONS, PYTHON_METADATA_KWARGS, } from './constants.js';
 export async function extractFromPythonSource(sourceCode, filePath) {
-    const parser = await getParser();
+    let parser;
+    try {
+        parser = await getParser();
+    }
+    catch (error) {
+        return {
+            results: [],
+            errors: [],
+            warnings: [
+                `${filePath}: Failed to initialize Python parser; skipping Python extraction. ${formatError(error)}`,
+            ],
+        };
+    }
     const tree = parser.parse(sourceCode);
     if (!tree) {
         return {
@@ -40,3 +52,6 @@ export async function extractFromPythonSource(sourceCode, filePath) {
 function prefixErrors(messages, filePath) {
     return messages.map((msg) => `${filePath}: ${msg}`);
 }
+function formatError(error) {
+    return error instanceof Error ? error.message : String(error);
+}

package/dist/parser.js

@@ -1,6 +1,8 @@
 import { Parser, Language } from 'web-tree-sitter';
 import { createRequire } from 'module';
 let parserPromise = null;
+const require = createRequire(import.meta.url);
+const isBun = typeof process !== 'undefined' && !!process.versions.bun;
 /**
  * Lazily initializes and returns a singleton tree-sitter Parser
  * configured for Python.
@@ -12,11 +14,32 @@ export function getParser() {
     return parserPromise;
 }
 async function initParser() {
-    await Parser.init();
+    const parserWasmPath = await resolveWasmPath('web-tree-sitter/web-tree-sitter.wasm');
+    await Parser.init({
+        locateFile(path) {
+            if (path === 'web-tree-sitter.wasm') {
+                return parserWasmPath;
+            }
+            return path;
+        },
+    });
     const parser = new Parser();
-    const require = createRequire(import.meta.url);
-    const wasmPath = require.resolve('tree-sitter-python/tree-sitter-python.wasm');
+    const wasmPath = await resolveWasmPath('tree-sitter-python/tree-sitter-python.wasm');
     const Python = await Language.load(wasmPath);
     parser.setLanguage(Python);
     return parser;
 }
+async function resolveWasmPath(specifier) {
+    if (!isBun) {
+        return require.resolve(specifier);
+    }
+    switch (specifier) {
+        case 'web-tree-sitter/web-tree-sitter.wasm':
+            return (await import('web-tree-sitter/web-tree-sitter.wasm')).default;
+        case 'tree-sitter-python/tree-sitter-python.wasm':
+            return (await import('tree-sitter-python/tree-sitter-python.wasm'))
+                .default;
+        default:
+            throw new Error(`Unknown WASM specifier for Bun: ${specifier}`);
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@generaltranslation/python-extractor",
-  "version": "0.2.15",
+  "version": "0.2.17",
   "description": "Python source code extraction for General Translation",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -47,7 +47,7 @@
   "dependencies": {
     "tree-sitter-python": "^0.25.0",
     "web-tree-sitter": "^0.26.6",
-    "generaltranslation": "8.2.10"
+    "generaltranslation": "8.2.11"
   },
   "scripts": {
     "build": "tsc",