From 6d6cb25cf46a2327d6cb2278385763abfa7a95a0 Mon Sep 17 00:00:00 2001 From: Ryan Cumming Date: Tue, 25 Jun 2019 07:40:06 +1000 Subject: Rich mapping of cargo watch output Currently we depend on the ASCII rendering string that `rustc` provides to populate Visual Studio Code's diagnostic. This has a number of shortcomings: 1. It's not a very good use of space in the error list 2. We can't jump to secondary spans (e.g. where a called function is defined) 3. We can't use Code Actions aka Quick Fix This moves all of the low-level parsing and mapping to a `rust_diagnostics.ts`. This uses some heuristics to map Rust diagnostics to VsCode: 1. As before, the Rust diagnostic message and primary span is used for the root diagnostic. However, we now just use the message instead of the rendered version. 2. Every secondary span is converted to "related information". This shows as child in the error list and can be jumped to. 3. Every child diagnostic is categorised in to three buckets: 1. If they have no span they're treated as another line of the root messages 2. If they have replacement text they're treated as a Code Action 3. If they have a span but no replacement text they're treated as related information (same as secondary spans). --- editors/code/src/commands/cargo_watch.ts | 186 ++++++++++++++++++++++--------- 1 file changed, 132 insertions(+), 54 deletions(-) (limited to 'editors/code/src/commands') diff --git a/editors/code/src/commands/cargo_watch.ts b/editors/code/src/commands/cargo_watch.ts index 13adf4c10..a662f7cc8 100644 --- a/editors/code/src/commands/cargo_watch.ts +++ b/editors/code/src/commands/cargo_watch.ts @@ -4,6 +4,10 @@ import * as path from 'path'; import * as vscode from 'vscode'; import { Server } from '../server'; import { terminate } from '../utils/processes'; +import { + mapRustDiagnosticToVsCode, + RustDiagnostic +} from '../utils/rust_diagnostics'; import { LineBuffer } from './line_buffer'; import { StatusDisplay } from './watch_status'; @@ -33,10 +37,17 @@ export function registerCargoWatchProvider( return provider; } -export class CargoWatchProvider implements vscode.Disposable { +export class CargoWatchProvider + implements vscode.Disposable, vscode.CodeActionProvider { private readonly diagnosticCollection: vscode.DiagnosticCollection; private readonly statusDisplay: StatusDisplay; private readonly outputChannel: vscode.OutputChannel; + + private codeActions: { + [fileUri: string]: vscode.CodeAction[]; + }; + private readonly codeActionDispose: vscode.Disposable; + private cargoProcess?: child_process.ChildProcess; constructor() { @@ -49,6 +60,16 @@ export class CargoWatchProvider implements vscode.Disposable { this.outputChannel = vscode.window.createOutputChannel( 'Cargo Watch Trace' ); + + // Register code actions for rustc's suggested fixes + this.codeActions = {}; + this.codeActionDispose = vscode.languages.registerCodeActionsProvider( + [{ scheme: 'file', language: 'rust' }], + this, + { + providedCodeActionKinds: [vscode.CodeActionKind.QuickFix] + } + ); } public start() { @@ -127,6 +148,14 @@ export class CargoWatchProvider implements vscode.Disposable { this.diagnosticCollection.dispose(); this.outputChannel.dispose(); this.statusDisplay.dispose(); + this.codeActionDispose.dispose(); + } + + public provideCodeActions( + document: vscode.TextDocument + ): vscode.ProviderResult> { + const documentActions = this.codeActions[document.uri.toString()]; + return documentActions || []; } private logInfo(line: string) { @@ -147,6 +176,7 @@ export class CargoWatchProvider implements vscode.Disposable { private parseLine(line: string) { if (line.startsWith('[Running')) { this.diagnosticCollection.clear(); + this.codeActions = {}; this.statusDisplay.show(); } @@ -154,34 +184,65 @@ export class CargoWatchProvider implements vscode.Disposable { this.statusDisplay.hide(); } - function getLevel(s: string): vscode.DiagnosticSeverity { - if (s === 'error') { - return vscode.DiagnosticSeverity.Error; + function areDiagnosticsEqual( + left: vscode.Diagnostic, + right: vscode.Diagnostic + ): boolean { + return ( + left.source === right.source && + left.severity === right.severity && + left.range.isEqual(right.range) && + left.message === right.message + ); + } + + function areCodeActionsEqual( + left: vscode.CodeAction, + right: vscode.CodeAction + ): boolean { + if ( + left.kind !== right.kind || + left.title !== right.title || + !left.edit || + !right.edit + ) { + return false; } - if (s.startsWith('warn')) { - return vscode.DiagnosticSeverity.Warning; + + const leftEditEntries = left.edit.entries(); + const rightEditEntries = right.edit.entries(); + + if (leftEditEntries.length !== leftEditEntries.length) { + return false; } - return vscode.DiagnosticSeverity.Information; - } - // Reference: - // https://github.com/rust-lang/rust/blob/master/src/libsyntax/json.rs - interface RustDiagnosticSpan { - line_start: number; - line_end: number; - column_start: number; - column_end: number; - is_primary: boolean; - file_name: string; - } + for (let i = 0; i < leftEditEntries.length; i++) { + const [leftUri, leftEdits] = leftEditEntries[i]; + const [rightUri, rightEdits] = rightEditEntries[i]; + + if (leftUri.toString() !== rightUri.toString()) { + return false; + } - interface RustDiagnostic { - spans: RustDiagnosticSpan[]; - rendered: string; - level: string; - code?: { - code: string; - }; + if (leftEdits.length !== rightEdits.length) { + return false; + } + + for (let j = 0; j < leftEdits.length; j++) { + const leftEdit = leftEdits[j]; + const rightEdit = rightEdits[j]; + + if (!leftEdit.range.isEqual(rightEdit.range)) { + return false; + } + + if (leftEdit.newText !== rightEdit.newText) { + return false; + } + } + } + + return true; } interface CargoArtifact { @@ -215,41 +276,58 @@ export class CargoWatchProvider implements vscode.Disposable { } else if (data.reason === 'compiler-message') { const msg = data.message as RustDiagnostic; - const spans = msg.spans.filter(o => o.is_primary); - - // We only handle primary span right now. - if (spans.length > 0) { - const o = spans[0]; + const mapResult = mapRustDiagnosticToVsCode(msg); + if (!mapResult) { + return; + } - const rendered = msg.rendered; - const level = getLevel(msg.level); - const range = new vscode.Range( - new vscode.Position(o.line_start - 1, o.column_start - 1), - new vscode.Position(o.line_end - 1, o.column_end - 1) - ); + const { location, diagnostic, codeActions } = mapResult; + const fileUri = location.uri; - const fileName = path.join( - vscode.workspace.rootPath!, - o.file_name - ); - const diagnostic = new vscode.Diagnostic( - range, - rendered, - level - ); + const diagnostics: vscode.Diagnostic[] = [ + ...(this.diagnosticCollection!.get(fileUri) || []) + ]; - diagnostic.source = 'rustc'; - diagnostic.code = msg.code ? msg.code.code : undefined; - diagnostic.relatedInformation = []; + // If we're building multiple targets it's possible we've already seen this diagnostic + const isDuplicate = diagnostics.some(d => + areDiagnosticsEqual(d, diagnostic) + ); - const fileUrl = vscode.Uri.file(fileName!); + if (isDuplicate) { + return; + } - const diagnostics: vscode.Diagnostic[] = [ - ...(this.diagnosticCollection!.get(fileUrl) || []) - ]; - diagnostics.push(diagnostic); + diagnostics.push(diagnostic); + this.diagnosticCollection!.set(fileUri, diagnostics); + + if (codeActions.length) { + const fileUriString = fileUri.toString(); + const existingActions = this.codeActions[fileUriString] || []; + + for (const newAction of codeActions) { + const existingAction = existingActions.find(existing => + areCodeActionsEqual(existing, newAction) + ); + + if (existingAction) { + if (!existingAction.diagnostics) { + existingAction.diagnostics = []; + } + // This action also applies to this diagnostic + existingAction.diagnostics.push(diagnostic); + } else { + newAction.diagnostics = [diagnostic]; + existingActions.push(newAction); + } + } - this.diagnosticCollection!.set(fileUrl, diagnostics); + // Have VsCode query us for the code actions + this.codeActions[fileUriString] = existingActions; + vscode.commands.executeCommand( + 'vscode.executeCodeActionProvider', + fileUri, + diagnostic.range + ); } } } -- cgit v1.2.3