about summary refs log tree commit diff
path: root/src/libsyntax/ext/trace_macros.rs
blob: c2d4de1b423c41d966e0d82b430202b2042e9a4b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
use codemap::span;
use ext::base::ext_ctxt;
use ast::tt_delim;
use parse::lexer::{new_tt_reader, reader};
use parse::parser::{parser, SOURCE_FILE};
use parse::common::parser_common;

fn expand_trace_macros(cx: ext_ctxt, sp: span,
                       tt: ~[ast::token_tree]) -> base::mac_result
{
    let sess = cx.parse_sess();
    let cfg = cx.cfg();
    let tt_rdr = new_tt_reader(cx.parse_sess().span_diagnostic,
                               cx.parse_sess().interner, None, tt);
    let rdr = tt_rdr as reader;
    let rust_parser = parser(sess, cfg, rdr.dup(), SOURCE_FILE);

    let arg = cx.str_of(rust_parser.parse_ident());
    match arg {
      ~"true"  => cx.set_trace_macros(true),
      ~"false" => cx.set_trace_macros(false),
      _ => cx.span_fatal(sp, ~"trace_macros! only accepts `true` or `false`")
    }
    let rust_parser = parser(sess, cfg, rdr.dup(), SOURCE_FILE);
    let result = rust_parser.parse_expr();
    base::mr_expr(result)
}