maud_macros 0.5.1

Compile-time HTML templates. 
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

#![crate_type = "dylib"]
#![feature(plugin_registrar, quote)]
#![feature(slice_patterns)]
#![feature(rustc_private)]

extern crate syntax;
extern crate rustc;
extern crate maud;

use syntax::ast::{Expr, TokenTree, TtToken};
use syntax::codemap::{DUMMY_SP, Span};
use syntax::ext::base::{ExtCtxt, MacEager, MacResult};
use syntax::parse::token;
use syntax::print::pprust;
use syntax::ptr::P;
use rustc::plugin::Registry;

mod parse;
mod render;

fn _expand_html(cx: &mut ExtCtxt, sp: Span, args: &[TokenTree]) -> P<Expr> {
    let (write, input) = parse::split_comma(cx, sp, args);
    parse::parse(cx, sp, write, input)
}

fn _expand_write_html(cx: &mut ExtCtxt, sp: Span, args: &[TokenTree]) -> P<Expr> {
    let (io_write, input) = parse::split_comma(cx, sp, args);
    let io_write = io_write.to_vec();
    let fmt_write = token::gensym_ident("__maud_utf8_writer");
    let fmt_write = vec![
        TtToken(DUMMY_SP, token::Ident(fmt_write, token::IdentStyle::Plain))];
    let expr = parse::parse(cx, sp, &fmt_write, input);
    quote_expr!(cx,
        match ::maud::Utf8Writer::new(&mut $io_write) {
            mut $fmt_write => {
                let _ = $expr;
                $fmt_write.into_result()
            }
        })
}

macro_rules! expand {
    ($name:ident $debug_name:ident $inner_fn:ident) => {
        fn $name<'cx>(cx: &'cx mut ExtCtxt, sp: Span, args: &[TokenTree])
            -> Box<MacResult + 'cx>
        {
            let expr = $inner_fn(cx, sp, args);
            MacEager::expr(expr)
        }

        fn $debug_name<'cx>(cx: &'cx mut ExtCtxt, sp: Span, args: &[TokenTree])
            -> Box<MacResult + 'cx>
        {
            let expr = $inner_fn(cx, sp, args);
            cx.span_note(sp, &format!("expansion:\n{}",
                                      pprust::expr_to_string(&expr)));
            MacEager::expr(expr)
        }
    }
}

expand!(expand_html expand_html_debug _expand_html);
expand!(expand_write_html expand_write_html_debug _expand_write_html);

#[plugin_registrar]
pub fn plugin_registrar(reg: &mut Registry) {
    reg.register_macro("html", expand_html);
    reg.register_macro("html_debug", expand_html_debug);
    reg.register_macro("write_html", expand_write_html);
    reg.register_macro("write_html_debug", expand_write_html_debug);
}