diff options
Diffstat (limited to 'crates/ra_mbe/src/syntax_bridge.rs')
-rw-r--r-- | crates/ra_mbe/src/syntax_bridge.rs | 46 |
1 files changed, 41 insertions, 5 deletions
diff --git a/crates/ra_mbe/src/syntax_bridge.rs b/crates/ra_mbe/src/syntax_bridge.rs index 19e09be80..6af3b1995 100644 --- a/crates/ra_mbe/src/syntax_bridge.rs +++ b/crates/ra_mbe/src/syntax_bridge.rs | |||
@@ -32,11 +32,11 @@ pub fn syntax_node_to_token_tree(node: &SyntaxNode) -> Option<(tt::Subtree, Toke | |||
32 | 32 | ||
33 | // The following items are what `rustc` macro can be parsed into : | 33 | // The following items are what `rustc` macro can be parsed into : |
34 | // link: https://github.com/rust-lang/rust/blob/9ebf47851a357faa4cd97f4b1dc7835f6376e639/src/libsyntax/ext/expand.rs#L141 | 34 | // link: https://github.com/rust-lang/rust/blob/9ebf47851a357faa4cd97f4b1dc7835f6376e639/src/libsyntax/ext/expand.rs#L141 |
35 | // * Expr(P<ast::Expr>) | 35 | // * Expr(P<ast::Expr>) -> token_tree_to_expr |
36 | // * Pat(P<ast::Pat>) | 36 | // * Pat(P<ast::Pat>) -> token_tree_to_pat |
37 | // * Ty(P<ast::Ty>) | 37 | // * Ty(P<ast::Ty>) -> token_tree_to_ty |
38 | // * Stmts(SmallVec<[ast::Stmt; 1]>) | 38 | // * Stmts(SmallVec<[ast::Stmt; 1]>) -> token_tree_to_stmts |
39 | // * Items(SmallVec<[P<ast::Item>; 1]>) | 39 | // * Items(SmallVec<[P<ast::Item>; 1]>) -> token_tree_to_items |
40 | // | 40 | // |
41 | // * TraitItems(SmallVec<[ast::TraitItem; 1]>) | 41 | // * TraitItems(SmallVec<[ast::TraitItem; 1]>) |
42 | // * ImplItems(SmallVec<[ast::ImplItem; 1]>) | 42 | // * ImplItems(SmallVec<[ast::ImplItem; 1]>) |
@@ -44,6 +44,42 @@ pub fn syntax_node_to_token_tree(node: &SyntaxNode) -> Option<(tt::Subtree, Toke | |||
44 | // | 44 | // |
45 | // | 45 | // |
46 | 46 | ||
47 | /// Parses the token tree (result of macro expansion) to an expression | ||
48 | pub fn token_tree_to_expr(tt: &tt::Subtree) -> TreeArc<ast::Expr> { | ||
49 | let token_source = SubtreeTokenSource::new(tt); | ||
50 | let mut tree_sink = TtTreeSink::new(token_source.querier()); | ||
51 | ra_parser::parse_expr(&token_source, &mut tree_sink); | ||
52 | let syntax = tree_sink.inner.finish(); | ||
53 | ast::Expr::cast(&syntax).unwrap().to_owned() | ||
54 | } | ||
55 | |||
56 | /// Parses the token tree (result of macro expansion) to a Pattern | ||
57 | pub fn token_tree_to_pat(tt: &tt::Subtree) -> TreeArc<ast::Pat> { | ||
58 | let token_source = SubtreeTokenSource::new(tt); | ||
59 | let mut tree_sink = TtTreeSink::new(token_source.querier()); | ||
60 | ra_parser::parse_pat(&token_source, &mut tree_sink); | ||
61 | let syntax = tree_sink.inner.finish(); | ||
62 | ast::Pat::cast(&syntax).unwrap().to_owned() | ||
63 | } | ||
64 | |||
65 | /// Parses the token tree (result of macro expansion) to a Type | ||
66 | pub fn token_tree_to_ty(tt: &tt::Subtree) -> TreeArc<ast::TypeRef> { | ||
67 | let token_source = SubtreeTokenSource::new(tt); | ||
68 | let mut tree_sink = TtTreeSink::new(token_source.querier()); | ||
69 | ra_parser::parse_ty(&token_source, &mut tree_sink); | ||
70 | let syntax = tree_sink.inner.finish(); | ||
71 | ast::TypeRef::cast(&syntax).unwrap().to_owned() | ||
72 | } | ||
73 | |||
74 | /// Parses the token tree (result of macro expansion) as a sequence of stmts | ||
75 | pub fn token_tree_to_macro_stmts(tt: &tt::Subtree) -> TreeArc<ast::MacroStmts> { | ||
76 | let token_source = SubtreeTokenSource::new(tt); | ||
77 | let mut tree_sink = TtTreeSink::new(token_source.querier()); | ||
78 | ra_parser::parse_macro_stmts(&token_source, &mut tree_sink); | ||
79 | let syntax = tree_sink.inner.finish(); | ||
80 | ast::MacroStmts::cast(&syntax).unwrap().to_owned() | ||
81 | } | ||
82 | |||
47 | /// Parses the token tree (result of macro expansion) as a sequence of items | 83 | /// Parses the token tree (result of macro expansion) as a sequence of items |
48 | pub fn token_tree_to_macro_items(tt: &tt::Subtree) -> TreeArc<ast::MacroItems> { | 84 | pub fn token_tree_to_macro_items(tt: &tt::Subtree) -> TreeArc<ast::MacroItems> { |
49 | let token_source = SubtreeTokenSource::new(tt); | 85 | let token_source = SubtreeTokenSource::new(tt); |