import * as vscode from 'vscode'; import * as lc from 'vscode-languageclient'; import { Config } from './config'; import { createClient } from './client'; export class Ctx { readonly config: Config; // Because we have "reload server" action, various listeners **will** face a // situation where the client is not ready yet, and should be prepared to // deal with it. // // Ideally, this should be replaced with async getter though. client: lc.LanguageClient | null = null; private extCtx: vscode.ExtensionContext; private onDidRestartHooks: Array<(client: lc.LanguageClient) => void> = []; constructor(extCtx: vscode.ExtensionContext) { this.config = new Config(extCtx); this.extCtx = extCtx; } async restartServer() { const old = this.client; if (old) { await old.stop(); } this.client = null; const client = createClient(this.config); this.pushCleanup(client.start()); await client.onReady(); this.client = client; for (const hook of this.onDidRestartHooks) { hook(client); } } get activeRustEditor(): vscode.TextEditor | undefined { const editor = vscode.window.activeTextEditor; return editor && editor.document.languageId === 'rust' ? editor : undefined; } registerCommand(name: string, factory: (ctx: Ctx) => Cmd) { const fullName = `rust-analyzer.${name}`; const cmd = factory(this); const d = vscode.commands.registerCommand(fullName, cmd); this.pushCleanup(d); } get subscriptions(): Disposable[] { return this.extCtx.subscriptions; } pushCleanup(d: Disposable) { this.extCtx.subscriptions.push(d); } onDidRestart(hook: (client: lc.LanguageClient) => void) { this.onDidRestartHooks.push(hook); } } export interface Disposable { dispose(): void; } export type Cmd = (...args: any[]) => unknown; export async function sendRequestWithRetry<R>( client: lc.LanguageClient, method: string, param: unknown, token?: vscode.CancellationToken, ): Promise<R> { for (const delay of [2, 4, 6, 8, 10, null]) { try { return await (token ? client.sendRequest(method, param, token) : client.sendRequest(method, param)); } catch (e) { if ( e.code === lc.ErrorCodes.ContentModified && delay !== null ) { await sleep(10 * (1 << delay)); continue; } throw e; } } throw 'unreachable'; } const sleep = (ms: number) => new Promise(resolve => setTimeout(resolve, ms));