diff options
Diffstat (limited to 'crates/ra_hir_expand/src/db.rs')
-rw-r--r-- | crates/ra_hir_expand/src/db.rs | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/crates/ra_hir_expand/src/db.rs b/crates/ra_hir_expand/src/db.rs new file mode 100644 index 000000000..7133b61db --- /dev/null +++ b/crates/ra_hir_expand/src/db.rs | |||
@@ -0,0 +1,46 @@ | |||
1 | use std::sync::Arc; | ||
2 | |||
3 | use ra_db::{salsa, SourceDatabase}; | ||
4 | use ra_syntax::{Parse, SyntaxNode}; | ||
5 | |||
6 | use crate::{ | ||
7 | ast_id_map::{AstIdMap, ErasedFileAstId}, | ||
8 | expand::{HirFileId, MacroCallId, MacroCallLoc, MacroDefId, MacroFile}, | ||
9 | }; | ||
10 | |||
11 | #[salsa::query_group(AstDatabaseStorage)] | ||
12 | pub trait AstDatabase: SourceDatabase { | ||
13 | fn ast_id_map(&self, file_id: HirFileId) -> Arc<AstIdMap>; | ||
14 | #[salsa::transparent] | ||
15 | fn ast_id_to_node(&self, file_id: HirFileId, ast_id: ErasedFileAstId) -> SyntaxNode; | ||
16 | |||
17 | #[salsa::transparent] | ||
18 | #[salsa::invoke(crate::expand::parse_or_expand_query)] | ||
19 | fn parse_or_expand(&self, file_id: HirFileId) -> Option<SyntaxNode>; | ||
20 | |||
21 | #[salsa::interned] | ||
22 | 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>>; | ||
25 | #[salsa::invoke(crate::expand::macro_def_query)] | ||
26 | 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>>; | ||
29 | #[salsa::invoke(crate::expand::macro_expand_query)] | ||
30 | fn macro_expand(&self, macro_call: MacroCallId) -> Result<Arc<tt::Subtree>, String>; | ||
31 | } | ||
32 | |||
33 | pub(crate) fn ast_id_map(db: &impl AstDatabase, file_id: HirFileId) -> Arc<AstIdMap> { | ||
34 | let map = | ||
35 | db.parse_or_expand(file_id).map_or_else(AstIdMap::default, |it| AstIdMap::from_source(&it)); | ||
36 | Arc::new(map) | ||
37 | } | ||
38 | |||
39 | pub(crate) fn ast_id_to_node( | ||
40 | db: &impl AstDatabase, | ||
41 | file_id: HirFileId, | ||
42 | ast_id: ErasedFileAstId, | ||
43 | ) -> SyntaxNode { | ||
44 | let node = db.parse_or_expand(file_id).unwrap(); | ||
45 | db.ast_id_map(file_id)[ast_id].to_node(&node) | ||
46 | } | ||