From e61ea796c6518bd5533630dfd911d16f3f6c2fdb Mon Sep 17 00:00:00 2001
From: Kyle Carberry <kyle@kwc.io>
Date: Mon, 22 Apr 2019 12:51:05 -0500
Subject: [PATCH] Bundle grammars (#563)

---
 packages/vscode/src/fill/vscodeTextmate.ts | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/packages/vscode/src/fill/vscodeTextmate.ts b/packages/vscode/src/fill/vscodeTextmate.ts
index 06a097fa..47b3cae6 100644
--- a/packages/vscode/src/fill/vscodeTextmate.ts
+++ b/packages/vscode/src/fill/vscodeTextmate.ts
@@ -2,6 +2,17 @@ import * as vscodeTextmate from "../../../../lib/vscode/node_modules/vscode-text
 
 const target = vscodeTextmate as typeof vscodeTextmate;
 
+const ctx = (require as any).context("../../../../lib/extensions", true, /.*\.tmLanguage.json$/);
+// Maps grammar scope to loaded grammar
+const scopeToGrammar = {} as any;
+
+ctx.keys().forEach((key: string) => {
+	const value = ctx(key);
+	if (value.scopeName) {
+		scopeToGrammar[value.scopeName] = value;
+	}
+});
+
 target.Registry = class Registry extends vscodeTextmate.Registry {
 	public constructor(opts: vscodeTextmate.RegistryOptions) {
 		super({
@@ -21,6 +32,13 @@ target.Registry = class Registry extends vscodeTextmate.Registry {
 					}).catch(reason => rej(reason));
 				});
 			},
+			loadGrammar: async (scopeName: string) => {
+				if (scopeToGrammar[scopeName]) {
+					return scopeToGrammar[scopeName];
+				}
+
+				return opts.loadGrammar(scopeName);
+			},
 		});
 	}
 };