brighterscript 1.0.0-alpha.23 → 1.0.0-alpha.25

package/dist/LanguageServer.js

@@ -6,7 +6,7 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
     return c > 3 && r && Object.defineProperty(target, key, r), r;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.CustomCommands = exports.LanguageServer = void 0;
+exports.NotificationName = exports.CustomCommands = exports.LanguageServer = void 0;
 require("array-flat-polyfill");
 const fastGlob = require("fast-glob");
 const path = require("path");
@@ -24,6 +24,7 @@ const KeyedThrottler_1 = require("./KeyedThrottler");
 const DiagnosticCollection_1 = require("./DiagnosticCollection");
 const reflection_1 = require("./astUtils/reflection");
 const SemanticTokenUtils_1 = require("./SemanticTokenUtils");
+const BusyStatusTracker_1 = require("./BusyStatusTracker");
 class LanguageServer {
     constructor() {
         this.connection = undefined;
@@ -53,6 +54,12 @@ class LanguageServer {
         this.validateThrottler = new Throttler_1.Throttler(0);
         this.sendDiagnosticsThrottler = new Throttler_1.Throttler(0);
         this.boundValidateAll = this.validateAll.bind(this);
+        this.busyStatusTracker = new BusyStatusTracker_1.BusyStatusTracker();
+        this.busyStatusIndex = -1;
+        /**
+         * A unique project counter to help distinguish log entries in lsp mode
+         */
+        this.projectCounter = 0;
         this.diagnosticCollection = new DiagnosticCollection_1.DiagnosticCollection();
     }
     createConnection() {
@@ -66,6 +73,10 @@ class LanguageServer {
         // Create a connection for the server. The connection uses Node's IPC as a transport.
         // Also include all preview / proposed LSP features.
         this.connection = this.createConnection();
+        // Send the current status of the busyStatusTracker anytime it changes
+        this.busyStatusTracker.on('change', (status) => {
+            this.sendBusyStatus(status);
+        });
         //listen to all of the output log events and pipe them into the debug channel in the extension
         this.loggerSubscription = Logger_1.Logger.subscribe((text) => {
             this.connection.tracer.log(text);
@@ -120,9 +131,17 @@ class LanguageServer {
         // Listen on the connection
         this.connection.listen();
     }
+    sendBusyStatus(status) {
+        this.busyStatusIndex = ++this.busyStatusIndex <= 0 ? 0 : this.busyStatusIndex;
+        this.connection.sendNotification(NotificationName.busyStatus, {
+            status: status,
+            timestamp: Date.now(),
+            index: this.busyStatusIndex,
+            activeRuns: [...this.busyStatusTracker.activeRuns]
+        });
+    }
     /**
      * Called when the client starts initialization
-     * @param params
      */
     onInitialize(params) {
         let clientCapabilities = params.capabilities;
@@ -169,11 +188,17 @@ class LanguageServer {
      */
     async getWorkspaceExcludeGlobs(workspaceFolder) {
         var _a;
-        //get any `files.exclude` globs to use to filter
-        let config = await this.connection.workspace.getConfiguration({
-            scopeUri: workspaceFolder,
-            section: 'files'
-        });
+        let config = {
+            exclude: {}
+        };
+        //if supported, ask vscode for the `files.exclude` configuration
+        if (this.hasConfigurationCapability) {
+            //get any `files.exclude` globs to use to filter
+            config = await this.connection.workspace.getConfiguration({
+                scopeUri: workspaceFolder,
+                section: 'files'
+            });
+        }
         return Object
             .keys((_a = config === null || config === void 0 ? void 0 : config.exclude) !== null && _a !== void 0 ? _a : {})
             .filter(x => { var _a; return (_a = config === null || config === void 0 ? void 0 : config.exclude) === null || _a === void 0 ? void 0 : _a[x]; })
@@ -206,10 +231,30 @@ class LanguageServer {
         if (files.length > 0) {
             return files.map(file => (0, util_1.standardizePath) `${path.dirname(file.src)}`);
         }
-        else {
-            //treat the workspace folder as a brightscript project itself
-            return [workspaceFolder];
+        //look for roku project folders
+        const rokuLikeDirs = (await Promise.all(
+        //find all folders containing a `manifest` file
+        (await roku_deploy_1.rokuDeploy.getFilePaths([
+            '**/manifest',
+            ...excludes
+            //is there at least one .bs|.brs file under the `/source` folder?
+        ], workspaceFolder)).map(async (manifestEntry) => {
+            const manifestDir = path.dirname(manifestEntry.src);
+            const files = await roku_deploy_1.rokuDeploy.getFilePaths([
+                'source/**/*.{brs,bs}',
+                ...excludes
+            ], manifestDir);
+            if (files.length > 0) {
+                return manifestDir;
+            }
+        })
+        //throw out nulls
+        )).filter(x => !!x);
+        if (rokuLikeDirs.length > 0) {
+            return rokuLikeDirs;
         }
+        //treat the workspace folder as a brightscript project itself
+        return [workspaceFolder];
     }
     /**
      * Find all folders with bsconfig.json files in them, and treat each as a project.
@@ -258,7 +303,6 @@ class LanguageServer {
     }
     /**
      * Called when the client has finished initializing
-     * @param params
      */
     async onInitialized() {
         let projectCreatedDeferred = new deferred_1.Deferred();
@@ -298,7 +342,6 @@ class LanguageServer {
         if (waitForFirstProject) {
             await this.initialProjectsCreated;
         }
-        let status;
         for (let project of this.getProjects()) {
             try {
                 await project.firstRunPromise;
@@ -311,7 +354,6 @@ class LanguageServer {
                 this.sendCriticalFailure(`BrighterScript language server failed to start: \n${e.message}`);
             }
         }
-        this.connection.sendNotification('build-status', status ? status : 'success');
     }
     /**
      * Event handler for when the program wants to load file contents.
@@ -332,11 +374,16 @@ class LanguageServer {
         else {
             scopeUri = vscode_uri_1.URI.file(workspacePath).toString();
         }
-        //look for config group called "brightscript"
-        let config = await this.connection.workspace.getConfiguration({
-            scopeUri: scopeUri,
-            section: 'brightscript'
-        });
+        let config = {
+            configFile: undefined
+        };
+        //if the client supports configuration, look for config group called "brightscript"
+        if (this.hasConfigurationCapability) {
+            config = await this.connection.workspace.getConfiguration({
+                scopeUri: scopeUri,
+                section: 'brightscript'
+            });
+        }
         let configFilePath;
         //if there's a setting, we need to find the file or show error if it can't be found
         if (config === null || config === void 0 ? void 0 : config.configFile) {
@@ -361,7 +408,12 @@ class LanguageServer {
         //no config file could be found
         return undefined;
     }
-    async createProject(projectPath, workspacePath = projectPath) {
+    /**
+     * @param projectPath path to the project
+     * @param workspacePath path to the workspace in which all project should reside or are referenced by
+     * @param projectNumber an optional project number to assign to the project. Used when reloading projects that should keep the same number
+     */
+    async createProject(projectPath, workspacePath = projectPath, projectNumber) {
         workspacePath !== null && workspacePath !== void 0 ? workspacePath : (workspacePath = projectPath);
         let project = this.projects.find((x) => x.projectPath === projectPath);
         //skip this project if we already have it
@@ -369,6 +421,9 @@ class LanguageServer {
             return;
         }
         let builder = new ProgramBuilder_1.ProgramBuilder();
+        projectNumber !== null && projectNumber !== void 0 ? projectNumber : (projectNumber = this.projectCounter++);
+        builder.logger.prefix = `[prj${projectNumber}]`;
+        builder.logger.log(`Created project #${projectNumber} for: "${projectPath}"`);
         //flush diagnostics every time the program finishes validating
         builder.plugins.add({
             name: 'bsc-language-server',
@@ -390,21 +445,11 @@ class LanguageServer {
             //config file doesn't exist...let `brighterscript` resolve the default way
             configFilePath = undefined;
         }
-        let firstRunPromise = builder.run({
-            cwd: cwd,
-            project: configFilePath,
-            watch: false,
-            createPackage: false,
-            deploy: false,
-            copyToStaging: false,
-            showDiagnosticsInConsole: false
-        });
-        firstRunPromise.catch((err) => {
-            console.error(err);
-        });
+        const firstRunDeferred = new deferred_1.Deferred();
         let newProject = {
+            projectNumber: projectNumber,
             builder: builder,
-            firstRunPromise: firstRunPromise,
+            firstRunPromise: firstRunDeferred.promise,
             projectPath: projectPath,
             workspacePath: workspacePath,
             isFirstRunComplete: false,
@@ -413,19 +458,31 @@ class LanguageServer {
             isStandaloneFileProject: false
         };
         this.projects.push(newProject);
-        await firstRunPromise.then(() => {
+        try {
+            await builder.run({
+                cwd: cwd,
+                project: configFilePath,
+                watch: false,
+                createPackage: false,
+                deploy: false,
+                copyToStaging: false,
+                showDiagnosticsInConsole: false
+            });
             newProject.isFirstRunComplete = true;
             newProject.isFirstRunSuccessful = true;
-        }).catch(() => {
+            firstRunDeferred.resolve();
+        }
+        catch (e) {
+            builder.logger.error(e);
+            firstRunDeferred.reject(e);
             newProject.isFirstRunComplete = true;
             newProject.isFirstRunSuccessful = false;
-        }).then(() => {
-            //if we found a deprecated brsconfig.json, add a diagnostic warning the user
-            if (configFilePath && path.basename(configFilePath) === 'brsconfig.json') {
-                builder.addDiagnostic(configFilePath, Object.assign(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.brsConfigJsonIsDeprecated()), { range: util_1.util.createRange(0, 0, 0, 0) }));
-                return this.sendDiagnostics();
-            }
-        });
+        }
+        //if we found a deprecated brsconfig.json, add a diagnostic warning the user
+        if (configFilePath && path.basename(configFilePath) === 'brsconfig.json') {
+            builder.addDiagnostic(configFilePath, Object.assign(Object.assign({}, DiagnosticMessages_1.DiagnosticMessages.brsConfigJsonIsDeprecated()), { range: util_1.util.createRange(0, 0, 0, 0) }));
+            return this.sendDiagnostics();
+        }
     }
     async createStandaloneFileProject(srcPath) {
         //skip this workspace if we already have it
@@ -458,6 +515,7 @@ class LanguageServer {
             console.error(err);
         });
         let newProject = {
+            projectNumber: this.projectCounter++,
             builder: builder,
             firstRunPromise: firstRunPromise,
             projectPath: srcPath,
@@ -486,7 +544,6 @@ class LanguageServer {
     }
     /**
      * Provide a list of completion items based on the current cursor position
-     * @param textDocumentPosition
      */
     async onCompletion(params) {
         //ensure programs are initialized
@@ -494,17 +551,24 @@ class LanguageServer {
         let filePath = util_1.util.uriToPath(params.textDocument.uri);
         //wait until the file has settled
         await this.keyedThrottler.onIdleOnce(filePath, true);
+        // make sure validation is complete
+        await this.validateAllThrottled();
+        //wait for the validation cycle to settle
+        await this.onValidateSettled();
         let completions = this
             .getProjects()
             .flatMap(workspace => workspace.builder.program.getCompletions(filePath, params.position));
+        //only send one completion if name and type are the same
+        let completionsMap = new Map();
         for (let completion of completions) {
             completion.commitCharacters = ['.'];
+            let key = `${completion.sortText}-${completion.label}-${completion.kind}`;
+            completionsMap.set(key, completion);
         }
-        return completions;
+        return [...completionsMap.values()];
     }
     /**
      * Provide a full completion item from the selection
-     * @param item
      */
     onCompletionResolve(item) {
         if (item.data === 1) {
@@ -518,6 +582,7 @@ class LanguageServer {
         return item;
     }
     async onCodeAction(params) {
+        var _a;
         //ensure programs are initialized
         await this.waitAllProjectFirstRuns();
         let srcPath = util_1.util.uriToPath(params.textDocument.uri);
@@ -531,7 +596,7 @@ class LanguageServer {
         //clone the diagnostics for each code action, since certain diagnostics can have circular reference properties that kill the language server if serialized
         for (const codeAction of codeActions) {
             if (codeAction.diagnostics) {
-                codeAction.diagnostics = codeAction.diagnostics.map(x => util_1.util.toDiagnostic(x));
+                codeAction.diagnostics = (_a = codeAction.diagnostics) === null || _a === void 0 ? void 0 : _a.map(x => util_1.util.toDiagnostic(x, params.textDocument.uri));
             }
         }
         return codeActions;
@@ -561,7 +626,7 @@ class LanguageServer {
             if (project.isStandaloneFileProject === false) {
                 this.removeProject(project);
                 //create a new workspace/brs program
-                await this.createProject(project.projectPath, project.workspacePath);
+                await this.createProject(project.projectPath, project.workspacePath, project.projectNumber);
                 //handle temp workspace
             }
             else {
@@ -638,12 +703,10 @@ class LanguageServer {
      * The CLIENT is in charge of what files to watch, so all client
      * implementations should ensure that all valid project
      * file types are watched (.brs,.bs,.xml,manifest, and any json/text/image files)
-     * @param params
      */
     async onDidChangeWatchedFiles(params) {
         //ensure programs are initialized
         await this.waitAllProjectFirstRuns();
-        this.connection.sendNotification('build-status', 'building');
         let projects = this.getProjects();
         //convert all file paths to absolute paths
         let changes = params.changes.map(x => {
@@ -712,37 +775,33 @@ class LanguageServer {
             //give every workspace the chance to handle file changes
             await Promise.all(projects.map((project) => this.handleFileChanges(project, changes)));
         }
-        this.connection.sendNotification('build-status', 'success');
     }
     /**
      * This only operates on files that match the specified files globs, so it is safe to throw
      * any file changes you receive with no unexpected side-effects
-     * @param changes
      */
     async handleFileChanges(project, changes) {
         //this loop assumes paths are both file paths and folder paths, which eliminates the need to detect.
         //All functions below can handle being given a file path AND a folder path, and will only operate on the one they are looking for
-        let consumeCount = 0;
         await Promise.all(changes.map(async (change) => {
             await this.keyedThrottler.run(change.srcPath, async () => {
-                consumeCount += await this.handleFileChange(project, change) ? 1 : 0;
+                if (await this.handleFileChange(project, change)) {
+                    await this.validateAllThrottled();
+                }
             });
         }));
-        if (consumeCount > 0) {
-            await this.validateAllThrottled();
-        }
     }
     /**
      * This only operates on files that match the specified files globs, so it is safe to throw
      * any file changes you receive with no unexpected side-effects
-     * @param changes
+     * @returns true if the file was handled by this project, false if it was not
      */
     async handleFileChange(project, change) {
         const { program, options, rootDir } = project.builder;
         //deleted
         if (change.type === node_1.FileChangeType.Deleted) {
             //try to act on this path as a directory
-            project.builder.program.removeFilesInFolder(change.srcPath);
+            project.builder.removeFilesInFolder(change.srcPath);
             //if this is a file loaded in the program, remove it
             if (program.hasFile(change.srcPath)) {
                 program.removeFile(change.srcPath);
@@ -787,14 +846,36 @@ class LanguageServer {
         }
     }
     async onHover(params) {
+        var _a;
         //ensure programs are initialized
         await this.waitAllProjectFirstRuns();
         const srcPath = util_1.util.uriToPath(params.textDocument.uri);
         let projects = this.getProjects();
-        let hovers = await Promise.all(Array.prototype.concat.call([], projects.map(async (x) => x.builder.program.getHover(srcPath, params.position))));
-        //return the first non-falsey hover. TODO is there a way to handle multiple hover results?
-        let hover = hovers.filter((x) => !!x)[0];
-        return hover;
+        let hovers = projects
+            //get hovers from all projects
+            .map((x) => x.builder.program.getHover(srcPath, params.position))
+            //flatten to a single list
+            .flat();
+        const contents = [
+            ...(hovers !== null && hovers !== void 0 ? hovers : [])
+                //pull all hover contents out into a flag array of strings
+                .map(x => {
+                return Array.isArray(x === null || x === void 0 ? void 0 : x.contents) ? x === null || x === void 0 ? void 0 : x.contents : [x === null || x === void 0 ? void 0 : x.contents];
+            }).flat()
+                //remove nulls
+                .filter(x => !!x)
+                //dedupe hovers across all projects
+                .reduce((set, content) => set.add(content), new Set()).values()
+        ];
+        if (contents.length > 0) {
+            let hover = {
+                //use the range from the first hover
+                range: (_a = hovers[0]) === null || _a === void 0 ? void 0 : _a.range,
+                //the contents of all hovers
+                contents: contents
+            };
+            return hover;
+        }
     }
     async onDocumentClose(event) {
         const { document } = event;
@@ -817,7 +898,6 @@ class LanguageServer {
             //throttle file processing. first call is run immediately, and then the last call is processed.
             await this.keyedThrottler.run(filePath, () => {
                 var _a;
-                this.connection.sendNotification('build-status', 'building');
                 let documentText = document.getText();
                 for (const project of this.getProjects()) {
                     //only add or replace existing files. All of the files in the project should
@@ -836,7 +916,7 @@ class LanguageServer {
             await this.validateAllThrottled();
         }
         catch (e) {
-            this.sendCriticalFailure(`Critical error parsing / validating ${filePath}: ${e.message}`);
+            this.sendCriticalFailure(`Critical error parsing/validating ${filePath}: ${e.message}`);
         }
     }
     async validateAll() {
@@ -846,13 +926,15 @@ class LanguageServer {
             await this.synchronizeStandaloneProjects();
             let projects = this.getProjects();
             //validate all programs
-            await Promise.all(projects.map((x) => x.builder.program.validate()));
+            await Promise.all(projects.map((project) => {
+                project.builder.program.validate();
+                return project;
+            }));
         }
         catch (e) {
             this.connection.console.error(e);
             this.sendCriticalFailure(`Critical error validating project: ${e.message}${(_a = e.stack) !== null && _a !== void 0 ? _a : ''}`);
         }
-        this.connection.sendNotification('build-status', 'success');
     }
     async onWorkspaceSymbol(params) {
         await this.waitAllProjectFirstRuns();
@@ -894,7 +976,7 @@ class LanguageServer {
         try {
             const signatures = util_1.util.flatMap(await Promise.all(this.getProjects().map(project => project.builder.program.getSignatureHelp(filepath, params.position))), c => c);
             const activeSignature = signatures.length > 0 ? 0 : null;
-            const activeParameter = activeSignature >= 0 ? (_a = signatures[activeSignature]) === null || _a === void 0 ? void 0 : _a.index : null;
+            const activeParameter = activeSignature !== null ? (_a = signatures[activeSignature]) === null || _a === void 0 ? void 0 : _a.index : null;
             let results = {
                 signatures: signatures.map((s) => s.signature),
                 activeSignature: activeSignature,
@@ -917,20 +999,35 @@ class LanguageServer {
         const srcPath = util_1.util.uriToPath(params.textDocument.uri);
         const results = util_1.util.flatMap(await Promise.all(this.getProjects().map(project => {
             return project.builder.program.getReferences(srcPath, position);
-        })), c => c);
+        })), c => c !== null && c !== void 0 ? c : []);
         return results.filter((r) => r);
     }
+    onValidateSettled() {
+        return Promise.all([
+            //wait for the validator to start running (or timeout if it never did)
+            this.validateThrottler.onRunOnce(100),
+            //wait for the validator to stop running (or resolve immediately if it's already idle)
+            this.validateThrottler.onIdleOnce(true)
+        ]);
+    }
     async onFullSemanticTokens(params) {
         await this.waitAllProjectFirstRuns();
+        //wait for the file to settle (in case there are multiple file changes in quick succession)
         await this.keyedThrottler.onIdleOnce(util_1.util.uriToPath(params.textDocument.uri), true);
+        // make sure validation is complete
+        await this.validateAllThrottled();
+        //wait for the validation cycle to settle
+        await this.onValidateSettled();
         const srcPath = util_1.util.uriToPath(params.textDocument.uri);
         for (const project of this.projects) {
             //find the first program that has this file, since it would be incredibly inefficient to generate semantic tokens for the same file multiple times.
             if (project.builder.program.hasFile(srcPath)) {
                 let semanticTokens = project.builder.program.getSemanticTokens(srcPath);
-                return {
-                    data: (0, SemanticTokenUtils_1.encodeSemanticTokens)(semanticTokens)
-                };
+                if (semanticTokens !== undefined) {
+                    return {
+                        data: (0, SemanticTokenUtils_1.encodeSemanticTokens)(semanticTokens)
+                    };
+                }
             }
         }
     }
@@ -941,9 +1038,10 @@ class LanguageServer {
             //Get only the changes to diagnostics since the last time we sent them to the client
             const patch = this.diagnosticCollection.getPatch(this.projects);
             for (let filePath in patch) {
-                const diagnostics = patch[filePath].map(d => util_1.util.toDiagnostic(d));
+                const uri = vscode_uri_1.URI.file(filePath).toString();
+                const diagnostics = patch[filePath].map(d => util_1.util.toDiagnostic(d, uri));
                 this.connection.sendDiagnostics({
-                    uri: vscode_uri_1.URI.file(filePath).toString(),
+                    uri: uri,
                     diagnostics: diagnostics
                 });
             }
@@ -958,9 +1056,6 @@ class LanguageServer {
             return result;
         }
     }
-    /**
-     * @param srcPath The absolute path to the source file on disk
-     */
     async transpileFile(srcPath) {
         //wait all program first runs
         await this.waitAllProjectFirstRuns();
@@ -981,22 +1076,35 @@ __decorate([
     AddStackToErrorMessage
 ], LanguageServer.prototype, "onInitialize", null);
 __decorate([
-    AddStackToErrorMessage
+    TrackBusyStatus
+], LanguageServer.prototype, "getProjectPaths", null);
+__decorate([
+    TrackBusyStatus
+], LanguageServer.prototype, "syncProjects", null);
+__decorate([
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onInitialized", null);
 __decorate([
-    AddStackToErrorMessage
+    TrackBusyStatus
+], LanguageServer.prototype, "createProject", null);
+__decorate([
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onCompletion", null);
 __decorate([
     AddStackToErrorMessage
 ], LanguageServer.prototype, "onCompletionResolve", null);
 __decorate([
-    AddStackToErrorMessage
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onCodeAction", null);
 __decorate([
     AddStackToErrorMessage
 ], LanguageServer.prototype, "onDidChangeConfiguration", null);
 __decorate([
-    AddStackToErrorMessage
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onDidChangeWatchedFiles", null);
 __decorate([
     AddStackToErrorMessage
@@ -1005,34 +1113,49 @@ __decorate([
     AddStackToErrorMessage
 ], LanguageServer.prototype, "onDocumentClose", null);
 __decorate([
-    AddStackToErrorMessage
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "validateTextDocument", null);
 __decorate([
-    AddStackToErrorMessage
+    TrackBusyStatus
+], LanguageServer.prototype, "validateAll", null);
+__decorate([
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onWorkspaceSymbol", null);
 __decorate([
-    AddStackToErrorMessage
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onDocumentSymbol", null);
 __decorate([
-    AddStackToErrorMessage
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onDefinition", null);
 __decorate([
-    AddStackToErrorMessage
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onSignatureHelp", null);
 __decorate([
-    AddStackToErrorMessage
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onReferences", null);
 __decorate([
-    AddStackToErrorMessage
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onFullSemanticTokens", null);
 __decorate([
-    AddStackToErrorMessage
+    AddStackToErrorMessage,
+    TrackBusyStatus
 ], LanguageServer.prototype, "onExecuteCommand", null);
 exports.LanguageServer = LanguageServer;
 var CustomCommands;
 (function (CustomCommands) {
     CustomCommands["TranspileFile"] = "TranspileFile";
 })(CustomCommands = exports.CustomCommands || (exports.CustomCommands = {}));
+var NotificationName;
+(function (NotificationName) {
+    NotificationName["busyStatus"] = "busyStatus";
+})(NotificationName = exports.NotificationName || (exports.NotificationName = {}));
 /**
  * Wraps a method. If there's an error (either sync or via a promise),
  * this appends the error's stack trace at the end of the error message so that the connection will
@@ -1064,4 +1187,16 @@ function AddStackToErrorMessage(target, propertyKey, descriptor) {
         }
     };
 }
+/**
+ * An annotation used to wrap the method in a busyStatus tracking call
+ */
+function TrackBusyStatus(target, propertyKey, descriptor) {
+    let originalMethod = descriptor.value;
+    //wrapping the original method
+    descriptor.value = function value(...args) {
+        return this.busyStatusTracker.run(() => {
+            return originalMethod.apply(this, args);
+        }, originalMethod.name);
+    };
+}
 //# sourceMappingURL=LanguageServer.js.map