diff --git a/crates/ra_ide_api/src/diagnostics.rs b/crates/ra_ide_api/src/diagnostics.rs index e1c5a1b3981..b9dc424c62c 100644 --- a/crates/ra_ide_api/src/diagnostics.rs +++ b/crates/ra_ide_api/src/diagnostics.rs @@ -1,6 +1,5 @@ use itertools::Itertools; use hir::{Problem, source_binder}; -use ra_ide_api_light::Severity; use ra_db::SourceDatabase; use ra_syntax::{ Location, SourceFile, SyntaxKind, TextRange, SyntaxNode, @@ -11,6 +10,12 @@ use ra_text_edit::{TextEdit, TextEditBuilder}; use crate::{Diagnostic, FileId, FileSystemEdit, SourceChange, SourceFileEdit, db::RootDatabase}; +#[derive(Debug, Copy, Clone)] +pub enum Severity { + Error, + WeakWarning, +} + pub(crate) fn diagnostics(db: &RootDatabase, file_id: FileId) -> Vec { let source_file = db.parse(file_id); let mut res = Vec::new(); diff --git a/crates/ra_ide_api/src/lib.rs b/crates/ra_ide_api/src/lib.rs index a838c30da75..99f18b6b8c3 100644 --- a/crates/ra_ide_api/src/lib.rs +++ b/crates/ra_ide_api/src/lib.rs @@ -37,6 +37,8 @@ mod line_index; mod folding_ranges; mod line_index_utils; mod join_lines; +mod typing; +mod matching_brace; #[cfg(test)] mod marks; @@ -69,10 +71,10 @@ pub use crate::{ line_index::{LineIndex, LineCol}, line_index_utils::translate_offset_with_edit, folding_ranges::{Fold, FoldKind}, + syntax_highlighting::HighlightedRange, + diagnostics::Severity, }; -pub use ra_ide_api_light::{ - HighlightedRange, Severity, StructureNode, LocalEdit, -}; +pub use ra_ide_api_light::StructureNode; pub use ra_db::{ Canceled, CrateGraph, CrateId, FileId, FilePosition, FileRange, SourceRootId, Edition @@ -266,7 +268,7 @@ impl Analysis { /// supported). pub fn matching_brace(&self, position: FilePosition) -> Option { let file = self.db.parse(position.file_id); - ra_ide_api_light::matching_brace(&file, position.offset) + matching_brace::matching_brace(&file, position.offset) } /// Returns a syntax tree represented as `String`, for debug purposes. @@ -294,9 +296,7 @@ impl Analysis { /// Returns an edit which should be applied when opening a new line, fixing /// up minor stuff like continuing the comment. pub fn on_enter(&self, position: FilePosition) -> Option { - let file = self.db.parse(position.file_id); - let edit = ra_ide_api_light::on_enter(&file, position.offset)?; - Some(SourceChange::from_local_edit(position.file_id, edit)) + typing::on_enter(&self.db, position) } /// Returns an edit which should be applied after `=` was typed. Primarily, @@ -304,15 +304,18 @@ impl Analysis { // FIXME: use a snippet completion instead of this hack here. pub fn on_eq_typed(&self, position: FilePosition) -> Option { let file = self.db.parse(position.file_id); - let edit = ra_ide_api_light::on_eq_typed(&file, position.offset)?; - Some(SourceChange::from_local_edit(position.file_id, edit)) + let edit = typing::on_eq_typed(&file, position.offset)?; + Some(SourceChange { + label: "add semicolon".to_string(), + source_file_edits: vec![SourceFileEdit { edit, file_id: position.file_id }], + file_system_edits: vec![], + cursor_position: None, + }) } /// Returns an edit which should be applied when a dot ('.') is typed on a blank line, indenting the line appropriately. pub fn on_dot_typed(&self, position: FilePosition) -> Option { - let file = self.db.parse(position.file_id); - let edit = ra_ide_api_light::on_dot_typed(&file, position.offset)?; - Some(SourceChange::from_local_edit(position.file_id, edit)) + typing::on_dot_typed(&self.db, position) } /// Returns a tree representation of symbols in the file. Useful to draw a @@ -434,18 +437,6 @@ impl Analysis { } } -impl SourceChange { - pub(crate) fn from_local_edit(file_id: FileId, edit: LocalEdit) -> SourceChange { - let file_edit = SourceFileEdit { file_id, edit: edit.edit }; - SourceChange { - label: edit.label, - source_file_edits: vec![file_edit], - file_system_edits: vec![], - cursor_position: edit.cursor_position.map(|offset| FilePosition { offset, file_id }), - } - } -} - #[test] fn analysis_is_send() { fn is_send() {} diff --git a/crates/ra_ide_api/src/matching_brace.rs b/crates/ra_ide_api/src/matching_brace.rs new file mode 100644 index 00000000000..d1405f14fcd --- /dev/null +++ b/crates/ra_ide_api/src/matching_brace.rs @@ -0,0 +1,45 @@ +use ra_syntax::{ + SourceFile, TextUnit, + algo::find_leaf_at_offset, + SyntaxKind::{self, *}, + ast::AstNode, +}; + +pub fn matching_brace(file: &SourceFile, offset: TextUnit) -> Option { + const BRACES: &[SyntaxKind] = + &[L_CURLY, R_CURLY, L_BRACK, R_BRACK, L_PAREN, R_PAREN, L_ANGLE, R_ANGLE]; + let (brace_node, brace_idx) = find_leaf_at_offset(file.syntax(), offset) + .filter_map(|node| { + let idx = BRACES.iter().position(|&brace| brace == node.kind())?; + Some((node, idx)) + }) + .next()?; + let parent = brace_node.parent()?; + let matching_kind = BRACES[brace_idx ^ 1]; + let matching_node = parent.children().find(|node| node.kind() == matching_kind)?; + Some(matching_node.range().start()) +} + +#[cfg(test)] +mod tests { + use test_utils::{add_cursor, assert_eq_text, extract_offset}; + + use super::*; + + #[test] + fn test_matching_brace() { + fn do_check(before: &str, after: &str) { + let (pos, before) = extract_offset(before); + let file = SourceFile::parse(&before); + let new_pos = match matching_brace(&file, pos) { + None => pos, + Some(pos) => pos, + }; + let actual = add_cursor(&before, new_pos); + assert_eq_text!(after, &actual); + } + + do_check("struct Foo { a: i32, }<|>", "struct Foo <|>{ a: i32, }"); + } + +} diff --git a/crates/ra_ide_api/src/snapshots/tests__highlighting.snap b/crates/ra_ide_api/src/snapshots/tests__highlighting.snap new file mode 100644 index 00000000000..72029e0ed27 --- /dev/null +++ b/crates/ra_ide_api/src/snapshots/tests__highlighting.snap @@ -0,0 +1,34 @@ +--- +created: "2019-03-23T16:20:31.394314144Z" +creator: insta@0.7.1 +source: crates/ra_ide_api/src/syntax_highlighting.rs +expression: result +--- +Ok( + [ + HighlightedRange { + range: [1; 11), + tag: "comment" + }, + HighlightedRange { + range: [12; 14), + tag: "keyword" + }, + HighlightedRange { + range: [15; 19), + tag: "function" + }, + HighlightedRange { + range: [29; 37), + tag: "macro" + }, + HighlightedRange { + range: [38; 50), + tag: "string" + }, + HighlightedRange { + range: [52; 54), + tag: "literal" + } + ] +) diff --git a/crates/ra_ide_api/src/syntax_highlighting.rs b/crates/ra_ide_api/src/syntax_highlighting.rs index fdd87bcfffc..a0c5e78ad79 100644 --- a/crates/ra_ide_api/src/syntax_highlighting.rs +++ b/crates/ra_ide_api/src/syntax_highlighting.rs @@ -1,12 +1,81 @@ -use ra_syntax::AstNode; +use rustc_hash::FxHashSet; + +use ra_syntax::{ast, AstNode, TextRange, Direction, SyntaxKind::*}; use ra_db::SourceDatabase; -use crate::{ - FileId, HighlightedRange, - db::RootDatabase, -}; +use crate::{FileId, db::RootDatabase}; + +#[derive(Debug)] +pub struct HighlightedRange { + pub range: TextRange, + pub tag: &'static str, +} pub(crate) fn highlight(db: &RootDatabase, file_id: FileId) -> Vec { let source_file = db.parse(file_id); - ra_ide_api_light::highlight(source_file.syntax()) + + // Visited nodes to handle highlighting priorities + let mut highlighted = FxHashSet::default(); + let mut res = Vec::new(); + for node in source_file.syntax().descendants() { + if highlighted.contains(&node) { + continue; + } + let tag = match node.kind() { + COMMENT => "comment", + STRING | RAW_STRING | RAW_BYTE_STRING | BYTE_STRING => "string", + ATTR => "attribute", + NAME_REF => "text", + NAME => "function", + INT_NUMBER | FLOAT_NUMBER | CHAR | BYTE => "literal", + LIFETIME => "parameter", + k if k.is_keyword() => "keyword", + _ => { + if let Some(macro_call) = ast::MacroCall::cast(node) { + if let Some(path) = macro_call.path() { + if let Some(segment) = path.segment() { + if let Some(name_ref) = segment.name_ref() { + highlighted.insert(name_ref.syntax()); + let range_start = name_ref.syntax().range().start(); + let mut range_end = name_ref.syntax().range().end(); + for sibling in path.syntax().siblings(Direction::Next) { + match sibling.kind() { + EXCL | IDENT => range_end = sibling.range().end(), + _ => (), + } + } + res.push(HighlightedRange { + range: TextRange::from_to(range_start, range_end), + tag: "macro", + }) + } + } + } + } + continue; + } + }; + res.push(HighlightedRange { range: node.range(), tag }) + } + res +} + +#[cfg(test)] +mod tests { + use insta::assert_debug_snapshot_matches; + + use crate::mock_analysis::single_file; + + #[test] + fn test_highlighting() { + let (analysis, file_id) = single_file( + r#" +// comment +fn main() {} + println!("Hello, {}!", 92); +"#, + ); + let result = analysis.highlight(file_id); + assert_debug_snapshot_matches!("highlighting", result); + } } diff --git a/crates/ra_ide_api_light/src/typing.rs b/crates/ra_ide_api/src/typing.rs similarity index 80% rename from crates/ra_ide_api_light/src/typing.rs rename to crates/ra_ide_api/src/typing.rs index c69270333c7..94b228466c5 100644 --- a/crates/ra_ide_api_light/src/typing.rs +++ b/crates/ra_ide_api/src/typing.rs @@ -5,30 +5,37 @@ use ra_syntax::{ ast::{self, AstToken}, }; use ra_fmt::leading_indent; -use crate::{LocalEdit, TextEditBuilder}; +use ra_text_edit::{TextEdit, TextEditBuilder}; +use ra_db::{FilePosition, SourceDatabase}; +use crate::{db::RootDatabase, SourceChange, SourceFileEdit}; -pub fn on_enter(file: &SourceFile, offset: TextUnit) -> Option { - let comment = - find_leaf_at_offset(file.syntax(), offset).left_biased().and_then(ast::Comment::cast)?; +pub(crate) fn on_enter(db: &RootDatabase, position: FilePosition) -> Option { + let file = db.parse(position.file_id); + let comment = find_leaf_at_offset(file.syntax(), position.offset) + .left_biased() + .and_then(ast::Comment::cast)?; if let ast::CommentFlavor::Multiline = comment.flavor() { return None; } let prefix = comment.prefix(); - if offset < comment.syntax().range().start() + TextUnit::of_str(prefix) + TextUnit::from(1) { + if position.offset + < comment.syntax().range().start() + TextUnit::of_str(prefix) + TextUnit::from(1) + { return None; } - let indent = node_indent(file, comment.syntax())?; + let indent = node_indent(&file, comment.syntax())?; let inserted = format!("\n{}{} ", indent, prefix); - let cursor_position = offset + TextUnit::of_str(&inserted); + let cursor_position = position.offset + TextUnit::of_str(&inserted); let mut edit = TextEditBuilder::default(); - edit.insert(offset, inserted); - Some(LocalEdit { + edit.insert(position.offset, inserted); + Some(SourceChange { label: "on enter".to_string(), - edit: edit.finish(), - cursor_position: Some(cursor_position), + source_file_edits: vec![SourceFileEdit { edit: edit.finish(), file_id: position.file_id }], + file_system_edits: vec![], + cursor_position: Some(FilePosition { offset: cursor_position, file_id: position.file_id }), }) } @@ -52,7 +59,7 @@ fn node_indent<'a>(file: &'a SourceFile, node: &SyntaxNode) -> Option<&'a str> { Some(&text[pos..]) } -pub fn on_eq_typed(file: &SourceFile, eq_offset: TextUnit) -> Option { +pub fn on_eq_typed(file: &SourceFile, eq_offset: TextUnit) -> Option { assert_eq!(file.syntax().text().char_at(eq_offset), Some('=')); let let_stmt: &ast::LetStmt = find_node_at_offset(file.syntax(), eq_offset)?; if let_stmt.has_semi() { @@ -72,17 +79,14 @@ pub fn on_eq_typed(file: &SourceFile, eq_offset: TextUnit) -> Option let offset = let_stmt.syntax().range().end(); let mut edit = TextEditBuilder::default(); edit.insert(offset, ";".to_string()); - Some(LocalEdit { - label: "add semicolon".to_string(), - edit: edit.finish(), - cursor_position: None, - }) + Some(edit.finish()) } -pub fn on_dot_typed(file: &SourceFile, dot_offset: TextUnit) -> Option { - assert_eq!(file.syntax().text().char_at(dot_offset), Some('.')); +pub(crate) fn on_dot_typed(db: &RootDatabase, position: FilePosition) -> Option { + let file = db.parse(position.file_id); + assert_eq!(file.syntax().text().char_at(position.offset), Some('.')); - let whitespace = find_leaf_at_offset(file.syntax(), dot_offset) + let whitespace = find_leaf_at_offset(file.syntax(), position.offset) .left_biased() .and_then(ast::Whitespace::cast)?; @@ -103,15 +107,18 @@ pub fn on_dot_typed(file: &SourceFile, dot_offset: TextUnit) -> Option Option Option { let (offset, before) = extract_offset(before); - let file = SourceFile::parse(&before); - let result = on_enter(&file, offset)?; - let actual = result.edit.apply(&before); - let actual = add_cursor(&actual, result.cursor_position.unwrap()); + let (analysis, file_id) = single_file(&before); + let result = analysis.on_enter(FilePosition { offset, file_id })?; + + assert_eq!(result.source_file_edits.len(), 1); + let actual = result.source_file_edits[0].edit.apply(&before); + let actual = add_cursor(&actual, result.cursor_position.unwrap().offset); Some(actual) } diff --git a/crates/ra_ide_api_light/src/lib.rs b/crates/ra_ide_api_light/src/lib.rs index f21a91e185c..df7f144b627 100644 --- a/crates/ra_ide_api_light/src/lib.rs +++ b/crates/ra_ide_api_light/src/lib.rs @@ -4,147 +4,9 @@ //! an edit or some auxiliary info. mod structure; -mod typing; -use rustc_hash::FxHashSet; -use ra_text_edit::TextEditBuilder; -use ra_syntax::{ - SourceFile, SyntaxNode, TextRange, TextUnit, Direction, - algo::find_leaf_at_offset, - SyntaxKind::{self, *}, - ast::{self, AstNode}, -}; +use ra_syntax::TextRange; pub use crate::{ structure::{file_structure, StructureNode}, - typing::{on_enter, on_dot_typed, on_eq_typed}, }; - -#[derive(Debug)] -pub struct LocalEdit { - pub label: String, - pub edit: ra_text_edit::TextEdit, - pub cursor_position: Option, -} - -#[derive(Debug)] -pub struct HighlightedRange { - pub range: TextRange, - pub tag: &'static str, -} - -#[derive(Debug, Copy, Clone)] -pub enum Severity { - Error, - WeakWarning, -} - -#[derive(Debug)] -pub struct Diagnostic { - pub range: TextRange, - pub msg: String, - pub severity: Severity, - pub fix: Option, -} - -pub fn matching_brace(file: &SourceFile, offset: TextUnit) -> Option { - const BRACES: &[SyntaxKind] = - &[L_CURLY, R_CURLY, L_BRACK, R_BRACK, L_PAREN, R_PAREN, L_ANGLE, R_ANGLE]; - let (brace_node, brace_idx) = find_leaf_at_offset(file.syntax(), offset) - .filter_map(|node| { - let idx = BRACES.iter().position(|&brace| brace == node.kind())?; - Some((node, idx)) - }) - .next()?; - let parent = brace_node.parent()?; - let matching_kind = BRACES[brace_idx ^ 1]; - let matching_node = parent.children().find(|node| node.kind() == matching_kind)?; - Some(matching_node.range().start()) -} - -pub fn highlight(root: &SyntaxNode) -> Vec { - // Visited nodes to handle highlighting priorities - let mut highlighted = FxHashSet::default(); - let mut res = Vec::new(); - for node in root.descendants() { - if highlighted.contains(&node) { - continue; - } - let tag = match node.kind() { - COMMENT => "comment", - STRING | RAW_STRING | RAW_BYTE_STRING | BYTE_STRING => "string", - ATTR => "attribute", - NAME_REF => "text", - NAME => "function", - INT_NUMBER | FLOAT_NUMBER | CHAR | BYTE => "literal", - LIFETIME => "parameter", - k if k.is_keyword() => "keyword", - _ => { - if let Some(macro_call) = ast::MacroCall::cast(node) { - if let Some(path) = macro_call.path() { - if let Some(segment) = path.segment() { - if let Some(name_ref) = segment.name_ref() { - highlighted.insert(name_ref.syntax()); - let range_start = name_ref.syntax().range().start(); - let mut range_end = name_ref.syntax().range().end(); - for sibling in path.syntax().siblings(Direction::Next) { - match sibling.kind() { - EXCL | IDENT => range_end = sibling.range().end(), - _ => (), - } - } - res.push(HighlightedRange { - range: TextRange::from_to(range_start, range_end), - tag: "macro", - }) - } - } - } - } - continue; - } - }; - res.push(HighlightedRange { range: node.range(), tag }) - } - res -} - -#[cfg(test)] -mod tests { - use ra_syntax::AstNode; - use insta::assert_debug_snapshot_matches; - - use test_utils::{add_cursor, assert_eq_text, extract_offset}; - - use super::*; - - #[test] - fn test_highlighting() { - let file = SourceFile::parse( - r#" -// comment -fn main() {} - println!("Hello, {}!", 92); -"#, - ); - let hls = highlight(file.syntax()); - assert_debug_snapshot_matches!("highlighting", hls); - } - - #[test] - fn test_matching_brace() { - fn do_check(before: &str, after: &str) { - let (pos, before) = extract_offset(before); - let file = SourceFile::parse(&before); - let new_pos = match matching_brace(&file, pos) { - None => pos, - Some(pos) => pos, - }; - let actual = add_cursor(&before, new_pos); - assert_eq_text!(after, &actual); - } - - do_check("struct Foo { a: i32, }<|>", "struct Foo <|>{ a: i32, }"); - } - -} diff --git a/crates/ra_ide_api_light/src/snapshots/tests__highlighting.snap b/crates/ra_ide_api_light/src/snapshots/tests__highlighting.snap deleted file mode 100644 index ef306a7a02f..00000000000 --- a/crates/ra_ide_api_light/src/snapshots/tests__highlighting.snap +++ /dev/null @@ -1,32 +0,0 @@ ---- -created: "2019-01-22T14:45:01.959724300+00:00" -creator: insta@0.4.0 -expression: hls -source: "crates\\ra_ide_api_light\\src\\lib.rs" ---- -[ - HighlightedRange { - range: [1; 11), - tag: "comment" - }, - HighlightedRange { - range: [12; 14), - tag: "keyword" - }, - HighlightedRange { - range: [15; 19), - tag: "function" - }, - HighlightedRange { - range: [29; 37), - tag: "macro" - }, - HighlightedRange { - range: [38; 50), - tag: "string" - }, - HighlightedRange { - range: [52; 54), - tag: "literal" - } -]