diff options
Diffstat (limited to 'crates/ra_hir_expand/src/db.rs')
-rw-r--r-- | crates/ra_hir_expand/src/db.rs | 80 |
1 files changed, 73 insertions, 7 deletions
diff --git a/crates/ra_hir_expand/src/db.rs b/crates/ra_hir_expand/src/db.rs index 7133b61db..912599e57 100644 --- a/crates/ra_hir_expand/src/db.rs +++ b/crates/ra_hir_expand/src/db.rs | |||
@@ -1,11 +1,13 @@ | |||
1 | use std::sync::Arc; | 1 | use std::sync::Arc; |
2 | 2 | ||
3 | use mbe::MacroRules; | ||
3 | use ra_db::{salsa, SourceDatabase}; | 4 | use ra_db::{salsa, SourceDatabase}; |
4 | use ra_syntax::{Parse, SyntaxNode}; | 5 | use ra_prof::profile; |
6 | use ra_syntax::{AstNode, Parse, SyntaxNode}; | ||
5 | 7 | ||
6 | use crate::{ | 8 | use crate::{ |
7 | ast_id_map::{AstIdMap, ErasedFileAstId}, | 9 | ast_id_map::{AstIdMap, ErasedFileAstId}, |
8 | expand::{HirFileId, MacroCallId, MacroCallLoc, MacroDefId, MacroFile}, | 10 | HirFileId, HirFileIdRepr, MacroCallId, MacroCallLoc, MacroDefId, MacroFile, MacroFileKind, |
9 | }; | 11 | }; |
10 | 12 | ||
11 | #[salsa::query_group(AstDatabaseStorage)] | 13 | #[salsa::query_group(AstDatabaseStorage)] |
@@ -15,18 +17,13 @@ pub trait AstDatabase: SourceDatabase { | |||
15 | fn ast_id_to_node(&self, file_id: HirFileId, ast_id: ErasedFileAstId) -> SyntaxNode; | 17 | fn ast_id_to_node(&self, file_id: HirFileId, ast_id: ErasedFileAstId) -> SyntaxNode; |
16 | 18 | ||
17 | #[salsa::transparent] | 19 | #[salsa::transparent] |
18 | #[salsa::invoke(crate::expand::parse_or_expand_query)] | ||
19 | fn parse_or_expand(&self, file_id: HirFileId) -> Option<SyntaxNode>; | 20 | fn parse_or_expand(&self, file_id: HirFileId) -> Option<SyntaxNode>; |
20 | 21 | ||
21 | #[salsa::interned] | 22 | #[salsa::interned] |
22 | fn intern_macro(&self, macro_call: MacroCallLoc) -> MacroCallId; | 23 | fn intern_macro(&self, macro_call: MacroCallLoc) -> MacroCallId; |
23 | #[salsa::invoke(crate::expand::macro_arg_query)] | ||
24 | fn macro_arg(&self, id: MacroCallId) -> Option<Arc<tt::Subtree>>; | 24 | fn macro_arg(&self, id: MacroCallId) -> Option<Arc<tt::Subtree>>; |
25 | #[salsa::invoke(crate::expand::macro_def_query)] | ||
26 | fn macro_def(&self, id: MacroDefId) -> Option<Arc<mbe::MacroRules>>; | 25 | fn macro_def(&self, id: MacroDefId) -> Option<Arc<mbe::MacroRules>>; |
27 | #[salsa::invoke(crate::expand::parse_macro_query)] | ||
28 | fn parse_macro(&self, macro_file: MacroFile) -> Option<Parse<SyntaxNode>>; | 26 | fn parse_macro(&self, macro_file: MacroFile) -> Option<Parse<SyntaxNode>>; |
29 | #[salsa::invoke(crate::expand::macro_expand_query)] | ||
30 | fn macro_expand(&self, macro_call: MacroCallId) -> Result<Arc<tt::Subtree>, String>; | 27 | fn macro_expand(&self, macro_call: MacroCallId) -> Result<Arc<tt::Subtree>, String>; |
31 | } | 28 | } |
32 | 29 | ||
@@ -44,3 +41,72 @@ pub(crate) fn ast_id_to_node( | |||
44 | let node = db.parse_or_expand(file_id).unwrap(); | 41 | let node = db.parse_or_expand(file_id).unwrap(); |
45 | db.ast_id_map(file_id)[ast_id].to_node(&node) | 42 | db.ast_id_map(file_id)[ast_id].to_node(&node) |
46 | } | 43 | } |
44 | |||
45 | pub(crate) fn macro_def(db: &impl AstDatabase, id: MacroDefId) -> Option<Arc<MacroRules>> { | ||
46 | let macro_call = id.ast_id.to_node(db); | ||
47 | let arg = macro_call.token_tree()?; | ||
48 | let (tt, _) = mbe::ast_to_token_tree(&arg).or_else(|| { | ||
49 | log::warn!("fail on macro_def to token tree: {:#?}", arg); | ||
50 | None | ||
51 | })?; | ||
52 | let rules = MacroRules::parse(&tt).ok().or_else(|| { | ||
53 | log::warn!("fail on macro_def parse: {:#?}", tt); | ||
54 | None | ||
55 | })?; | ||
56 | Some(Arc::new(rules)) | ||
57 | } | ||
58 | |||
59 | pub(crate) fn macro_arg(db: &impl AstDatabase, id: MacroCallId) -> Option<Arc<tt::Subtree>> { | ||
60 | let loc = db.lookup_intern_macro(id); | ||
61 | let macro_call = loc.ast_id.to_node(db); | ||
62 | let arg = macro_call.token_tree()?; | ||
63 | let (tt, _) = mbe::ast_to_token_tree(&arg)?; | ||
64 | Some(Arc::new(tt)) | ||
65 | } | ||
66 | |||
67 | pub(crate) fn macro_expand( | ||
68 | db: &impl AstDatabase, | ||
69 | id: MacroCallId, | ||
70 | ) -> Result<Arc<tt::Subtree>, String> { | ||
71 | let loc = db.lookup_intern_macro(id); | ||
72 | let macro_arg = db.macro_arg(id).ok_or("Fail to args in to tt::TokenTree")?; | ||
73 | |||
74 | let macro_rules = db.macro_def(loc.def).ok_or("Fail to find macro definition")?; | ||
75 | let tt = macro_rules.expand(¯o_arg).map_err(|err| format!("{:?}", err))?; | ||
76 | // Set a hard limit for the expanded tt | ||
77 | let count = tt.count(); | ||
78 | if count > 65536 { | ||
79 | return Err(format!("Total tokens count exceed limit : count = {}", count)); | ||
80 | } | ||
81 | Ok(Arc::new(tt)) | ||
82 | } | ||
83 | |||
84 | pub(crate) fn parse_or_expand(db: &impl AstDatabase, file_id: HirFileId) -> Option<SyntaxNode> { | ||
85 | match file_id.0 { | ||
86 | HirFileIdRepr::FileId(file_id) => Some(db.parse(file_id).tree().syntax().clone()), | ||
87 | HirFileIdRepr::MacroFile(macro_file) => { | ||
88 | db.parse_macro(macro_file).map(|it| it.syntax_node()) | ||
89 | } | ||
90 | } | ||
91 | } | ||
92 | |||
93 | pub(crate) fn parse_macro( | ||
94 | db: &impl AstDatabase, | ||
95 | macro_file: MacroFile, | ||
96 | ) -> Option<Parse<SyntaxNode>> { | ||
97 | let _p = profile("parse_macro_query"); | ||
98 | let macro_call_id = macro_file.macro_call_id; | ||
99 | let tt = db | ||
100 | .macro_expand(macro_call_id) | ||
101 | .map_err(|err| { | ||
102 | // Note: | ||
103 | // The final goal we would like to make all parse_macro success, | ||
104 | // such that the following log will not call anyway. | ||
105 | log::warn!("fail on macro_parse: (reason: {})", err,); | ||
106 | }) | ||
107 | .ok()?; | ||
108 | match macro_file.macro_file_kind { | ||
109 | MacroFileKind::Items => mbe::token_tree_to_items(&tt).ok().map(Parse::to_syntax), | ||
110 | MacroFileKind::Expr => mbe::token_tree_to_expr(&tt).ok().map(Parse::to_syntax), | ||
111 | } | ||
112 | } | ||