From 7870338137d9941d7a3c54128f9b1ba21590ee54 Mon Sep 17 00:00:00 2001
From: Anton Kosyakov <anton.kosyakov@typefox.io>
Date: Mon, 27 Jan 2020 07:51:50 +0000
Subject: [PATCH] [vscode] fix #6907: register grammars only when all languages
 are registered

otherwise not yet registered languages are not considered as embedded languages for grammars

Signed-off-by: Anton Kosyakov <anton.kosyakov@typefox.io>
---
 .../browser/plugin-contribution-handler.ts    | 42 ++++++++++++-------
 1 file changed, 27 insertions(+), 15 deletions(-)

diff --git a/packages/plugin-ext/src/main/browser/plugin-contribution-handler.ts b/packages/plugin-ext/src/main/browser/plugin-contribution-handler.ts
index e0366a9113b24..bb7d1488185c4 100644
--- a/packages/plugin-ext/src/main/browser/plugin-contribution-handler.ts
+++ b/packages/plugin-ext/src/main/browser/plugin-contribution-handler.ts
@@ -108,12 +108,17 @@ export class PluginContributionHandler {
         if (!contributions) {
             return Disposable.NULL;
         }
-        const toDispose = new DisposableCollection();
+        const toDispose = new DisposableCollection(Disposable.create(() => { /* mark as not disposed */ }));
+        // tslint:disable-next-line:no-any
+        const logError = (message: string, ...args: any[]) => console.error(`[${clientId}][${plugin.metadata.model.id}]: ${message}`, ...args);
         const pushContribution = (id: string, contribute: () => Disposable) => {
+            if (toDispose.disposed) {
+                return;
+            }
             try {
                 toDispose.push(contribute());
             } catch (e) {
-                console.error(`[${clientId}][${plugin.metadata.model.id}]: Failed to load '${id}' contribution.`, e);
+                logError(`Failed to load '${id}' contribution.`, e);
             }
         };
 
@@ -181,7 +186,6 @@ export class PluginContributionHandler {
                         });
                     }
                 }
-
                 pushContribution(`grammar.textmate.scope.${grammar.scope}`, () => this.grammarsRegistry.registerTextmateGrammarScope(grammar.scope, {
                     async getGrammarDefinition(): Promise<GrammarDefinition> {
                         return {
@@ -193,17 +197,22 @@ export class PluginContributionHandler {
                     getInjections: (scopeName: string) =>
                         this.injections.get(scopeName)!
                 }));
-                const language = grammar.language;
-                if (language) {
-                    pushContribution(`grammar.language.${language}.scope`, () => this.grammarsRegistry.mapLanguageIdToTextmateGrammar(language, grammar.scope));
-                    pushContribution(`grammar.language.${language}.configuration`, () => this.grammarsRegistry.registerGrammarConfiguration(language, {
-                        embeddedLanguages: this.convertEmbeddedLanguages(grammar.embeddedLanguages),
-                        tokenTypes: this.convertTokenTypes(grammar.tokenTypes)
-                    }));
-                    pushContribution(`grammar.language.${language}.activation`,
-                        () => this.onDidActivateLanguage(language, () => this.monacoTextmateService.activateLanguage(language))
-                    );
-                }
+
+                // load grammars on next tick to await registration of languages from all plugins in current tick
+                // see https://github.com/eclipse-theia/theia/issues/6907#issuecomment-578600243
+                setTimeout(() => {
+                    const language = grammar.language;
+                    if (language) {
+                        pushContribution(`grammar.language.${language}.scope`, () => this.grammarsRegistry.mapLanguageIdToTextmateGrammar(language, grammar.scope));
+                        pushContribution(`grammar.language.${language}.configuration`, () => this.grammarsRegistry.registerGrammarConfiguration(language, {
+                            embeddedLanguages: this.convertEmbeddedLanguages(grammar.embeddedLanguages, logError),
+                            tokenTypes: this.convertTokenTypes(grammar.tokenTypes)
+                        }));
+                        pushContribution(`grammar.language.${language}.activation`,
+                            () => this.onDidActivateLanguage(language, () => this.monacoTextmateService.activateLanguage(language))
+                        );
+                    }
+                });
             }
         }
 
@@ -442,7 +451,7 @@ export class PluginContributionHandler {
         return result;
     }
 
-    private convertEmbeddedLanguages(languages?: ScopeMap): IEmbeddedLanguagesMap | undefined {
+    private convertEmbeddedLanguages(languages: ScopeMap | undefined, logError: (error: string) => void): IEmbeddedLanguagesMap | undefined {
         if (typeof languages === 'undefined' || languages === null) {
             return undefined;
         }
@@ -455,6 +464,9 @@ export class PluginContributionHandler {
             const scope = scopes[i];
             const langId = languages[scope];
             result[scope] = getEncodedLanguageId(langId);
+            if (!result[scope]) {
+                logError(`Language for '${scope}' not found.`);
+            }
         }
         return result;
     }