expand.rs 67.1 KB
Newer Older
S
Steven Fackler 已提交
1
// Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
2 3 4 5 6 7 8 9 10
// 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.

11
use ast::{Block, Crate, DeclKind, PatKind};
12
use ast::{Local, Ident, Mac_, Name};
13
use ast::{MacStmtStyle, Mrk, Stmt, StmtKind, ItemKind};
14
use ast::TokenTree;
15
use ast;
16
use ext::mtwt;
17
use ext::build::AstBuilder;
18
use attr;
19
use attr::{AttrMetaMethods, WithAttrs, ThinAttributesExt};
20
use codemap;
21
use codemap::{Span, Spanned, ExpnInfo, NameAndSpan, MacroBang, MacroAttribute};
P
Patrick Walton 已提交
22
use ext::base::*;
23
use feature_gate::{self, Features};
J
John Clements 已提交
24
use fold;
25
use fold::*;
M
Marvin Löbel 已提交
26
use util::move_map::MoveMap;
27
use parse;
28
use parse::token::{fresh_mark, fresh_name, intern, keywords};
29 30
use ptr::P;
use util::small_vector::SmallVector;
J
John Clements 已提交
31
use visit;
H
Huon Wilson 已提交
32
use visit::Visitor;
33
use std_inject;
34

35
use std::collections::HashSet;
36
use std::env;
37

J
Jeffrey Seyfried 已提交
38
// A trait for AST nodes and AST node lists into which macro invocations may expand.
39
trait MacroGenerable: Sized {
J
Jeffrey Seyfried 已提交
40
    // Expand the given MacResult using its appropriate `make_*` method.
41
    fn make_with<'a>(result: Box<MacResult + 'a>) -> Option<Self>;
J
Jeffrey Seyfried 已提交
42 43

    // Fold this node or list of nodes using the given folder.
44
    fn fold_with<F: Folder>(self, folder: &mut F) -> Self;
J
Jeffrey Seyfried 已提交
45 46

    // Return a placeholder expansion to allow compilation to continue after an erroring expansion.
47
    fn dummy(span: Span) -> Self;
J
Jeffrey Seyfried 已提交
48 49

    // The user-friendly name of the node type (e.g. "expression", "item", etc.) for diagnostics.
50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79
    fn kind_name() -> &'static str;
}

macro_rules! impl_macro_generable {
    ($($ty:ty: $kind_name:expr, .$make:ident, $(.$fold:ident)* $(lift .$fold_elt:ident)*,
               |$span:ident| $dummy:expr;)*) => { $(
        impl MacroGenerable for $ty {
            fn kind_name() -> &'static str { $kind_name }
            fn make_with<'a>(result: Box<MacResult + 'a>) -> Option<Self> { result.$make() }
            fn fold_with<F: Folder>(self, folder: &mut F) -> Self {
                $( folder.$fold(self) )*
                $( self.into_iter().flat_map(|item| folder. $fold_elt (item)).collect() )*
            }
            fn dummy($span: Span) -> Self { $dummy }
        }
    )* }
}

impl_macro_generable! {
    P<ast::Expr>: "expression", .make_expr, .fold_expr, |span| DummyResult::raw_expr(span);
    P<ast::Pat>:  "pattern",    .make_pat,  .fold_pat,  |span| P(DummyResult::raw_pat(span));
    P<ast::Ty>:   "type",       .make_ty,   .fold_ty,   |span| DummyResult::raw_ty(span);
    SmallVector<ast::ImplItem>:
        "impl item", .make_impl_items, lift .fold_impl_item, |_span| SmallVector::zero();
    SmallVector<P<ast::Item>>:
        "item",      .make_items,      lift .fold_item,      |_span| SmallVector::zero();
    SmallVector<ast::Stmt>:
        "statement", .make_stmts,      lift .fold_stmt,      |_span| SmallVector::zero();
}

80
pub fn expand_expr(e: P<ast::Expr>, fld: &mut MacroExpander) -> P<ast::Expr> {
81
    return e.and_then(|ast::Expr {id, node, span, attrs}| match node {
82

83 84
        // expr_mac should really be expr_ext or something; it's the
        // entry-point for all syntax extensions.
85
        ast::ExprKind::Mac(mac) => {
86
            expand_mac_invoc(mac, None, attrs.into_attr_vec(), span, fld)
87
        }
88

89
        ast::ExprKind::While(cond, body, opt_ident) => {
P
Pythoner6 已提交
90 91
            let cond = fld.fold_expr(cond);
            let (body, opt_ident) = expand_loop_block(body, opt_ident, fld);
92
            fld.cx.expr(span, ast::ExprKind::While(cond, body, opt_ident))
93
                .with_attrs(fold_thin_attrs(attrs, fld))
P
Pythoner6 已提交
94 95
        }

96
        ast::ExprKind::WhileLet(pat, expr, body, opt_ident) => {
N
Nick Cameron 已提交
97 98
            let pat = fld.fold_pat(pat);
            let expr = fld.fold_expr(expr);
99 100 101 102 103 104 105 106 107 108 109

            // Hygienic renaming of the body.
            let ((body, opt_ident), mut rewritten_pats) =
                rename_in_scope(vec![pat],
                                fld,
                                (body, opt_ident),
                                |rename_fld, fld, (body, opt_ident)| {
                expand_loop_block(rename_fld.fold_block(body), opt_ident, fld)
            });
            assert!(rewritten_pats.len() == 1);

110 111
            let wl = ast::ExprKind::WhileLet(rewritten_pats.remove(0), expr, body, opt_ident);
            fld.cx.expr(span, wl).with_attrs(fold_thin_attrs(attrs, fld))
112 113
        }

114
        ast::ExprKind::Loop(loop_block, opt_ident) => {
115
            let (loop_block, opt_ident) = expand_loop_block(loop_block, opt_ident, fld);
116
            fld.cx.expr(span, ast::ExprKind::Loop(loop_block, opt_ident))
117
                .with_attrs(fold_thin_attrs(attrs, fld))
E
Edward Wang 已提交
118 119
        }

120
        ast::ExprKind::ForLoop(pat, head, body, opt_ident) => {
121
            let pat = fld.fold_pat(pat);
122 123 124 125 126 127 128 129 130 131 132

            // Hygienic renaming of the for loop body (for loop binds its pattern).
            let ((body, opt_ident), mut rewritten_pats) =
                rename_in_scope(vec![pat],
                                fld,
                                (body, opt_ident),
                                |rename_fld, fld, (body, opt_ident)| {
                expand_loop_block(rename_fld.fold_block(body), opt_ident, fld)
            });
            assert!(rewritten_pats.len() == 1);

133
            let head = fld.fold_expr(head);
134 135
            let fl = ast::ExprKind::ForLoop(rewritten_pats.remove(0), head, body, opt_ident);
            fld.cx.expr(span, fl).with_attrs(fold_thin_attrs(attrs, fld))
136 137
        }

138
        ast::ExprKind::IfLet(pat, sub_expr, body, else_opt) => {
139 140 141 142 143 144 145 146 147 148 149 150 151 152
            let pat = fld.fold_pat(pat);

            // Hygienic renaming of the body.
            let (body, mut rewritten_pats) =
                rename_in_scope(vec![pat],
                                fld,
                                body,
                                |rename_fld, fld, body| {
                fld.fold_block(rename_fld.fold_block(body))
            });
            assert!(rewritten_pats.len() == 1);

            let else_opt = else_opt.map(|else_opt| fld.fold_expr(else_opt));
            let sub_expr = fld.fold_expr(sub_expr);
153 154
            let il = ast::ExprKind::IfLet(rewritten_pats.remove(0), sub_expr, body, else_opt);
            fld.cx.expr(span, il).with_attrs(fold_thin_attrs(attrs, fld))
155 156
        }

157
        ast::ExprKind::Closure(capture_clause, fn_decl, block, fn_decl_span) => {
158
            let (rewritten_fn_decl, rewritten_block)
159
                = expand_and_rename_fn_decl_and_block(fn_decl, block, fld);
160
            let new_node = ast::ExprKind::Closure(capture_clause,
161 162 163 164 165 166 167
                                                  rewritten_fn_decl,
                                                  rewritten_block,
                                                  fld.new_span(fn_decl_span));
            P(ast::Expr{ id:id,
                         node: new_node,
                         span: fld.new_span(span),
                         attrs: fold_thin_attrs(attrs, fld) })
168 169
        }

170 171 172 173
        _ => {
            P(noop_fold_expr(ast::Expr {
                id: id,
                node: node,
174
                span: span,
175
                attrs: attrs
176 177
            }, fld))
        }
178
    });
179 180
}

