rust-analyzer/editors/code/src/ctx.ts

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)
    }
}