agentHost Adopt new tree-sitter (#308352)

* agentHost Adopt new tree-sitter

* Bump in web

* Fix test

* fix tests

Author: roblourens

package-lock.json

@@ -109,7 +109,7 @@
     "@vscode/spdlog": "^0.15.8",
     "@vscode/sqlite3": "5.1.12-vscode",
     "@vscode/sudo-prompt": "9.3.2",
-    "@vscode/tree-sitter-wasm": "^0.3.0",
+    "@vscode/tree-sitter-wasm": "^0.3.1",
     "@vscode/vscode-languagedetection": "1.0.23",
     "@vscode/windows-mutex": "^0.5.0",
     "@vscode/windows-process-tree": "^0.6.0",

package.json

@@ -16,7 +16,7 @@
     "@vscode/ripgrep": "^1.17.1",
     "@vscode/spdlog": "^0.15.8",
     "@vscode/sqlite3": "5.1.12-vscode",
-    "@vscode/tree-sitter-wasm": "^0.3.0",
+    "@vscode/tree-sitter-wasm": "^0.3.1",
     "@vscode/vscode-languagedetection": "1.0.23",
     "@vscode/windows-process-tree": "^0.6.0",
     "@vscode/windows-registry": "^1.2.0",

remote/package-lock.json

@@ -7,7 +7,7 @@
     "@microsoft/1ds-post-js": "^3.2.13",
     "@vscode/codicons": "^0.0.46-1",
     "@vscode/iconv-lite-umd": "0.7.1",
-    "@vscode/tree-sitter-wasm": "^0.3.0",
+    "@vscode/tree-sitter-wasm": "^0.3.1",
     "@vscode/vscode-languagedetection": "1.0.23",
     "@xterm/addon-clipboard": "^0.3.0-beta.197",
     "@xterm/addon-image": "^0.10.0-beta.197",

remote/package.json

@@ -172,7 +172,11 @@ export class CommandAutoApprover extends Disposable {
 
 	private async _initTreeSitter(): Promise<void> {
 		try {
-			const TreeSitter = await import('@vscode/tree-sitter-wasm');
+			const { default: TreeSitter } = (await import('@vscode/tree-sitter-wasm'));
+
+			if (this._store.isDisposed) {
+				return;
+			}
 
 			// Resolve WASM files from node_modules
 			const moduleRoot = URI.joinPath(FileAccess.asFileUri(''), '..', 'node_modules', '@vscode', 'tree-sitter-wasm', 'wasm');
@@ -184,14 +188,33 @@ export class CommandAutoApprover extends Disposable {
 				}
 			});
 
+			if (this._store.isDisposed) {
+				return;
+			}
+
 			const parser = new TreeSitter.Parser();
-			this._register(toDisposable(() => parser.delete()));
+			this._register(toDisposable(() => {
+				try {
+					parser.delete();
+				} catch {
+					// WASM memory may already be freed
+				}
+			}));
 
 			// Load bash grammar
 			const bashWasmPath = URI.joinPath(moduleRoot, 'tree-sitter-bash.wasm').fsPath;
 			const bashWasm = await fs.promises.readFile(bashWasmPath);
+
+			if (this._store.isDisposed) {
+				return;
+			}
+
 			const bashLanguage = await TreeSitter.Language.load(new Uint8Array(bashWasm.buffer, bashWasm.byteOffset, bashWasm.byteLength));
 
+			if (this._store.isDisposed) {
+				return;
+			}
+
 			this._parser = parser;
 			this._bashLanguage = bashLanguage;
 			this._queryClass = TreeSitter.Query;