181 182
/// Expand a macro invocation. Returns the result of expansion.
fn expand_mac_invoc<T>(mac: ast::Mac, ident: Option<Ident>, attrs: Vec<ast::Attribute>, span: Span,
183 184 185
                       fld: &mut MacroExpander) -> T
    where T: MacroGenerable,
{
186 187
    // It would almost certainly be cleaner to pass the whole macro invocation in,
    // rather than pulling it apart and marking the tts and the ctxt separately.
188 189 190
    let Mac_ { path, tts, .. } = mac.node;
    let mark = fresh_mark();

191
    fn mac_result<'a>(path: &ast::Path, ident: Option<Ident>, tts: Vec<TokenTree>, mark: Mrk,
192 193
                      attrs: Vec<ast::Attribute>, call_site: Span, fld: &'a mut MacroExpander)
                      -> Option<Box<MacResult + 'a>> {
194 195 196 197 198 199 200
        // Detect use of feature-gated or invalid attributes on macro invoations
        // since they will not be detected after macro expansion.
        for attr in attrs.iter() {
            feature_gate::check_attribute(&attr, &fld.cx.parse_sess.span_diagnostic,
                                          &fld.cx.parse_sess.codemap(),
                                          &fld.cx.ecfg.features.unwrap());
        }
201 202 203 204 205 206 207 208 209 210 211 212

        if path.segments.len() > 1 {
            fld.cx.span_err(path.span, "expected macro name without module separators");
            return None;
        }

        let extname = path.segments[0].identifier.name;
        let extension = if let Some(extension) = fld.cx.syntax_env.find(extname) {
            extension
        } else {
            let mut err = fld.cx.struct_span_err(path.span,
                                                 &format!("macro undefined: '{}!'", &extname));
213
            fld.cx.suggest_macro_name(&extname.as_str(), &mut err);
N
Nick Cameron 已提交
214
            err.emit();
215 216
            return None;
        };
J
John Clements 已提交
217

J
Jeffrey Seyfried 已提交
218
        let ident = ident.unwrap_or(keywords::Invalid.ident());
219
        match *extension {
220
            NormalTT(ref expandfun, exp_span, allow_internal_unstable) => {
221 222 223 224 225 226 227
                if ident.name != keywords::Invalid.name() {
                    let msg =
                        format!("macro {}! expects no ident argument, given '{}'", extname, ident);
                    fld.cx.span_err(path.span, &msg);
                    return None;
                }

228
                fld.cx.bt_push(ExpnInfo {
229 230 231 232 233 234 235
                    call_site: call_site,
                    callee: NameAndSpan {
                        format: MacroBang(extname),
                        span: exp_span,
                        allow_internal_unstable: allow_internal_unstable,
                    },
                });
236 237 238 239 240 241 242

                // The span that we pass to the expanders we want to
                // be the root of the call stack. That's the most
                // relevant span and it's the actual invocation of
                // the macro.
                let mac_span = fld.cx.original_span();

243 244
                let marked_tts = mark_tts(&tts[..], mark);
                Some(expandfun.expand(fld.cx, mac_span, &marked_tts))
245
            }
246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303

            IdentTT(ref expander, tt_span, allow_internal_unstable) => {
                if ident.name == keywords::Invalid.name() {
                    fld.cx.span_err(path.span,
                                    &format!("macro {}! expects an ident argument", extname));
                    return None;
                };

                fld.cx.bt_push(ExpnInfo {
                    call_site: call_site,
                    callee: NameAndSpan {
                        format: MacroBang(extname),
                        span: tt_span,
                        allow_internal_unstable: allow_internal_unstable,
                    }
                });

                let marked_tts = mark_tts(&tts, mark);
                Some(expander.expand(fld.cx, call_site, ident, marked_tts))
            }

            MacroRulesTT => {
                if ident.name == keywords::Invalid.name() {
                    fld.cx.span_err(path.span,
                                    &format!("macro {}! expects an ident argument", extname));
                    return None;
                };

                fld.cx.bt_push(ExpnInfo {
                    call_site: call_site,
                    callee: NameAndSpan {
                        format: MacroBang(extname),
                        span: None,
                        // `macro_rules!` doesn't directly allow unstable
                        // (this is orthogonal to whether the macro it creates allows it)
                        allow_internal_unstable: false,
                    }
                });

                // DON'T mark before expansion.
                fld.cx.insert_macro(ast::MacroDef {
                    ident: ident,
                    id: ast::DUMMY_NODE_ID,
                    span: call_site,
                    imported_from: None,
                    use_locally: true,
                    body: tts,
                    export: attr::contains_name(&attrs, "macro_export"),
                    allow_internal_unstable: attr::contains_name(&attrs, "allow_internal_unstable"),
                    attrs: attrs,
                });

                // macro_rules! has a side effect but expands to nothing.
                fld.cx.bt_pop();
                None
            }

            MultiDecorator(..) | MultiModifier(..) => {
304
                fld.cx.span_err(path.span,
305
                                &format!("`{}` can only be used in attributes", extname));
306
                None
J
John Clements 已提交
307 308 309
            }
        }
    }
310

311
    let opt_expanded = T::make_with(match mac_result(&path, ident, tts, mark, attrs, span, fld) {
312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328
        Some(result) => result,
        None => return T::dummy(span),
    });

    let expanded = if let Some(expanded) = opt_expanded {
        expanded
    } else {
        let msg = format!("non-{kind} macro in {kind} position: {name}",
                          name = path.segments[0].identifier.name, kind = T::kind_name());
        fld.cx.span_err(path.span, &msg);
        return T::dummy(span);
    };

    let marked = expanded.fold_with(&mut Marker { mark: mark });
    let fully_expanded = marked.fold_with(fld);
    fld.cx.bt_pop();
    fully_expanded
J
John Clements 已提交
329 330
}

331 332 333 334 335
/// Rename loop label and expand its loop body
///
/// The renaming procedure for loop is different in the sense that the loop
/// body is in a block enclosed by loop head so the renaming of loop label
/// must be propagated to the enclosed context.
336 337 338
fn expand_loop_block(loop_block: P<Block>,
                     opt_ident: Option<Ident>,
                     fld: &mut MacroExpander) -> (P<Block>, Option<Ident>) {
E
Edward Wang 已提交
339 340
    match opt_ident {
        Some(label) => {
341
            let new_label = fresh_name(label);
E
Edward Wang 已提交
342
            let rename = (label, new_label);
343 344 345 346 347 348

            // The rename *must not* be added to the pending list of current
            // syntax context otherwise an unrelated `break` or `continue` in
            // the same context will pick that up in the deferred renaming pass
            // and be renamed incorrectly.
            let mut rename_list = vec!(rename);
349
            let mut rename_fld = IdentRenamer{renames: &mut rename_list};
350 351 352 353 354
            let renamed_ident = rename_fld.fold_ident(label);

            // The rename *must* be added to the enclosed syntax context for
            // `break` or `continue` to pick up because by definition they are
            // in a block enclosed by loop head.
355 356
            fld.cx.syntax_env.push_frame();
            fld.cx.syntax_env.info().pending_renames.push(rename);
357
            let expanded_block = expand_block_elts(loop_block, fld);
358
            fld.cx.syntax_env.pop_frame();
359 360

            (expanded_block, Some(renamed_ident))
E
Edward Wang 已提交
361
        }
362
        None => (fld.fold_block(loop_block), opt_ident)
E
Edward Wang 已提交
363 364 365
    }
}

366 367
// eval $e with a new exts frame.
// must be a macro so that $e isn't evaluated too early.
368
macro_rules! with_exts_frame {
369
    ($extsboxexpr:expr,$macros_escape:expr,$e:expr) =>
S
Steven Fackler 已提交
370 371
    ({$extsboxexpr.push_frame();
      $extsboxexpr.info().macros_escape = $macros_escape;
J
John Clements 已提交
372
      let result = $e;
S
Steven Fackler 已提交
373
      $extsboxexpr.pop_frame();
J
John Clements 已提交
374 375
      result
     })
376
}
J
John Clements 已提交
377

378
// When we enter a module, record it, for the sake of `module!`
379 380
pub fn expand_item(it: P<ast::Item>, fld: &mut MacroExpander)
                   -> SmallVector<P<ast::Item>> {
381
    expand_annotatable(Annotatable::Item(it), fld)
382
        .into_iter().map(|i| i.expect_item()).collect()
383 384
}

385 386
/// Expand item_kind
fn expand_item_kind(item: ast::ItemKind, fld: &mut MacroExpander) -> ast::ItemKind {
387
    match item {
388
        ast::ItemKind::Fn(decl, unsafety, constness, abi, generics, body) => {
J
John Clements 已提交
389
            let (rewritten_fn_decl, rewritten_body)
390
                = expand_and_rename_fn_decl_and_block(decl, body, fld);
M
Marvin Löbel 已提交
391
            let expanded_generics = fold::noop_fold_generics(generics,fld);
392
            ast::ItemKind::Fn(rewritten_fn_decl, unsafety, constness, abi,
393
                        expanded_generics, rewritten_body)
J
John Clements 已提交
394
        }
395
        _ => noop_fold_item_kind(item, fld)
J
John Clements 已提交
396 397 398
    }
}

399 400
// does this attribute list contain "macro_use" ?
fn contains_macro_use(fld: &mut MacroExpander, attrs: &[ast::Attribute]) -> bool {
401
    for attr in attrs {
402 403
        let mut is_use = attr.check_name("macro_use");
        if attr.check_name("macro_escape") {
N
Nick Cameron 已提交
404 405 406
            let mut err =
                fld.cx.struct_span_warn(attr.span,
                                        "macro_escape is a deprecated synonym for macro_use");
407
            is_use = true;
408
            if let ast::AttrStyle::Inner = attr.node.style {
409 410
                err.help("consider an outer attribute, \
                          #[macro_use] mod ...").emit();
N
Nick Cameron 已提交
411 412
            } else {
                err.emit();
413 414 415 416 417
            }
        };

        if is_use {
            match attr.node.value.node {
418
                ast::MetaItemKind::Word(..) => (),
419 420 421 422 423 424
                _ => fld.cx.span_err(attr.span, "arguments to macro_use are not allowed here"),
            }
            return true;
        }
    }
    false
J
John Clements 已提交
425 426
}

J
John Clements 已提交
427
/// Expand a stmt
428
fn expand_stmt(stmt: Stmt, fld: &mut MacroExpander) -> SmallVector<Stmt> {
429 430 431 432 433 434 435
    // perform all pending renames
    let stmt = {
        let pending_renames = &mut fld.cx.syntax_env.info().pending_renames;
        let mut rename_fld = IdentRenamer{renames:pending_renames};
        rename_fld.fold_stmt(stmt).expect_one("rename_fold didn't return one value")
    };

436
    let (mac, style, attrs) = match stmt.node {
437
        StmtKind::Mac(mac, style, attrs) => (mac, style, attrs),
438
        _ => return expand_non_macro_stmt(stmt, fld)
439
    };
440

441
    let mut fully_expanded: SmallVector<ast::Stmt> =
442
        expand_mac_invoc(mac.unwrap(), None, attrs.into_attr_vec(), stmt.span, fld);
443 444 445

    // If this is a macro invocation with a semicolon, then apply that
    // semicolon to the final statement produced by expansion.
446
    if style == MacStmtStyle::Semicolon {
447
        if let Some(stmt) = fully_expanded.pop() {
448 449 450 451 452 453 454
            let new_stmt = Spanned {
                node: match stmt.node {
                    StmtKind::Expr(e, stmt_id) => StmtKind::Semi(e, stmt_id),
                    _ => stmt.node /* might already have a semi */
                },
                span: stmt.span
            };
455
            fully_expanded.push(new_stmt);
456
        }
457
    }
458 459

    fully_expanded
460 461
}

