]>
Commit | Line | Data |
---|---|---|
0531ce1d XL |
1 | // Copyright 2018 The Rust Project Developers. See the COPYRIGHT |
2 | // file at the top-level directory of this distribution and at | |
3 | // http://rust-lang.org/COPYRIGHT. | |
4 | // | |
5 | // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or | |
6 | // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license | |
7 | // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your | |
8 | // option. This file may not be copied, modified, or distributed | |
9 | // except according to those terms. | |
10 | ||
11 | use syntax::ast::*; | |
12 | use syntax::codemap::Spanned; | |
13 | use syntax::ext::base::*; | |
14 | use syntax::ext::build::AstBuilder; | |
15 | use syntax::parse::token; | |
16 | use syntax::print::pprust; | |
83c7162d | 17 | use syntax::symbol::Symbol; |
0531ce1d XL |
18 | use syntax::tokenstream::{TokenStream, TokenTree}; |
19 | use syntax_pos::{Span, DUMMY_SP}; | |
20 | ||
21 | pub fn expand_assert<'cx>( | |
22 | cx: &'cx mut ExtCtxt, | |
23 | sp: Span, | |
24 | tts: &[TokenTree], | |
25 | ) -> Box<MacResult + 'cx> { | |
26 | let mut parser = cx.new_parser_from_tts(tts); | |
27 | let cond_expr = panictry!(parser.parse_expr()); | |
28 | let custom_msg_args = if parser.eat(&token::Comma) { | |
29 | let ts = parser.parse_tokens(); | |
30 | if !ts.is_empty() { | |
31 | Some(ts) | |
32 | } else { | |
33 | None | |
34 | } | |
35 | } else { | |
36 | None | |
37 | }; | |
38 | ||
83c7162d | 39 | let sp = sp.apply_mark(cx.current_expansion.mark); |
0531ce1d | 40 | let panic_call = Mac_ { |
83c7162d | 41 | path: Path::from_ident(Ident::new(Symbol::intern("panic"), sp)), |
0531ce1d XL |
42 | tts: if let Some(ts) = custom_msg_args { |
43 | ts.into() | |
44 | } else { | |
45 | TokenStream::from(TokenTree::Token( | |
46 | DUMMY_SP, | |
47 | token::Literal( | |
48 | token::Lit::Str_(Name::intern(&format!( | |
49 | "assertion failed: {}", | |
50 | pprust::expr_to_string(&cond_expr).escape_debug() | |
51 | ))), | |
52 | None, | |
53 | ), | |
54 | )).into() | |
55 | }, | |
94b46f34 | 56 | delim: MacDelimiter::Parenthesis, |
0531ce1d XL |
57 | }; |
58 | let if_expr = cx.expr_if( | |
59 | sp, | |
60 | cx.expr(sp, ExprKind::Unary(UnOp::Not, cond_expr)), | |
61 | cx.expr( | |
62 | sp, | |
63 | ExprKind::Mac(Spanned { | |
64 | span: sp, | |
65 | node: panic_call, | |
66 | }), | |
67 | ), | |
68 | None, | |
69 | ); | |
70 | MacEager::expr(if_expr) | |
71 | } |