More second command to Ctx

4 files changed, 14 insertions, 11 deletions

diff --git a/editors/code/src/commands/analyzer_status.ts b/editors/code/src/commands/analyzer_status.ts
index 6e92c50ef..c9d32fe07 100644
--- a/editors/code/src/commands/analyzer_status.ts
+++ b/editors/code/src/commands/analyzer_status.ts
@@ -1,8 +1,8 @@
 import * as vscode from 'vscode';
-import { Ctx } from '../ctx';
+import { Ctx, Cmd } from '../ctx';
 // Shows status of rust-analyzer (for debugging)
 
-export function analyzerStatus(ctx: Ctx) {
+export function analyzerStatus(ctx: Ctx): Cmd {
     let poller: NodeJS.Timer | null = null;
     const tdcp = new TextDocumentContentProvider(ctx);
 
diff --git a/editors/code/src/commands/index.ts b/editors/code/src/commands/index.ts
index ec1995396..ed56f5a4e 100644
--- a/editors/code/src/commands/index.ts
+++ b/editors/code/src/commands/index.ts
@@ -1,3 +1,5 @@
+import { Ctx, Cmd } from '../ctx'
+
 import { analyzerStatus } from './analyzer_status';
 import * as applySourceChange from './apply_source_change';
 import * as expandMacro from './expand_macro';
@@ -9,6 +11,10 @@ import * as parentModule from './parent_module';
 import * as runnables from './runnables';
 import * as syntaxTree from './syntaxTree';
 
+function collectGarbage(ctx: Ctx): Cmd {
+    return async () => { ctx.client.sendRequest<null>('rust-analyzer/collectGarbage', null) }
+}
+
 export {
     analyzerStatus,
     applySourceChange,
@@ -20,4 +26,5 @@ export {
     syntaxTree,
     onEnter,
     inlayHints,
+    collectGarbage
 };
diff --git a/editors/code/src/ctx.ts b/editors/code/src/ctx.ts
index 8581667b4..9dd2b7d4f 100644
--- a/editors/code/src/ctx.ts
+++ b/editors/code/src/ctx.ts
@@ -16,7 +16,7 @@ export class Ctx {
 
     registerCommand(
         name: string,
-        factory: (ctx: Ctx) => () => Promise<vscode.TextEditor>,
+        factory: (ctx: Ctx) => Cmd,
     ) {
         const fullName = `rust-analyzer.${name}`
         const cmd = factory(this);
@@ -28,3 +28,5 @@ export class Ctx {
         this.extCtx.subscriptions.push(d)
     }
 }
+
+export type Cmd = (...args: any[]) => any;
diff --git a/editors/code/src/main.ts b/editors/code/src/main.ts
index 048b9bbd4..9500219ca 100644
--- a/editors/code/src/main.ts
+++ b/editors/code/src/main.ts
@@ -15,11 +15,8 @@ let ctx!: Ctx;
 
 export async function activate(context: vscode.ExtensionContext) {
     ctx = new Ctx(context);
-    ctx.registerCommand(
-        'analyzerStatus',
-        commands.analyzerStatus
-    );
-
+    ctx.registerCommand('analyzerStatus', commands.analyzerStatus);
+    ctx.registerCommand('collectGarbage', commands.collectGarbage);
 
     function disposeOnDeactivation(disposable: vscode.Disposable) {
         context.subscriptions.push(disposable);
@@ -58,9 +55,6 @@ export async function activate(context: vscode.ExtensionContext) {
     }
 
     // Commands are requests from vscode to the language server
-    registerCommand('rust-analyzer.collectGarbage', () =>
-        Server.client.sendRequest<null>('rust-analyzer/collectGarbage', null),
-    );
     registerCommand(
         'rust-analyzer.matchingBrace',
         commands.matchingBrace.handle,