mirror of
https://github.com/rust-lang/rust-analyzer
synced 2024-11-15 17:28:09 +00:00
Move highlighting and matching_brace
This commit is contained in:
parent
a656b891fb
commit
a3711e08dc
7 changed files with 166 additions and 164 deletions
|
@ -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<Diagnostic> {
|
||||
let source_file = db.parse(file_id);
|
||||
let mut res = Vec::new();
|
||||
|
|
|
@ -38,6 +38,7 @@ mod folding_ranges;
|
|||
mod line_index_utils;
|
||||
mod join_lines;
|
||||
mod typing;
|
||||
mod matching_brace;
|
||||
|
||||
#[cfg(test)]
|
||||
mod marks;
|
||||
|
@ -70,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,
|
||||
};
|
||||
pub use ra_ide_api_light::StructureNode;
|
||||
pub use ra_db::{
|
||||
Canceled, CrateGraph, CrateId, FileId, FilePosition, FileRange, SourceRootId,
|
||||
Edition
|
||||
|
@ -267,7 +268,7 @@ impl Analysis {
|
|||
/// supported).
|
||||
pub fn matching_brace(&self, position: FilePosition) -> Option<TextUnit> {
|
||||
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.
|
||||
|
|
45
crates/ra_ide_api/src/matching_brace.rs
Normal file
45
crates/ra_ide_api/src/matching_brace.rs
Normal file
|
@ -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<TextUnit> {
|
||||
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, }");
|
||||
}
|
||||
|
||||
}
|
34
crates/ra_ide_api/src/snapshots/tests__highlighting.snap
Normal file
34
crates/ra_ide_api/src/snapshots/tests__highlighting.snap
Normal file
|
@ -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"
|
||||
}
|
||||
]
|
||||
)
|
|
@ -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<HighlightedRange> {
|
||||
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);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -5,128 +5,8 @@
|
|||
|
||||
mod structure;
|
||||
|
||||
use rustc_hash::FxHashSet;
|
||||
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},
|
||||
};
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct HighlightedRange {
|
||||
pub range: TextRange,
|
||||
pub tag: &'static str,
|
||||
}
|
||||
|
||||
#[derive(Debug, Copy, Clone)]
|
||||
pub enum Severity {
|
||||
Error,
|
||||
WeakWarning,
|
||||
}
|
||||
|
||||
pub fn matching_brace(file: &SourceFile, offset: TextUnit) -> Option<TextUnit> {
|
||||
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<HighlightedRange> {
|
||||
// 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, }");
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
@ -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"
|
||||
}
|
||||
]
|
Loading…
Reference in a new issue