Fix fallout in tests.
This commit is contained in:
parent
b7eed53b55
commit
51104e5ca6
@ -18,10 +18,10 @@ extern crate syntax_pos;
|
|||||||
extern crate rustc;
|
extern crate rustc;
|
||||||
extern crate rustc_plugin;
|
extern crate rustc_plugin;
|
||||||
|
|
||||||
use syntax::parse::token::{self, str_to_ident, NtExpr, NtPat};
|
use syntax::parse::token::{str_to_ident, NtExpr, NtPat};
|
||||||
use syntax::ast::{Pat};
|
use syntax::ast::{Pat};
|
||||||
use syntax::tokenstream::{TokenTree};
|
use syntax::tokenstream::{TokenTree};
|
||||||
use syntax::ext::base::{ExtCtxt, MacResult, DummyResult, MacEager};
|
use syntax::ext::base::{ExtCtxt, MacResult, MacEager};
|
||||||
use syntax::ext::build::AstBuilder;
|
use syntax::ext::build::AstBuilder;
|
||||||
use syntax::ext::tt::macro_parser::{MatchedSeq, MatchedNonterminal};
|
use syntax::ext::tt::macro_parser::{MatchedSeq, MatchedNonterminal};
|
||||||
use syntax::ext::tt::macro_parser::{Success, Failure, Error};
|
use syntax::ext::tt::macro_parser::{Success, Failure, Error};
|
||||||
@ -30,35 +30,12 @@ use syntax::ptr::P;
|
|||||||
use syntax_pos::Span;
|
use syntax_pos::Span;
|
||||||
use rustc_plugin::Registry;
|
use rustc_plugin::Registry;
|
||||||
|
|
||||||
fn expand_mbe_matches(cx: &mut ExtCtxt, sp: Span, args: &[TokenTree])
|
fn expand_mbe_matches(cx: &mut ExtCtxt, _: Span, args: &[TokenTree])
|
||||||
-> Box<MacResult + 'static> {
|
-> Box<MacResult + 'static> {
|
||||||
|
|
||||||
let mbe_matcher = quote_matcher!(cx, $matched:expr, $($pat:pat)|+);
|
let mbe_matcher = quote_matcher!(cx, $matched:expr, $($pat:pat)|+);
|
||||||
|
let map = match TokenTree::parse(cx, &mbe_matcher, args) {
|
||||||
let mac_expr = match TokenTree::parse(cx, &mbe_matcher[..], args) {
|
Success(map) => map,
|
||||||
Success(map) => {
|
|
||||||
match (&*map[&str_to_ident("matched")], &*map[&str_to_ident("pat")]) {
|
|
||||||
(&MatchedNonterminal(NtExpr(ref matched_expr)),
|
|
||||||
&MatchedSeq(ref pats, seq_sp)) => {
|
|
||||||
let pats: Vec<P<Pat>> = pats.iter().map(|pat_nt|
|
|
||||||
if let &MatchedNonterminal(NtPat(ref pat)) = &**pat_nt {
|
|
||||||
pat.clone()
|
|
||||||
} else {
|
|
||||||
unreachable!()
|
|
||||||
}
|
|
||||||
).collect();
|
|
||||||
let arm = cx.arm(seq_sp, pats, cx.expr_bool(seq_sp, true));
|
|
||||||
|
|
||||||
quote_expr!(cx,
|
|
||||||
match $matched_expr {
|
|
||||||
$arm
|
|
||||||
_ => false
|
|
||||||
}
|
|
||||||
)
|
|
||||||
}
|
|
||||||
_ => unreachable!()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Failure(_, tok) => {
|
Failure(_, tok) => {
|
||||||
panic!("expected Success, but got Failure: {}", parse_failure_msg(tok));
|
panic!("expected Success, but got Failure: {}", parse_failure_msg(tok));
|
||||||
}
|
}
|
||||||
@ -67,6 +44,34 @@ fn expand_mbe_matches(cx: &mut ExtCtxt, sp: Span, args: &[TokenTree])
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
let matched_nt = match *map[&str_to_ident("matched")] {
|
||||||
|
MatchedNonterminal(ref nt) => nt.clone(),
|
||||||
|
_ => unreachable!(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let mac_expr = match (&*matched_nt, &*map[&str_to_ident("pat")]) {
|
||||||
|
(&NtExpr(ref matched_expr), &MatchedSeq(ref pats, seq_sp)) => {
|
||||||
|
let pats: Vec<P<Pat>> = pats.iter().map(|pat_nt| {
|
||||||
|
match **pat_nt {
|
||||||
|
MatchedNonterminal(ref nt) => match **nt {
|
||||||
|
NtPat(ref pat) => pat.clone(),
|
||||||
|
_ => unreachable!(),
|
||||||
|
},
|
||||||
|
_ => unreachable!(),
|
||||||
|
}
|
||||||
|
}).collect();
|
||||||
|
let arm = cx.arm(seq_sp, pats, cx.expr_bool(seq_sp, true));
|
||||||
|
|
||||||
|
quote_expr!(cx,
|
||||||
|
match $matched_expr {
|
||||||
|
$arm
|
||||||
|
_ => false
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
_ => unreachable!()
|
||||||
|
};
|
||||||
|
|
||||||
MacEager::expr(mac_expr)
|
MacEager::expr(mac_expr)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user