rust/crates/rust-analyzer/src/diagnostics.rs

87 lines
2.6 KiB
Rust
Raw Normal View History

//! Book keeping for keeping diagnostics easily in sync with the client.
pub(crate) mod to_proto;
2020-02-18 05:25:26 -06:00
2020-06-25 15:45:35 -05:00
use std::{collections::HashMap, mem, sync::Arc};
2020-02-18 05:25:26 -06:00
2020-05-17 17:11:40 -05:00
use lsp_types::{Diagnostic, Range};
use ra_ide::FileId;
2020-06-25 15:45:35 -05:00
use rustc_hash::FxHashSet;
2020-05-17 17:11:40 -05:00
use crate::lsp_ext;
2020-06-25 15:45:35 -05:00
pub(crate) type CheckFixes = Arc<HashMap<FileId, Vec<Fix>>>;
2020-06-16 15:26:33 -05:00
#[derive(Debug, Default, Clone)]
pub struct DiagnosticsConfig {
pub warnings_as_info: Vec<String>,
pub warnings_as_hint: Vec<String>,
}
#[derive(Debug, Default, Clone)]
2020-06-25 15:45:35 -05:00
pub(crate) struct DiagnosticCollection {
pub(crate) native: HashMap<FileId, Vec<Diagnostic>>,
pub(crate) check: HashMap<FileId, Vec<Diagnostic>>,
pub(crate) check_fixes: CheckFixes,
changes: FxHashSet<FileId>,
}
#[derive(Debug, Clone)]
2020-06-25 15:45:35 -05:00
pub(crate) struct Fix {
pub(crate) range: Range,
pub(crate) action: lsp_ext::CodeAction,
}
impl DiagnosticCollection {
2020-06-25 15:45:35 -05:00
pub(crate) fn clear_check(&mut self) {
Arc::make_mut(&mut self.check_fixes).clear();
2020-06-25 15:45:35 -05:00
self.changes.extend(self.check.drain().map(|(key, _value)| key))
}
2020-06-25 15:45:35 -05:00
pub(crate) fn add_check_diagnostic(
&mut self,
file_id: FileId,
diagnostic: Diagnostic,
2020-05-17 17:11:40 -05:00
fixes: Vec<lsp_ext::CodeAction>,
) {
let diagnostics = self.check.entry(file_id).or_default();
for existing_diagnostic in diagnostics.iter() {
if are_diagnostics_equal(&existing_diagnostic, &diagnostic) {
return;
}
}
let check_fixes = Arc::make_mut(&mut self.check_fixes);
check_fixes
.entry(file_id)
.or_default()
.extend(fixes.into_iter().map(|action| Fix { range: diagnostic.range, action }));
diagnostics.push(diagnostic);
2020-06-25 15:45:35 -05:00
self.changes.insert(file_id);
}
2020-06-25 15:45:35 -05:00
pub(crate) fn set_native_diagnostics(&mut self, file_id: FileId, diagnostics: Vec<Diagnostic>) {
self.native.insert(file_id, diagnostics);
2020-06-25 15:45:35 -05:00
self.changes.insert(file_id);
}
2020-06-25 15:45:35 -05:00
pub(crate) fn diagnostics_for(&self, file_id: FileId) -> impl Iterator<Item = &Diagnostic> {
let native = self.native.get(&file_id).into_iter().flatten();
let check = self.check.get(&file_id).into_iter().flatten();
native.chain(check)
}
2020-06-25 15:45:35 -05:00
pub(crate) fn take_changes(&mut self) -> Option<FxHashSet<FileId>> {
if self.changes.is_empty() {
return None;
}
2020-06-25 15:45:35 -05:00
Some(mem::take(&mut self.changes))
}
}
fn are_diagnostics_equal(left: &Diagnostic, right: &Diagnostic) -> bool {
left.source == right.source
&& left.severity == right.severity
&& left.range == right.range
&& left.message == right.message
}