feat: CallStack: add reload grammars from disk #10

Author: RedCMD

package.json

@@ -500,6 +500,11 @@
 					"group": "navigation@0",
 					"when": "view == TextMate"
 				},
+				{
+					"command": "textmate.reload.grammars",
+					"group": "navigation@1",
+					"when": "view == TextMate"
+				},
 				{
 					"command": "textmate.tree-view",
 					"group": "navigation@2",
@@ -588,6 +593,12 @@
 				"category": "TextMate",
 				"icon": "$(refresh)"
 			},
+			{
+				"command": "textmate.reload.grammars",
+				"title": "Reload Grammars from Disk",
+				"category": "TextMate",
+				"icon": "$(sync)"
+			},
 			{
 				"command": "textmate.follow.cursor",
 				"title": "Follow Cursor",

src/Providers/TreeDataProvider.ts

@@ -2,7 +2,7 @@ import * as vscode from 'vscode';
 import * as vscodeOniguruma from 'vscode-oniguruma';
 // import * as vscodeTextmate from 'vscode-textmate';
 import { stringify } from "../extension";
-import { getScopeName, grammarLanguages, tokenizeFile } from "../TextMate";
+import { getScopeName, grammarLanguages, initTextMate, tokenizeFile } from "../TextMate";
 import { getTrees, queryNode, toRange } from "../TreeSitter";
 import { IGrammar, IMatchResult, IToken, RegExpSource, RuleId, endRuleId, whileRuleId } from "../ITextMate";
 import { IRawGrammar } from "../textmate/main";
@@ -854,6 +854,7 @@ export function initCallStackView(context: vscode.ExtensionContext): void {
 	context.subscriptions.push(
 		vscode.commands.registerTextEditorCommand("textmate.callstack", CallStackView),
 		vscode.commands.registerCommand("textmate.refresh", refresh),
+		vscode.commands.registerCommand("textmate.reload.grammars", reloadGrammars),
 		vscode.commands.registerCommand("textmate.find", find),
 		vscode.commands.registerCommand("textmate.copytoclipboard.grammar", copyGrammarToClipBoard),
 		vscode.commands.registerCommand("textmate.call.details", callDetails),
@@ -1033,6 +1034,14 @@ async function refresh(element?: element) {
 	onDidChangeTreeDataCall.fire(undefined);
 }
 
+async function reloadGrammars(element?: element) {
+	// vscode.window.showInformationMessage(`reloadGrammars\n${JSON.stringify(element)}`);
+
+	initTextMate();
+
+	onDidChangeTreeData.fire(undefined);
+}
+
 async function find(element?: element) {
 	// vscode.window.showInformationMessage(`find\n${JSON.stringify(element)}`);
 	await treeView.reveal(undefined, { focus: true });

src/TextMate.ts

@@ -36,6 +36,9 @@ export const grammarLanguages: {
 };
 
 function parseExtensions() {
+	grammarLanguages.languageId = {};
+	grammarLanguages.scopeName = {};
+
 	for (const extension of vscode.extensions.all as IRelaxedExtension[]) {
 		// if (extension?.isActive === false) {
 		// 	continue;
@@ -181,10 +184,10 @@ async function onigLibInterface(): vscodeTextmate.RegistryOptions['onigLib'] {
 }
 
 let registry: vscodeTextmate.Registry;
-export function initTextMate(context: vscode.ExtensionContext) {
+export function initTextMate(context?: vscode.ExtensionContext) {
 	parseExtensions();
 
-	context.subscriptions.push(vscode.extensions.onDidChange(parseExtensions));
+	context?.subscriptions.push(vscode.extensions.onDidChange(parseExtensions));
 
 	// https://vscode.dev/github/microsoft/vscode/blob/main/src/vs/workbench/services/textMate/common/TMGrammarFactory.ts#L27
 	const options: vscodeTextmate.RegistryOptions = {