qquote.rs

// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

#![allow(unused_imports)]
// ignore-cross-compile

#![feature(quote, rustc_private)]

extern crate syntax;
extern crate syntax_pos;

use syntax::source_map::FilePathMapping;
use syntax::print::pprust::*;
use syntax::symbol::Symbol;
use syntax_pos::DUMMY_SP;

fn main() {
    syntax::with_globals(|| run());
}

fn run() {
    let ps = syntax::parse::ParseSess::new(FilePathMapping::empty());
    let mut resolver = syntax::ext::base::DummyResolver;
    let mut cx = syntax::ext::base::ExtCtxt::new(
        &ps,
        syntax::ext::expand::ExpansionConfig::default("qquote".to_string()),
        &mut resolver);
    let cx = &mut cx;

    macro_rules! check {
        ($f: ident, $($e: expr),+; $expect: expr) => ({
            $(assert_eq!($f(&$e), $expect);)+
        });
    }

    let abc = quote_expr!(cx, 23);
    check!(expr_to_string, abc, *quote_expr!(cx, $abc); "23");

    let ty = quote_ty!(cx, isize);
    check!(ty_to_string, ty, *quote_ty!(cx, $ty); "isize");

    let item = quote_item!(cx, static x: $ty = 10;).unwrap();
    check!(item_to_string, item, quote_item!(cx, $item).unwrap(); "static x: isize = 10;");

    let twenty: u16 = 20;
    let stmt = quote_stmt!(cx, let x = $twenty;).unwrap();
    check!(stmt_to_string, stmt, quote_stmt!(cx, $stmt).unwrap(); "let x = 20u16;");

    let pat = quote_pat!(cx, Some(_));
    check!(pat_to_string, pat, *quote_pat!(cx, $pat); "Some(_)");

    let expr = quote_expr!(cx, (x, y));
    let arm = quote_arm!(cx, (ref x, ref y) => $expr,);
    check!(arm_to_string, arm, quote_arm!(cx, $arm); " (ref x, ref y) => (x, y),");

    let attr = quote_attr!(cx, #![cfg(foo = "bar")]);
    check!(attribute_to_string, attr, quote_attr!(cx, $attr); r#"#![cfg(foo = "bar")]"#);

    // quote_arg!

    let arg = quote_arg!(cx, foo: i32);
    check!(arg_to_string, arg, quote_arg!(cx, $arg); "foo: i32");

    let function = quote_item!(cx, fn f($arg) { }).unwrap();
    check!(item_to_string, function; "fn f(foo: i32) { }");

    let args = vec![arg, quote_arg!(cx, bar: u32)];
    let args = &args[..];
    let function = quote_item!(cx, fn f($args) { }).unwrap();
    check!(item_to_string, function; "fn f(foo: i32, bar: u32) { }");

    // quote_block!

    let block = quote_block!(cx, { $stmt let y = 40u32; });
    check!(block_to_string, block, *quote_block!(cx, $block); "{ let x = 20u16; let y = 40u32; }");

    let function = quote_item!(cx, fn f() $block).unwrap();
    check!(item_to_string, function; "fn f() { let x = 20u16; let y = 40u32; }");

    // quote_path!

    let path = quote_path!(cx, ::syntax::ptr::P<MetaItem>);
    check!(path_to_string, path, quote_path!(cx, $path); "::syntax::ptr::P<MetaItem>");

    let ty = quote_ty!(cx, $path);
    check!(ty_to_string, ty; "::syntax::ptr::P<MetaItem>");

    // quote_meta_item!

    let meta = quote_meta_item!(cx, cfg(foo = "bar"));
    check!(meta_item_to_string, meta, quote_meta_item!(cx, $meta); r#"cfg(foo = "bar")"#);

    let attr = quote_attr!(cx, #![$meta]);
    check!(attribute_to_string, attr; r#"#![cfg(foo = "bar")]"#);
}