462 463
// expand a non-macro stmt. this is essentially the fallthrough for
// expand_stmt, above.
464
fn expand_non_macro_stmt(Spanned {node, span: stmt_span}: Stmt, fld: &mut MacroExpander)
465
                         -> SmallVector<Stmt> {
466
    // is it a let?
467
    match node {
468
        StmtKind::Decl(decl, node_id) => decl.and_then(|Spanned {node: decl, span}| match decl {
469
            DeclKind::Local(local) => {
470
                // take it apart:
471
                let rewritten_local = local.map(|Local {id, pat, ty, init, span, attrs}| {
472
                    // expand the ty since TyKind::FixedLengthVec contains an Expr
473
                    // and thus may have a macro use
S
Seo Sanghyeon 已提交
474
                    let expanded_ty = ty.map(|t| fld.fold_ty(t));
475
                    // expand the pat (it might contain macro uses):
476
                    let expanded_pat = fld.fold_pat(pat);
477
                    // find the PatIdents in the pattern:
478 479 480 481
                    // oh dear heaven... this is going to include the enum
                    // names, as well... but that should be okay, as long as
                    // the new names are gensyms for the old ones.
                    // generate fresh names, push them to a new pending list
S
Seo Sanghyeon 已提交
482
                    let idents = pattern_bindings(&expanded_pat);
483
                    let mut new_pending_renames =
484
                        idents.iter().map(|ident| (*ident, fresh_name(*ident))).collect();
485 486
                    // rewrite the pattern using the new names (the old
                    // ones have already been applied):
487
                    let rewritten_pat = {
488 489
                        // nested binding to allow borrow to expire:
                        let mut rename_fld = IdentRenamer{renames: &mut new_pending_renames};
490 491 492
                        rename_fld.fold_pat(expanded_pat)
                    };
                    // add them to the existing pending renames:
493
                    fld.cx.syntax_env.info().pending_renames
494
                          .extend(new_pending_renames);
495 496 497 498 499 500
                    Local {
                        id: id,
                        ty: expanded_ty,
                        pat: rewritten_pat,
                        // also, don't forget to expand the init:
                        init: init.map(|e| fld.fold_expr(e)),
501
                        span: span,
502
                        attrs: fold::fold_thin_attrs(attrs, fld),
503 504
                    }
                });
505
                SmallVector::one(Spanned {
506
                    node: StmtKind::Decl(P(Spanned {
507
                            node: DeclKind::Local(rewritten_local),
508 509 510 511
                            span: span
                        }),
                        node_id),
                    span: stmt_span
512
                })
S
Steven Fackler 已提交
513
            }
514 515
            _ => {
                noop_fold_stmt(Spanned {
516
                    node: StmtKind::Decl(P(Spanned {
517 518 519 520 521 522 523 524 525 526 527 528 529 530
                            node: decl,
                            span: span
                        }),
                        node_id),
                    span: stmt_span
                }, fld)
            }
        }),
        _ => {
            noop_fold_stmt(Spanned {
                node: node,
                span: stmt_span
            }, fld)
        }
531 532 533
    }
}

J
John Clements 已提交
534
// expand the arm of a 'match', renaming for macro hygiene
535
fn expand_arm(arm: ast::Arm, fld: &mut MacroExpander) -> ast::Arm {
536
    // expand pats... they might contain macro uses:
537
    let expanded_pats = arm.pats.move_map(|pat| fld.fold_pat(pat));
538
    if expanded_pats.is_empty() {
S
Steve Klabnik 已提交
539
        panic!("encountered match arm with 0 patterns");
J
John Clements 已提交
540
    }
541

542
    // apply renaming and then expansion to the guard and the body:
543 544 545 546 547 548 549 550 551 552
    let ((rewritten_guard, rewritten_body), rewritten_pats) =
        rename_in_scope(expanded_pats,
                        fld,
                        (arm.guard, arm.body),
                        |rename_fld, fld, (ag, ab)|{
        let rewritten_guard = ag.map(|g| fld.fold_expr(rename_fld.fold_expr(g)));
        let rewritten_body = fld.fold_expr(rename_fld.fold_expr(ab));
        (rewritten_guard, rewritten_body)
    });

J
John Clements 已提交
553
    ast::Arm {
554
        attrs: fold::fold_attrs(arm.attrs, fld),
555 556 557
        pats: rewritten_pats,
        guard: rewritten_guard,
        body: rewritten_body,
J
John Clements 已提交
558
    }
J
John Clements 已提交
559 560
}

561 562 563 564 565 566 567 568 569
fn rename_in_scope<X, F>(pats: Vec<P<ast::Pat>>,
                         fld: &mut MacroExpander,
                         x: X,
                         f: F)
                         -> (X, Vec<P<ast::Pat>>)
    where F: Fn(&mut IdentRenamer, &mut MacroExpander, X) -> X
{
    // all of the pats must have the same set of bindings, so use the
    // first one to extract them and generate new names:
S
Seo Sanghyeon 已提交
570
    let idents = pattern_bindings(&pats[0]);
571 572 573 574 575 576 577 578 579
    let new_renames = idents.into_iter().map(|id| (id, fresh_name(id))).collect();
    // apply the renaming, but only to the PatIdents:
    let mut rename_pats_fld = PatIdentRenamer{renames:&new_renames};
    let rewritten_pats = pats.move_map(|pat| rename_pats_fld.fold_pat(pat));

    let mut rename_fld = IdentRenamer{ renames:&new_renames };
    (f(&mut rename_fld, fld, x), rewritten_pats)
}

580
/// A visitor that extracts the PatKind::Ident (binding) paths
581 582
/// from a given thingy and puts them in a mutable
/// array
583
#[derive(Clone)]
584
struct PatIdentFinder {
585
    ident_accumulator: Vec<ast::Ident>
586 587
}

588
impl<'v> Visitor<'v> for PatIdentFinder {
589
    fn visit_pat(&mut self, pattern: &ast::Pat) {
590
        match *pattern {
591
            ast::Pat { id: _, node: PatKind::Ident(_, ref path1, ref inner), span: _ } => {
592
                self.ident_accumulator.push(path1.node);
593
                // visit optional subpattern of PatKind::Ident:
594
                if let Some(ref subpat) = *inner {
S
Seo Sanghyeon 已提交
595
                    self.visit_pat(subpat)
J
John Clements 已提交
596 597
                }
            }
598
            // use the default traversal for non-PatIdents
599
            _ => visit::walk_pat(self, pattern)
600 601 602 603
        }
    }
}

604
/// find the PatKind::Ident paths in a pattern
605
fn pattern_bindings(pat: &ast::Pat) -> Vec<ast::Ident> {
606
    let mut name_finder = PatIdentFinder{ident_accumulator:Vec::new()};
607
    name_finder.visit_pat(pat);
608
    name_finder.ident_accumulator
609 610
}

611
/// find the PatKind::Ident paths in a
J
John Clements 已提交
612 613
fn fn_decl_arg_bindings(fn_decl: &ast::FnDecl) -> Vec<ast::Ident> {
    let mut pat_idents = PatIdentFinder{ident_accumulator:Vec::new()};
614
    for arg in &fn_decl.inputs {
S
Seo Sanghyeon 已提交
615
        pat_idents.visit_pat(&arg.pat);
J
John Clements 已提交
616 617 618 619
    }
    pat_idents.ident_accumulator
}

J
John Clements 已提交
620
// expand a block. pushes a new exts_frame, then calls expand_block_elts
621
pub fn expand_block(blk: P<Block>, fld: &mut MacroExpander) -> P<Block> {
622
    // see note below about treatment of exts table
623
    with_exts_frame!(fld.cx.syntax_env,false,
624
                     expand_block_elts(blk, fld))
625 626
}

J
John Clements 已提交
627
// expand the elements of a block.
628
pub fn expand_block_elts(b: P<Block>, fld: &mut MacroExpander) -> P<Block> {
629
    b.map(|Block {id, stmts, expr, rules, span}| {
A
Aaron Turon 已提交
630
        let new_stmts = stmts.into_iter().flat_map(|x| {
631 632
            // perform pending renames and expand macros in the statement
            fld.fold_stmt(x).into_iter()
633
        }).collect();
634 635 636 637 638 639 640 641 642 643 644 645 646 647 648
        let new_expr = expr.map(|x| {
            let expr = {
                let pending_renames = &mut fld.cx.syntax_env.info().pending_renames;
                let mut rename_fld = IdentRenamer{renames:pending_renames};
                rename_fld.fold_expr(x)
            };
            fld.fold_expr(expr)
        });
        Block {
            id: fld.new_id(id),
            stmts: new_stmts,
            expr: new_expr,
            rules: rules,
            span: span
        }
649
    })
J
John Clements 已提交
650 651
}

652 653
fn expand_pat(p: P<ast::Pat>, fld: &mut MacroExpander) -> P<ast::Pat> {
    match p.node {
654
        PatKind::Mac(_) => {}
655
        _ => return noop_fold_pat(p, fld)
K
Keegan McAllister 已提交
656
    }
657 658
    p.and_then(|ast::Pat {node, span, ..}| {
        match node {
659
            PatKind::Mac(mac) => expand_mac_invoc(mac, None, Vec::new(), span, fld),
660 661 662
            _ => unreachable!()
        }
    })
K
Keegan McAllister 已提交
663 664
}

J
John Clements 已提交
665 666 667
/// A tree-folder that applies every rename in its (mutable) list
/// to every identifier, including both bindings and varrefs
/// (and lots of things that will turn out to be neither)
668
pub struct IdentRenamer<'a> {
669
    renames: &'a mtwt::RenameList,
670 671
}

672
impl<'a> Folder for IdentRenamer<'a> {
673
    fn fold_ident(&mut self, id: Ident) -> Ident {
674
        Ident::new(id.name, mtwt::apply_renames(self.renames, id.ctxt))
675
    }
K
Keegan McAllister 已提交
676 677
    fn fold_mac(&mut self, mac: ast::Mac) -> ast::Mac {
        fold::noop_fold_mac(mac, self)
J
John Clements 已提交
678
    }
679 680
}

J
John Clements 已提交
681
/// A tree-folder that applies every rename in its list to
682
/// the idents that are in PatKind::Ident patterns. This is more narrowly
J
John Clements 已提交
683 684 685 686 687 688 689
/// focused than IdentRenamer, and is needed for FnDecl,
/// where we want to rename the args but not the fn name or the generics etc.
pub struct PatIdentRenamer<'a> {
    renames: &'a mtwt::RenameList,
}

impl<'a> Folder for PatIdentRenamer<'a> {
690
    fn fold_pat(&mut self, pat: P<ast::Pat>) -> P<ast::Pat> {
J
John Clements 已提交
691
        match pat.node {
692
            PatKind::Ident(..) => {},
693 694 695 696
            _ => return noop_fold_pat(pat, self)
        }

        pat.map(|ast::Pat {id, node, span}| match node {
697
            PatKind::Ident(binding_mode, Spanned{span: sp, node: ident}, sub) => {
698 699
                let new_ident = Ident::new(ident.name,
                                           mtwt::apply_renames(self.renames, ident.ctxt));
J
John Clements 已提交
700
                let new_node =
701
                    PatKind::Ident(binding_mode,
702
                                  Spanned{span: self.new_span(sp), node: new_ident},
J
John Clements 已提交
703
                                  sub.map(|p| self.fold_pat(p)));
704 705
                ast::Pat {
                    id: id,
J
John Clements 已提交
706
                    node: new_node,
707
                    span: self.new_span(span)
J
John Clements 已提交
708 709
                }
            },
710 711
            _ => unreachable!()
        })
712
    }
K
Keegan McAllister 已提交
713 714
    fn fold_mac(&mut self, mac: ast::Mac) -> ast::Mac {
        fold::noop_fold_mac(mac, self)
J
John Clements 已提交
715
    }
K
Kevin Atkinson 已提交
716 717
}

718 719 720 721 722 723 724
fn expand_annotatable(a: Annotatable,
                      fld: &mut MacroExpander)
                      -> SmallVector<Annotatable> {
    let a = expand_item_multi_modifier(a, fld);

    let mut decorator_items = SmallVector::zero();
    let mut new_attrs = Vec::new();
725
    expand_decorators(a.clone(), fld, &mut decorator_items, &mut new_attrs);
726 727 728

    let mut new_items: SmallVector<Annotatable> = match a {
        Annotatable::Item(it) => match it.node {
729
            ast::ItemKind::Mac(..) => {
730 731 732 733 734 735 736
                let new_items: SmallVector<P<ast::Item>> = it.and_then(|it| match it.node {
                    ItemKind::Mac(mac) =>
                        expand_mac_invoc(mac, Some(it.ident), it.attrs, it.span, fld),
                    _ => unreachable!(),
                });

                new_items.into_iter().map(|i| Annotatable::Item(i)).collect()
737
            }
738
            ast::ItemKind::Mod(_) | ast::ItemKind::ForeignMod(_) => {
739
                let valid_ident =
740
                    it.ident.name != keywords::Invalid.name();
741 742 743 744

                if valid_ident {
                    fld.cx.mod_push(it.ident);
                }
745
                let macro_use = contains_macro_use(fld, &new_attrs[..]);
746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761
                let result = with_exts_frame!(fld.cx.syntax_env,
                                              macro_use,
                                              noop_fold_item(it, fld));
                if valid_ident {
                    fld.cx.mod_pop();
                }
                result.into_iter().map(|i| Annotatable::Item(i)).collect()
            },
            _ => {
                let it = P(ast::Item {
                    attrs: new_attrs,
                    ..(*it).clone()
                });
                noop_fold_item(it, fld).into_iter().map(|i| Annotatable::Item(i)).collect()
            }
        },
762

763
        Annotatable::TraitItem(it) => match it.node {
764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781
            ast::TraitItemKind::Method(_, Some(_)) => {
                let ti = it.unwrap();
                SmallVector::one(ast::TraitItem {
                    id: ti.id,
                    ident: ti.ident,
                    attrs: ti.attrs,
                    node: match ti.node  {
                        ast::TraitItemKind::Method(sig, Some(body)) => {
                            let (sig, body) = expand_and_rename_method(sig, body, fld);
                            ast::TraitItemKind::Method(sig, Some(body))
                        }
                        _ => unreachable!()
                    },
                    span: fld.new_span(ti.span)
                })
            }
            _ => fold::noop_fold_trait_item(it.unwrap(), fld)
        }.into_iter().map(|ti| Annotatable::TraitItem(P(ti))).collect(),
782

783
        Annotatable::ImplItem(ii) => {
784 785
            expand_impl_item(ii.unwrap(), fld).into_iter().
                map(|ii| Annotatable::ImplItem(P(ii))).collect()
786 787 788
        }
    };

789
    new_items.push_all(decorator_items);
790 791 792
    new_items
}

793 794 795 796 797 798 799 800
// Partition a set of attributes into one kind of attribute, and other kinds.
macro_rules! partition {
    ($fn_name: ident, $variant: ident) => {
        #[allow(deprecated)] // The `allow` is needed because the `Modifier` variant might be used.
        fn $fn_name(attrs: &[ast::Attribute],
                    fld: &MacroExpander)
                     -> (Vec<ast::Attribute>, Vec<ast::Attribute>) {
            attrs.iter().cloned().partition(|attr| {
801
                match fld.cx.syntax_env.find(intern(&attr.name())) {
802 803 804 805 806 807 808
                    Some(rc) => match *rc {
                        $variant(..) => true,
                        _ => false
                    },
                    _ => false
                }
            })
809
        }
810
    }
811 812
}

813 814 815 816 817 818 819 820 821
partition!(multi_modifiers, MultiModifier);


fn expand_decorators(a: Annotatable,
                     fld: &mut MacroExpander,
                     decorator_items: &mut SmallVector<Annotatable>,
                     new_attrs: &mut Vec<ast::Attribute>)
{
    for attr in a.attrs() {
M
Manish Goregaokar 已提交
822
        let mname = intern(&attr.name());
823
        match fld.cx.syntax_env.find(mname) {
824
            Some(rc) => match *rc {
825 826 827 828 829 830
                MultiDecorator(ref dec) => {
                    attr::mark_used(&attr);

                    fld.cx.bt_push(ExpnInfo {
                        call_site: attr.span,
                        callee: NameAndSpan {
M
Manish Goregaokar 已提交
831
                            format: MacroAttribute(mname),
832 833 834 835 836 837 838 839 840 841 842 843 844
                            span: Some(attr.span),
                            // attributes can do whatever they like,
                            // for now.
                            allow_internal_unstable: true,
                        }
                    });

                    // we'd ideally decorator_items.push_all(expand_annotatable(ann, fld)),
                    // but that double-mut-borrows fld
                    let mut items: SmallVector<Annotatable> = SmallVector::zero();
                    dec.expand(fld.cx,
                               attr.span,
                               &attr.node.value,
845
                               &a,
846 847 848 849 850 851 852
                               &mut |ann| items.push(ann));
                    decorator_items.extend(items.into_iter()
                        .flat_map(|ann| expand_annotatable(ann, fld).into_iter()));

                    fld.cx.bt_pop();
                }
                _ => new_attrs.push((*attr).clone()),
853
            },
854
            _ => new_attrs.push((*attr).clone()),
855
        }
856
    }
857 858 859 860 861 862 863 864 865 866 867 868 869 870
}

fn expand_item_multi_modifier(mut it: Annotatable,
                              fld: &mut MacroExpander)
                              -> Annotatable {
    let (modifiers, other_attrs) = multi_modifiers(it.attrs(), fld);

    // Update the attrs, leave everything else alone. Is this mutation really a good idea?
    it = it.fold_attrs(other_attrs);

    if modifiers.is_empty() {
        return it
    }

871
    for attr in &modifiers {
M
Manish Goregaokar 已提交
872
        let mname = intern(&attr.name());
873

874
        match fld.cx.syntax_env.find(mname) {
875 876 877 878 879 880
            Some(rc) => match *rc {
                MultiModifier(ref mac) => {
                    attr::mark_used(attr);
                    fld.cx.bt_push(ExpnInfo {
                        call_site: attr.span,
                        callee: NameAndSpan {
M
Manish Goregaokar 已提交
881
                            format: MacroAttribute(mname),
882
                            span: Some(attr.span),
883 884 885
                            // attributes can do whatever they like,
                            // for now
                            allow_internal_unstable: true,
886 887
                        }
                    });
888
                    it = mac.expand(fld.cx, attr.span, &attr.node.value, it);
889 890 891 892 893 894 895 896
                    fld.cx.bt_pop();
                }
                _ => unreachable!()
            },
            _ => unreachable!()
        }
    }

897
    // Expansion may have added new ItemKind::Modifiers.
898 899 900
    expand_item_multi_modifier(it, fld)
}

901 902
fn expand_impl_item(ii: ast::ImplItem, fld: &mut MacroExpander)
                 -> SmallVector<ast::ImplItem> {
903
    match ii.node {
904
        ast::ImplItemKind::Method(..) => SmallVector::one(ast::ImplItem {
905 906 907 908
            id: ii.id,
            ident: ii.ident,
            attrs: ii.attrs,
            vis: ii.vis,
909
            defaultness: ii.defaultness,
L
Leo Testard 已提交
910
            node: match ii.node {
911
                ast::ImplItemKind::Method(sig, body) => {
912
                    let (sig, body) = expand_and_rename_method(sig, body, fld);
913
                    ast::ImplItemKind::Method(sig, body)
914 915 916 917
                }
                _ => unreachable!()
            },
            span: fld.new_span(ii.span)
918
        }),
L
Leo Testard 已提交
919
        ast::ImplItemKind::Macro(mac) => {
920
            expand_mac_invoc(mac, None, ii.attrs, ii.span, fld)
921
        }
922
        _ => fold::noop_fold_impl_item(ii, fld)
923
    }
924 925
}

J
John Clements 已提交
926 927 928
/// Given a fn_decl and a block and a MacroExpander, expand the fn_decl, then use the
/// PatIdents in its arguments to perform renaming in the FnDecl and
/// the block, returning both the new FnDecl and the new Block.
929
fn expand_and_rename_fn_decl_and_block(fn_decl: P<ast::FnDecl>, block: P<ast::Block>,
J
John Clements 已提交
930
                                       fld: &mut MacroExpander)
