src/libsyntax/ext/concat_idents.rs

   1 // Copyright 2012 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 ast;
  12 use codemap::Span;
  13 use ext::base::*;
  14 use ext::base;
  15 use owned_slice::OwnedSlice;
  16 use parse::token;
  17 use parse::token::{str_to_ident};
  18 use ptr::P;
  19
  20 pub fn expand_syntax_ext<'cx>(cx: &mut ExtCtxt, sp: Span, tts: &[ast::TokenTree])
  21                               -> Box<base::MacResult+'cx> {
  22     let mut res_str = String::new();
  23     for (i, e) in tts.iter().enumerate() {
  24         if i & 1 == 1 {
  25             match *e {
  26                 ast::TTTok(_, token::COMMA) => (),
  27                 _ => {
  28                     cx.span_err(sp, "concat_idents! expecting comma.");
  29                     return DummyResult::expr(sp);
  30                 }
  31             }
  32         } else {
  33             match *e {
  34                 ast::TTTok(_, token::IDENT(ident,_)) => {
  35                     res_str.push_str(token::get_ident(ident).get())
  36                 }
  37                 _ => {
  38                     cx.span_err(sp, "concat_idents! requires ident args.");
  39                     return DummyResult::expr(sp);
  40                 }
  41             }
  42         }
  43     }
  44     let res = str_to_ident(res_str.as_slice());
  45
  46     let e = P(ast::Expr {
  47         id: ast::DUMMY_NODE_ID,
  48         node: ast::ExprPath(
  49             ast::Path {
  50                  span: sp,
  51                  global: false,
  52                  segments: vec!(
  53                     ast::PathSegment {
  54                         identifier: res,
  55                         lifetimes: Vec::new(),
  56                         types: OwnedSlice::empty(),
  57                     }
  58                 )
  59             }
  60         ),
  61         span: sp,
  62     });
  63     MacExpr::new(e)
  64 }