aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_syntax/src/ast
diff options
context:
space:
mode:
authorKirill Bulatov <[email protected]>2020-02-12 14:44:52 +0000
committerKirill Bulatov <[email protected]>2020-02-12 14:44:52 +0000
commit848c5762667101c8a017e561da2fa7e39b212929 (patch)
tree26a6acb0eb0d4099003ed455f89fde4d196a363a /crates/ra_syntax/src/ast
parent1596b31698acd1ca8fe25a1b699bef4a9a6feb1d (diff)
Introduce AttrKind
Diffstat (limited to 'crates/ra_syntax/src/ast')
-rw-r--r--crates/ra_syntax/src/ast/extensions.rs15
1 files changed, 12 insertions, 3 deletions
diff --git a/crates/ra_syntax/src/ast/extensions.rs b/crates/ra_syntax/src/ast/extensions.rs
index a297ae110..44de4af89 100644
--- a/crates/ra_syntax/src/ast/extensions.rs
+++ b/crates/ra_syntax/src/ast/extensions.rs
@@ -37,6 +37,12 @@ fn text_of_first_token(node: &SyntaxNode) -> &SmolStr {
37 node.green().children().next().and_then(|it| it.into_token()).unwrap().text() 37 node.green().children().next().and_then(|it| it.into_token()).unwrap().text()
38} 38}
39 39
40#[derive(Debug, Clone, PartialEq, Eq)]
41pub enum AttrKind {
42 Inner,
43 Outer,
44}
45
40impl ast::Attr { 46impl ast::Attr {
41 pub fn as_simple_atom(&self) -> Option<SmolStr> { 47 pub fn as_simple_atom(&self) -> Option<SmolStr> {
42 match self.input() { 48 match self.input() {
@@ -72,13 +78,16 @@ impl ast::Attr {
72 } 78 }
73 } 79 }
74 80
75 pub fn is_inner_attribute(&self) -> bool { 81 pub fn kind(&self) -> AttrKind {
76 let first_token = self.syntax().first_token(); 82 let first_token = self.syntax().first_token();
77 let first_token_kind = first_token.as_ref().map(SyntaxToken::kind); 83 let first_token_kind = first_token.as_ref().map(SyntaxToken::kind);
78 let second_token_kind = 84 let second_token_kind =
79 first_token.and_then(|token| token.next_token()).as_ref().map(SyntaxToken::kind); 85 first_token.and_then(|token| token.next_token()).as_ref().map(SyntaxToken::kind);
80 return first_token_kind == Some(SyntaxKind::POUND) 86
81 && second_token_kind == Some(SyntaxKind::EXCL); 87 match (first_token_kind, second_token_kind) {
88 (Some(SyntaxKind::POUND), Some(SyntaxKind::EXCL)) => AttrKind::Inner,
89 _ => AttrKind::Outer,
90 }
82 } 91 }
83} 92}
84 93