931
                                       -> (P<ast::FnDecl>, P<ast::Block>) {
J
John Clements 已提交
932
    let expanded_decl = fld.fold_fn_decl(fn_decl);
S
Seo Sanghyeon 已提交
933
    let idents = fn_decl_arg_bindings(&expanded_decl);
J
John Clements 已提交
934
    let renames =
935
        idents.iter().map(|id| (*id,fresh_name(*id))).collect();
J
John Clements 已提交
936 937
    // first, a renamer for the PatIdents, for the fn_decl:
    let mut rename_pat_fld = PatIdentRenamer{renames: &renames};
938
    let rewritten_fn_decl = rename_pat_fld.fold_fn_decl(expanded_decl);
J
John Clements 已提交
939 940 941 942 943 944
    // now, a renamer for *all* idents, for the body:
    let mut rename_fld = IdentRenamer{renames: &renames};
    let rewritten_body = fld.fold_block(rename_fld.fold_block(block));
    (rewritten_fn_decl,rewritten_body)
}

945 946 947 948 949 950 951 952 953 954
fn expand_and_rename_method(sig: ast::MethodSig, body: P<ast::Block>,
                            fld: &mut MacroExpander)
                            -> (ast::MethodSig, P<ast::Block>) {
    let (rewritten_fn_decl, rewritten_body)
        = expand_and_rename_fn_decl_and_block(sig.decl, body, fld);
    (ast::MethodSig {
        generics: fld.fold_generics(sig.generics),
        abi: sig.abi,
        explicit_self: fld.fold_explicit_self(sig.explicit_self),
        unsafety: sig.unsafety,
N
Niko Matsakis 已提交
955
        constness: sig.constness,
956 957 958 959
        decl: rewritten_fn_decl
    }, rewritten_body)
}

960 961
pub fn expand_type(t: P<ast::Ty>, fld: &mut MacroExpander) -> P<ast::Ty> {
    let t = match t.node.clone() {
962
        ast::TyKind::Mac(mac) => {
J
Jared Roesch 已提交
963
            if fld.cx.ecfg.features.unwrap().type_macros {
964
                expand_mac_invoc(mac, None, Vec::new(), t.span, fld)
J
Jared Roesch 已提交
965 966 967 968 969
            } else {
                feature_gate::emit_feature_err(
                    &fld.cx.parse_sess.span_diagnostic,
                    "type_macros",
                    t.span,
970 971
                    feature_gate::GateIssue::Language,
                    "type macros are experimental");
J
Jared Roesch 已提交
972 973

                DummyResult::raw_ty(t.span)
J
Jared Roesch 已提交
974
            }
975 976 977
        }
        _ => t
    };
J
Jared Roesch 已提交
978

979 980 981
    fold::noop_fold_ty(t, fld)
}

J
John Clements 已提交
982
/// A tree-folder that performs macro expansion
983
pub struct MacroExpander<'a, 'b:'a> {
984
    pub cx: &'a mut ExtCtxt<'b>,
N
Nick Cameron 已提交
985 986 987 988
}

impl<'a, 'b> MacroExpander<'a, 'b> {
    pub fn new(cx: &'a mut ExtCtxt<'b>) -> MacroExpander<'a, 'b> {
989
        MacroExpander { cx: cx }
N
Nick Cameron 已提交
990
    }
991 992
}

E
Eduard Burtescu 已提交
993
impl<'a, 'b> Folder for MacroExpander<'a, 'b> {
994 995 996 997 998
    fn fold_crate(&mut self, c: Crate) -> Crate {
        self.cx.filename = Some(self.cx.parse_sess.codemap().span_to_filename(c.span));
        noop_fold_crate(c, self)
    }

999
    fn fold_expr(&mut self, expr: P<ast::Expr>) -> P<ast::Expr> {
1000
        expand_expr(expr, self)
1001 1002
    }

1003
    fn fold_pat(&mut self, pat: P<ast::Pat>) -> P<ast::Pat> {
K
Keegan McAllister 已提交
1004 1005 1006
        expand_pat(pat, self)
    }

1007
    fn fold_item(&mut self, item: P<ast::Item>) -> SmallVector<P<ast::Item>> {
1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028
        use std::mem::replace;
        let result;
        if let ast::ItemKind::Mod(ast::Mod { inner, .. }) = item.node {
            if item.span.contains(inner) {
                self.push_mod_path(item.ident, &item.attrs);
                result = expand_item(item, self);
                self.pop_mod_path();
            } else {
                let filename = if inner != codemap::DUMMY_SP {
                    Some(self.cx.parse_sess.codemap().span_to_filename(inner))
                } else { None };
                let orig_filename = replace(&mut self.cx.filename, filename);
                let orig_mod_path_stack = replace(&mut self.cx.mod_path_stack, Vec::new());
                result = expand_item(item, self);
                self.cx.filename = orig_filename;
                self.cx.mod_path_stack = orig_mod_path_stack;
            }
        } else {
            result = expand_item(item, self);
        }
        result
1029 1030
    }

1031 1032
    fn fold_item_kind(&mut self, item: ast::ItemKind) -> ast::ItemKind {
        expand_item_kind(item, self)
J
John Clements 已提交
1033 1034
    }

1035
    fn fold_stmt(&mut self, stmt: ast::Stmt) -> SmallVector<ast::Stmt> {
1036
        expand_stmt(stmt, self)
1037 1038
    }

S
Steven Fackler 已提交
1039
    fn fold_block(&mut self, block: P<Block>) -> P<Block> {
1040 1041 1042 1043
        let was_in_block = ::std::mem::replace(&mut self.cx.in_block, true);
        let result = expand_block(block, self);
        self.cx.in_block = was_in_block;
        result
1044 1045
    }

1046
    fn fold_arm(&mut self, arm: ast::Arm) -> ast::Arm {
J
John Clements 已提交
1047 1048 1049
        expand_arm(arm, self)
    }

1050 1051
    fn fold_trait_item(&mut self, i: ast::TraitItem) -> SmallVector<ast::TraitItem> {
        expand_annotatable(Annotatable::TraitItem(P(i)), self)
1052
            .into_iter().map(|i| i.expect_trait_item()).collect()
1053 1054
    }

1055 1056
    fn fold_impl_item(&mut self, i: ast::ImplItem) -> SmallVector<ast::ImplItem> {
        expand_annotatable(Annotatable::ImplItem(P(i)), self)
1057
            .into_iter().map(|i| i.expect_impl_item()).collect()
1058 1059
    }

1060 1061 1062 1063
    fn fold_ty(&mut self, ty: P<ast::Ty>) -> P<ast::Ty> {
        expand_type(ty, self)
    }

S
Steven Fackler 已提交
1064
    fn new_span(&mut self, span: Span) -> Span {
1065 1066
        new_span(self.cx, span)
    }
1067 1068
}

1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083
impl<'a, 'b> MacroExpander<'a, 'b> {
    fn push_mod_path(&mut self, id: Ident, attrs: &[ast::Attribute]) {
        let default_path = id.name.as_str();
        let file_path = match ::attr::first_attr_value_str_by_name(attrs, "path") {
            Some(d) => d,
            None => default_path,
        };
        self.cx.mod_path_stack.push(file_path)
    }

    fn pop_mod_path(&mut self) {
        self.cx.mod_path_stack.pop().unwrap();
    }
}

J
John Clements 已提交
1084
fn new_span(cx: &ExtCtxt, sp: Span) -> Span {
1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119
    debug!("new_span(sp={:?})", sp);

    if cx.codemap().more_specific_trace(sp.expn_id, cx.backtrace()) {
        // If the span we are looking at has a backtrace that has more
        // detail than our current backtrace, then we keep that
        // backtrace.  Honestly, I have no idea if this makes sense,
        // because I have no idea why we are stripping the backtrace
        // below. But the reason I made this change is because, in
        // deriving, we were generating attributes with a specific
        // backtrace, which was essential for `#[structural_match]` to
        // be properly supported, but these backtraces were being
        // stripped and replaced with a null backtrace. Sort of
        // unclear why this is the case. --nmatsakis
        debug!("new_span: keeping trace from {:?} because it is more specific",
               sp.expn_id);
        sp
    } else {
        // This discards information in the case of macro-defining macros.
        //
        // The comment above was originally added in
        // b7ec2488ff2f29681fe28691d20fd2c260a9e454 in Feb 2012. I
        // *THINK* the reason we are doing this is because we want to
        // replace the backtrace of the macro contents with the
        // backtrace that contains the macro use. But it's pretty
        // unclear to me. --nmatsakis
        let sp1 = Span {
            lo: sp.lo,
            hi: sp.hi,
            expn_id: cx.backtrace(),
        };
        debug!("new_span({:?}) = {:?}", sp, sp1);
        if sp.expn_id.into_u32() == 0 && env::var_os("NDM").is_some() {
            panic!("NDM");
        }
        sp1
J
John Clements 已提交
1120 1121 1122
    }
}

1123
pub struct ExpansionConfig<'feat> {
1124
    pub crate_name: String,
1125
    pub features: Option<&'feat Features>,
P
Paul Collier 已提交
1126
    pub recursion_limit: usize,
1127
    pub trace_mac: bool,
1128 1129
}

1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142
macro_rules! feature_tests {
    ($( fn $getter:ident = $field:ident, )*) => {
        $(
            pub fn $getter(&self) -> bool {
                match self.features {
                    Some(&Features { $field: true, .. }) => true,
                    _ => false,
                }
            }
        )*
    }
}

1143 1144
impl<'feat> ExpansionConfig<'feat> {
    pub fn default(crate_name: String) -> ExpansionConfig<'static> {
1145 1146
        ExpansionConfig {
            crate_name: crate_name,
1147
            features: None,
1148
            recursion_limit: 64,
1149
            trace_mac: false,
1150 1151
        }
    }
1152

