114 lines
2.8 KiB
TypeScript
114 lines
2.8 KiB
TypeScript
import * as vscode from 'vscode';
|
|
|
|
import { Ctx } from './ctx';
|
|
|
|
const spinnerFrames = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
|
|
|
|
export function activateStatusDisplay(ctx: Ctx) {
|
|
const statusDisplay = new StatusDisplay(ctx.config.cargoWatchOptions.command);
|
|
ctx.pushCleanup(statusDisplay);
|
|
ctx.onNotification('$/progress', params => statusDisplay.handleProgressNotification(params));
|
|
}
|
|
|
|
class StatusDisplay implements vscode.Disposable {
|
|
packageName?: string;
|
|
|
|
private i = 0;
|
|
private statusBarItem: vscode.StatusBarItem;
|
|
private command: string;
|
|
private timer?: NodeJS.Timeout;
|
|
|
|
constructor(command: string) {
|
|
this.statusBarItem = vscode.window.createStatusBarItem(
|
|
vscode.StatusBarAlignment.Left,
|
|
10,
|
|
);
|
|
this.command = command;
|
|
this.statusBarItem.hide();
|
|
}
|
|
|
|
show() {
|
|
this.packageName = undefined;
|
|
|
|
this.timer =
|
|
this.timer ||
|
|
setInterval(() => {
|
|
if (this.packageName) {
|
|
this.statusBarItem!.text = `cargo ${this.command} [${
|
|
this.packageName
|
|
}] ${this.frame()}`;
|
|
} else {
|
|
this.statusBarItem!.text = `cargo ${
|
|
this.command
|
|
} ${this.frame()}`;
|
|
}
|
|
}, 300);
|
|
|
|
this.statusBarItem.show();
|
|
}
|
|
|
|
hide() {
|
|
if (this.timer) {
|
|
clearInterval(this.timer);
|
|
this.timer = undefined;
|
|
}
|
|
|
|
this.statusBarItem.hide();
|
|
}
|
|
|
|
dispose() {
|
|
if (this.timer) {
|
|
clearInterval(this.timer);
|
|
this.timer = undefined;
|
|
}
|
|
|
|
this.statusBarItem.dispose();
|
|
}
|
|
|
|
handleProgressNotification(params: ProgressParams) {
|
|
const { token, value } = params;
|
|
if (token !== 'rustAnalyzer/cargoWatcher') {
|
|
return;
|
|
}
|
|
|
|
switch (value.kind) {
|
|
case 'begin':
|
|
this.show();
|
|
break;
|
|
|
|
case 'report':
|
|
if (value.message) {
|
|
this.packageName = value.message;
|
|
}
|
|
break;
|
|
|
|
case 'end':
|
|
this.hide();
|
|
break;
|
|
}
|
|
}
|
|
|
|
private frame() {
|
|
return spinnerFrames[(this.i = ++this.i % spinnerFrames.length)];
|
|
}
|
|
}
|
|
|
|
// FIXME: Replace this once vscode-languageclient is updated to LSP 3.15
|
|
interface ProgressParams {
|
|
token: string;
|
|
value: WorkDoneProgress;
|
|
}
|
|
|
|
enum WorkDoneProgressKind {
|
|
Begin = 'begin',
|
|
Report = 'report',
|
|
End = 'end',
|
|
}
|
|
|
|
interface WorkDoneProgress {
|
|
kind: WorkDoneProgressKind;
|
|
message?: string;
|
|
cancelable?: boolean;
|
|
percentage?: string;
|
|
}
|