1 files changed, 30 insertions, 0 deletions

diff --git a/editors/code/src/ctx.ts b/editors/code/src/ctx.ts
new file mode 100644
index 000000000..8581667b4
--- /dev/null
+++ b/editors/code/src/ctx.ts
@@ -0,0 +1,30 @@
+import * as vscode from 'vscode';
+import * as lc from 'vscode-languageclient';
+import { Server } from './server';
+
+
+export class Ctx {
+    private extCtx: vscode.ExtensionContext
+
+    constructor(extCtx: vscode.ExtensionContext) {
+        this.extCtx = extCtx
+    }
+
+    get client(): lc.LanguageClient {
+        return Server.client
+    }
+
+    registerCommand(
+        name: string,
+        factory: (ctx: Ctx) => () => Promise<vscode.TextEditor>,
+    ) {
+        const fullName = `rust-analyzer.${name}`
+        const cmd = factory(this);
+        const d = vscode.commands.registerCommand(fullName, cmd);
+        this.pushCleanup(d);
+    }
+
+    pushCleanup(d: { dispose(): any }) {
+        this.extCtx.subscriptions.push(d)
+    }
+}