1153
    feature_tests! {
1154 1155 1156 1157 1158
        fn enable_quotes = quote,
        fn enable_asm = asm,
        fn enable_log_syntax = log_syntax,
        fn enable_concat_idents = concat_idents,
        fn enable_trace_macros = trace_macros,
1159
        fn enable_allow_internal_unstable = allow_internal_unstable,
1160 1161
        fn enable_custom_derive = custom_derive,
        fn enable_pushpop_unsafe = pushpop_unsafe,
1162
    }
1163 1164
}

1165 1166 1167 1168 1169
pub fn expand_crate(mut cx: ExtCtxt,
                    // these are the macros being imported to this crate:
                    imported_macros: Vec<ast::MacroDef>,
                    user_exts: Vec<NamedSyntaxExtension>,
                    c: Crate) -> (Crate, HashSet<Name>) {
A
Alex Crichton 已提交
1170 1171 1172 1173 1174 1175 1176
    if std_inject::no_core(&c) {
        cx.crate_root = None;
    } else if std_inject::no_std(&c) {
        cx.crate_root = Some("core");
    } else {
        cx.crate_root = Some("std");
    }
1177 1178
    let ret = {
        let mut expander = MacroExpander::new(&mut cx);
1179

1180 1181 1182
        for def in imported_macros {
            expander.cx.insert_macro(def);
        }
1183

1184 1185 1186
        for (name, extension) in user_exts {
            expander.cx.syntax_env.insert(name, extension);
        }
1187

N
Nick Cameron 已提交
1188
        let err_count = cx.parse_sess.span_diagnostic.err_count();
1189 1190
        let mut ret = expander.fold_crate(c);
        ret.exported_macros = expander.cx.exported_macros.clone();
N
Nick Cameron 已提交
1191 1192 1193 1194 1195

        if cx.parse_sess.span_diagnostic.err_count() > err_count {
            cx.parse_sess.span_diagnostic.abort_if_errors();
        }

1196 1197 1198
        ret
    };
    return (ret, cx.syntax_env.names);
1199
}
J
John Clements 已提交
1200

1201 1202 1203 1204 1205 1206 1207 1208
// HYGIENIC CONTEXT EXTENSION:
// all of these functions are for walking over
// ASTs and making some change to the context of every
// element that has one. a CtxtFn is a trait-ified
// version of a closure in (SyntaxContext -> SyntaxContext).
// the ones defined here include:
// Marker - add a mark to a context

E
Eduard Burtescu 已提交
1209 1210
// A Marker adds the given mark to the syntax context
struct Marker { mark: Mrk }
1211

E
Eduard Burtescu 已提交
1212
impl Folder for Marker {
1213
    fn fold_ident(&mut self, id: Ident) -> Ident {
1214
        ast::Ident::new(id.name, mtwt::apply_mark(self.mark, id.ctxt))
1215
    }
1216
    fn fold_mac(&mut self, Spanned {node, span}: ast::Mac) -> ast::Mac {
1217
        Spanned {
1218 1219 1220 1221
            node: Mac_ {
                path: self.fold_path(node.path),
                tts: self.fold_tts(&node.tts),
                ctxt: mtwt::apply_mark(self.mark, node.ctxt),
1222 1223
            },
            span: span,
1224
        }
1225 1226 1227
    }
}

1228
// apply a given mark to the given token trees. Used prior to expansion of a macro.
1229
fn mark_tts(tts: &[TokenTree], m: Mrk) -> Vec<TokenTree> {
M
Marvin Löbel 已提交
1230
    noop_fold_tts(tts, &mut Marker{mark:m})
1231 1232
}

J
John Clements 已提交
1233 1234
/// Check that there are no macro invocations left in the AST:
pub fn check_for_macros(sess: &parse::ParseSess, krate: &ast::Crate) {
1235
    visit::walk_crate(&mut MacroExterminator{sess:sess}, krate);
J
John Clements 已提交
1236 1237 1238 1239 1240 1241 1242
}

/// A visitor that ensures that no macro invocations remain in an AST.
struct MacroExterminator<'a>{
    sess: &'a parse::ParseSess
}

1243
impl<'a, 'v> Visitor<'v> for MacroExterminator<'a> {
K
Keegan McAllister 已提交
1244 1245
    fn visit_mac(&mut self, mac: &ast::Mac) {
        self.sess.span_diagnostic.span_bug(mac.span,
J
John Clements 已提交
1246 1247 1248 1249 1250 1251
                                           "macro exterminator: expected AST \
                                           with no macro invocations");
    }
}


J
John Clements 已提交
1252
#[cfg(test)]
1253
mod tests {
A
Alex Crichton 已提交
1254
    use super::{pattern_bindings, expand_crate};
1255
    use super::{PatIdentFinder, IdentRenamer, PatIdentRenamer, ExpansionConfig};
1256
    use ast;
J
Jorge Aparicio 已提交
1257
    use ast::Name;
1258
    use codemap;
S
Seo Sanghyeon 已提交
1259
    use ext::base::ExtCtxt;
1260
    use ext::mtwt;
1261
    use fold::Folder;
1262
    use parse;
1263
    use parse::token::{self, keywords};
E
Eduard Burtescu 已提交
1264
    use util::parser_testing::{string_to_parser};
1265
    use util::parser_testing::{string_to_pat, string_to_crate, strs_to_idents};
1266
    use visit;
1267 1268 1269 1270 1271
    use visit::Visitor;

    // a visitor that extracts the paths
    // from a given thingy and puts them in a mutable
    // array (passed in to the traversal)
1272
    #[derive(Clone)]
1273
    struct PathExprFinderContext {
1274
        path_accumulator: Vec<ast::Path> ,
1275 1276
    }

1277
    impl<'v> Visitor<'v> for PathExprFinderContext {
1278
        fn visit_expr(&mut self, expr: &ast::Expr) {
1279
            if let ast::ExprKind::Path(None, ref p) = expr.node {
1280
                self.path_accumulator.push(p.clone());
1281
            }
1282
            visit::walk_expr(self, expr);
1283 1284 1285
        }
    }

1286 1287 1288
    // find the variable references in a crate
    fn crate_varrefs(the_crate : &ast::Crate) -> Vec<ast::Path> {
        let mut path_finder = PathExprFinderContext{path_accumulator:Vec::new()};
1289
        visit::walk_crate(&mut path_finder, the_crate);
1290
        path_finder.path_accumulator
1291
    }
J
John Clements 已提交
1292

1293 1294
    /// A Visitor that extracts the identifiers from a thingy.
    // as a side note, I'm starting to want to abstract over these....
1295
    struct IdentFinder {
1296 1297
        ident_accumulator: Vec<ast::Ident>
    }
1298

1299
    impl<'v> Visitor<'v> for IdentFinder {
1300
        fn visit_ident(&mut self, _: codemap::Span, id: ast::Ident){
1301 1302 1303 1304 1305 1306 1307
            self.ident_accumulator.push(id);
        }
    }

    /// Find the idents in a crate
    fn crate_idents(the_crate: &ast::Crate) -> Vec<ast::Ident> {
        let mut ident_finder = IdentFinder{ident_accumulator: Vec::new()};
1308
        visit::walk_crate(&mut ident_finder, the_crate);
1309 1310
        ident_finder.ident_accumulator
    }
1311

J
John Clements 已提交
1312 1313 1314
    // these following tests are quite fragile, in that they don't test what
    // *kind* of failure occurs.

1315
    fn test_ecfg() -> ExpansionConfig<'static> {
1316 1317 1318
        ExpansionConfig::default("test".to_string())
    }

J
John Clements 已提交
1319
    // make sure that macros can't escape fns
1320
    #[should_panic]
J
John Clements 已提交
1321
    #[test] fn macros_cant_escape_fns_test () {
1322
        let src = "fn bogus() {macro_rules! z (() => (3+4));}\
1323
                   fn inty() -> i32 { z!() }".to_string();
1324
        let sess = parse::ParseSess::new();
J
John Clements 已提交
1325
        let crate_ast = parse::parse_crate_from_source_str(
1326
            "<test>".to_string(),
1327
            src,
1328
            Vec::new(), &sess).unwrap();
J
John Clements 已提交
1329
        // should fail:
S
Seo Sanghyeon 已提交
1330 1331 1332
        let mut gated_cfgs = vec![];
        let ecx = ExtCtxt::new(&sess, vec![], test_ecfg(), &mut gated_cfgs);
        expand_crate(ecx, vec![], vec![], crate_ast);
J
John Clements 已提交
1333 1334
    }

J
John Clements 已提交
1335
    // make sure that macros can't escape modules
1336
    #[should_panic]
J
John Clements 已提交
1337
    #[test] fn macros_cant_escape_mods_test () {
1338
        let src = "mod foo {macro_rules! z (() => (3+4));}\
1339
                   fn inty() -> i32 { z!() }".to_string();
1340
        let sess = parse::ParseSess::new();
J
John Clements 已提交
1341
        let crate_ast = parse::parse_crate_from_source_str(
1342
            "<test>".to_string(),
1343
            src,
1344
            Vec::new(), &sess).unwrap();
S
Seo Sanghyeon 已提交
1345 1346 1347
        let mut gated_cfgs = vec![];
        let ecx = ExtCtxt::new(&sess, vec![], test_ecfg(), &mut gated_cfgs);
        expand_crate(ecx, vec![], vec![], crate_ast);
J
John Clements 已提交
1348 1349
    }

1350
    // macro_use modules should allow macros to escape
J
John Clements 已提交
1351
    #[test] fn macros_can_escape_flattened_mods_test () {
1352
        let src = "#[macro_use] mod foo {macro_rules! z (() => (3+4));}\
1353
                   fn inty() -> i32 { z!() }".to_string();
1354
        let sess = parse::ParseSess::new();
J
John Clements 已提交
1355
        let crate_ast = parse::parse_crate_from_source_str(
1356
            "<test>".to_string(),
1357
            src,
1358
            Vec::new(), &sess).unwrap();
S
Seo Sanghyeon 已提交
1359 1360 1361
        let mut gated_cfgs = vec![];
        let ecx = ExtCtxt::new(&sess, vec![], test_ecfg(), &mut gated_cfgs);
        expand_crate(ecx, vec![], vec![], crate_ast);
J
John Clements 已提交
1362 1363
    }

