Allows plugin to register commands

microsoft · May 27, 2021 · b411c08 · b411c08
1 parent 0e1df66
commit b411c08
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/server/project.ts b/src/server/project.ts
@@ -90,6 +90,7 @@ namespace ts.server {
     export interface PluginModule {
         create(createInfo: PluginCreateInfo): LanguageService;
         getExternalFiles?(proj: Project): string[];
+        getProtocolHandlers?(): [command: string, handler: (request: protocol.Request) => HandlerResponse][]
         onConfigurationChanged?(config: any): void;
     }
 
@@ -240,6 +241,15 @@ namespace ts.server {
             return result.module;
         }
 
+        public setupCustomProtocolHandlers (cb: (command: string, handler: (request: protocol.Request) => HandlerResponse) => void) {
+            this.plugins.forEach(plugin => {
+                const handlers = plugin.module.getProtocolHandlers?.();
+                handlers?.forEach(([name, handler]) => {
+                    cb(name, handler)
+                })
+            })
+        }
+
         /*@internal*/
         readonly currentDirectory: string;
 

diff --git a/src/server/session.ts b/src/server/session.ts
@@ -839,6 +839,7 @@ namespace ts.server {
                     break;
                 case ProjectLoadingFinishEvent:
                     const { project: finishProject } = event.data;
+                    finishProject.setupCustomProtocolHandlers(this.addProtocolHandler.bind(this));
                     this.event<protocol.ProjectLoadingFinishEventBody>({ projectName: finishProject.getProjectName() }, ProjectLoadingFinishEvent);
                     break;
                 case LargeFileReferencedEvent: