@kusto/monaco-kusto 4.0.6 → 4.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@kusto/monaco-kusto",
-    "version": "4.0.6",
+    "version": "4.1.3",
     "description": "CSL, KQL plugin for the Monaco Editor",
     "author": {
         "name": "Microsoft"
@@ -55,7 +55,7 @@
     },
     "dependencies": {
         "@kusto/language-service": "2.0.0-beta.0",
-        "@kusto/language-service-next": "0.0.51"
+        "@kusto/language-service-next": "0.0.52"
     },
     "peerDependencies": {
         "monaco-editor": "^0.24.0"

package/release/dev/kustoWorker.js

@@ -6358,6 +6358,7 @@ define('vs/language/kusto/languageService/kustoLanguageService',["require", "exp
             kind: classification.Kind,
         }); });
     }
+    ;
     /**
      * Kusto Language service translates the kusto object model (transpiled from C# by Bridge.Net)
      * to the vscode language server types, which are used by vscode language extensions.
@@ -6511,12 +6512,46 @@ define('vs/language/kusto/languageService/kustoLanguageService',["require", "exp
                 _d);
             this._schemaCache = {};
             this._kustoJsSchema = KustoLanguageService.convertToKustoJsSchema(schema);
-            this._kustoJsSchemaV2 = this.convertToKustoJsSchemaV2(schema);
+            this.__kustoJsSchemaV2 = this.convertToKustoJsSchemaV2(schema);
             this._schema = schema;
+            this._clustersSetInGlobalState = new Set();
+            this._nonEmptyDatabaseSetInGlobalState = new Set(); // used to remove clusters that are already in the global state 
             this.configure(languageSettings);
             this._newlineAppendPipePolicy = new Kusto.Data.IntelliSense.ApplyPolicy();
             this._newlineAppendPipePolicy.Text = '\n| ';
         }
+        KustoLanguageService.prototype.createDatabaseUniqueName = function (clusterName, databaseName) {
+            return clusterName + "_" + databaseName;
+        };
+        Object.defineProperty(KustoLanguageService.prototype, "_kustoJsSchemaV2", {
+            /**
+             * A getter for __kustoJsSchemaV2
+             */
+            get: function () {
+                return this.__kustoJsSchemaV2;
+            },
+            /**
+             * A setter for _kustoJsSchemaV2. After a schema (global state) is set, create 2 sets of cluster and database names.
+             */
+            set: function (globalState) {
+                this.__kustoJsSchemaV2 = globalState;
+                this._clustersSetInGlobalState.clear();
+                this._nonEmptyDatabaseSetInGlobalState.clear();
+                // create 2 Sets with cluster names and database names based on the updated Global State. 
+                for (var i = 0; i < globalState.Clusters.Count; i++) {
+                    var clusterSymbol = this._kustoJsSchemaV2.Clusters.getItem(i);
+                    this._clustersSetInGlobalState.add(clusterSymbol.Name);
+                    for (var i2 = 0; i2 < clusterSymbol.Databases.Count; i2++) {
+                        var databaseSymbol = clusterSymbol.Databases.getItem(i2);
+                        if (databaseSymbol.Tables.Count > 0) { // only include database with tables
+                            this._nonEmptyDatabaseSetInGlobalState.add(this.createDatabaseUniqueName(clusterSymbol.Name, databaseSymbol.Name));
+                        }
+                    }
+                }
+            },
+            enumerable: false,
+            configurable: true
+        });
         KustoLanguageService.prototype.configure = function (languageSettings) {
             this._languageSettings = languageSettings;
             // Since we're still reverting to V1 intellisense for control commands, we need to update the rules provider
@@ -6526,15 +6561,43 @@ define('vs/language/kusto/languageService/kustoLanguageService',["require", "exp
         KustoLanguageService.prototype.doComplete = function (document, position) {
             return this.isIntellisenseV2() ? this.doCompleteV2(document, position) : this.doCompleteV1(document, position);
         };
+        /**
+         * important: Only use during development to test Global State.
+         * Prints clusters, databases and tables that are currently in the GlobalState.
+         */
+        KustoLanguageService.prototype.debugGlobalState = function (globals) {
+            // iterate over clusters
+            console.log("globals.Clusters.Count: " + globals.Clusters.Count);
+            for (var i = 0; i < globals.Clusters.Count; i++) {
+                var cluster = globals.Clusters.getItem(i);
+                console.log("cluster: " + cluster.Name);
+                // iterate over databases
+                console.log("cluster.Databases.Count: " + cluster.Databases.Count);
+                for (var i2 = 0; i2 < cluster.Databases.Count; i2++) {
+                    var database = cluster.Databases.getItem(i2);
+                    console.log("cluster.database: [" + cluster.Name + "].[" + database.Name + "]");
+                    // iterate over tables
+                    console.log("cluster.Databases.Tables.Count: " + database.Tables.Count);
+                    for (var i3 = 0; i3 < database.Tables.Count; i3++) {
+                        var table = database.Tables.getItem(i3);
+                        console.log("cluster.database.table: [" + cluster.Name + "].[" + database.Name + "].[" + table.Name + "]");
+                    }
+                }
+            }
+        };
         KustoLanguageService.prototype.doCompleteV2 = function (document, position) {
             var _this = this;
             if (!document) {
                 return Promise.resolve(ls.CompletionList.create([]));
             }
             var script = this.parseDocumentV2(document);
+            // print cluster/database/tables from CodeScript.Globals
+            // this.debugGlobalState(script.Globals);
+            // get current command
             var cursorOffset = document.offsetAt(position);
-            var currentcommand = this.getCurrentCommandV2(script, cursorOffset);
-            var completionItems = currentcommand.Service.GetCompletionItems(cursorOffset);
+            var currentCommand = script.GetBlockAtPosition(cursorOffset);
+            // get completion items
+            var completionItems = currentCommand.Service.GetCompletionItems(cursorOffset);
             var disabledItems = this.disabledCompletionItemsV2;
             if (this._languageSettings.disabledCompletionItems) {
                 this._languageSettings.disabledCompletionItems.map(function (item) {
@@ -6701,6 +6764,62 @@ define('vs/language/kusto/languageService/kustoLanguageService',["require", "exp
                 });
             });
         };
+        KustoLanguageService.prototype.getClusterReferences = function (document, cursorOffset) {
+            var _a;
+            var script = this.parseDocumentV2(document);
+            var currentBlock = this.getCurrentCommandV2(script, cursorOffset);
+            var clusterReferences = (_a = currentBlock === null || currentBlock === void 0 ? void 0 : currentBlock.Service) === null || _a === void 0 ? void 0 : _a.GetClusterReferences();
+            if (!clusterReferences) {
+                return Promise.resolve([]);
+            }
+            var newClustersReferences = [];
+            var newClustersReferencesSet = new Set(); // used to remove duplicates        
+            // Keep only unique clusters that aren't already exist in the Global State
+            for (var i = 0; i < clusterReferences.Count; i++) {
+                var clusterReference = clusterReferences.getItem(i);
+                var clusterHostName = Kusto.Language.KustoFacts.GetHostName(clusterReference.Cluster);
+                // ignore duplicates
+                if (newClustersReferencesSet.has(clusterHostName)) {
+                    continue;
+                }
+                newClustersReferencesSet.add(clusterHostName);
+                // ignore references that are already in the GlobalState.            
+                if (!this._clustersSetInGlobalState.has(clusterHostName)) {
+                    newClustersReferences.push({ clusterName: clusterHostName });
+                }
+            }
+            return Promise.resolve(newClustersReferences);
+        };
+        KustoLanguageService.prototype.getDatabaseReferences = function (document, cursorOffset) {
+            var _a;
+            var script = this.parseDocumentV2(document);
+            var currentBlock = this.getCurrentCommandV2(script, cursorOffset);
+            var databasesReferences = (_a = currentBlock === null || currentBlock === void 0 ? void 0 : currentBlock.Service) === null || _a === void 0 ? void 0 : _a.GetDatabaseReferences();
+            if (!databasesReferences) {
+                return Promise.resolve([]);
+            }
+            var newDatabasesReferences = [];
+            var newDatabasesReferencesSet = new Set();
+            for (var i1 = 0; i1 < databasesReferences.Count; i1++) {
+                var databaseReference = databasesReferences.getItem(i1);
+                var clusterHostName = Kusto.Language.KustoFacts.GetHostName(databaseReference.Cluster);
+                // ignore duplicates
+                var databaseReferenceUniqueId = this.createDatabaseUniqueName(clusterHostName, databaseReference.Database);
+                if (newDatabasesReferencesSet.has(databaseReferenceUniqueId)) {
+                    continue;
+                }
+                newDatabasesReferencesSet.add(databaseReferenceUniqueId);
+                // ignore references that are already in the GlobalState.
+                var foundInGlobalState = this._nonEmptyDatabaseSetInGlobalState.has(databaseReferenceUniqueId);
+                if (!foundInGlobalState) {
+                    newDatabasesReferences.push({
+                        databaseName: databaseReference.Database,
+                        clusterName: databaseReference.Cluster
+                    });
+                }
+            }
+            return Promise.resolve(newDatabasesReferences);
+        };
         KustoLanguageService.prototype.doValidation = function (document, changeIntervals) {
             var _this = this;
             // didn't implement validation for v1.
@@ -6841,6 +6960,43 @@ define('vs/language/kusto/languageService/kustoLanguageService',["require", "exp
                     : false;
             });
         };
+        KustoLanguageService.prototype.addClusterToSchema = function (document, clusterName, databaseNames) {
+            var clusterNameOnly = Kusto.Language.KustoFacts.GetHostName(clusterName);
+            var cluster = this._kustoJsSchemaV2.GetCluster$1(clusterNameOnly);
+            if (cluster) {
+                // add databases that are not already in the cluster.
+                databaseNames
+                    .filter(function (databaseName) { return !cluster.GetDatabase(databaseName); })
+                    .map(function (databaseName) {
+                    var symbol = new sym.DatabaseSymbol.$ctor1(databaseName, undefined, false);
+                    cluster = cluster.AddDatabase(symbol);
+                });
+            }
+            if (!cluster) {
+                var databaseSymbols = databaseNames
+                    .map(function (databaseName) {
+                    var symbol = new sym.DatabaseSymbol.$ctor1(databaseName, undefined, false);
+                    return symbol;
+                });
+                var databaseSymbolsList = KustoLanguageService.toBridgeList(databaseSymbols);
+                cluster = new sym.ClusterSymbol.$ctor1(clusterNameOnly, databaseSymbolsList, false);
+            }
+            this._kustoJsSchemaV2 = this._kustoJsSchemaV2.AddOrReplaceCluster(cluster);
+            this._script = k2.CodeScript.From$1(document.getText(), this._kustoJsSchemaV2);
+            return Promise.resolve();
+        };
+        KustoLanguageService.prototype.addDatabaseToSchema = function (document, clusterName, databaseSchema) {
+            var clusterHostName = Kusto.Language.KustoFacts.GetHostName(clusterName);
+            var cluster = this._kustoJsSchemaV2.GetCluster$1(clusterHostName);
+            if (!cluster) {
+                cluster = new sym.ClusterSymbol.$ctor1(clusterHostName, null, false);
+            }
+            var databaseSymbol = KustoLanguageService.convertToDatabaseSymbol(databaseSchema);
+            cluster = cluster.AddOrUpdateDatabase(databaseSymbol);
+            this._kustoJsSchemaV2 = this._kustoJsSchemaV2.AddOrReplaceCluster(cluster);
+            this._script = k2.CodeScript.From$1(document.getText(), this._kustoJsSchemaV2);
+            return Promise.resolve();
+        };
         KustoLanguageService.prototype.setSchema = function (schema) {
             var _this = this;
             this._schema = schema;
@@ -7555,7 +7711,7 @@ define('vs/language/kusto/languageService/kustoLanguageService',["require", "exp
             var argumentType = new sym.TableSymbol.ctor(param.columns.map(function (col) { return KustoLanguageService.createColumnSymbol(col); }));
             return new sym.Parameter.$ctor2(param.name, argumentType);
         };
-        KustoLanguageService.convertToDatabaseSymbol = function (db, globalState, addFunctions) {
+        KustoLanguageService.convertToDatabaseSymbol = function (db) {
             var createFunctionSymbol = function (fn) {
                 var parameters = fn.inputParameters.map(function (param) {
                     return KustoLanguageService.createParameter(param);
@@ -7612,7 +7768,7 @@ define('vs/language/kusto/languageService/kustoLanguageService',["require", "exp
                     cachedDb.database.majorVersion < db.majorVersion ||
                     (shouldIncludeFunctions && !cachedDb.includesFunctions)) {
                     // only add functions for the database in context (it's very time consuming)
-                    var databaseSymbol_1 = KustoLanguageService.convertToDatabaseSymbol(db, globalState, shouldIncludeFunctions);
+                    var databaseSymbol_1 = KustoLanguageService.convertToDatabaseSymbol(db);
                     cached[db.name] = { database: db, symbol: databaseSymbol_1, includesFunctions: shouldIncludeFunctions };
                 }
                 var databaseSymbol = cached[db.name].symbol;
@@ -7874,6 +8030,22 @@ define('vs/language/kusto/kustoWorker',["require", "exports", "./languageService
         KustoWorker.prototype.setSchema = function (schema) {
             return this._languageService.setSchema(schema);
         };
+        KustoWorker.prototype.addClusterToSchema = function (uri, clusterName, databasesNames) {
+            var document = this._getTextDocument(uri);
+            if (!document) {
+                console.error("addClusterToSchema: document is " + document + ". uri is " + uri);
+                return Promise.resolve();
+            }
+            return this._languageService.addClusterToSchema(document, clusterName, databasesNames);
+        };
+        KustoWorker.prototype.addDatabaseToSchema = function (uri, clusterName, databaseSchema) {
+            var document = this._getTextDocument(uri);
+            if (!document) {
+                console.error("addDatabaseToSchema: document is " + document + ". uri is " + uri);
+                return Promise.resolve();
+            }
+            return this._languageService.addDatabaseToSchema(document, clusterName, databaseSchema);
+        };
         KustoWorker.prototype.setSchemaFromShowSchema = function (schema, clusterConnectionString, databaseInContextName) {
             return this._languageService.setSchemaFromShowSchema(schema, clusterConnectionString, databaseInContextName);
         };
@@ -8044,6 +8216,20 @@ define('vs/language/kusto/kustoWorker',["require", "exports", "./languageService
         KustoWorker.prototype.setParameters = function (parameters) {
             return this._languageService.setParameters(parameters);
         };
+        KustoWorker.prototype.getClusterReferences = function (uri, cursorOffset) {
+            var document = this._getTextDocument(uri);
+            if (!document) {
+                return Promise.resolve(null);
+            }
+            return this._languageService.getClusterReferences(document, cursorOffset);
+        };
+        KustoWorker.prototype.getDatabaseReferences = function (uri, cursorOffset) {
+            var document = this._getTextDocument(uri);
+            if (!document) {
+                return Promise.resolve(null);
+            }
+            return this._languageService.getDatabaseReferences(document, cursorOffset);
+        };
         KustoWorker.prototype._getTextDocument = function (uri) {
             var models = this._ctx.getMirrorModels();
             for (var _i = 0, models_1 = models; _i < models_1.length; _i++) {

package/release/esm/kustoWorker.js

@@ -11,6 +11,22 @@ var KustoWorker = /** @class */ (function () {
     KustoWorker.prototype.setSchema = function (schema) {
         return this._languageService.setSchema(schema);
     };
+    KustoWorker.prototype.addClusterToSchema = function (uri, clusterName, databasesNames) {
+        var document = this._getTextDocument(uri);
+        if (!document) {
+            console.error("addClusterToSchema: document is " + document + ". uri is " + uri);
+            return Promise.resolve();
+        }
+        return this._languageService.addClusterToSchema(document, clusterName, databasesNames);
+    };
+    KustoWorker.prototype.addDatabaseToSchema = function (uri, clusterName, databaseSchema) {
+        var document = this._getTextDocument(uri);
+        if (!document) {
+            console.error("addDatabaseToSchema: document is " + document + ". uri is " + uri);
+            return Promise.resolve();
+        }
+        return this._languageService.addDatabaseToSchema(document, clusterName, databaseSchema);
+    };
     KustoWorker.prototype.setSchemaFromShowSchema = function (schema, clusterConnectionString, databaseInContextName) {
         return this._languageService.setSchemaFromShowSchema(schema, clusterConnectionString, databaseInContextName);
     };
@@ -181,6 +197,20 @@ var KustoWorker = /** @class */ (function () {
     KustoWorker.prototype.setParameters = function (parameters) {
         return this._languageService.setParameters(parameters);
     };
+    KustoWorker.prototype.getClusterReferences = function (uri, cursorOffset) {
+        var document = this._getTextDocument(uri);
+        if (!document) {
+            return Promise.resolve(null);
+        }
+        return this._languageService.getClusterReferences(document, cursorOffset);
+    };
+    KustoWorker.prototype.getDatabaseReferences = function (uri, cursorOffset) {
+        var document = this._getTextDocument(uri);
+        if (!document) {
+            return Promise.resolve(null);
+        }
+        return this._languageService.getDatabaseReferences(document, cursorOffset);
+    };
     KustoWorker.prototype._getTextDocument = function (uri) {
         var models = this._ctx.getMirrorModels();
         for (var _i = 0, models_1 = models; _i < models_1.length; _i++) {

package/release/esm/languageService/kustoLanguageService.js

@@ -96,6 +96,7 @@ function toClassifiedRange(k2Classifications) {
         kind: classification.Kind,
     }); });
 }
+;
 /**
  * Kusto Language service translates the kusto object model (transpiled from C# by Bridge.Net)
  * to the vscode language server types, which are used by vscode language extensions.
@@ -249,12 +250,46 @@ var KustoLanguageService = /** @class */ (function () {
             _d);
         this._schemaCache = {};
         this._kustoJsSchema = KustoLanguageService.convertToKustoJsSchema(schema);
-        this._kustoJsSchemaV2 = this.convertToKustoJsSchemaV2(schema);
+        this.__kustoJsSchemaV2 = this.convertToKustoJsSchemaV2(schema);
         this._schema = schema;
+        this._clustersSetInGlobalState = new Set();
+        this._nonEmptyDatabaseSetInGlobalState = new Set(); // used to remove clusters that are already in the global state 
         this.configure(languageSettings);
         this._newlineAppendPipePolicy = new Kusto.Data.IntelliSense.ApplyPolicy();
         this._newlineAppendPipePolicy.Text = '\n| ';
     }
+    KustoLanguageService.prototype.createDatabaseUniqueName = function (clusterName, databaseName) {
+        return clusterName + "_" + databaseName;
+    };
+    Object.defineProperty(KustoLanguageService.prototype, "_kustoJsSchemaV2", {
+        /**
+         * A getter for __kustoJsSchemaV2
+         */
+        get: function () {
+            return this.__kustoJsSchemaV2;
+        },
+        /**
+         * A setter for _kustoJsSchemaV2. After a schema (global state) is set, create 2 sets of cluster and database names.
+         */
+        set: function (globalState) {
+            this.__kustoJsSchemaV2 = globalState;
+            this._clustersSetInGlobalState.clear();
+            this._nonEmptyDatabaseSetInGlobalState.clear();
+            // create 2 Sets with cluster names and database names based on the updated Global State. 
+            for (var i = 0; i < globalState.Clusters.Count; i++) {
+                var clusterSymbol = this._kustoJsSchemaV2.Clusters.getItem(i);
+                this._clustersSetInGlobalState.add(clusterSymbol.Name);
+                for (var i2 = 0; i2 < clusterSymbol.Databases.Count; i2++) {
+                    var databaseSymbol = clusterSymbol.Databases.getItem(i2);
+                    if (databaseSymbol.Tables.Count > 0) { // only include database with tables
+                        this._nonEmptyDatabaseSetInGlobalState.add(this.createDatabaseUniqueName(clusterSymbol.Name, databaseSymbol.Name));
+                    }
+                }
+            }
+        },
+        enumerable: false,
+        configurable: true
+    });
     KustoLanguageService.prototype.configure = function (languageSettings) {
         this._languageSettings = languageSettings;
         // Since we're still reverting to V1 intellisense for control commands, we need to update the rules provider
@@ -264,15 +299,43 @@ var KustoLanguageService = /** @class */ (function () {
     KustoLanguageService.prototype.doComplete = function (document, position) {
         return this.isIntellisenseV2() ? this.doCompleteV2(document, position) : this.doCompleteV1(document, position);
     };
+    /**
+     * important: Only use during development to test Global State.
+     * Prints clusters, databases and tables that are currently in the GlobalState.
+     */
+    KustoLanguageService.prototype.debugGlobalState = function (globals) {
+        // iterate over clusters
+        console.log("globals.Clusters.Count: " + globals.Clusters.Count);
+        for (var i = 0; i < globals.Clusters.Count; i++) {
+            var cluster = globals.Clusters.getItem(i);
+            console.log("cluster: " + cluster.Name);
+            // iterate over databases
+            console.log("cluster.Databases.Count: " + cluster.Databases.Count);
+            for (var i2 = 0; i2 < cluster.Databases.Count; i2++) {
+                var database = cluster.Databases.getItem(i2);
+                console.log("cluster.database: [" + cluster.Name + "].[" + database.Name + "]");
+                // iterate over tables
+                console.log("cluster.Databases.Tables.Count: " + database.Tables.Count);
+                for (var i3 = 0; i3 < database.Tables.Count; i3++) {
+                    var table = database.Tables.getItem(i3);
+                    console.log("cluster.database.table: [" + cluster.Name + "].[" + database.Name + "].[" + table.Name + "]");
+                }
+            }
+        }
+    };
     KustoLanguageService.prototype.doCompleteV2 = function (document, position) {
         var _this = this;
         if (!document) {
             return Promise.resolve(ls.CompletionList.create([]));
         }
         var script = this.parseDocumentV2(document);
+        // print cluster/database/tables from CodeScript.Globals
+        // this.debugGlobalState(script.Globals);
+        // get current command
         var cursorOffset = document.offsetAt(position);
-        var currentcommand = this.getCurrentCommandV2(script, cursorOffset);
-        var completionItems = currentcommand.Service.GetCompletionItems(cursorOffset);
+        var currentCommand = script.GetBlockAtPosition(cursorOffset);
+        // get completion items
+        var completionItems = currentCommand.Service.GetCompletionItems(cursorOffset);
         var disabledItems = this.disabledCompletionItemsV2;
         if (this._languageSettings.disabledCompletionItems) {
             this._languageSettings.disabledCompletionItems.map(function (item) {
@@ -439,6 +502,62 @@ var KustoLanguageService = /** @class */ (function () {
             });
         });
     };
+    KustoLanguageService.prototype.getClusterReferences = function (document, cursorOffset) {
+        var _a;
+        var script = this.parseDocumentV2(document);
+        var currentBlock = this.getCurrentCommandV2(script, cursorOffset);
+        var clusterReferences = (_a = currentBlock === null || currentBlock === void 0 ? void 0 : currentBlock.Service) === null || _a === void 0 ? void 0 : _a.GetClusterReferences();
+        if (!clusterReferences) {
+            return Promise.resolve([]);
+        }
+        var newClustersReferences = [];
+        var newClustersReferencesSet = new Set(); // used to remove duplicates        
+        // Keep only unique clusters that aren't already exist in the Global State
+        for (var i = 0; i < clusterReferences.Count; i++) {
+            var clusterReference = clusterReferences.getItem(i);
+            var clusterHostName = Kusto.Language.KustoFacts.GetHostName(clusterReference.Cluster);
+            // ignore duplicates
+            if (newClustersReferencesSet.has(clusterHostName)) {
+                continue;
+            }
+            newClustersReferencesSet.add(clusterHostName);
+            // ignore references that are already in the GlobalState.            
+            if (!this._clustersSetInGlobalState.has(clusterHostName)) {
+                newClustersReferences.push({ clusterName: clusterHostName });
+            }
+        }
+        return Promise.resolve(newClustersReferences);
+    };
+    KustoLanguageService.prototype.getDatabaseReferences = function (document, cursorOffset) {
+        var _a;
+        var script = this.parseDocumentV2(document);
+        var currentBlock = this.getCurrentCommandV2(script, cursorOffset);
+        var databasesReferences = (_a = currentBlock === null || currentBlock === void 0 ? void 0 : currentBlock.Service) === null || _a === void 0 ? void 0 : _a.GetDatabaseReferences();
+        if (!databasesReferences) {
+            return Promise.resolve([]);
+        }
+        var newDatabasesReferences = [];
+        var newDatabasesReferencesSet = new Set();
+        for (var i1 = 0; i1 < databasesReferences.Count; i1++) {
+            var databaseReference = databasesReferences.getItem(i1);
+            var clusterHostName = Kusto.Language.KustoFacts.GetHostName(databaseReference.Cluster);
+            // ignore duplicates
+            var databaseReferenceUniqueId = this.createDatabaseUniqueName(clusterHostName, databaseReference.Database);
+            if (newDatabasesReferencesSet.has(databaseReferenceUniqueId)) {
+                continue;
+            }
+            newDatabasesReferencesSet.add(databaseReferenceUniqueId);
+            // ignore references that are already in the GlobalState.
+            var foundInGlobalState = this._nonEmptyDatabaseSetInGlobalState.has(databaseReferenceUniqueId);
+            if (!foundInGlobalState) {
+                newDatabasesReferences.push({
+                    databaseName: databaseReference.Database,
+                    clusterName: databaseReference.Cluster
+                });
+            }
+        }
+        return Promise.resolve(newDatabasesReferences);
+    };
     KustoLanguageService.prototype.doValidation = function (document, changeIntervals) {
         var _this = this;
         // didn't implement validation for v1.
@@ -579,6 +698,43 @@ var KustoLanguageService = /** @class */ (function () {
                 : false;
         });
     };
+    KustoLanguageService.prototype.addClusterToSchema = function (document, clusterName, databaseNames) {
+        var clusterNameOnly = Kusto.Language.KustoFacts.GetHostName(clusterName);
+        var cluster = this._kustoJsSchemaV2.GetCluster$1(clusterNameOnly);
+        if (cluster) {
+            // add databases that are not already in the cluster.
+            databaseNames
+                .filter(function (databaseName) { return !cluster.GetDatabase(databaseName); })
+                .map(function (databaseName) {
+                var symbol = new sym.DatabaseSymbol.$ctor1(databaseName, undefined, false);
+                cluster = cluster.AddDatabase(symbol);
+            });
+        }
+        if (!cluster) {
+            var databaseSymbols = databaseNames
+                .map(function (databaseName) {
+                var symbol = new sym.DatabaseSymbol.$ctor1(databaseName, undefined, false);
+                return symbol;
+            });
+            var databaseSymbolsList = KustoLanguageService.toBridgeList(databaseSymbols);
+            cluster = new sym.ClusterSymbol.$ctor1(clusterNameOnly, databaseSymbolsList, false);
+        }
+        this._kustoJsSchemaV2 = this._kustoJsSchemaV2.AddOrReplaceCluster(cluster);
+        this._script = k2.CodeScript.From$1(document.getText(), this._kustoJsSchemaV2);
+        return Promise.resolve();
+    };
+    KustoLanguageService.prototype.addDatabaseToSchema = function (document, clusterName, databaseSchema) {
+        var clusterHostName = Kusto.Language.KustoFacts.GetHostName(clusterName);
+        var cluster = this._kustoJsSchemaV2.GetCluster$1(clusterHostName);
+        if (!cluster) {
+            cluster = new sym.ClusterSymbol.$ctor1(clusterHostName, null, false);
+        }
+        var databaseSymbol = KustoLanguageService.convertToDatabaseSymbol(databaseSchema);
+        cluster = cluster.AddOrUpdateDatabase(databaseSymbol);
+        this._kustoJsSchemaV2 = this._kustoJsSchemaV2.AddOrReplaceCluster(cluster);
+        this._script = k2.CodeScript.From$1(document.getText(), this._kustoJsSchemaV2);
+        return Promise.resolve();
+    };
     KustoLanguageService.prototype.setSchema = function (schema) {
         var _this = this;
         this._schema = schema;
@@ -1293,7 +1449,7 @@ var KustoLanguageService = /** @class */ (function () {
         var argumentType = new sym.TableSymbol.ctor(param.columns.map(function (col) { return KustoLanguageService.createColumnSymbol(col); }));
         return new sym.Parameter.$ctor2(param.name, argumentType);
     };
-    KustoLanguageService.convertToDatabaseSymbol = function (db, globalState, addFunctions) {
+    KustoLanguageService.convertToDatabaseSymbol = function (db) {
         var createFunctionSymbol = function (fn) {
             var parameters = fn.inputParameters.map(function (param) {
                 return KustoLanguageService.createParameter(param);
@@ -1350,7 +1506,7 @@ var KustoLanguageService = /** @class */ (function () {
                 cachedDb.database.majorVersion < db.majorVersion ||
                 (shouldIncludeFunctions && !cachedDb.includesFunctions)) {
                 // only add functions for the database in context (it's very time consuming)
-                var databaseSymbol_1 = KustoLanguageService.convertToDatabaseSymbol(db, globalState, shouldIncludeFunctions);
+                var databaseSymbol_1 = KustoLanguageService.convertToDatabaseSymbol(db);
                 cached[db.name] = { database: db, symbol: databaseSymbol_1, includesFunctions: shouldIncludeFunctions };
             }
             var databaseSymbol = cached[db.name].symbol;

package/release/esm/monaco.d.ts

@@ -119,6 +119,50 @@ declare module monaco.languages.kusto {
         doCurrentCommandFormat(uri: string, caretPosition: ls.Position): Promise<ls.TextEdit[]>;
         doValidation(uri: string, intervals: { start: number; end: number }[]): Promise<ls.Diagnostic[]>;
         setParameters(parameters: ScalarParameter[]): void;
+        /**
+         * Get all the database references from the current command. 
+         * If database's schema is already cached in previous calls to setSchema or addDatabaseToSchema it will not be returned.
+         * This method should be used to get all the cross-databases in a command, then schema for the database should be fetched and added with addDatabaseToSchema.
+         * @example
+         * If the current command includes: cluster('help').database('Samples') 
+         * getDatabaseReferences will return [{ clusterName: 'help', databaseName 'Samples' }]
+         */
+        getDatabaseReferences(uri: string, cursorOffset: number): Promise<DatabaseReference[]>;
+        /**
+         * Get all the cluster references from the current command.
+         * If cluster's schema is already cached it will not be returned.
+         * This method should be used to get all the cross-clusters in a command, then schema for the cluster should be fetched and added with addClusterToSchema.
+         * cluster name is returned exactly as written in the KQL `cluster(<cluster name>)` function.
+         * @example
+         * If the current command includes: cluster('help')
+         * it returns [{ clusterName: 'help' }]
+         * @example
+         * If the current command includes: cluster('https://demo11.westus.kusto.windows.net')
+         * getClusterReferences will return [{ clusterName: 'https://demo11.westus.kusto.windows.net' }]
+         */
+        getClusterReferences(uri: string, cursorOffset: number): Promise<ClusterReference[]>;
+        /**
+         * Adds a database's scheme. Useful with getDatabaseReferences to load schema for cross-cluster commands.
+         * @param clusterName the name of the cluster as returned from getDatabaseReferences/getClusterReferences.
+         * @example
+         * - User enters cluster('help').database('Samples')
+         * - hosting app calls getDatabaseReferences which returns [{ clusterName: 'help', databaseName: 'Samples' }]. 
+         * - hosting app fetches the database Schema from https://help.kusto.windows.net
+         * - hosting app calls 'addDatabaseToSchema' with the database's schema.
+         * - now, when user types cluster('help').database('Samples') then the auto complete list will show all the tables.
+         */
+        addDatabaseToSchema(uri: string, clusterName: string, databaseSchema: Database): Promise<void>;
+        /**
+         * Adds a cluster's databases to the schema. Useful when used with getClusterReferences in cross-cluster commands.
+         * @param clusterName the name of the cluster as returned in getClusterReferences.
+         * @example
+         * - User enters cluster('help')
+         * - hosting app calls getClusterReferences which returns [{ clusterName: 'help' }]. 
+         * - hosting app fetches the list of databases from https://help.kusto.windows.net
+         * - hosting app calls addClusterToSchema with the list of databases.
+         * - now, when user type `cluster('help').database(` then the auto complete list will show all the databases.
+         */
+        addClusterToSchema(uri: string, clusterName: string, databasesNames: string[]): Promise<void>;
     }
 
     /**
@@ -234,6 +278,15 @@ declare module monaco.languages.kusto {
         location: { startOffset: number; endOffset: number };
     }
 
+    export interface DatabaseReference {
+        databaseName: string;
+        clusterName: string; 
+    };
+
+    export interface ClusterReference {
+        clusterName: string; 
+    }
+
     export type RenderOptionKeys = keyof RenderOptions;
 
     export type OnDidProvideCompletionItems = (list: ls.CompletionList) => Promise<ls.CompletionList>;