1364
    fn expand_crate_str(crate_str: String) -> ast::Crate {
1365
        let ps = parse::ParseSess::new();
1366
        let crate_ast = panictry!(string_to_parser(&ps, crate_str).parse_crate_mod());
1367
        // the cfg argument actually does matter, here...
S
Seo Sanghyeon 已提交
1368 1369 1370
        let mut gated_cfgs = vec![];
        let ecx = ExtCtxt::new(&ps, vec![], test_ecfg(), &mut gated_cfgs);
        expand_crate(ecx, vec![], vec![], crate_ast).0
1371 1372
    }

1373 1374
    // find the pat_ident paths in a crate
    fn crate_bindings(the_crate : &ast::Crate) -> Vec<ast::Ident> {
1375
        let mut name_finder = PatIdentFinder{ident_accumulator:Vec::new()};
1376
        visit::walk_crate(&mut name_finder, the_crate);
1377 1378 1379
        name_finder.ident_accumulator
    }

1380
    #[test] fn macro_tokens_should_match(){
1381
        expand_crate_str(
1382
            "macro_rules! m((a)=>(13)) ;fn main(){m!(a);}".to_string());
1383 1384
    }

1385 1386 1387
    // should be able to use a bound identifier as a literal in a macro definition:
    #[test] fn self_macro_parsing(){
        expand_crate_str(
1388 1389
            "macro_rules! foo ((zz) => (287;));
            fn f(zz: i32) {foo!(zz);}".to_string()
1390 1391 1392
            );
    }

1393 1394 1395 1396 1397 1398 1399
    // renaming tests expand a crate and then check that the bindings match
    // the right varrefs. The specification of the test case includes the
    // text of the crate, and also an array of arrays.  Each element in the
    // outer array corresponds to a binding in the traversal of the AST
    // induced by visit.  Each of these arrays contains a list of indexes,
    // interpreted as the varrefs in the varref traversal that this binding
    // should match.  So, for instance, in a program with two bindings and
1400
    // three varrefs, the array [[1, 2], [0]] would indicate that the first
1401 1402 1403
    // binding should match the second two varrefs, and the second binding
    // should match the first varref.
    //
J
John Clements 已提交
1404 1405 1406 1407 1408 1409
    // Put differently; this is a sparse representation of a boolean matrix
    // indicating which bindings capture which identifiers.
    //
    // Note also that this matrix is dependent on the implicit ordering of
    // the bindings and the varrefs discovered by the name-finder and the path-finder.
    //
1410 1411
    // The comparisons are done post-mtwt-resolve, so we're comparing renamed
    // names; differences in marks don't matter any more.
J
John Clements 已提交
1412
    //
1413
    // oog... I also want tests that check "bound-identifier-=?". That is,
J
John Clements 已提交
1414 1415 1416 1417 1418
    // not just "do these have the same name", but "do they have the same
    // name *and* the same marks"? Understanding this is really pretty painful.
    // in principle, you might want to control this boolean on a per-varref basis,
    // but that would make things even harder to understand, and might not be
    // necessary for thorough testing.
P
Paul Collier 已提交
1419
    type RenamingTest = (&'static str, Vec<Vec<usize>>, bool);
1420

1421 1422
    #[test]
    fn automatic_renaming () {
1423 1424
        let tests: Vec<RenamingTest> =
            vec!(// b & c should get new names throughout, in the expr too:
1425
                ("fn a() -> i32 { let b = 13; let c = b; b+c }",
1426
                 vec!(vec!(0,1),vec!(2)), false),
J
John Clements 已提交
1427
                // both x's should be renamed (how is this causing a bug?)
1428
                ("fn main () {let x: i32 = 13;x;}",
1429
                 vec!(vec!(0)), false),
1430
                // the use of b after the + should be renamed, the other one not:
1431
                ("macro_rules! f (($x:ident) => (b + $x)); fn a() -> i32 { let b = 13; f!(b)}",
1432
                 vec!(vec!(1)), false),
J
John Clements 已提交
1433
                // the b before the plus should not be renamed (requires marks)
1434
                ("macro_rules! f (($x:ident) => ({let b=9; ($x + b)})); fn a() -> i32 { f!(b)}",
1435
                 vec!(vec!(1)), false),
1436 1437 1438
                // the marks going in and out of letty should cancel, allowing that $x to
                // capture the one following the semicolon.
                // this was an awesome test case, and caught a *lot* of bugs.
1439 1440
                ("macro_rules! letty(($x:ident) => (let $x = 15;));
                  macro_rules! user(($x:ident) => ({letty!($x); $x}));
1441
                  fn main() -> i32 {user!(z)}",
1442 1443
                 vec!(vec!(0)), false)
                );
1444 1445
        for (idx,s) in tests.iter().enumerate() {
            run_renaming_test(s,idx);
1446 1447 1448
        }
    }

1449 1450 1451 1452 1453
    // no longer a fixme #8062: this test exposes a *potential* bug; our system does
    // not behave exactly like MTWT, but a conversation with Matthew Flatt
    // suggests that this can only occur in the presence of local-expand, which
    // we have no plans to support. ... unless it's needed for item hygiene....
    #[ignore]
R
Richo Healey 已提交
1454 1455
    #[test]
    fn issue_8062(){
1456 1457 1458 1459 1460 1461 1462 1463 1464 1465
        run_renaming_test(
            &("fn main() {let hrcoo = 19; macro_rules! getx(()=>(hrcoo)); getx!();}",
              vec!(vec!(0)), true), 0)
    }

    // FIXME #6994:
    // the z flows into and out of two macros (g & f) along one path, and one
    // (just g) along the other, so the result of the whole thing should
    // be "let z_123 = 3; z_123"
    #[ignore]
R
Richo Healey 已提交
1466 1467
    #[test]
    fn issue_6994(){
1468 1469
        run_renaming_test(
            &("macro_rules! g (($x:ident) =>
1470
              ({macro_rules! f(($y:ident)=>({let $y=3;$x}));f!($x)}));
1471 1472 1473
              fn a(){g!(z)}",
              vec!(vec!(0)),false),
            0)
1474 1475
    }

1476
    // match variable hygiene. Should expand into
1477
    // fn z() {match 8 {x_1 => {match 9 {x_2 | x_2 if x_2 == x_1 => x_2 + x_1}}}}
R
Richo Healey 已提交
1478 1479
    #[test]
    fn issue_9384(){
1480
        run_renaming_test(
1481
            &("macro_rules! bad_macro (($ex:expr) => ({match 9 {x | x if x == $ex => x + $ex}}));
1482
              fn z() {match 8 {x => bad_macro!(x)}}",
1483
              // NB: the third "binding" is the repeat of the second one.
1484
              vec!(vec!(1,3),vec!(0,2),vec!(0,2)),
1485 1486
              true),
            0)
1487 1488
    }

1489
    // interpolated nodes weren't getting labeled.
1490 1491
    // should expand into
    // fn main(){let g1_1 = 13; g1_1}}
R
Richo Healey 已提交
1492 1493
    #[test]
    fn pat_expand_issue_15221(){
1494
        run_renaming_test(
1495 1496
            &("macro_rules! inner ( ($e:pat ) => ($e));
              macro_rules! outer ( ($e:pat ) => (inner!($e)));
1497 1498 1499 1500 1501 1502
              fn main() { let outer!(g) = 13; g;}",
              vec!(vec!(0)),
              true),
            0)
    }

1503
    // create a really evil test case where a $x appears inside a binding of $x
J
Joseph Crail 已提交
1504
    // but *shouldn't* bind because it was inserted by a different macro....
1505 1506
    // can't write this test case until we have macro-generating macros.

1507
    // method arg hygiene
1508
    // method expands to fn get_x(&self_0, x_1: i32) {self_0 + self_2 + x_3 + x_1}
R
Richo Healey 已提交
1509 1510
    #[test]
    fn method_arg_hygiene(){
1511
        run_renaming_test(
1512 1513
            &("macro_rules! inject_x (()=>(x));
              macro_rules! inject_self (()=>(self));
1514
              struct A;
1515
              impl A{fn get_x(&self, x: i32) {self + inject_self!() + inject_x!() + x;} }",
1516 1517 1518 1519 1520
              vec!(vec!(0),vec!(3)),
              true),
            0)
    }

J
John Clements 已提交
1521 1522
    // ooh, got another bite?
    // expands to struct A; impl A {fn thingy(&self_1) {self_1;}}
R
Richo Healey 已提交
1523 1524
    #[test]
    fn method_arg_hygiene_2(){
J
John Clements 已提交
1525 1526 1527
        run_renaming_test(
            &("struct A;
              macro_rules! add_method (($T:ty) =>
1528 1529
              (impl $T {  fn thingy(&self) {self;} }));
              add_method!(A);",
J
John Clements 已提交
1530 1531 1532 1533 1534
              vec!(vec!(0)),
              true),
            0)
    }

J
John Clements 已提交
1535
    // item fn hygiene
1536
    // expands to fn q(x_1: i32){fn g(x_2: i32){x_2 + x_1};}
R
Richo Healey 已提交
1537 1538
    #[test]
    fn issue_9383(){
1539
        run_renaming_test(
1540 1541
            &("macro_rules! bad_macro (($ex:expr) => (fn g(x: i32){ x + $ex }));
              fn q(x: i32) { bad_macro!(x); }",
1542
              vec!(vec!(1),vec!(0)),true),
1543
            0)
1544
    }
1545

1546
    // closure arg hygiene (ExprKind::Closure)
1547
    // expands to fn f(){(|x_1 : i32| {(x_2 + x_1)})(3);}
R
Richo Healey 已提交
1548 1549
    #[test]
    fn closure_arg_hygiene(){
1550
        run_renaming_test(
1551
            &("macro_rules! inject_x (()=>(x));
1552
            fn f(){(|x : i32| {(inject_x!() + x)})(3);}",
1553 1554 1555 1556 1557
              vec!(vec!(1)),
              true),
            0)
    }

1558
    // macro_rules in method position. Sadly, unimplemented.
