language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-09-29 13:25:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

vscode: minor refactorings

Browse source
This commit is contained in:
Veetaha 2020-02-17 01:47:14 +02:00
parent 31ae646448
commit bd113623a0
3 changed files with 28 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

45
editors/code/src/inlay_hints.ts
View file

@ -38,7 +38,7 @@ interface InlayHintsParams {
interface InlayHint {
range: vscode.Range;
kind: string;
kind: "TypeHint" | "ParameterHint";
label: string;
}
@ -55,7 +55,7 @@ const parameterHintDecorationType = vscode.window.createTextEditorDecorationType
});
class HintsUpdater {
private pending: Map<string, vscode.CancellationTokenSource> = new Map();
private pending = new Map<string, vscode.CancellationTokenSource>();
private ctx: Ctx;
private enabled: boolean;
@ -64,30 +64,36 @@ class HintsUpdater {
this.enabled = ctx.config.displayInlayHints;
}
async setEnabled(enabled: boolean) {
async setEnabled(enabled: boolean): Promise<void> {
if (this.enabled == enabled) return;
this.enabled = enabled;
if (this.enabled) {
await this.refresh();
} else {
this.allEditors.forEach(it => {
this.setTypeDecorations(it, []);
this.setParameterDecorations(it, []);
});
return await this.refresh();
}
this.allEditors.forEach(it => {
this.setTypeDecorations(it, []);
this.setParameterDecorations(it, []);
});
}
async refresh() {
if (!this.enabled) return;
const promises = this.allEditors.map(it => this.refreshEditor(it));
await Promise.all(promises);
await Promise.all(this.allEditors.map(it => this.refreshEditor(it)));
}
private get allEditors(): vscode.TextEditor[] {
return vscode.window.visibleTextEditors.filter(
editor => editor.document.languageId === 'rust',
);
}
private async refreshEditor(editor: vscode.TextEditor): Promise<void> {
const newHints = await this.queryHints(editor.document.uri.toString());
if (newHints == null) return;
const newTypeDecorations = newHints.filter(hint => hint.kind === 'TypeHint')
const newTypeDecorations = newHints
.filter(hint => hint.kind === 'TypeHint')
.map(hint => ({
range: hint.range,
renderOptions: {
@ -98,7 +104,8 @@ class HintsUpdater {
}));
this.setTypeDecorations(editor, newTypeDecorations);
const newParameterDecorations = newHints.filter(hint => hint.kind === 'ParameterHint')
const newParameterDecorations = newHints
.filter(hint => hint.kind === 'ParameterHint')
.map(hint => ({
range: hint.range,
renderOptions: {
@ -110,12 +117,6 @@ class HintsUpdater {
this.setParameterDecorations(editor, newParameterDecorations);
}
private get allEditors(): vscode.TextEditor[] {
return vscode.window.visibleTextEditors.filter(
editor => editor.document.languageId === 'rust',
);
}
private setTypeDecorations(
editor: vscode.TextEditor,
decorations: vscode.DecorationOptions[],
@ -139,12 +140,14 @@ class HintsUpdater {
private async queryHints(documentUri: string): Promise<InlayHint[] | null> {
const client = this.ctx.client;
if (!client) return null;
const request: InlayHintsParams = {
textDocument: { uri: documentUri },
};
const tokenSource = new vscode.CancellationTokenSource();
const prev = this.pending.get(documentUri);
if (prev) prev.cancel();
const prevHintsRequest = this.pending.get(documentUri);
prevHintsRequest?.cancel();
this.pending.set(documentUri, tokenSource);
try {
return await sendRequestWithRetry<InlayHint[] | null>(