internal: move block to prefix entry point

This commit is contained in:
Aleksey Kladov 2021-12-27 15:39:17 +03:00
parent 350d5dc152
commit 519ee21bcb
3 changed files with 12 additions and 6 deletions

View file

@ -695,7 +695,11 @@ fn match_meta_var(kind: &str, input: &mut TtIter) -> ExpandResult<Option<Fragmen
"ty" => ParserEntryPoint::Type,
"pat" | "pat_param" => ParserEntryPoint::Pattern, // FIXME: edition2021
"stmt" => ParserEntryPoint::Statement,
"block" => ParserEntryPoint::Block,
"block" => {
return input
.expect_fragment2(parser::PrefixEntryPoint::Block)
.map(|tt| tt.map(Fragment::Tokens));
}
"meta" => ParserEntryPoint::MetaItem,
"item" => ParserEntryPoint::Item,
_ => {
@ -725,7 +729,7 @@ fn match_meta_var(kind: &str, input: &mut TtIter) -> ExpandResult<Option<Fragmen
.map_err(|()| err!())
}
// `vis` is optional
"vis" => Ok(input.eat_vis()),
"vis" => Ok(input.expect_fragment2(parser::PrefixEntryPoint::Vis).value),
_ => Err(ExpandError::UnexpectedToken),
};
return tt_result.map(|it| it.map(Fragment::Tokens)).into();
@ -894,10 +898,6 @@ fn expect_lifetime(&mut self) -> Result<tt::TokenTree, ()> {
.into())
}
fn eat_vis(&mut self) -> Option<tt::TokenTree> {
self.expect_fragment2(parser::PrefixEntryPoint::Vis).value
}
fn eat_char(&mut self, c: char) -> Option<tt::TokenTree> {
let mut fork = self.clone();
match fork.expect_char(c) {

View file

@ -53,6 +53,10 @@ pub(crate) mod prefix {
pub(crate) fn vis(p: &mut Parser) {
let _ = opt_visibility(p, false);
}
pub(crate) fn block(p: &mut Parser) {
expressions::block_expr(p);
}
}
}

View file

@ -52,12 +52,14 @@
#[derive(Debug)]
pub enum PrefixEntryPoint {
Vis,
Block,
}
impl PrefixEntryPoint {
pub fn parse(&self, input: &Input) -> Output {
let entry_point: fn(&'_ mut parser::Parser) = match self {
PrefixEntryPoint::Vis => grammar::entry::prefix::vis,
PrefixEntryPoint::Block => grammar::entry::prefix::block,
};
let mut p = parser::Parser::new(input);
entry_point(&mut p);