From 9fcad829807a0306fbf4eb2ebc1603a11a6df182 Mon Sep 17 00:00:00 2001 From: James Leitch Date: Tue, 20 Apr 2021 20:03:35 -0700 Subject: Diagnostic Remap Path Prefixes added. --- crates/rust-analyzer/src/diagnostics/to_proto.rs | 67 +++++++++++------------- 1 file changed, 32 insertions(+), 35 deletions(-) (limited to 'crates/rust-analyzer/src/diagnostics') diff --git a/crates/rust-analyzer/src/diagnostics/to_proto.rs b/crates/rust-analyzer/src/diagnostics/to_proto.rs index 8b01a7e5d..08303a781 100644 --- a/crates/rust-analyzer/src/diagnostics/to_proto.rs +++ b/crates/rust-analyzer/src/diagnostics/to_proto.rs @@ -44,8 +44,12 @@ fn is_dummy_macro_file(file_name: &str) -> bool { } /// Converts a Rust span to a LSP location -fn location(workspace_root: &Path, span: &DiagnosticSpan) -> lsp_types::Location { - let file_name = resolve_path(workspace_root, &span.file_name); +fn location( + config: &DiagnosticsMapConfig, + workspace_root: &Path, + span: &DiagnosticSpan, +) -> lsp_types::Location { + let file_name = resolve_path(config, workspace_root, &span.file_name); let uri = url_from_abs_path(&file_name); // FIXME: this doesn't handle UTF16 offsets correctly @@ -61,54 +65,46 @@ fn location(workspace_root: &Path, span: &DiagnosticSpan) -> lsp_types::Location /// /// This takes locations pointing into the standard library, or generally outside the current /// workspace into account and tries to avoid those, in case macros are involved. -fn primary_location(workspace_root: &Path, span: &DiagnosticSpan) -> lsp_types::Location { +fn primary_location( + config: &DiagnosticsMapConfig, + workspace_root: &Path, + span: &DiagnosticSpan, +) -> lsp_types::Location { let span_stack = std::iter::successors(Some(span), |span| Some(&span.expansion.as_ref()?.span)); for span in span_stack.clone() { - let abs_path = resolve_path(workspace_root, &span.file_name); + let abs_path = resolve_path(config, workspace_root, &span.file_name); if !is_dummy_macro_file(&span.file_name) && abs_path.starts_with(workspace_root) { - return location(workspace_root, span); + return location(config, workspace_root, span); } } // Fall back to the outermost macro invocation if no suitable span comes up. let last_span = span_stack.last().unwrap(); - location(workspace_root, last_span) + location(config, workspace_root, last_span) } /// Converts a secondary Rust span to a LSP related information /// /// If the span is unlabelled this will return `None`. fn diagnostic_related_information( + config: &DiagnosticsMapConfig, workspace_root: &Path, span: &DiagnosticSpan, ) -> Option { let message = span.label.clone()?; - let location = location(workspace_root, span); + let location = location(config, workspace_root, span); Some(lsp_types::DiagnosticRelatedInformation { location, message }) } -/// Resolves paths mimicking VSCode's behavior when `file_name` starts -/// with the root directory component, which does not discard the base -/// path. If this relative path exists, use it, otherwise fall back -/// to the existing Rust behavior of path joining. -fn resolve_path(workspace_root: &Path, file_name: &str) -> PathBuf { - let file_name = Path::new(file_name); - - // Test path with VSCode's path join behavior. - let vscode_path = { - let mut result = PathBuf::from(workspace_root); - result.extend(file_name.components().skip_while(|component| match component { - std::path::Component::RootDir => true, - _ => false, - })); - result - }; - if vscode_path.exists() { - return vscode_path; +/// Resolves paths applying any matching path prefix remappings, and then +/// joining the path to the workspace root. +fn resolve_path(config: &DiagnosticsMapConfig, workspace_root: &Path, file_name: &str) -> PathBuf { + match config.remap_path_prefixes.iter().find(|(from, _)| file_name.starts_with(*from)) { + Some((from, to)) => { + workspace_root.join(format!("{}{}", to, file_name.strip_prefix(from).unwrap())) + } + None => workspace_root.join(file_name), } - - // Default to Rust's path join behavior. - workspace_root.join(file_name) } struct SubDiagnostic { @@ -122,6 +118,7 @@ enum MappedRustChildDiagnostic { } fn map_rust_child_diagnostic( + config: &DiagnosticsMapConfig, workspace_root: &Path, rd: &flycheck::Diagnostic, ) -> MappedRustChildDiagnostic { @@ -135,7 +132,7 @@ fn map_rust_child_diagnostic( let mut edit_map: HashMap> = HashMap::new(); for &span in &spans { if let Some(suggested_replacement) = &span.suggested_replacement { - let location = location(workspace_root, span); + let location = location(config, workspace_root, span); let edit = lsp_types::TextEdit::new(location.range, suggested_replacement.clone()); edit_map.entry(location.uri).or_default().push(edit); } @@ -144,7 +141,7 @@ fn map_rust_child_diagnostic( if edit_map.is_empty() { MappedRustChildDiagnostic::SubDiagnostic(SubDiagnostic { related: lsp_types::DiagnosticRelatedInformation { - location: location(workspace_root, spans[0]), + location: location(config, workspace_root, spans[0]), message: rd.message.clone(), }, suggested_fix: None, @@ -152,7 +149,7 @@ fn map_rust_child_diagnostic( } else { MappedRustChildDiagnostic::SubDiagnostic(SubDiagnostic { related: lsp_types::DiagnosticRelatedInformation { - location: location(workspace_root, spans[0]), + location: location(config, workspace_root, spans[0]), message: rd.message.clone(), }, suggested_fix: Some(lsp_ext::CodeAction { @@ -217,7 +214,7 @@ pub(crate) fn map_rust_diagnostic_to_lsp( let mut tags = Vec::new(); for secondary_span in rd.spans.iter().filter(|s| !s.is_primary) { - let related = diagnostic_related_information(workspace_root, secondary_span); + let related = diagnostic_related_information(config, workspace_root, secondary_span); if let Some(related) = related { subdiagnostics.push(SubDiagnostic { related, suggested_fix: None }); } @@ -225,7 +222,7 @@ pub(crate) fn map_rust_diagnostic_to_lsp( let mut message = rd.message.clone(); for child in &rd.children { - let child = map_rust_child_diagnostic(workspace_root, &child); + let child = map_rust_child_diagnostic(config, workspace_root, &child); match child { MappedRustChildDiagnostic::SubDiagnostic(sub) => { subdiagnostics.push(sub); @@ -269,7 +266,7 @@ pub(crate) fn map_rust_diagnostic_to_lsp( primary_spans .iter() .flat_map(|primary_span| { - let primary_location = primary_location(workspace_root, &primary_span); + let primary_location = primary_location(config, workspace_root, &primary_span); let mut message = message.clone(); if needs_primary_span_label { @@ -299,7 +296,7 @@ pub(crate) fn map_rust_diagnostic_to_lsp( // generated that code. let is_in_macro_call = i != 0; - let secondary_location = location(workspace_root, &span); + let secondary_location = location(config, workspace_root, &span); if secondary_location == primary_location { continue; } -- cgit v1.2.3