R
Richo Healey 已提交
1559 1560
    #[test]
    fn macro_in_method_posn(){
1561
        expand_crate_str(
1562
            "macro_rules! my_method (() => (fn thirteen(&self) -> i32 {13}));
1563
            struct A;
1564
            impl A{ my_method!(); }
1565 1566 1567
            fn f(){A.thirteen;}".to_string());
    }

1568 1569
    // another nested macro
    // expands to impl Entries {fn size_hint(&self_1) {self_1;}
R
Richo Healey 已提交
1570 1571
    #[test]
    fn item_macro_workaround(){
1572 1573 1574 1575
        run_renaming_test(
            &("macro_rules! item { ($i:item) => {$i}}
              struct Entries;
              macro_rules! iterator_impl {
1576
              () => { item!( impl Entries { fn size_hint(&self) { self;}});}}
1577 1578 1579 1580 1581
              iterator_impl! { }",
              vec!(vec!(0)), true),
            0)
    }

J
John Clements 已提交
1582
    // run one of the renaming tests
P
Paul Collier 已提交
1583
    fn run_renaming_test(t: &RenamingTest, test_idx: usize) {
1584
        let invalid_name = keywords::Invalid.name();
J
John Clements 已提交
1585
        let (teststr, bound_connections, bound_ident_check) = match *t {
1586
            (ref str,ref conns, bic) => (str.to_string(), conns.clone(), bic)
1587
        };
1588
        let cr = expand_crate_str(teststr.to_string());
1589 1590
        let bindings = crate_bindings(&cr);
        let varrefs = crate_varrefs(&cr);
1591

1592 1593 1594
        // must be one check clause for each binding:
        assert_eq!(bindings.len(),bound_connections.len());
        for (binding_idx,shouldmatch) in bound_connections.iter().enumerate() {
1595 1596
            let binding_name = mtwt::resolve(bindings[binding_idx]);
            let binding_marks = mtwt::marksof(bindings[binding_idx].ctxt, invalid_name);
1597
            // shouldmatch can't name varrefs that don't exist:
1598
            assert!((shouldmatch.is_empty()) ||
1599 1600
                    (varrefs.len() > *shouldmatch.iter().max().unwrap()));
            for (idx,varref) in varrefs.iter().enumerate() {
1601
                let print_hygiene_debug_info = || {
J
John Clements 已提交
1602 1603 1604
                    // good lord, you can't make a path with 0 segments, can you?
                    let final_varref_ident = match varref.segments.last() {
                        Some(pathsegment) => pathsegment.identifier,
S
Steve Klabnik 已提交
1605
                        None => panic!("varref with 0 path segments?")
J
John Clements 已提交
1606 1607 1608 1609 1610
                    };
                    let varref_name = mtwt::resolve(final_varref_ident);
                    let varref_idents : Vec<ast::Ident>
                        = varref.segments.iter().map(|s| s.identifier)
                        .collect();
A
Alex Crichton 已提交
1611
                    println!("varref #{}: {:?}, resolves to {}",idx, varref_idents, varref_name);
1612
                    println!("varref's first segment's string: \"{}\"", final_varref_ident);
J
John Clements 已提交
1613
                    println!("binding #{}: {}, resolves to {}",
1614
                             binding_idx, bindings[binding_idx], binding_name);
J
John Clements 已提交
1615 1616
                    mtwt::with_sctable(|x| mtwt::display_sctable(x));
                };
1617 1618
                if shouldmatch.contains(&idx) {
                    // it should be a path of length 1, and it should
J
John Clements 已提交
1619
                    // be free-identifier=? or bound-identifier=? to the given binding
1620
                    assert_eq!(varref.segments.len(),1);
1621 1622
                    let varref_name = mtwt::resolve(varref.segments[0].identifier);
                    let varref_marks = mtwt::marksof(varref.segments[0]
1623 1624 1625
                                                           .identifier
                                                           .ctxt,
                                                     invalid_name);
1626
                    if !(varref_name==binding_name) {
1627
                        println!("uh oh, should match but doesn't:");
J
John Clements 已提交
1628
                        print_hygiene_debug_info();
1629 1630
                    }
                    assert_eq!(varref_name,binding_name);
1631
                    if bound_ident_check {
J
John Clements 已提交
1632 1633
                        // we're checking bound-identifier=?, and the marks
                        // should be the same, too:
J
John Clements 已提交
1634 1635
                        assert_eq!(varref_marks,binding_marks.clone());
                    }
1636
                } else {
1637
                    let varref_name = mtwt::resolve(varref.segments[0].identifier);
1638
                    let fail = (varref.segments.len() == 1)
1639
                        && (varref_name == binding_name);
1640
                    // temp debugging:
1641
                    if fail {
1642 1643 1644 1645
                        println!("failure on test {}",test_idx);
                        println!("text of test case: \"{}\"", teststr);
                        println!("");
                        println!("uh oh, matches but shouldn't:");
J
John Clements 已提交
1646
                        print_hygiene_debug_info();
1647 1648 1649 1650
                    }
                    assert!(!fail);
                }
            }
1651 1652
        }
    }
1653

R
Richo Healey 已提交
1654 1655
    #[test]
    fn fmt_in_macro_used_inside_module_macro() {
1656 1657 1658
        let crate_str = "macro_rules! fmt_wrap(($b:expr)=>($b.to_string()));
macro_rules! foo_module (() => (mod generated { fn a() { let xx = 147; fmt_wrap!(xx);}}));
foo_module!();
1659
".to_string();
J
John Clements 已提交
1660 1661
        let cr = expand_crate_str(crate_str);
        // find the xx binding
1662
        let bindings = crate_bindings(&cr);
1663
        let cxbinds: Vec<&ast::Ident> =
1664
            bindings.iter().filter(|b| b.name.as_str() == "xx").collect();
1665
        let cxbinds: &[&ast::Ident] = &cxbinds[..];
1666 1667
        let cxbind = match (cxbinds.len(), cxbinds.get(0)) {
            (1, Some(b)) => *b,
S
Steve Klabnik 已提交
1668
            _ => panic!("expected just one binding for ext_cx")
J
John Clements 已提交
1669
        };
1670
        let resolved_binding = mtwt::resolve(*cxbind);
1671
        let varrefs = crate_varrefs(&cr);
1672

J
John Clements 已提交
1673
        // the xx binding should bind all of the xx varrefs:
1674
        for (idx,v) in varrefs.iter().filter(|p| {
1675
            p.segments.len() == 1
1676
            && p.segments[0].identifier.name.as_str() == "xx"
1677
        }).enumerate() {
1678
            if mtwt::resolve(v.segments[0].identifier) != resolved_binding {
1679
                println!("uh oh, xx binding didn't match xx varref:");
L
Luqman Aden 已提交
1680 1681 1682
                println!("this is xx varref \\# {}", idx);
                println!("binding: {}", cxbind);
                println!("resolves to: {}", resolved_binding);
1683
                println!("varref: {}", v.segments[0].identifier);
L
Luqman Aden 已提交
1684
                println!("resolves to: {}",
1685
                         mtwt::resolve(v.segments[0].identifier));
1686
                mtwt::with_sctable(|x| mtwt::display_sctable(x));
J
John Clements 已提交
1687
            }
1688
            assert_eq!(mtwt::resolve(v.segments[0].identifier),
1689
                       resolved_binding);
J
John Clements 已提交
1690 1691 1692
        };
    }

J
John Clements 已提交
1693 1694
    #[test]
    fn pat_idents(){
1695
        let pat = string_to_pat(
1696
            "(a,Foo{x:c @ (b,9),y:Bar(4,d)})".to_string());
S
Seo Sanghyeon 已提交
1697
        let idents = pattern_bindings(&pat);
1698
        assert_eq!(idents, strs_to_idents(vec!("a","c","b","d")));
J
John Clements 已提交
1699
    }
J
John Clements 已提交
1700

1701 1702 1703
    // test the list of identifier patterns gathered by the visitor. Note that
    // 'None' is listed as an identifier pattern because we don't yet know that
    // it's the name of a 0-ary variant, and that 'i' appears twice in succession.
1704
    #[test]
1705
    fn crate_bindings_test(){
1706
        let the_crate = string_to_crate("fn main (a: i32) -> i32 {|b| {
1707
        match 34 {None => 3, Some(i) | i => j, Foo{k:z,l:y} => \"banana\"}} }".to_string());
1708 1709
        let idents = crate_bindings(&the_crate);
        assert_eq!(idents, strs_to_idents(vec!("a","b","None","i","i","z","y")));
1710 1711
    }

1712 1713 1714
    // test the IdentRenamer directly
    #[test]
    fn ident_renamer_test () {
1715
        let the_crate = string_to_crate("fn f(x: i32){let x = x; x}".to_string());
1716 1717
        let f_ident = token::str_to_ident("f");
        let x_ident = token::str_to_ident("x");
1718
        let int_ident = token::str_to_ident("i32");
C
Corey Richardson 已提交
1719
        let renames = vec!((x_ident,Name(16)));
1720 1721 1722 1723
        let mut renamer = IdentRenamer{renames: &renames};
        let renamed_crate = renamer.fold_crate(the_crate);
        let idents = crate_idents(&renamed_crate);
        let resolved : Vec<ast::Name> = idents.iter().map(|id| mtwt::resolve(*id)).collect();
1724
        assert_eq!(resolved, [f_ident.name,Name(16),int_ident.name,Name(16),Name(16),Name(16)]);
1725 1726 1727 1728 1729
    }

    // test the PatIdentRenamer; only PatIdents get renamed
    #[test]
    fn pat_ident_renamer_test () {
1730
        let the_crate = string_to_crate("fn f(x: i32){let x = x; x}".to_string());
1731 1732
        let f_ident = token::str_to_ident("f");
        let x_ident = token::str_to_ident("x");
1733
        let int_ident = token::str_to_ident("i32");
C
Corey Richardson 已提交
1734
        let renames = vec!((x_ident,Name(16)));
1735 1736 1737 1738 1739
        let mut renamer = PatIdentRenamer{renames: &renames};
        let renamed_crate = renamer.fold_crate(the_crate);
        let idents = crate_idents(&renamed_crate);
        let resolved : Vec<ast::Name> = idents.iter().map(|id| mtwt::resolve(*id)).collect();
        let x_name = x_ident.name;
1740
        assert_eq!(resolved, [f_ident.name,Name(16),int_ident.name,Name(16),x_name,x_name]);
1741
    }
J
John Clements 已提交
1742
}