2014-02-11 22:43:23 -08:00
|
|
|
// Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
|
2012-12-03 16:48:01 -08:00
|
|
|
// 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.
|
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
use ast::{Block, Crate, DeclLocal, ExprMac, PatMac};
|
2015-09-20 16:15:37 -04:00
|
|
|
use ast::{Local, Ident, Mac_};
|
2014-11-14 09:18:10 -08:00
|
|
|
use ast::{ItemMac, MacStmtWithSemicolon, Mrk, Stmt, StmtDecl, StmtMac};
|
|
|
|
use ast::{StmtExpr, StmtSemi};
|
2014-02-25 04:47:19 +08:00
|
|
|
use ast::TokenTree;
|
2012-12-23 17:41:37 -05:00
|
|
|
use ast;
|
2014-02-25 04:47:19 +08:00
|
|
|
use ext::mtwt;
|
2013-09-08 22:46:20 +10:00
|
|
|
use ext::build::AstBuilder;
|
2013-02-11 08:36:42 -08:00
|
|
|
use attr;
|
2013-07-19 21:51:37 +10:00
|
|
|
use attr::AttrMetaMethods;
|
2013-02-25 14:11:21 -05:00
|
|
|
use codemap;
|
2015-08-27 05:16:05 +05:30
|
|
|
use codemap::{Span, Spanned, ExpnInfo, NameAndSpan, MacroBang, MacroAttribute};
|
2012-09-04 11:37:29 -07:00
|
|
|
use ext::base::*;
|
2015-07-13 17:10:44 -07:00
|
|
|
use feature_gate::{self, Features, GatedCfg};
|
2014-07-02 23:16:01 -07:00
|
|
|
use fold;
|
2012-12-23 17:41:37 -05:00
|
|
|
use fold::*;
|
2013-02-25 14:11:21 -05:00
|
|
|
use parse;
|
2014-01-31 15:26:59 -08:00
|
|
|
use parse::token::{fresh_mark, fresh_name, intern};
|
2014-09-13 19:06:01 +03:00
|
|
|
use ptr::P;
|
|
|
|
use util::small_vector::SmallVector;
|
2013-06-04 14:56:33 -07:00
|
|
|
use visit;
|
2013-06-09 02:21:11 +10:00
|
|
|
use visit::Visitor;
|
2014-09-07 14:57:26 -07:00
|
|
|
use std_inject;
|
2011-08-05 13:06:11 -07:00
|
|
|
|
2015-01-27 01:22:12 +01:00
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
pub fn expand_expr(e: P<ast::Expr>, fld: &mut MacroExpander) -> P<ast::Expr> {
|
2015-02-12 11:30:16 +01:00
|
|
|
let expr_span = e.span;
|
2015-06-05 14:21:32 +02:00
|
|
|
return e.and_then(|ast::Expr {id, node, span}| match node {
|
2015-01-27 01:22:12 +01:00
|
|
|
|
2013-02-17 22:20:36 -08:00
|
|
|
// expr_mac should really be expr_ext or something; it's the
|
|
|
|
// entry-point for all syntax extensions.
|
2014-08-24 19:08:48 -07:00
|
|
|
ast::ExprMac(mac) => {
|
2014-09-13 19:06:01 +03:00
|
|
|
let expanded_expr = match expand_mac_invoc(mac, span,
|
|
|
|
|r| r.make_expr(),
|
|
|
|
mark_expr, fld) {
|
2014-07-12 15:00:23 -07:00
|
|
|
Some(expr) => expr,
|
|
|
|
None => {
|
2014-09-13 19:06:01 +03:00
|
|
|
return DummyResult::raw_expr(span);
|
2014-07-12 15:00:23 -07:00
|
|
|
}
|
|
|
|
};
|
2014-01-05 18:02:57 -08:00
|
|
|
|
2014-07-12 15:00:23 -07:00
|
|
|
// Keep going, outside-in.
|
2014-09-13 19:06:01 +03:00
|
|
|
let fully_expanded = fld.fold_expr(expanded_expr);
|
2015-07-08 14:30:18 +12:00
|
|
|
let span = fld.new_span(span);
|
2014-07-12 15:00:23 -07:00
|
|
|
fld.cx.bt_pop();
|
2014-01-05 18:02:57 -08:00
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
fully_expanded.map(|e| ast::Expr {
|
2014-07-12 15:00:23 -07:00
|
|
|
id: ast::DUMMY_NODE_ID,
|
2014-09-13 19:06:01 +03:00
|
|
|
node: e.node,
|
2015-07-08 14:29:43 +12:00
|
|
|
span: span,
|
2014-09-13 19:06:01 +03:00
|
|
|
})
|
2013-02-17 22:20:36 -08:00
|
|
|
}
|
2013-07-29 17:25:00 -07:00
|
|
|
|
2015-09-24 18:00:08 +03:00
|
|
|
ast::ExprInPlace(placer, value_expr) => {
|
2015-02-12 11:30:16 +01:00
|
|
|
// Ensure feature-gate is enabled
|
|
|
|
feature_gate::check_for_placement_in(
|
|
|
|
fld.cx.ecfg.features,
|
|
|
|
&fld.cx.parse_sess.span_diagnostic,
|
|
|
|
expr_span);
|
|
|
|
|
2015-09-29 13:17:46 +13:00
|
|
|
let placer = fld.fold_expr(placer);
|
2015-01-27 01:22:12 +01:00
|
|
|
let value_expr = fld.fold_expr(value_expr);
|
2015-09-29 13:46:01 +13:00
|
|
|
fld.cx.expr(span, ast::ExprInPlace(placer, value_expr))
|
2015-01-27 01:22:12 +01:00
|
|
|
}
|
|
|
|
|
2014-07-25 20:12:51 -04:00
|
|
|
ast::ExprWhile(cond, body, opt_ident) => {
|
|
|
|
let cond = fld.fold_expr(cond);
|
|
|
|
let (body, opt_ident) = expand_loop_block(body, opt_ident, fld);
|
2014-09-13 19:06:01 +03:00
|
|
|
fld.cx.expr(span, ast::ExprWhile(cond, body, opt_ident))
|
2014-07-25 20:12:51 -04:00
|
|
|
}
|
|
|
|
|
2014-10-02 22:45:46 -04:00
|
|
|
ast::ExprWhileLet(pat, expr, body, opt_ident) => {
|
2015-09-28 17:24:42 +13:00
|
|
|
let pat = fld.fold_pat(pat);
|
|
|
|
let expr = fld.fold_expr(expr);
|
2015-09-30 12:19:45 +13:00
|
|
|
|
|
|
|
// 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);
|
|
|
|
|
|
|
|
fld.cx.expr(span, ast::ExprWhileLet(rewritten_pats.remove(0), expr, body, opt_ident))
|
2014-08-24 19:08:48 -07:00
|
|
|
}
|
|
|
|
|
2014-02-15 16:54:32 +08:00
|
|
|
ast::ExprLoop(loop_block, opt_ident) => {
|
2014-02-25 04:47:19 +08:00
|
|
|
let (loop_block, opt_ident) = expand_loop_block(loop_block, opt_ident, fld);
|
2014-09-13 19:06:01 +03:00
|
|
|
fld.cx.expr(span, ast::ExprLoop(loop_block, opt_ident))
|
2014-02-15 16:54:32 +08:00
|
|
|
}
|
|
|
|
|
2014-07-21 20:54:28 -07:00
|
|
|
ast::ExprForLoop(pat, head, body, opt_ident) => {
|
2015-09-29 13:17:46 +13:00
|
|
|
let pat = fld.fold_pat(pat);
|
2015-09-30 12:19:45 +13:00
|
|
|
|
|
|
|
// 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);
|
|
|
|
|
2014-07-21 20:54:28 -07:00
|
|
|
let head = fld.fold_expr(head);
|
2015-09-30 12:19:45 +13:00
|
|
|
fld.cx.expr(span, ast::ExprForLoop(rewritten_pats.remove(0), head, body, opt_ident))
|
|
|
|
}
|
|
|
|
|
|
|
|
ast::ExprIfLet(pat, sub_expr, body, else_opt) => {
|
|
|
|
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);
|
|
|
|
fld.cx.expr(span, ast::ExprIfLet(rewritten_pats.remove(0), sub_expr, body, else_opt))
|
2014-07-21 20:54:28 -07:00
|
|
|
}
|
|
|
|
|
2015-02-03 11:34:05 -05:00
|
|
|
ast::ExprClosure(capture_clause, fn_decl, block) => {
|
2014-07-03 17:20:25 -07:00
|
|
|
let (rewritten_fn_decl, rewritten_block)
|
2014-09-13 19:06:01 +03:00
|
|
|
= expand_and_rename_fn_decl_and_block(fn_decl, block, fld);
|
2014-11-19 11:18:17 -05:00
|
|
|
let new_node = ast::ExprClosure(capture_clause,
|
2014-07-23 12:43:29 -07:00
|
|
|
rewritten_fn_decl,
|
|
|
|
rewritten_block);
|
2015-08-31 17:06:30 +12:00
|
|
|
P(ast::Expr{id:id, node: new_node, span: fld.new_span(span)})
|
2014-07-03 17:20:25 -07:00
|
|
|
}
|
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
_ => {
|
|
|
|
P(noop_fold_expr(ast::Expr {
|
|
|
|
id: id,
|
|
|
|
node: node,
|
|
|
|
span: span
|
|
|
|
}, fld))
|
|
|
|
}
|
2015-06-05 14:21:32 +02:00
|
|
|
});
|
2011-07-06 15:22:23 -07:00
|
|
|
}
|
|
|
|
|
2014-07-12 15:00:23 -07:00
|
|
|
/// Expand a (not-ident-style) macro invocation. Returns the result
|
|
|
|
/// of expansion and the mark which must be applied to the result.
|
|
|
|
/// Our current interface doesn't allow us to apply the mark to the
|
|
|
|
/// result until after calling make_expr, make_items, etc.
|
2015-07-08 14:29:43 +12:00
|
|
|
fn expand_mac_invoc<T, F, G>(mac: ast::Mac,
|
|
|
|
span: codemap::Span,
|
2014-12-08 13:28:32 -05:00
|
|
|
parse_thunk: F,
|
|
|
|
mark_thunk: G,
|
|
|
|
fld: &mut MacroExpander)
|
|
|
|
-> Option<T> where
|
2015-02-12 10:41:47 -05:00
|
|
|
F: for<'a> FnOnce(Box<MacResult+'a>) -> Option<T>,
|
2014-12-08 13:28:32 -05:00
|
|
|
G: FnOnce(T, Mrk) -> T,
|
2014-08-27 21:46:52 -04:00
|
|
|
{
|
2015-09-20 16:15:37 -04:00
|
|
|
// 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. This also goes
|
|
|
|
// for the other three macro invocation chunks of code
|
|
|
|
// in this file.
|
|
|
|
|
|
|
|
let Mac_ { path: pth, tts, .. } = mac.node;
|
|
|
|
if pth.segments.len() > 1 {
|
|
|
|
fld.cx.span_err(pth.span,
|
|
|
|
"expected macro name without module \
|
|
|
|
separators");
|
|
|
|
// let compilation continue
|
|
|
|
return None;
|
|
|
|
}
|
|
|
|
let extname = pth.segments[0].identifier.name;
|
2015-09-24 23:05:02 +03:00
|
|
|
match fld.cx.syntax_env.find(extname) {
|
2015-09-20 16:15:37 -04:00
|
|
|
None => {
|
|
|
|
fld.cx.span_err(
|
|
|
|
pth.span,
|
|
|
|
&format!("macro undefined: '{}!'",
|
|
|
|
&extname));
|
2014-07-12 15:00:23 -07:00
|
|
|
|
2015-09-20 16:15:37 -04:00
|
|
|
// let compilation continue
|
|
|
|
None
|
|
|
|
}
|
|
|
|
Some(rc) => match *rc {
|
|
|
|
NormalTT(ref expandfun, exp_span, allow_internal_unstable) => {
|
|
|
|
fld.cx.bt_push(ExpnInfo {
|
|
|
|
call_site: span,
|
|
|
|
callee: NameAndSpan {
|
|
|
|
format: MacroBang(extname),
|
|
|
|
span: exp_span,
|
|
|
|
allow_internal_unstable: allow_internal_unstable,
|
|
|
|
},
|
|
|
|
});
|
|
|
|
let fm = fresh_mark();
|
|
|
|
let marked_before = mark_tts(&tts[..], fm);
|
|
|
|
|
|
|
|
// 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();
|
|
|
|
|
|
|
|
let opt_parsed = {
|
|
|
|
let expanded = expandfun.expand(fld.cx,
|
|
|
|
mac_span,
|
|
|
|
&marked_before[..]);
|
|
|
|
parse_thunk(expanded)
|
|
|
|
};
|
|
|
|
let parsed = match opt_parsed {
|
|
|
|
Some(e) => e,
|
|
|
|
None => {
|
2014-07-19 21:34:24 +02:00
|
|
|
fld.cx.span_err(
|
|
|
|
pth.span,
|
2015-09-20 16:15:37 -04:00
|
|
|
&format!("non-expression macro in expression position: {}",
|
|
|
|
extname
|
|
|
|
));
|
|
|
|
return None;
|
2014-07-19 21:34:24 +02:00
|
|
|
}
|
2015-09-20 16:15:37 -04:00
|
|
|
};
|
|
|
|
Some(mark_thunk(parsed,fm))
|
|
|
|
}
|
|
|
|
_ => {
|
|
|
|
fld.cx.span_err(
|
|
|
|
pth.span,
|
|
|
|
&format!("'{}' is not a tt-style macro",
|
|
|
|
extname));
|
|
|
|
None
|
2014-07-12 15:00:23 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-06-09 13:12:30 -07:00
|
|
|
/// 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.
|
2014-02-25 04:47:19 +08:00
|
|
|
fn expand_loop_block(loop_block: P<Block>,
|
|
|
|
opt_ident: Option<Ident>,
|
|
|
|
fld: &mut MacroExpander) -> (P<Block>, Option<Ident>) {
|
2014-02-15 16:54:32 +08:00
|
|
|
match opt_ident {
|
|
|
|
Some(label) => {
|
2015-09-24 23:05:02 +03:00
|
|
|
let new_label = fresh_name(label);
|
2014-02-15 16:54:32 +08:00
|
|
|
let rename = (label, new_label);
|
2014-02-25 04:47:19 +08:00
|
|
|
|
|
|
|
// 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);
|
2014-06-26 13:18:52 -07:00
|
|
|
let mut rename_fld = IdentRenamer{renames: &mut rename_list};
|
2014-02-25 04:47:19 +08:00
|
|
|
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.
|
2014-07-19 21:34:24 +02:00
|
|
|
fld.cx.syntax_env.push_frame();
|
|
|
|
fld.cx.syntax_env.info().pending_renames.push(rename);
|
2014-09-13 19:06:01 +03:00
|
|
|
let expanded_block = expand_block_elts(loop_block, fld);
|
2014-07-19 21:34:24 +02:00
|
|
|
fld.cx.syntax_env.pop_frame();
|
2014-02-25 04:47:19 +08:00
|
|
|
|
|
|
|
(expanded_block, Some(renamed_ident))
|
2014-02-15 16:54:32 +08:00
|
|
|
}
|
2014-02-25 04:47:19 +08:00
|
|
|
None => (fld.fold_block(loop_block), opt_ident)
|
2014-02-15 16:54:32 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-07-04 12:05:43 -07:00
|
|
|
// eval $e with a new exts frame.
|
|
|
|
// must be a macro so that $e isn't evaluated too early.
|
2014-11-14 09:18:10 -08:00
|
|
|
macro_rules! with_exts_frame {
|
2013-05-08 15:27:29 -07:00
|
|
|
($extsboxexpr:expr,$macros_escape:expr,$e:expr) =>
|
2013-12-30 17:45:39 -08:00
|
|
|
({$extsboxexpr.push_frame();
|
|
|
|
$extsboxexpr.info().macros_escape = $macros_escape;
|
2013-02-26 10:15:29 -08:00
|
|
|
let result = $e;
|
2013-12-30 17:45:39 -08:00
|
|
|
$extsboxexpr.pop_frame();
|
2013-02-26 10:15:29 -08:00
|
|
|
result
|
|
|
|
})
|
2014-11-14 09:18:10 -08:00
|
|
|
}
|
2013-02-26 10:15:29 -08:00
|
|
|
|
2012-07-27 11:19:21 -07:00
|
|
|
// When we enter a module, record it, for the sake of `module!`
|
2014-09-13 19:06:01 +03:00
|
|
|
pub fn expand_item(it: P<ast::Item>, fld: &mut MacroExpander)
|
|
|
|
-> SmallVector<P<ast::Item>> {
|
2015-09-01 13:16:03 +12:00
|
|
|
let it = expand_item_multi_modifier(Annotatable::Item(it), fld);
|
2014-02-27 23:49:25 -08:00
|
|
|
|
2015-09-01 13:16:03 +12:00
|
|
|
expand_annotatable(it, fld)
|
2014-12-02 10:07:41 -08:00
|
|
|
.into_iter().map(|i| i.expect_item()).collect()
|
2012-05-18 10:00:49 -07:00
|
|
|
}
|
|
|
|
|
2014-07-04 12:19:15 -07:00
|
|
|
/// Expand item_underscore
|
2014-09-13 19:06:01 +03:00
|
|
|
fn expand_item_underscore(item: ast::Item_, fld: &mut MacroExpander) -> ast::Item_ {
|
|
|
|
match item {
|
2015-02-25 22:05:07 +02:00
|
|
|
ast::ItemFn(decl, unsafety, constness, abi, generics, body) => {
|
2014-07-04 12:19:15 -07:00
|
|
|
let (rewritten_fn_decl, rewritten_body)
|
2014-09-13 19:06:01 +03:00
|
|
|
= expand_and_rename_fn_decl_and_block(decl, body, fld);
|
2014-07-25 21:00:33 +02:00
|
|
|
let expanded_generics = fold::noop_fold_generics(generics,fld);
|
2015-02-25 22:05:07 +02:00
|
|
|
ast::ItemFn(rewritten_fn_decl, unsafety, constness, abi,
|
|
|
|
expanded_generics, rewritten_body)
|
2014-07-04 12:19:15 -07:00
|
|
|
}
|
2014-09-13 19:06:01 +03:00
|
|
|
_ => noop_fold_item_underscore(item, fld)
|
2014-07-04 12:19:15 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-12-18 20:48:26 -08:00
|
|
|
// does this attribute list contain "macro_use" ?
|
|
|
|
fn contains_macro_use(fld: &mut MacroExpander, attrs: &[ast::Attribute]) -> bool {
|
2015-01-31 12:20:46 -05:00
|
|
|
for attr in attrs {
|
2014-12-18 20:48:26 -08:00
|
|
|
let mut is_use = attr.check_name("macro_use");
|
|
|
|
if attr.check_name("macro_escape") {
|
|
|
|
fld.cx.span_warn(attr.span, "macro_escape is a deprecated synonym for macro_use");
|
|
|
|
is_use = true;
|
2015-10-01 18:03:34 +02:00
|
|
|
if let ast::AttrStyle::Inner = attr.node.style {
|
2015-02-24 16:07:54 +02:00
|
|
|
fld.cx.fileline_help(attr.span, "consider an outer attribute, \
|
2014-12-18 20:48:26 -08:00
|
|
|
#[macro_use] mod ...");
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
if is_use {
|
|
|
|
match attr.node.value.node {
|
|
|
|
ast::MetaWord(..) => (),
|
|
|
|
_ => fld.cx.span_err(attr.span, "arguments to macro_use are not allowed here"),
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
false
|
2013-02-26 10:15:29 -08:00
|
|
|
}
|
|
|
|
|
2012-07-27 11:19:21 -07:00
|
|
|
// Support for item-position macro invocations, exactly the same
|
|
|
|
// logic as for expression-position macro invocations.
|
2014-09-15 18:27:28 -07:00
|
|
|
pub fn expand_item_mac(it: P<ast::Item>,
|
|
|
|
fld: &mut MacroExpander) -> SmallVector<P<ast::Item>> {
|
2015-09-20 16:15:37 -04:00
|
|
|
let (extname, path_span, tts, span, attrs, ident) = it.and_then(|it| match it.node {
|
|
|
|
ItemMac(codemap::Spanned { node: Mac_ { path, tts, .. }, .. }) =>
|
|
|
|
(path.segments[0].identifier.name, path.span, tts, it.span, it.attrs, it.ident),
|
2013-12-27 17:43:01 -07:00
|
|
|
_ => fld.cx.span_bug(it.span, "invalid item macro invocation")
|
2015-09-20 16:15:37 -04:00
|
|
|
});
|
2012-11-17 07:41:36 -05:00
|
|
|
|
2013-06-29 05:59:08 -07:00
|
|
|
let fm = fresh_mark();
|
2014-09-15 16:09:09 -07:00
|
|
|
let items = {
|
2015-09-24 23:05:02 +03:00
|
|
|
let expanded = match fld.cx.syntax_env.find(extname) {
|
2014-08-27 21:46:52 -04:00
|
|
|
None => {
|
2014-09-13 19:06:01 +03:00
|
|
|
fld.cx.span_err(path_span,
|
2015-01-07 11:58:31 -05:00
|
|
|
&format!("macro undefined: '{}!'",
|
2015-07-28 18:07:20 +02:00
|
|
|
extname));
|
2014-08-27 21:46:52 -04:00
|
|
|
// let compilation continue
|
|
|
|
return SmallVector::zero();
|
|
|
|
}
|
2012-11-17 07:41:36 -05:00
|
|
|
|
2014-08-27 21:46:52 -04:00
|
|
|
Some(rc) => match *rc {
|
2015-09-20 06:45:19 +05:30
|
|
|
NormalTT(ref expander, tt_span, allow_internal_unstable) => {
|
|
|
|
if ident.name != parse::token::special_idents::invalid.name {
|
2014-08-27 21:46:52 -04:00
|
|
|
fld.cx
|
2014-09-13 19:06:01 +03:00
|
|
|
.span_err(path_span,
|
2015-02-20 14:08:14 -05:00
|
|
|
&format!("macro {}! expects no ident argument, given '{}'",
|
2015-07-28 18:07:20 +02:00
|
|
|
extname,
|
2015-09-20 06:45:19 +05:30
|
|
|
ident));
|
2014-08-27 21:46:52 -04:00
|
|
|
return SmallVector::zero();
|
|
|
|
}
|
|
|
|
fld.cx.bt_push(ExpnInfo {
|
2015-09-20 06:45:19 +05:30
|
|
|
call_site: span,
|
2014-08-27 21:46:52 -04:00
|
|
|
callee: NameAndSpan {
|
2015-08-27 05:41:53 +05:30
|
|
|
format: MacroBang(extname),
|
2015-09-20 06:45:19 +05:30
|
|
|
span: tt_span,
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
allow_internal_unstable: allow_internal_unstable,
|
2014-08-27 21:46:52 -04:00
|
|
|
}
|
|
|
|
});
|
|
|
|
// mark before expansion:
|
2015-02-18 14:48:57 -05:00
|
|
|
let marked_before = mark_tts(&tts[..], fm);
|
2015-09-20 06:45:19 +05:30
|
|
|
expander.expand(fld.cx, span, &marked_before[..])
|
2014-07-19 21:34:24 +02:00
|
|
|
}
|
2015-09-20 06:45:19 +05:30
|
|
|
IdentTT(ref expander, tt_span, allow_internal_unstable) => {
|
|
|
|
if ident.name == parse::token::special_idents::invalid.name {
|
2014-09-13 19:06:01 +03:00
|
|
|
fld.cx.span_err(path_span,
|
2015-01-07 11:58:31 -05:00
|
|
|
&format!("macro {}! expects an ident argument",
|
2015-07-28 18:07:20 +02:00
|
|
|
extname));
|
2014-08-27 21:46:52 -04:00
|
|
|
return SmallVector::zero();
|
2014-07-19 21:34:24 +02:00
|
|
|
}
|
2014-08-27 21:46:52 -04:00
|
|
|
fld.cx.bt_push(ExpnInfo {
|
2015-09-20 06:45:19 +05:30
|
|
|
call_site: span,
|
2014-08-27 21:46:52 -04:00
|
|
|
callee: NameAndSpan {
|
2015-08-27 05:41:53 +05:30
|
|
|
format: MacroBang(extname),
|
2015-09-20 06:45:19 +05:30
|
|
|
span: tt_span,
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
allow_internal_unstable: allow_internal_unstable,
|
2014-08-27 21:46:52 -04:00
|
|
|
}
|
|
|
|
});
|
|
|
|
// mark before expansion:
|
2015-02-18 14:48:57 -05:00
|
|
|
let marked_tts = mark_tts(&tts[..], fm);
|
2015-09-20 06:45:19 +05:30
|
|
|
expander.expand(fld.cx, span, ident, marked_tts)
|
2013-02-21 00:16:31 -08:00
|
|
|
}
|
2014-09-15 16:09:09 -07:00
|
|
|
MacroRulesTT => {
|
2015-09-20 06:45:19 +05:30
|
|
|
if ident.name == parse::token::special_idents::invalid.name {
|
2015-11-05 15:30:32 +01:00
|
|
|
fld.cx.span_err(path_span, "macro_rules! expects an ident argument");
|
2014-08-27 21:46:52 -04:00
|
|
|
return SmallVector::zero();
|
2014-07-19 21:34:24 +02:00
|
|
|
}
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
|
2014-08-27 21:46:52 -04:00
|
|
|
fld.cx.bt_push(ExpnInfo {
|
2015-09-20 06:45:19 +05:30
|
|
|
call_site: span,
|
2014-08-27 21:46:52 -04:00
|
|
|
callee: NameAndSpan {
|
2015-08-27 05:41:53 +05:30
|
|
|
format: MacroBang(extname),
|
2014-09-15 16:09:09 -07:00
|
|
|
span: None,
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
// `macro_rules!` doesn't directly allow
|
|
|
|
// unstable (this is orthogonal to whether
|
|
|
|
// the macro it creates allows it)
|
|
|
|
allow_internal_unstable: false,
|
2014-08-27 21:46:52 -04:00
|
|
|
}
|
|
|
|
});
|
2014-09-15 16:09:09 -07:00
|
|
|
// DON'T mark before expansion.
|
2014-12-30 19:10:46 -08:00
|
|
|
|
2015-09-20 06:45:19 +05:30
|
|
|
let allow_internal_unstable = attr::contains_name(&attrs,
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
"allow_internal_unstable");
|
|
|
|
|
|
|
|
// ensure any #[allow_internal_unstable]s are
|
|
|
|
// detected (including nested macro definitions
|
|
|
|
// etc.)
|
|
|
|
if allow_internal_unstable && !fld.cx.ecfg.enable_allow_internal_unstable() {
|
|
|
|
feature_gate::emit_feature_err(
|
|
|
|
&fld.cx.parse_sess.span_diagnostic,
|
|
|
|
"allow_internal_unstable",
|
2015-09-20 06:45:19 +05:30
|
|
|
span,
|
2015-09-04 16:37:22 -07:00
|
|
|
feature_gate::GateIssue::Language,
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
feature_gate::EXPLAIN_ALLOW_INTERNAL_UNSTABLE)
|
|
|
|
}
|
|
|
|
|
2015-09-20 06:45:19 +05:30
|
|
|
let export = attr::contains_name(&attrs, "macro_export");
|
2014-12-30 19:10:46 -08:00
|
|
|
let def = ast::MacroDef {
|
2015-09-20 06:45:19 +05:30
|
|
|
ident: ident,
|
|
|
|
attrs: attrs,
|
2014-12-30 19:10:46 -08:00
|
|
|
id: ast::DUMMY_NODE_ID,
|
2015-09-20 06:45:19 +05:30
|
|
|
span: span,
|
2014-12-30 19:10:46 -08:00
|
|
|
imported_from: None,
|
2015-09-20 06:45:19 +05:30
|
|
|
export: export,
|
2015-01-02 12:50:45 -08:00
|
|
|
use_locally: true,
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
allow_internal_unstable: allow_internal_unstable,
|
2014-12-30 19:10:46 -08:00
|
|
|
body: tts,
|
|
|
|
};
|
2015-01-01 16:37:47 -08:00
|
|
|
fld.cx.insert_macro(def);
|
2014-09-15 16:09:09 -07:00
|
|
|
|
|
|
|
// macro_rules! has a side effect but expands to nothing.
|
|
|
|
fld.cx.bt_pop();
|
|
|
|
return SmallVector::zero();
|
2014-08-27 21:46:52 -04:00
|
|
|
}
|
|
|
|
_ => {
|
2015-09-20 06:45:19 +05:30
|
|
|
fld.cx.span_err(span,
|
2015-01-07 11:58:31 -05:00
|
|
|
&format!("{}! is not legal in item position",
|
2015-07-28 18:07:20 +02:00
|
|
|
extname));
|
2014-07-19 21:34:24 +02:00
|
|
|
return SmallVector::zero();
|
2013-02-21 00:16:31 -08:00
|
|
|
}
|
2014-07-07 09:54:08 -07:00
|
|
|
}
|
2014-08-27 21:46:52 -04:00
|
|
|
};
|
|
|
|
|
2014-09-15 16:09:09 -07:00
|
|
|
expanded.make_items()
|
2012-11-17 07:41:36 -05:00
|
|
|
};
|
2012-11-08 23:12:45 -05:00
|
|
|
|
2014-09-15 16:09:09 -07:00
|
|
|
let items = match items {
|
|
|
|
Some(items) => {
|
2014-09-14 20:27:36 -07:00
|
|
|
items.into_iter()
|
2014-08-27 21:46:52 -04:00
|
|
|
.map(|i| mark_item(i, fm))
|
2014-09-14 20:27:36 -07:00
|
|
|
.flat_map(|i| fld.fold_item(i).into_iter())
|
2014-08-27 21:46:52 -04:00
|
|
|
.collect()
|
|
|
|
}
|
2014-09-15 16:09:09 -07:00
|
|
|
None => {
|
2014-09-13 19:06:01 +03:00
|
|
|
fld.cx.span_err(path_span,
|
2015-01-07 11:58:31 -05:00
|
|
|
&format!("non-item macro in item position: {}",
|
2015-07-28 18:07:20 +02:00
|
|
|
extname));
|
2014-08-27 21:46:52 -04:00
|
|
|
return SmallVector::zero();
|
2014-04-15 22:00:14 +10:00
|
|
|
}
|
2012-11-17 07:41:36 -05:00
|
|
|
};
|
2014-08-27 21:46:52 -04:00
|
|
|
|
2013-12-27 17:43:01 -07:00
|
|
|
fld.cx.bt_pop();
|
2014-09-13 19:06:01 +03:00
|
|
|
items
|
2012-07-05 12:10:33 -07:00
|
|
|
}
|
|
|
|
|
2014-07-12 15:00:23 -07:00
|
|
|
/// Expand a stmt
|
2015-04-07 08:21:18 -05:00
|
|
|
fn expand_stmt(stmt: P<Stmt>, fld: &mut MacroExpander) -> SmallVector<P<Stmt>> {
|
|
|
|
let stmt = stmt.and_then(|stmt| stmt);
|
|
|
|
let (mac, style) = match stmt.node {
|
2014-11-14 09:18:10 -08:00
|
|
|
StmtMac(mac, style) => (mac, style),
|
2015-04-07 08:21:18 -05:00
|
|
|
_ => return expand_non_macro_stmt(stmt, fld)
|
2012-12-12 12:25:40 -08:00
|
|
|
};
|
2015-04-07 08:21:18 -05:00
|
|
|
|
|
|
|
let maybe_new_items =
|
|
|
|
expand_mac_invoc(mac.and_then(|m| m), stmt.span,
|
|
|
|
|r| r.make_stmts(),
|
|
|
|
|stmts, mark| stmts.move_map(|m| mark_stmt(m, mark)),
|
|
|
|
fld);
|
|
|
|
|
2015-04-10 21:50:23 -05:00
|
|
|
let mut fully_expanded = match maybe_new_items {
|
2015-04-07 08:21:18 -05:00
|
|
|
Some(stmts) => {
|
|
|
|
// Keep going, outside-in.
|
2015-04-11 00:59:00 -05:00
|
|
|
let new_items = stmts.into_iter().flat_map(|s| {
|
2015-04-07 08:21:18 -05:00
|
|
|
fld.fold_stmt(s).into_iter()
|
2015-04-11 00:59:00 -05:00
|
|
|
}).collect();
|
|
|
|
fld.cx.bt_pop();
|
|
|
|
new_items
|
2012-11-17 07:41:36 -05:00
|
|
|
}
|
2015-04-07 08:21:18 -05:00
|
|
|
None => SmallVector::zero()
|
2012-11-21 18:49:42 -05:00
|
|
|
};
|
|
|
|
|
2015-04-07 08:21:18 -05:00
|
|
|
// If this is a macro invocation with a semicolon, then apply that
|
|
|
|
// semicolon to the final statement produced by expansion.
|
2014-11-14 09:18:10 -08:00
|
|
|
if style == MacStmtWithSemicolon {
|
2015-04-10 23:06:34 -05:00
|
|
|
if let Some(stmt) = fully_expanded.pop() {
|
|
|
|
let new_stmt = stmt.map(|Spanned {node, span}| {
|
|
|
|
Spanned {
|
|
|
|
node: match node {
|
|
|
|
StmtExpr(e, stmt_id) => StmtSemi(e, stmt_id),
|
|
|
|
_ => node /* might already have a semi */
|
|
|
|
},
|
|
|
|
span: span
|
|
|
|
}
|
|
|
|
});
|
|
|
|
fully_expanded.push(new_stmt);
|
2015-04-10 21:50:23 -05:00
|
|
|
}
|
2014-09-13 19:06:01 +03:00
|
|
|
}
|
2015-04-10 21:50:23 -05:00
|
|
|
|
|
|
|
fully_expanded
|
2012-11-19 20:31:22 -05:00
|
|
|
}
|
|
|
|
|
2013-06-25 11:40:51 -07:00
|
|
|
// expand a non-macro stmt. this is essentially the fallthrough for
|
|
|
|
// expand_stmt, above.
|
2014-09-13 19:06:01 +03:00
|
|
|
fn expand_non_macro_stmt(Spanned {node, span: stmt_span}: Stmt, fld: &mut MacroExpander)
|
|
|
|
-> SmallVector<P<Stmt>> {
|
2013-06-25 11:40:51 -07:00
|
|
|
// is it a let?
|
2014-09-13 19:06:01 +03:00
|
|
|
match node {
|
|
|
|
StmtDecl(decl, node_id) => decl.and_then(|Spanned {node: decl, span}| match decl {
|
|
|
|
DeclLocal(local) => {
|
|
|
|
// take it apart:
|
2015-07-25 22:43:35 +02:00
|
|
|
let rewritten_local = local.map(|Local {id, pat, ty, init, span}| {
|
2014-07-19 00:06:43 -07:00
|
|
|
// expand the ty since TyFixedLengthVec contains an Expr
|
|
|
|
// and thus may have a macro use
|
2015-01-02 20:55:31 +09:00
|
|
|
let expanded_ty = ty.map(|t| fld.fold_ty(t));
|
2014-06-26 16:49:13 -07:00
|
|
|
// expand the pat (it might contain macro uses):
|
2014-01-03 15:08:48 -08:00
|
|
|
let expanded_pat = fld.fold_pat(pat);
|
2014-07-04 12:05:43 -07:00
|
|
|
// find the PatIdents in the pattern:
|
2014-01-03 15:08:48 -08:00
|
|
|
// 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
|
2014-07-07 16:35:15 -07:00
|
|
|
let idents = pattern_bindings(&*expanded_pat);
|
2014-06-26 14:39:54 -07:00
|
|
|
let mut new_pending_renames =
|
2015-09-24 23:05:02 +03:00
|
|
|
idents.iter().map(|ident| (*ident, fresh_name(*ident))).collect();
|
2014-06-26 16:49:13 -07:00
|
|
|
// rewrite the pattern using the new names (the old
|
|
|
|
// ones have already been applied):
|
2014-01-03 15:08:48 -08:00
|
|
|
let rewritten_pat = {
|
2014-06-26 16:49:13 -07:00
|
|
|
// nested binding to allow borrow to expire:
|
|
|
|
let mut rename_fld = IdentRenamer{renames: &mut new_pending_renames};
|
2014-01-03 15:08:48 -08:00
|
|
|
rename_fld.fold_pat(expanded_pat)
|
|
|
|
};
|
|
|
|
// add them to the existing pending renames:
|
2014-10-14 23:05:01 -07:00
|
|
|
fld.cx.syntax_env.info().pending_renames
|
2015-06-10 17:22:20 +01:00
|
|
|
.extend(new_pending_renames);
|
2014-09-13 19:06:01 +03:00
|
|
|
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)),
|
2014-01-03 15:08:48 -08:00
|
|
|
span: span
|
2014-09-13 19:06:01 +03:00
|
|
|
}
|
|
|
|
});
|
|
|
|
SmallVector::one(P(Spanned {
|
|
|
|
node: StmtDecl(P(Spanned {
|
|
|
|
node: DeclLocal(rewritten_local),
|
|
|
|
span: span
|
|
|
|
}),
|
|
|
|
node_id),
|
|
|
|
span: stmt_span
|
|
|
|
}))
|
2013-12-30 17:45:39 -08:00
|
|
|
}
|
2014-09-13 19:06:01 +03:00
|
|
|
_ => {
|
|
|
|
noop_fold_stmt(Spanned {
|
|
|
|
node: StmtDecl(P(Spanned {
|
|
|
|
node: decl,
|
|
|
|
span: span
|
|
|
|
}),
|
|
|
|
node_id),
|
|
|
|
span: stmt_span
|
|
|
|
}, fld)
|
|
|
|
}
|
|
|
|
}),
|
|
|
|
_ => {
|
|
|
|
noop_fold_stmt(Spanned {
|
|
|
|
node: node,
|
|
|
|
span: stmt_span
|
|
|
|
}, fld)
|
|
|
|
}
|
2013-06-25 11:40:51 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-07-04 12:19:15 -07:00
|
|
|
// expand the arm of a 'match', renaming for macro hygiene
|
2014-09-13 19:06:01 +03:00
|
|
|
fn expand_arm(arm: ast::Arm, fld: &mut MacroExpander) -> ast::Arm {
|
2014-06-26 16:49:13 -07:00
|
|
|
// expand pats... they might contain macro uses:
|
2014-09-13 19:06:01 +03:00
|
|
|
let expanded_pats = arm.pats.move_map(|pat| fld.fold_pat(pat));
|
2015-03-24 16:53:34 -07:00
|
|
|
if expanded_pats.is_empty() {
|
2014-10-09 15:17:22 -04:00
|
|
|
panic!("encountered match arm with 0 patterns");
|
2014-06-25 15:20:01 -07:00
|
|
|
}
|
2015-09-30 12:19:45 +13:00
|
|
|
|
2014-06-26 16:49:13 -07:00
|
|
|
// apply renaming and then expansion to the guard and the body:
|
2015-09-30 12:19:45 +13:00
|
|
|
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)
|
|
|
|
});
|
|
|
|
|
2014-06-26 13:19:34 -07:00
|
|
|
ast::Arm {
|
2015-02-09 09:01:15 -08:00
|
|
|
attrs: fold::fold_attrs(arm.attrs, fld),
|
2014-06-26 16:49:13 -07:00
|
|
|
pats: rewritten_pats,
|
|
|
|
guard: rewritten_guard,
|
|
|
|
body: rewritten_body,
|
2014-06-26 16:53:33 -07:00
|
|
|
}
|
2014-06-25 15:20:01 -07:00
|
|
|
}
|
|
|
|
|
2015-09-30 12:19:45 +13:00
|
|
|
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:
|
|
|
|
let idents = pattern_bindings(&*pats[0]);
|
|
|
|
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)
|
|
|
|
}
|
|
|
|
|
2014-07-04 12:05:43 -07:00
|
|
|
/// A visitor that extracts the PatIdent (binding) paths
|
|
|
|
/// from a given thingy and puts them in a mutable
|
|
|
|
/// array
|
2015-01-03 22:54:18 -05:00
|
|
|
#[derive(Clone)]
|
2014-07-04 12:05:43 -07:00
|
|
|
struct PatIdentFinder {
|
2014-09-12 13:10:30 +03:00
|
|
|
ident_accumulator: Vec<ast::Ident>
|
2013-07-19 18:42:11 -07:00
|
|
|
}
|
|
|
|
|
2014-09-10 01:54:36 +03:00
|
|
|
impl<'v> Visitor<'v> for PatIdentFinder {
|
2014-09-12 13:10:30 +03:00
|
|
|
fn visit_pat(&mut self, pattern: &ast::Pat) {
|
2013-07-19 18:42:11 -07:00
|
|
|
match *pattern {
|
2014-07-04 12:05:43 -07:00
|
|
|
ast::Pat { id: _, node: ast::PatIdent(_, ref path1, ref inner), span: _ } => {
|
2014-06-30 18:02:14 -07:00
|
|
|
self.ident_accumulator.push(path1.node);
|
2014-07-04 12:05:43 -07:00
|
|
|
// visit optional subpattern of PatIdent:
|
2015-01-31 12:20:46 -05:00
|
|
|
if let Some(ref subpat) = *inner {
|
2014-09-12 13:10:30 +03:00
|
|
|
self.visit_pat(&**subpat)
|
2013-06-04 14:56:33 -07:00
|
|
|
}
|
|
|
|
}
|
2014-07-04 12:05:43 -07:00
|
|
|
// use the default traversal for non-PatIdents
|
2014-09-12 13:10:30 +03:00
|
|
|
_ => visit::walk_pat(self, pattern)
|
2013-07-19 18:42:11 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-07-04 12:05:43 -07:00
|
|
|
/// find the PatIdent paths in a pattern
|
2014-09-12 13:10:30 +03:00
|
|
|
fn pattern_bindings(pat: &ast::Pat) -> Vec<ast::Ident> {
|
2014-07-04 12:05:43 -07:00
|
|
|
let mut name_finder = PatIdentFinder{ident_accumulator:Vec::new()};
|
2014-09-12 13:10:30 +03:00
|
|
|
name_finder.visit_pat(pat);
|
2014-06-26 14:39:54 -07:00
|
|
|
name_finder.ident_accumulator
|
2013-06-29 05:59:08 -07:00
|
|
|
}
|
|
|
|
|
2014-07-04 12:19:15 -07:00
|
|
|
/// find the PatIdent paths in a
|
|
|
|
fn fn_decl_arg_bindings(fn_decl: &ast::FnDecl) -> Vec<ast::Ident> {
|
|
|
|
let mut pat_idents = PatIdentFinder{ident_accumulator:Vec::new()};
|
2015-01-31 12:20:46 -05:00
|
|
|
for arg in &fn_decl.inputs {
|
2014-09-12 13:10:30 +03:00
|
|
|
pat_idents.visit_pat(&*arg.pat);
|
2014-07-04 12:19:15 -07:00
|
|
|
}
|
|
|
|
pat_idents.ident_accumulator
|
|
|
|
}
|
|
|
|
|
2013-07-05 13:57:53 -07:00
|
|
|
// expand a block. pushes a new exts_frame, then calls expand_block_elts
|
2014-09-13 19:06:01 +03:00
|
|
|
pub fn expand_block(blk: P<Block>, fld: &mut MacroExpander) -> P<Block> {
|
2013-06-25 11:40:51 -07:00
|
|
|
// see note below about treatment of exts table
|
2014-07-19 21:34:24 +02:00
|
|
|
with_exts_frame!(fld.cx.syntax_env,false,
|
2013-12-27 17:43:01 -07:00
|
|
|
expand_block_elts(blk, fld))
|
2013-06-25 11:40:51 -07:00
|
|
|
}
|
|
|
|
|
2013-07-05 13:57:53 -07:00
|
|
|
// expand the elements of a block.
|
2014-09-13 19:06:01 +03:00
|
|
|
pub fn expand_block_elts(b: P<Block>, fld: &mut MacroExpander) -> P<Block> {
|
2015-01-13 17:30:17 +02:00
|
|
|
b.map(|Block {id, stmts, expr, rules, span}| {
|
2014-09-14 20:27:36 -07:00
|
|
|
let new_stmts = stmts.into_iter().flat_map(|x| {
|
2014-06-26 13:18:52 -07:00
|
|
|
// perform all pending renames
|
2014-02-07 17:28:07 -05:00
|
|
|
let renamed_stmt = {
|
2014-07-19 21:34:24 +02:00
|
|
|
let pending_renames = &mut fld.cx.syntax_env.info().pending_renames;
|
2014-06-26 13:18:52 -07:00
|
|
|
let mut rename_fld = IdentRenamer{renames:pending_renames};
|
2014-09-13 19:06:01 +03:00
|
|
|
rename_fld.fold_stmt(x).expect_one("rename_fold didn't return one value")
|
2014-02-07 17:28:07 -05:00
|
|
|
};
|
2014-06-26 13:18:52 -07:00
|
|
|
// expand macros in the statement
|
2014-09-14 20:27:36 -07:00
|
|
|
fld.fold_stmt(renamed_stmt).into_iter()
|
2014-02-07 17:28:07 -05:00
|
|
|
}).collect();
|
2014-09-13 19:06:01 +03:00
|
|
|
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
|
|
|
|
}
|
2013-12-01 00:00:39 +02:00
|
|
|
})
|
2013-05-17 09:41:04 -07:00
|
|
|
}
|
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
fn expand_pat(p: P<ast::Pat>, fld: &mut MacroExpander) -> P<ast::Pat> {
|
|
|
|
match p.node {
|
|
|
|
PatMac(_) => {}
|
|
|
|
_ => return noop_fold_pat(p, fld)
|
2014-05-19 13:59:35 -07:00
|
|
|
}
|
2014-09-13 19:06:01 +03:00
|
|
|
p.map(|ast::Pat {node, span, ..}| {
|
|
|
|
let (pth, tts) = match node {
|
2015-09-20 16:15:37 -04:00
|
|
|
PatMac(mac) => (mac.node.path, mac.node.tts),
|
2014-09-13 19:06:01 +03:00
|
|
|
_ => unreachable!()
|
|
|
|
};
|
2015-01-28 01:01:48 +00:00
|
|
|
if pth.segments.len() > 1 {
|
2014-09-13 19:06:01 +03:00
|
|
|
fld.cx.span_err(pth.span, "expected macro name without module separators");
|
|
|
|
return DummyResult::raw_pat(span);
|
2014-05-19 13:59:35 -07:00
|
|
|
}
|
2015-07-28 18:07:20 +02:00
|
|
|
let extname = pth.segments[0].identifier.name;
|
2015-09-24 23:05:02 +03:00
|
|
|
let marked_after = match fld.cx.syntax_env.find(extname) {
|
2014-09-13 19:06:01 +03:00
|
|
|
None => {
|
|
|
|
fld.cx.span_err(pth.span,
|
2015-01-07 11:58:31 -05:00
|
|
|
&format!("macro undefined: '{}!'",
|
2015-07-28 18:07:20 +02:00
|
|
|
extname));
|
2014-09-13 19:06:01 +03:00
|
|
|
// let compilation continue
|
|
|
|
return DummyResult::raw_pat(span);
|
|
|
|
}
|
2014-05-19 13:59:35 -07:00
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
Some(rc) => match *rc {
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
NormalTT(ref expander, tt_span, allow_internal_unstable) => {
|
2014-09-13 19:06:01 +03:00
|
|
|
fld.cx.bt_push(ExpnInfo {
|
|
|
|
call_site: span,
|
|
|
|
callee: NameAndSpan {
|
2015-08-27 05:41:53 +05:30
|
|
|
format: MacroBang(extname),
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
span: tt_span,
|
|
|
|
allow_internal_unstable: allow_internal_unstable,
|
2014-09-13 19:06:01 +03:00
|
|
|
}
|
|
|
|
});
|
2014-05-19 13:59:35 -07:00
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
let fm = fresh_mark();
|
2015-02-18 14:48:57 -05:00
|
|
|
let marked_before = mark_tts(&tts[..], fm);
|
2014-09-17 19:01:33 +03:00
|
|
|
let mac_span = fld.cx.original_span();
|
Fallout from fixing Issue 25199.
There are two interesting kinds of breakage illustrated here:
1. `Box<Trait>` in many contexts is treated as `Box<Trait + 'static>`,
due to [RFC 599]. However, in a type like `&'a Box<Trait>`, the
`Box<Trait>` type will be expanded to `Box<Trait + 'a>`, again due
to [RFC 599]. This, combined with the fix to Issue 25199, leads to
a borrowck problem due the combination of this function signature
(in src/libstd/net/parser.rs):
```rust
fn read_or<T>(&mut self, parsers: &mut [Box<FnMut(&mut Parser) -> Option<T>>]) -> Option<T>;
```
with this call site (again in src/libstd/net/parser.rs):
```rust
fn read_ip_addr(&mut self) -> Option<IpAddr> {
let ipv4_addr = |p: &mut Parser| p.read_ipv4_addr().map(|v4| IpAddr::V4(v4));
let ipv6_addr = |p: &mut Parser| p.read_ipv6_addr().map(|v6| IpAddr::V6(v6));
self.read_or(&mut [Box::new(ipv4_addr), Box::new(ipv6_addr)])
}
```
yielding borrowck errors like:
```
parser.rs:265:27: 265:69 error: borrowed value does not live long enough
parser.rs:265 self.read_or(&mut [Box::new(ipv4_addr), Box::new(ipv6_addr)])
^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
```
(full log at: https://gist.github.com/pnkfelix/e2e80f1a71580f5d3103 )
The issue here is perhaps subtle: the `parsers` argument is
inferred to be taking a slice of boxed objects with the implicit
lifetime bound attached to the `self` parameter to `read_or`.
Meanwhile, the fix to Issue 25199 (added in a forth-coming commit)
is forcing us to assume that each boxed object may have a
destructor that could refer to state of that lifetime, and
*therefore* that inferred lifetime is required to outlive the boxed
object itself.
In this case, the relevant boxed object here is not going to make
any such references; I believe it is just an artifact of how the
expression was built that it is not assigned type:
`Box<FnMut(&mut Parser) -> Option<T> + 'static>`.
(i.e., mucking with the expression is probably one way to fix this
problem).
But the other way to fix it, adopted here, is to change the
`read_or` method type to force make the (presumably-intended)
`'static` bound explicit on the boxed `FnMut` object.
(Note: this is still just the *first* example of breakage.)
2. In `macro_rules.rs`, the `TTMacroExpander` trait defines a method
with signature:
```rust
fn expand<'cx>(&self, cx: &'cx mut ExtCtxt, ...) -> Box<MacResult+'cx>;
```
taking a `&'cx mut ExtCtxt` as an argument and returning a
`Box<MacResult'cx>`.
The fix to Issue 25199 (added in aforementioned forth-coming
commit) assumes that a value of type `Box<MacResult+'cx>` may, in
its destructor, refer to a reference of lifetime `'cx`; thus the
`'cx` lifetime is forced to outlive the returned value.
Meanwhile, within `expand.rs`, the old code was doing:
```rust
match expander.expand(fld.cx, ...).make_pat() { ... => immutable borrow of fld.cx ... }
```
The problem is that the `'cx` lifetime, inferred for the
`expander.expand` call, has now been extended so that it has to
outlive the temporary R-value returned by `expanded.expand`. But
call is also reborrowing `fld.cx` *mutably*, which means that this
reborrow must end before any immutable borrow of `fld.cx`; but
there is one of those within the match body. (Note that the
temporary R-values for the input expression to `match` all live as
long as the whole `match` expression itself (see Issue #3511 and PR
#11585).
To address this, I moved the construction of the pat value into its
own `let`-statement, so that the `Box<MacResult>` will only live
for as long as the initializing expression for the `let`-statement,
and thus allow the subsequent immutable borrow within the `match`.
[RFC 599]: https://github.com/rust-lang/rfcs/blob/master/text/0599-default-object-bound.md
2015-05-08 14:48:26 +02:00
|
|
|
let pat = expander.expand(fld.cx,
|
|
|
|
mac_span,
|
|
|
|
&marked_before[..]).make_pat();
|
|
|
|
let expanded = match pat {
|
2014-09-13 19:06:01 +03:00
|
|
|
Some(e) => e,
|
|
|
|
None => {
|
|
|
|
fld.cx.span_err(
|
|
|
|
pth.span,
|
2015-01-07 11:58:31 -05:00
|
|
|
&format!(
|
2014-09-13 19:06:01 +03:00
|
|
|
"non-pattern macro in pattern position: {}",
|
2015-07-28 18:07:20 +02:00
|
|
|
extname
|
2015-02-20 14:08:14 -05:00
|
|
|
)
|
2014-09-13 19:06:01 +03:00
|
|
|
);
|
|
|
|
return DummyResult::raw_pat(span);
|
|
|
|
}
|
|
|
|
};
|
2014-07-19 21:34:24 +02:00
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
// mark after:
|
|
|
|
mark_pat(expanded,fm)
|
|
|
|
}
|
|
|
|
_ => {
|
|
|
|
fld.cx.span_err(span,
|
2015-01-07 11:58:31 -05:00
|
|
|
&format!("{}! is not legal in pattern position",
|
2015-07-28 18:07:20 +02:00
|
|
|
extname));
|
2014-09-13 19:06:01 +03:00
|
|
|
return DummyResult::raw_pat(span);
|
|
|
|
}
|
2014-07-19 21:34:24 +02:00
|
|
|
}
|
2014-09-13 19:06:01 +03:00
|
|
|
};
|
2014-05-19 13:59:35 -07:00
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
let fully_expanded =
|
|
|
|
fld.fold_pat(marked_after).node.clone();
|
|
|
|
fld.cx.bt_pop();
|
2014-05-19 13:59:35 -07:00
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
ast::Pat {
|
|
|
|
id: ast::DUMMY_NODE_ID,
|
|
|
|
node: fully_expanded,
|
|
|
|
span: span
|
|
|
|
}
|
|
|
|
})
|
2014-05-19 13:59:35 -07:00
|
|
|
}
|
|
|
|
|
2014-07-02 23:17:10 -07:00
|
|
|
/// 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)
|
2014-03-02 15:26:39 -08:00
|
|
|
pub struct IdentRenamer<'a> {
|
2014-07-02 22:38:30 -07:00
|
|
|
renames: &'a mtwt::RenameList,
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
|
|
|
|
2014-01-09 15:05:33 +02:00
|
|
|
impl<'a> Folder for IdentRenamer<'a> {
|
2014-02-25 04:47:19 +08:00
|
|
|
fn fold_ident(&mut self, id: Ident) -> Ident {
|
2015-09-24 23:05:02 +03:00
|
|
|
Ident::new(id.name, mtwt::apply_renames(self.renames, id.ctxt))
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
2015-01-02 13:39:05 -08:00
|
|
|
fn fold_mac(&mut self, mac: ast::Mac) -> ast::Mac {
|
|
|
|
fold::noop_fold_mac(mac, self)
|
2014-07-09 14:48:12 -07:00
|
|
|
}
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
|
|
|
|
2014-07-02 23:16:01 -07:00
|
|
|
/// A tree-folder that applies every rename in its list to
|
|
|
|
/// the idents that are in PatIdent patterns. This is more narrowly
|
|
|
|
/// 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> {
|
2014-09-13 19:06:01 +03:00
|
|
|
fn fold_pat(&mut self, pat: P<ast::Pat>) -> P<ast::Pat> {
|
2014-07-02 23:16:01 -07:00
|
|
|
match pat.node {
|
2014-09-13 19:06:01 +03:00
|
|
|
ast::PatIdent(..) => {},
|
|
|
|
_ => return noop_fold_pat(pat, self)
|
|
|
|
}
|
|
|
|
|
|
|
|
pat.map(|ast::Pat {id, node, span}| match node {
|
|
|
|
ast::PatIdent(binding_mode, Spanned{span: sp, node: ident}, sub) => {
|
2015-09-24 23:05:02 +03:00
|
|
|
let new_ident = Ident::new(ident.name,
|
|
|
|
mtwt::apply_renames(self.renames, ident.ctxt));
|
2014-07-02 23:16:01 -07:00
|
|
|
let new_node =
|
|
|
|
ast::PatIdent(binding_mode,
|
2014-09-13 19:06:01 +03:00
|
|
|
Spanned{span: self.new_span(sp), node: new_ident},
|
2014-07-02 23:16:01 -07:00
|
|
|
sub.map(|p| self.fold_pat(p)));
|
2014-09-13 19:06:01 +03:00
|
|
|
ast::Pat {
|
|
|
|
id: id,
|
2014-07-02 23:16:01 -07:00
|
|
|
node: new_node,
|
2014-09-13 19:06:01 +03:00
|
|
|
span: self.new_span(span)
|
2014-07-02 23:16:01 -07:00
|
|
|
}
|
|
|
|
},
|
2014-09-13 19:06:01 +03:00
|
|
|
_ => unreachable!()
|
|
|
|
})
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
2015-01-02 13:39:05 -08:00
|
|
|
fn fold_mac(&mut self, mac: ast::Mac) -> ast::Mac {
|
|
|
|
fold::noop_fold_mac(mac, self)
|
2014-07-09 14:48:12 -07:00
|
|
|
}
|
2012-02-04 18:37:24 -07:00
|
|
|
}
|
|
|
|
|
2014-12-02 10:07:41 -08:00
|
|
|
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();
|
2015-04-28 17:34:39 +12:00
|
|
|
expand_decorators(a.clone(), fld, &mut decorator_items, &mut new_attrs);
|
2014-12-02 10:07:41 -08:00
|
|
|
|
|
|
|
let mut new_items: SmallVector<Annotatable> = match a {
|
|
|
|
Annotatable::Item(it) => match it.node {
|
|
|
|
ast::ItemMac(..) => {
|
|
|
|
expand_item_mac(it, fld).into_iter().map(|i| Annotatable::Item(i)).collect()
|
|
|
|
}
|
|
|
|
ast::ItemMod(_) | ast::ItemForeignMod(_) => {
|
|
|
|
let valid_ident =
|
|
|
|
it.ident.name != parse::token::special_idents::invalid.name;
|
|
|
|
|
|
|
|
if valid_ident {
|
|
|
|
fld.cx.mod_push(it.ident);
|
|
|
|
}
|
2015-02-18 14:48:57 -05:00
|
|
|
let macro_use = contains_macro_use(fld, &new_attrs[..]);
|
2014-12-02 10:07:41 -08:00
|
|
|
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()
|
|
|
|
}
|
|
|
|
},
|
2015-03-11 23:38:58 +02:00
|
|
|
|
2015-03-10 12:28:44 +02:00
|
|
|
Annotatable::TraitItem(it) => match it.node {
|
2015-03-11 23:38:58 +02:00
|
|
|
ast::MethodTraitItem(_, Some(_)) => SmallVector::one(it.map(|ti| ast::TraitItem {
|
|
|
|
id: ti.id,
|
|
|
|
ident: ti.ident,
|
|
|
|
attrs: ti.attrs,
|
|
|
|
node: match ti.node {
|
|
|
|
ast::MethodTraitItem(sig, Some(body)) => {
|
|
|
|
let (sig, body) = expand_and_rename_method(sig, body, fld);
|
|
|
|
ast::MethodTraitItem(sig, Some(body))
|
|
|
|
}
|
2015-03-10 12:28:44 +02:00
|
|
|
_ => unreachable!()
|
2015-03-11 23:38:58 +02:00
|
|
|
},
|
|
|
|
span: fld.new_span(ti.span)
|
|
|
|
})),
|
|
|
|
_ => fold::noop_fold_trait_item(it, fld)
|
|
|
|
}.into_iter().map(Annotatable::TraitItem).collect(),
|
|
|
|
|
2015-03-10 12:28:44 +02:00
|
|
|
Annotatable::ImplItem(ii) => {
|
2015-03-11 23:38:58 +02:00
|
|
|
expand_impl_item(ii, fld).into_iter().map(Annotatable::ImplItem).collect()
|
2014-12-02 10:07:41 -08:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2015-01-20 20:28:36 +01:00
|
|
|
new_items.push_all(decorator_items);
|
2014-12-02 10:07:41 -08:00
|
|
|
new_items
|
|
|
|
}
|
|
|
|
|
2015-04-28 17:34:39 +12:00
|
|
|
// 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| {
|
2015-09-24 23:05:02 +03:00
|
|
|
match fld.cx.syntax_env.find(intern(&attr.name())) {
|
2015-04-28 17:34:39 +12:00
|
|
|
Some(rc) => match *rc {
|
|
|
|
$variant(..) => true,
|
|
|
|
_ => false
|
|
|
|
},
|
|
|
|
_ => false
|
|
|
|
}
|
|
|
|
})
|
2014-12-02 10:07:41 -08:00
|
|
|
}
|
2015-04-28 17:34:39 +12:00
|
|
|
}
|
2014-12-02 10:07:41 -08:00
|
|
|
}
|
|
|
|
|
2015-04-28 17:34:39 +12:00
|
|
|
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() {
|
2015-08-27 05:41:53 +05:30
|
|
|
let mname = intern(&attr.name());
|
2015-09-24 23:05:02 +03:00
|
|
|
match fld.cx.syntax_env.find(mname) {
|
2014-12-02 10:07:41 -08:00
|
|
|
Some(rc) => match *rc {
|
2015-04-28 17:34:39 +12:00
|
|
|
MultiDecorator(ref dec) => {
|
|
|
|
attr::mark_used(&attr);
|
|
|
|
|
|
|
|
fld.cx.bt_push(ExpnInfo {
|
|
|
|
call_site: attr.span,
|
|
|
|
callee: NameAndSpan {
|
2015-08-27 05:41:53 +05:30
|
|
|
format: MacroAttribute(mname),
|
2015-04-28 17:34:39 +12:00
|
|
|
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,
|
2015-05-22 21:10:14 +05:30
|
|
|
&a,
|
2015-04-28 17:34:39 +12:00
|
|
|
&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()),
|
2014-12-02 10:07:41 -08:00
|
|
|
},
|
2015-04-28 17:34:39 +12:00
|
|
|
_ => new_attrs.push((*attr).clone()),
|
2014-12-02 10:07:41 -08:00
|
|
|
}
|
2015-04-28 17:34:39 +12:00
|
|
|
}
|
2014-12-02 10:07:41 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
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
|
|
|
|
}
|
|
|
|
|
2015-01-31 12:20:46 -05:00
|
|
|
for attr in &modifiers {
|
2015-08-27 05:41:53 +05:30
|
|
|
let mname = intern(&attr.name());
|
2014-12-02 10:07:41 -08:00
|
|
|
|
2015-09-24 23:05:02 +03:00
|
|
|
match fld.cx.syntax_env.find(mname) {
|
2014-12-02 10:07:41 -08:00
|
|
|
Some(rc) => match *rc {
|
|
|
|
MultiModifier(ref mac) => {
|
|
|
|
attr::mark_used(attr);
|
|
|
|
fld.cx.bt_push(ExpnInfo {
|
|
|
|
call_site: attr.span,
|
|
|
|
callee: NameAndSpan {
|
2015-08-27 05:41:53 +05:30
|
|
|
format: MacroAttribute(mname),
|
2015-04-28 17:34:39 +12:00
|
|
|
span: Some(attr.span),
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
// attributes can do whatever they like,
|
|
|
|
// for now
|
|
|
|
allow_internal_unstable: true,
|
2014-12-02 10:07:41 -08:00
|
|
|
}
|
|
|
|
});
|
|
|
|
it = mac.expand(fld.cx, attr.span, &*attr.node.value, it);
|
|
|
|
fld.cx.bt_pop();
|
|
|
|
}
|
|
|
|
_ => unreachable!()
|
|
|
|
},
|
|
|
|
_ => unreachable!()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Expansion may have added new ItemModifiers.
|
|
|
|
expand_item_multi_modifier(it, fld)
|
|
|
|
}
|
|
|
|
|
2015-03-11 23:38:58 +02:00
|
|
|
fn expand_impl_item(ii: P<ast::ImplItem>, fld: &mut MacroExpander)
|
2015-03-10 12:28:44 +02:00
|
|
|
-> SmallVector<P<ast::ImplItem>> {
|
|
|
|
match ii.node {
|
2015-03-11 23:38:58 +02:00
|
|
|
ast::MethodImplItem(..) => SmallVector::one(ii.map(|ii| ast::ImplItem {
|
|
|
|
id: ii.id,
|
|
|
|
ident: ii.ident,
|
|
|
|
attrs: ii.attrs,
|
|
|
|
vis: ii.vis,
|
|
|
|
node: match ii.node {
|
|
|
|
ast::MethodImplItem(sig, body) => {
|
|
|
|
let (sig, body) = expand_and_rename_method(sig, body, fld);
|
|
|
|
ast::MethodImplItem(sig, body)
|
|
|
|
}
|
|
|
|
_ => unreachable!()
|
|
|
|
},
|
|
|
|
span: fld.new_span(ii.span)
|
|
|
|
})),
|
|
|
|
ast::MacImplItem(_) => {
|
2015-03-10 12:28:44 +02:00
|
|
|
let (span, mac) = ii.and_then(|ii| match ii.node {
|
2015-03-11 23:38:58 +02:00
|
|
|
ast::MacImplItem(mac) => (ii.span, mac),
|
2015-03-10 12:28:44 +02:00
|
|
|
_ => unreachable!()
|
|
|
|
});
|
2015-03-11 23:38:58 +02:00
|
|
|
let maybe_new_items =
|
2015-03-10 12:28:44 +02:00
|
|
|
expand_mac_invoc(mac, span,
|
2015-03-11 23:38:58 +02:00
|
|
|
|r| r.make_impl_items(),
|
|
|
|
|meths, mark| meths.move_map(|m| mark_impl_item(m, mark)),
|
2014-07-13 09:35:48 -07:00
|
|
|
fld);
|
|
|
|
|
2015-03-11 23:38:58 +02:00
|
|
|
match maybe_new_items {
|
|
|
|
Some(impl_items) => {
|
2014-12-12 17:34:28 +01:00
|
|
|
// expand again if necessary
|
2015-03-11 23:38:58 +02:00
|
|
|
let new_items = impl_items.into_iter().flat_map(|ii| {
|
|
|
|
expand_impl_item(ii, fld).into_iter()
|
|
|
|
}).collect();
|
2014-12-12 17:34:28 +01:00
|
|
|
fld.cx.bt_pop();
|
2015-03-11 23:38:58 +02:00
|
|
|
new_items
|
2014-12-12 17:34:28 +01:00
|
|
|
}
|
2014-07-13 09:35:48 -07:00
|
|
|
None => SmallVector::zero()
|
2014-12-12 17:34:28 +01:00
|
|
|
}
|
2014-07-11 21:22:11 -07:00
|
|
|
}
|
2015-03-11 23:38:58 +02:00
|
|
|
_ => fold::noop_fold_impl_item(ii, fld)
|
2015-03-05 04:48:54 +02:00
|
|
|
}
|
2014-07-04 11:24:28 -07:00
|
|
|
}
|
|
|
|
|
2014-07-04 12:19:15 -07:00
|
|
|
/// 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.
|
2014-09-13 19:06:01 +03:00
|
|
|
fn expand_and_rename_fn_decl_and_block(fn_decl: P<ast::FnDecl>, block: P<ast::Block>,
|
2014-07-04 12:19:15 -07:00
|
|
|
fld: &mut MacroExpander)
|
2015-03-11 23:38:58 +02:00
|
|
|
-> (P<ast::FnDecl>, P<ast::Block>) {
|
2014-07-04 12:19:15 -07:00
|
|
|
let expanded_decl = fld.fold_fn_decl(fn_decl);
|
2014-07-07 16:35:15 -07:00
|
|
|
let idents = fn_decl_arg_bindings(&*expanded_decl);
|
2014-07-04 12:19:15 -07:00
|
|
|
let renames =
|
2015-09-24 23:05:02 +03:00
|
|
|
idents.iter().map(|id| (*id,fresh_name(*id))).collect();
|
2014-07-04 12:19:15 -07:00
|
|
|
// first, a renamer for the PatIdents, for the fn_decl:
|
|
|
|
let mut rename_pat_fld = PatIdentRenamer{renames: &renames};
|
2014-09-13 19:06:01 +03:00
|
|
|
let rewritten_fn_decl = rename_pat_fld.fold_fn_decl(expanded_decl);
|
2014-07-04 12:19:15 -07:00
|
|
|
// 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)
|
|
|
|
}
|
|
|
|
|
2015-03-11 23:38:58 +02:00
|
|
|
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,
|
2015-05-05 08:47:04 -04:00
|
|
|
constness: sig.constness,
|
2015-03-11 23:38:58 +02:00
|
|
|
decl: rewritten_fn_decl
|
|
|
|
}, rewritten_body)
|
|
|
|
}
|
|
|
|
|
2015-07-25 21:54:19 -07:00
|
|
|
pub fn expand_type(t: P<ast::Ty>, fld: &mut MacroExpander) -> P<ast::Ty> {
|
|
|
|
let t = match t.node.clone() {
|
|
|
|
ast::Ty_::TyMac(mac) => {
|
2015-07-28 09:00:32 -07:00
|
|
|
if fld.cx.ecfg.features.unwrap().type_macros {
|
|
|
|
let expanded_ty = match expand_mac_invoc(mac, t.span,
|
|
|
|
|r| r.make_ty(),
|
|
|
|
mark_ty,
|
|
|
|
fld) {
|
|
|
|
Some(ty) => ty,
|
|
|
|
None => {
|
|
|
|
return DummyResult::raw_ty(t.span);
|
|
|
|
}
|
|
|
|
};
|
2015-07-25 21:54:19 -07:00
|
|
|
|
2015-07-28 09:00:32 -07:00
|
|
|
// Keep going, outside-in.
|
|
|
|
let fully_expanded = fld.fold_ty(expanded_ty);
|
|
|
|
fld.cx.bt_pop();
|
|
|
|
|
|
|
|
fully_expanded.map(|t| ast::Ty {
|
|
|
|
id: ast::DUMMY_NODE_ID,
|
|
|
|
node: t.node,
|
|
|
|
span: t.span,
|
|
|
|
})
|
|
|
|
} else {
|
|
|
|
feature_gate::emit_feature_err(
|
|
|
|
&fld.cx.parse_sess.span_diagnostic,
|
|
|
|
"type_macros",
|
|
|
|
t.span,
|
2015-09-04 16:37:22 -07:00
|
|
|
feature_gate::GateIssue::Language,
|
|
|
|
"type macros are experimental");
|
2015-08-04 16:04:26 -07:00
|
|
|
|
|
|
|
DummyResult::raw_ty(t.span)
|
2015-07-28 09:00:32 -07:00
|
|
|
}
|
2015-07-25 21:54:19 -07:00
|
|
|
}
|
|
|
|
_ => t
|
|
|
|
};
|
2015-07-28 09:00:32 -07:00
|
|
|
|
2015-07-25 21:54:19 -07:00
|
|
|
fold::noop_fold_ty(t, fld)
|
|
|
|
}
|
|
|
|
|
2014-07-02 23:16:01 -07:00
|
|
|
/// A tree-folder that performs macro expansion
|
2014-08-27 21:46:52 -04:00
|
|
|
pub struct MacroExpander<'a, 'b:'a> {
|
2014-03-27 15:39:48 -07:00
|
|
|
pub cx: &'a mut ExtCtxt<'b>,
|
2014-12-14 15:42:41 +13:00
|
|
|
}
|
|
|
|
|
|
|
|
impl<'a, 'b> MacroExpander<'a, 'b> {
|
|
|
|
pub fn new(cx: &'a mut ExtCtxt<'b>) -> MacroExpander<'a, 'b> {
|
2015-04-03 17:13:52 +13:00
|
|
|
MacroExpander { cx: cx }
|
2014-12-14 15:42:41 +13:00
|
|
|
}
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
|
|
|
|
2014-03-09 16:54:34 +02:00
|
|
|
impl<'a, 'b> Folder for MacroExpander<'a, 'b> {
|
2014-09-13 19:06:01 +03:00
|
|
|
fn fold_expr(&mut self, expr: P<ast::Expr>) -> P<ast::Expr> {
|
2013-12-27 17:43:01 -07:00
|
|
|
expand_expr(expr, self)
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
fn fold_pat(&mut self, pat: P<ast::Pat>) -> P<ast::Pat> {
|
2014-05-19 13:59:35 -07:00
|
|
|
expand_pat(pat, self)
|
|
|
|
}
|
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
fn fold_item(&mut self, item: P<ast::Item>) -> SmallVector<P<ast::Item>> {
|
2015-04-03 17:13:52 +13:00
|
|
|
expand_item(item, self)
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
fn fold_item_underscore(&mut self, item: ast::Item_) -> ast::Item_ {
|
2014-07-04 12:19:15 -07:00
|
|
|
expand_item_underscore(item, self)
|
|
|
|
}
|
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
fn fold_stmt(&mut self, stmt: P<ast::Stmt>) -> SmallVector<P<ast::Stmt>> {
|
2015-04-07 08:21:18 -05:00
|
|
|
expand_stmt(stmt, self)
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
|
|
|
|
2013-12-27 20:34:51 -07:00
|
|
|
fn fold_block(&mut self, block: P<Block>) -> P<Block> {
|
2014-09-13 19:06:01 +03:00
|
|
|
expand_block(block, self)
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
|
|
|
|
2014-09-13 19:06:01 +03:00
|
|
|
fn fold_arm(&mut self, arm: ast::Arm) -> ast::Arm {
|
2014-06-25 15:20:01 -07:00
|
|
|
expand_arm(arm, self)
|
|
|
|
}
|
|
|
|
|
2015-03-10 12:28:44 +02:00
|
|
|
fn fold_trait_item(&mut self, i: P<ast::TraitItem>) -> SmallVector<P<ast::TraitItem>> {
|
|
|
|
expand_annotatable(Annotatable::TraitItem(i), self)
|
|
|
|
.into_iter().map(|i| i.expect_trait_item()).collect()
|
2014-12-02 10:07:41 -08:00
|
|
|
}
|
|
|
|
|
2015-03-10 12:28:44 +02:00
|
|
|
fn fold_impl_item(&mut self, i: P<ast::ImplItem>) -> SmallVector<P<ast::ImplItem>> {
|
|
|
|
expand_annotatable(Annotatable::ImplItem(i), self)
|
|
|
|
.into_iter().map(|i| i.expect_impl_item()).collect()
|
2014-07-04 11:24:28 -07:00
|
|
|
}
|
|
|
|
|
2015-07-25 21:54:19 -07:00
|
|
|
fn fold_ty(&mut self, ty: P<ast::Ty>) -> P<ast::Ty> {
|
|
|
|
expand_type(ty, self)
|
|
|
|
}
|
|
|
|
|
2013-12-27 20:34:51 -07:00
|
|
|
fn new_span(&mut self, span: Span) -> Span {
|
2013-08-29 12:10:02 -07:00
|
|
|
new_span(self.cx, span)
|
|
|
|
}
|
2013-07-16 15:05:50 +10:00
|
|
|
}
|
|
|
|
|
2014-07-02 23:16:01 -07:00
|
|
|
fn new_span(cx: &ExtCtxt, sp: Span) -> Span {
|
|
|
|
/* this discards information in the case of macro-defining macros */
|
|
|
|
Span {
|
|
|
|
lo: sp.lo,
|
|
|
|
hi: sp.hi,
|
2014-09-17 19:01:33 +03:00
|
|
|
expn_id: cx.backtrace(),
|
2014-07-02 23:16:01 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-02-15 21:30:45 +01:00
|
|
|
pub struct ExpansionConfig<'feat> {
|
2014-06-06 13:21:18 -07:00
|
|
|
pub crate_name: String,
|
2015-02-15 21:30:45 +01:00
|
|
|
pub features: Option<&'feat Features>,
|
2015-01-17 23:33:05 +00:00
|
|
|
pub recursion_limit: usize,
|
2015-04-14 15:36:38 +02:00
|
|
|
pub trace_mac: bool,
|
2014-09-26 17:14:23 -07:00
|
|
|
}
|
|
|
|
|
2015-03-06 12:56:28 -08:00
|
|
|
macro_rules! feature_tests {
|
|
|
|
($( fn $getter:ident = $field:ident, )*) => {
|
|
|
|
$(
|
|
|
|
pub fn $getter(&self) -> bool {
|
|
|
|
match self.features {
|
|
|
|
Some(&Features { $field: true, .. }) => true,
|
|
|
|
_ => false,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
)*
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-02-15 21:30:45 +01:00
|
|
|
impl<'feat> ExpansionConfig<'feat> {
|
|
|
|
pub fn default(crate_name: String) -> ExpansionConfig<'static> {
|
2014-09-26 17:14:23 -07:00
|
|
|
ExpansionConfig {
|
|
|
|
crate_name: crate_name,
|
2015-02-15 21:30:45 +01:00
|
|
|
features: None,
|
2014-10-01 11:38:54 +02:00
|
|
|
recursion_limit: 64,
|
2015-04-14 15:36:38 +02:00
|
|
|
trace_mac: false,
|
2014-09-26 17:14:23 -07:00
|
|
|
}
|
|
|
|
}
|
2015-02-15 21:30:45 +01:00
|
|
|
|
2015-03-06 12:56:28 -08:00
|
|
|
feature_tests! {
|
|
|
|
fn enable_quotes = allow_quote,
|
|
|
|
fn enable_asm = allow_asm,
|
|
|
|
fn enable_log_syntax = allow_log_syntax,
|
|
|
|
fn enable_concat_idents = allow_concat_idents,
|
|
|
|
fn enable_trace_macros = allow_trace_macros,
|
|
|
|
fn enable_allow_internal_unstable = allow_internal_unstable,
|
2015-03-06 13:15:54 -08:00
|
|
|
fn enable_custom_derive = allow_custom_derive,
|
2015-06-05 08:31:27 +02:00
|
|
|
fn enable_pushpop_unsafe = allow_pushpop_unsafe,
|
Add #[allow_internal_unstable] to track stability for macros better.
Unstable items used in a macro expansion will now always trigger
stability warnings, *unless* the unstable items are directly inside a
macro marked with `#[allow_internal_unstable]`. IOW, the compiler warns
unless the span of the unstable item is a subspan of the definition of a
macro marked with that attribute.
E.g.
#[allow_internal_unstable]
macro_rules! foo {
($e: expr) => {{
$e;
unstable(); // no warning
only_called_by_foo!();
}}
}
macro_rules! only_called_by_foo {
() => { unstable() } // warning
}
foo!(unstable()) // warning
The unstable inside `foo` is fine, due to the attribute. But the
`unstable` inside `only_called_by_foo` is not, since that macro doesn't
have the attribute, and the `unstable` passed into `foo` is also not
fine since it isn't contained in the macro itself (that is, even though
it is only used directly in the macro).
In the process this makes the stability tracking much more precise,
e.g. previously `println!("{}", unstable())` got no warning, but now it
does. As such, this is a bug fix that may cause [breaking-change]s.
The attribute is definitely feature gated, since it explicitly allows
side-stepping the feature gating system.
2015-03-01 14:09:28 +11:00
|
|
|
}
|
2014-02-28 23:17:38 -08:00
|
|
|
}
|
|
|
|
|
2015-02-15 21:30:45 +01:00
|
|
|
pub fn expand_crate<'feat>(parse_sess: &parse::ParseSess,
|
|
|
|
cfg: ExpansionConfig<'feat>,
|
|
|
|
// these are the macros being imported to this crate:
|
|
|
|
imported_macros: Vec<ast::MacroDef>,
|
|
|
|
user_exts: Vec<NamedSyntaxExtension>,
|
2015-07-13 17:10:44 -07:00
|
|
|
feature_gated_cfgs: &mut Vec<GatedCfg>,
|
2015-02-15 21:30:45 +01:00
|
|
|
c: Crate) -> Crate {
|
2015-07-13 17:10:44 -07:00
|
|
|
let mut cx = ExtCtxt::new(parse_sess, c.config.clone(), cfg,
|
|
|
|
feature_gated_cfgs);
|
2015-07-29 17:01:14 -07:00
|
|
|
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");
|
|
|
|
}
|
2014-09-07 14:57:26 -07:00
|
|
|
|
2014-12-14 15:42:41 +13:00
|
|
|
let mut expander = MacroExpander::new(&mut cx);
|
2013-08-29 12:10:02 -07:00
|
|
|
|
2015-01-31 20:03:04 -05:00
|
|
|
for def in imported_macros {
|
2015-01-01 16:37:47 -08:00
|
|
|
expander.cx.insert_macro(def);
|
2014-05-24 16:16:10 -07:00
|
|
|
}
|
|
|
|
|
2015-01-31 20:03:04 -05:00
|
|
|
for (name, extension) in user_exts {
|
2014-07-19 21:34:24 +02:00
|
|
|
expander.cx.syntax_env.insert(name, extension);
|
2014-05-24 16:16:10 -07:00
|
|
|
}
|
|
|
|
|
2014-07-10 15:41:11 -07:00
|
|
|
let mut ret = expander.fold_crate(c);
|
|
|
|
ret.exported_macros = expander.cx.exported_macros.clone();
|
2013-07-29 01:12:41 -07:00
|
|
|
parse_sess.span_diagnostic.handler().abort_if_errors();
|
|
|
|
return ret;
|
2011-07-06 15:22:23 -07:00
|
|
|
}
|
2013-02-26 10:15:29 -08:00
|
|
|
|
2013-07-13 10:18:37 -07:00
|
|
|
// 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
|
|
|
|
|
2014-02-07 00:38:33 +02:00
|
|
|
// A Marker adds the given mark to the syntax context
|
|
|
|
struct Marker { mark: Mrk }
|
2013-07-09 15:56:21 -07:00
|
|
|
|
2014-02-07 00:38:33 +02:00
|
|
|
impl Folder for Marker {
|
2014-02-25 04:47:19 +08:00
|
|
|
fn fold_ident(&mut self, id: Ident) -> Ident {
|
2015-09-24 23:05:02 +03:00
|
|
|
ast::Ident::new(id.name, mtwt::apply_mark(self.mark, id.ctxt))
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
2014-09-13 19:06:01 +03:00
|
|
|
fn fold_mac(&mut self, Spanned {node, span}: ast::Mac) -> ast::Mac {
|
2013-08-29 12:10:02 -07:00
|
|
|
Spanned {
|
2015-09-20 16:15:37 -04:00
|
|
|
node: Mac_ {
|
|
|
|
path: self.fold_path(node.path),
|
|
|
|
tts: self.fold_tts(&node.tts),
|
|
|
|
ctxt: mtwt::apply_mark(self.mark, node.ctxt),
|
2014-09-13 19:06:01 +03:00
|
|
|
},
|
|
|
|
span: span,
|
2013-08-29 12:10:02 -07:00
|
|
|
}
|
2013-07-09 15:56:21 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-29 05:59:08 -07:00
|
|
|
// apply a given mark to the given token trees. Used prior to expansion of a macro.
|
2014-02-28 13:09:09 -08:00
|
|
|
fn mark_tts(tts: &[TokenTree], m: Mrk) -> Vec<TokenTree> {
|
2014-07-25 21:00:33 +02:00
|
|
|
noop_fold_tts(tts, &mut Marker{mark:m})
|
2013-06-29 05:59:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// apply a given mark to the given expr. Used following the expansion of a macro.
|
2014-09-13 19:06:01 +03:00
|
|
|
fn mark_expr(expr: P<ast::Expr>, m: Mrk) -> P<ast::Expr> {
|
2014-06-26 15:37:18 -07:00
|
|
|
Marker{mark:m}.fold_expr(expr)
|
2013-06-29 05:59:08 -07:00
|
|
|
}
|
|
|
|
|
2014-05-19 13:59:35 -07:00
|
|
|
// apply a given mark to the given pattern. Used following the expansion of a macro.
|
2014-09-13 19:06:01 +03:00
|
|
|
fn mark_pat(pat: P<ast::Pat>, m: Mrk) -> P<ast::Pat> {
|
2014-06-26 15:37:18 -07:00
|
|
|
Marker{mark:m}.fold_pat(pat)
|
2014-05-19 13:59:35 -07:00
|
|
|
}
|
|
|
|
|
2013-06-29 05:59:08 -07:00
|
|
|
// apply a given mark to the given stmt. Used following the expansion of a macro.
|
2015-04-07 08:21:18 -05:00
|
|
|
fn mark_stmt(stmt: P<ast::Stmt>, m: Mrk) -> P<ast::Stmt> {
|
|
|
|
Marker{mark:m}.fold_stmt(stmt)
|
2014-07-13 09:35:48 -07:00
|
|
|
.expect_one("marking a stmt didn't return exactly one stmt")
|
2013-06-29 05:59:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// apply a given mark to the given item. Used following the expansion of a macro.
|
2014-09-13 19:06:01 +03:00
|
|
|
fn mark_item(expr: P<ast::Item>, m: Mrk) -> P<ast::Item> {
|
2014-06-26 15:37:18 -07:00
|
|
|
Marker{mark:m}.fold_item(expr)
|
2014-07-13 09:35:48 -07:00
|
|
|
.expect_one("marking an item didn't return exactly one item")
|
|
|
|
}
|
|
|
|
|
|
|
|
// apply a given mark to the given item. Used following the expansion of a macro.
|
2015-03-11 23:38:58 +02:00
|
|
|
fn mark_impl_item(ii: P<ast::ImplItem>, m: Mrk) -> P<ast::ImplItem> {
|
2015-03-10 12:28:44 +02:00
|
|
|
Marker{mark:m}.fold_impl_item(ii)
|
2015-03-11 23:38:58 +02:00
|
|
|
.expect_one("marking an impl item didn't return exactly one impl item")
|
2013-06-29 05:59:08 -07:00
|
|
|
}
|
|
|
|
|
2015-07-25 21:54:19 -07:00
|
|
|
fn mark_ty(ty: P<ast::Ty>, m: Mrk) -> P<ast::Ty> {
|
|
|
|
Marker { mark: m }.fold_ty(ty)
|
|
|
|
}
|
|
|
|
|
2014-07-09 16:41:13 -07:00
|
|
|
/// Check that there are no macro invocations left in the AST:
|
|
|
|
pub fn check_for_macros(sess: &parse::ParseSess, krate: &ast::Crate) {
|
2014-09-12 13:10:30 +03:00
|
|
|
visit::walk_crate(&mut MacroExterminator{sess:sess}, krate);
|
2014-07-09 16:41:13 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
/// A visitor that ensures that no macro invocations remain in an AST.
|
|
|
|
struct MacroExterminator<'a>{
|
|
|
|
sess: &'a parse::ParseSess
|
|
|
|
}
|
|
|
|
|
2014-09-10 01:54:36 +03:00
|
|
|
impl<'a, 'v> Visitor<'v> for MacroExterminator<'a> {
|
2015-01-02 13:39:05 -08:00
|
|
|
fn visit_mac(&mut self, mac: &ast::Mac) {
|
|
|
|
self.sess.span_diagnostic.span_bug(mac.span,
|
2014-07-09 16:41:13 -07:00
|
|
|
"macro exterminator: expected AST \
|
|
|
|
with no macro invocations");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2013-02-26 10:15:29 -08:00
|
|
|
#[cfg(test)]
|
2015-04-24 17:30:41 +02:00
|
|
|
mod tests {
|
2015-01-07 18:53:58 -08:00
|
|
|
use super::{pattern_bindings, expand_crate};
|
2014-09-26 17:14:23 -07:00
|
|
|
use super::{PatIdentFinder, IdentRenamer, PatIdentRenamer, ExpansionConfig};
|
2013-02-25 14:11:21 -05:00
|
|
|
use ast;
|
2015-01-26 22:56:50 -05:00
|
|
|
use ast::Name;
|
2013-02-25 14:11:21 -05:00
|
|
|
use codemap;
|
2014-02-25 04:47:19 +08:00
|
|
|
use ext::mtwt;
|
2014-07-04 12:18:37 -07:00
|
|
|
use fold::Folder;
|
2013-02-25 14:11:21 -05:00
|
|
|
use parse;
|
2013-07-14 15:25:04 -04:00
|
|
|
use parse::token;
|
2014-03-09 16:54:34 +02:00
|
|
|
use util::parser_testing::{string_to_parser};
|
2013-11-11 17:35:25 -08:00
|
|
|
use util::parser_testing::{string_to_pat, string_to_crate, strs_to_idents};
|
2013-06-29 05:59:08 -07:00
|
|
|
use visit;
|
2013-10-28 19:20:26 +11:00
|
|
|
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)
|
2015-01-03 22:54:18 -05:00
|
|
|
#[derive(Clone)]
|
2014-06-24 17:03:49 -07:00
|
|
|
struct PathExprFinderContext {
|
2014-02-28 13:09:09 -08:00
|
|
|
path_accumulator: Vec<ast::Path> ,
|
2013-10-28 19:20:26 +11:00
|
|
|
}
|
|
|
|
|
2014-09-10 01:54:36 +03:00
|
|
|
impl<'v> Visitor<'v> for PathExprFinderContext {
|
2014-09-12 13:10:30 +03:00
|
|
|
fn visit_expr(&mut self, expr: &ast::Expr) {
|
2015-02-17 19:29:13 +02:00
|
|
|
if let ast::ExprPath(None, ref p) = expr.node {
|
|
|
|
self.path_accumulator.push(p.clone());
|
2013-10-28 19:20:26 +11:00
|
|
|
}
|
2015-02-17 19:29:13 +02:00
|
|
|
visit::walk_expr(self, expr);
|
2013-10-28 19:20:26 +11:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-06-26 14:39:54 -07:00
|
|
|
// 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()};
|
2014-09-12 13:10:30 +03:00
|
|
|
visit::walk_crate(&mut path_finder, the_crate);
|
2014-06-26 14:39:54 -07:00
|
|
|
path_finder.path_accumulator
|
2013-10-28 19:20:26 +11:00
|
|
|
}
|
2013-02-26 10:15:29 -08:00
|
|
|
|
2014-07-04 12:18:37 -07:00
|
|
|
/// A Visitor that extracts the identifiers from a thingy.
|
|
|
|
// as a side note, I'm starting to want to abstract over these....
|
2014-09-12 13:10:30 +03:00
|
|
|
struct IdentFinder {
|
2014-07-04 12:18:37 -07:00
|
|
|
ident_accumulator: Vec<ast::Ident>
|
|
|
|
}
|
2014-06-26 14:39:54 -07:00
|
|
|
|
2014-09-10 01:54:36 +03:00
|
|
|
impl<'v> Visitor<'v> for IdentFinder {
|
2014-09-12 13:10:30 +03:00
|
|
|
fn visit_ident(&mut self, _: codemap::Span, id: ast::Ident){
|
2014-07-04 12:18:37 -07:00
|
|
|
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()};
|
2014-09-12 13:10:30 +03:00
|
|
|
visit::walk_crate(&mut ident_finder, the_crate);
|
2014-07-04 12:18:37 -07:00
|
|
|
ident_finder.ident_accumulator
|
|
|
|
}
|
2014-06-26 14:39:54 -07:00
|
|
|
|
2013-02-26 10:15:29 -08:00
|
|
|
// these following tests are quite fragile, in that they don't test what
|
|
|
|
// *kind* of failure occurs.
|
|
|
|
|
2015-02-15 21:30:45 +01:00
|
|
|
fn test_ecfg() -> ExpansionConfig<'static> {
|
2014-09-26 17:14:23 -07:00
|
|
|
ExpansionConfig::default("test".to_string())
|
|
|
|
}
|
|
|
|
|
2014-06-23 18:45:51 -07:00
|
|
|
// make sure that macros can't escape fns
|
2015-01-31 15:08:25 -08:00
|
|
|
#[should_panic]
|
2013-02-26 10:15:29 -08:00
|
|
|
#[test] fn macros_cant_escape_fns_test () {
|
2014-11-14 09:18:10 -08:00
|
|
|
let src = "fn bogus() {macro_rules! z (() => (3+4));}\
|
2015-01-17 23:55:21 +00:00
|
|
|
fn inty() -> i32 { z!() }".to_string();
|
2015-05-13 23:00:17 +03:00
|
|
|
let sess = parse::ParseSess::new();
|
2013-02-26 10:15:29 -08:00
|
|
|
let crate_ast = parse::parse_crate_from_source_str(
|
2014-05-25 03:17:19 -07:00
|
|
|
"<test>".to_string(),
|
2013-06-13 03:02:55 +10:00
|
|
|
src,
|
2014-03-17 09:55:41 +02:00
|
|
|
Vec::new(), &sess);
|
2013-02-26 10:15:29 -08:00
|
|
|
// should fail:
|
2015-07-13 17:10:44 -07:00
|
|
|
expand_crate(&sess,test_ecfg(),vec!(),vec!(), &mut vec![], crate_ast);
|
2013-02-26 10:15:29 -08:00
|
|
|
}
|
|
|
|
|
2014-06-23 18:45:51 -07:00
|
|
|
// make sure that macros can't escape modules
|
2015-01-31 15:08:25 -08:00
|
|
|
#[should_panic]
|
2013-02-26 10:15:29 -08:00
|
|
|
#[test] fn macros_cant_escape_mods_test () {
|
2014-11-14 09:18:10 -08:00
|
|
|
let src = "mod foo {macro_rules! z (() => (3+4));}\
|
2015-01-17 23:55:21 +00:00
|
|
|
fn inty() -> i32 { z!() }".to_string();
|
2015-05-13 23:00:17 +03:00
|
|
|
let sess = parse::ParseSess::new();
|
2013-02-26 10:15:29 -08:00
|
|
|
let crate_ast = parse::parse_crate_from_source_str(
|
2014-05-25 03:17:19 -07:00
|
|
|
"<test>".to_string(),
|
2013-06-13 03:02:55 +10:00
|
|
|
src,
|
2014-03-17 09:55:41 +02:00
|
|
|
Vec::new(), &sess);
|
2015-07-13 17:10:44 -07:00
|
|
|
expand_crate(&sess,test_ecfg(),vec!(),vec!(), &mut vec![], crate_ast);
|
2013-02-26 10:15:29 -08:00
|
|
|
}
|
|
|
|
|
2014-12-18 20:48:26 -08:00
|
|
|
// macro_use modules should allow macros to escape
|
2013-02-26 10:15:29 -08:00
|
|
|
#[test] fn macros_can_escape_flattened_mods_test () {
|
2014-12-18 20:48:26 -08:00
|
|
|
let src = "#[macro_use] mod foo {macro_rules! z (() => (3+4));}\
|
2015-01-17 23:55:21 +00:00
|
|
|
fn inty() -> i32 { z!() }".to_string();
|
2015-05-13 23:00:17 +03:00
|
|
|
let sess = parse::ParseSess::new();
|
2013-02-26 10:15:29 -08:00
|
|
|
let crate_ast = parse::parse_crate_from_source_str(
|
2014-05-25 03:17:19 -07:00
|
|
|
"<test>".to_string(),
|
2013-06-13 03:02:55 +10:00
|
|
|
src,
|
2014-03-17 09:55:41 +02:00
|
|
|
Vec::new(), &sess);
|
2015-07-13 17:10:44 -07:00
|
|
|
expand_crate(&sess, test_ecfg(), vec!(), vec!(), &mut vec![], crate_ast);
|
2013-02-26 10:15:29 -08:00
|
|
|
}
|
|
|
|
|
2014-05-22 16:57:53 -07:00
|
|
|
fn expand_crate_str(crate_str: String) -> ast::Crate {
|
2015-05-13 23:00:17 +03:00
|
|
|
let ps = parse::ParseSess::new();
|
2015-03-28 21:58:51 +00:00
|
|
|
let crate_ast = panictry!(string_to_parser(&ps, crate_str).parse_crate_mod());
|
2013-06-25 11:40:51 -07:00
|
|
|
// the cfg argument actually does matter, here...
|
2015-07-13 17:10:44 -07:00
|
|
|
expand_crate(&ps,test_ecfg(),vec!(),vec!(), &mut vec![], crate_ast)
|
2013-07-09 16:00:41 -07:00
|
|
|
}
|
|
|
|
|
2014-06-26 14:39:54 -07:00
|
|
|
// find the pat_ident paths in a crate
|
|
|
|
fn crate_bindings(the_crate : &ast::Crate) -> Vec<ast::Ident> {
|
2014-07-04 12:05:43 -07:00
|
|
|
let mut name_finder = PatIdentFinder{ident_accumulator:Vec::new()};
|
2014-09-12 13:10:30 +03:00
|
|
|
visit::walk_crate(&mut name_finder, the_crate);
|
2014-06-26 14:39:54 -07:00
|
|
|
name_finder.ident_accumulator
|
|
|
|
}
|
|
|
|
|
2013-07-09 15:10:16 -07:00
|
|
|
#[test] fn macro_tokens_should_match(){
|
2014-05-07 16:33:43 -07:00
|
|
|
expand_crate_str(
|
2014-11-14 09:18:10 -08:00
|
|
|
"macro_rules! m((a)=>(13)) ;fn main(){m!(a);}".to_string());
|
2013-07-09 15:10:16 -07:00
|
|
|
}
|
|
|
|
|
2014-07-07 17:43:09 -07:00
|
|
|
// should be able to use a bound identifier as a literal in a macro definition:
|
|
|
|
#[test] fn self_macro_parsing(){
|
|
|
|
expand_crate_str(
|
2015-01-17 23:55:21 +00:00
|
|
|
"macro_rules! foo ((zz) => (287;));
|
|
|
|
fn f(zz: i32) {foo!(zz);}".to_string()
|
2014-07-07 17:43:09 -07:00
|
|
|
);
|
|
|
|
}
|
|
|
|
|
2013-06-29 05:59:08 -07:00
|
|
|
// 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
|
2015-05-02 16:25:49 -04:00
|
|
|
// three varrefs, the array [[1, 2], [0]] would indicate that the first
|
2013-06-29 05:59:08 -07:00
|
|
|
// binding should match the second two varrefs, and the second binding
|
|
|
|
// should match the first varref.
|
|
|
|
//
|
2014-06-23 18:45:51 -07:00
|
|
|
// 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.
|
|
|
|
//
|
2013-06-29 05:59:08 -07:00
|
|
|
// The comparisons are done post-mtwt-resolve, so we're comparing renamed
|
|
|
|
// names; differences in marks don't matter any more.
|
2013-07-12 18:35:47 -07:00
|
|
|
//
|
2013-11-11 17:35:25 -08:00
|
|
|
// oog... I also want tests that check "bound-identifier-=?". That is,
|
2013-07-12 18:35:47 -07:00
|
|
|
// 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.
|
2015-01-17 23:33:05 +00:00
|
|
|
type RenamingTest = (&'static str, Vec<Vec<usize>>, bool);
|
2013-06-29 05:59:08 -07:00
|
|
|
|
2013-06-25 11:40:51 -07:00
|
|
|
#[test]
|
|
|
|
fn automatic_renaming () {
|
2014-02-28 13:09:09 -08:00
|
|
|
let tests: Vec<RenamingTest> =
|
|
|
|
vec!(// b & c should get new names throughout, in the expr too:
|
2015-01-17 23:55:21 +00:00
|
|
|
("fn a() -> i32 { let b = 13; let c = b; b+c }",
|
2014-02-28 13:09:09 -08:00
|
|
|
vec!(vec!(0,1),vec!(2)), false),
|
2013-06-07 15:01:28 -07:00
|
|
|
// both x's should be renamed (how is this causing a bug?)
|
2015-01-17 23:55:21 +00:00
|
|
|
("fn main () {let x: i32 = 13;x;}",
|
2014-02-28 13:09:09 -08:00
|
|
|
vec!(vec!(0)), false),
|
2013-06-29 05:59:08 -07:00
|
|
|
// the use of b after the + should be renamed, the other one not:
|
2015-01-17 23:55:21 +00:00
|
|
|
("macro_rules! f (($x:ident) => (b + $x)); fn a() -> i32 { let b = 13; f!(b)}",
|
2014-02-28 13:09:09 -08:00
|
|
|
vec!(vec!(1)), false),
|
2013-06-06 11:21:02 -07:00
|
|
|
// the b before the plus should not be renamed (requires marks)
|
2015-01-17 23:55:21 +00:00
|
|
|
("macro_rules! f (($x:ident) => ({let b=9; ($x + b)})); fn a() -> i32 { f!(b)}",
|
2014-02-28 13:09:09 -08:00
|
|
|
vec!(vec!(1)), false),
|
2013-06-29 05:59:08 -07:00
|
|
|
// 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.
|
2014-11-14 09:18:10 -08:00
|
|
|
("macro_rules! letty(($x:ident) => (let $x = 15;));
|
|
|
|
macro_rules! user(($x:ident) => ({letty!($x); $x}));
|
2015-01-17 23:55:21 +00:00
|
|
|
fn main() -> i32 {user!(z)}",
|
2013-11-11 17:35:25 -08:00
|
|
|
vec!(vec!(0)), false)
|
|
|
|
);
|
2013-09-05 13:28:57 -07:00
|
|
|
for (idx,s) in tests.iter().enumerate() {
|
|
|
|
run_renaming_test(s,idx);
|
2013-06-29 05:59:08 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-11-11 17:35:25 -08:00
|
|
|
// 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]
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn issue_8062(){
|
2013-11-11 17:35:25 -08:00
|
|
|
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]
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn issue_6994(){
|
2013-11-11 17:35:25 -08:00
|
|
|
run_renaming_test(
|
|
|
|
&("macro_rules! g (($x:ident) =>
|
2014-11-14 09:18:10 -08:00
|
|
|
({macro_rules! f(($y:ident)=>({let $y=3;$x}));f!($x)}));
|
2013-11-11 17:35:25 -08:00
|
|
|
fn a(){g!(z)}",
|
|
|
|
vec!(vec!(0)),false),
|
|
|
|
0)
|
2014-06-24 16:34:03 -07:00
|
|
|
}
|
|
|
|
|
2014-06-27 16:10:23 -07:00
|
|
|
// match variable hygiene. Should expand into
|
2014-06-26 15:37:45 -07:00
|
|
|
// fn z() {match 8 {x_1 => {match 9 {x_2 | x_2 if x_2 == x_1 => x_2 + x_1}}}}
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn issue_9384(){
|
2014-06-24 16:34:03 -07:00
|
|
|
run_renaming_test(
|
2014-11-14 09:18:10 -08:00
|
|
|
&("macro_rules! bad_macro (($ex:expr) => ({match 9 {x | x if x == $ex => x + $ex}}));
|
2014-06-26 15:37:45 -07:00
|
|
|
fn z() {match 8 {x => bad_macro!(x)}}",
|
2014-06-24 16:34:03 -07:00
|
|
|
// NB: the third "binding" is the repeat of the second one.
|
2014-06-26 15:37:45 -07:00
|
|
|
vec!(vec!(1,3),vec!(0,2),vec!(0,2)),
|
2014-06-24 16:34:03 -07:00
|
|
|
true),
|
|
|
|
0)
|
2013-11-11 17:35:25 -08:00
|
|
|
}
|
|
|
|
|
2014-06-27 16:10:23 -07:00
|
|
|
// interpolated nodes weren't getting labeled.
|
2014-06-27 13:10:47 -07:00
|
|
|
// should expand into
|
|
|
|
// fn main(){let g1_1 = 13; g1_1}}
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn pat_expand_issue_15221(){
|
2014-06-27 13:10:47 -07:00
|
|
|
run_renaming_test(
|
2014-11-14 09:18:10 -08:00
|
|
|
&("macro_rules! inner ( ($e:pat ) => ($e));
|
|
|
|
macro_rules! outer ( ($e:pat ) => (inner!($e)));
|
2014-06-27 13:10:47 -07:00
|
|
|
fn main() { let outer!(g) = 13; g;}",
|
|
|
|
vec!(vec!(0)),
|
|
|
|
true),
|
|
|
|
0)
|
|
|
|
}
|
|
|
|
|
2013-11-11 17:35:25 -08:00
|
|
|
// create a really evil test case where a $x appears inside a binding of $x
|
2014-07-02 21:27:07 -04:00
|
|
|
// but *shouldn't* bind because it was inserted by a different macro....
|
2013-11-11 17:35:25 -08:00
|
|
|
// can't write this test case until we have macro-generating macros.
|
|
|
|
|
2014-07-04 10:57:17 -07:00
|
|
|
// method arg hygiene
|
2015-01-17 23:55:21 +00:00
|
|
|
// method expands to fn get_x(&self_0, x_1: i32) {self_0 + self_2 + x_3 + x_1}
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn method_arg_hygiene(){
|
2014-07-04 10:57:17 -07:00
|
|
|
run_renaming_test(
|
2014-11-14 09:18:10 -08:00
|
|
|
&("macro_rules! inject_x (()=>(x));
|
|
|
|
macro_rules! inject_self (()=>(self));
|
2014-07-04 10:57:17 -07:00
|
|
|
struct A;
|
2015-01-17 23:55:21 +00:00
|
|
|
impl A{fn get_x(&self, x: i32) {self + inject_self!() + inject_x!() + x;} }",
|
2014-07-04 10:57:17 -07:00
|
|
|
vec!(vec!(0),vec!(3)),
|
|
|
|
true),
|
|
|
|
0)
|
|
|
|
}
|
|
|
|
|
2014-07-07 09:53:41 -07:00
|
|
|
// ooh, got another bite?
|
|
|
|
// expands to struct A; impl A {fn thingy(&self_1) {self_1;}}
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn method_arg_hygiene_2(){
|
2014-07-07 09:53:41 -07:00
|
|
|
run_renaming_test(
|
|
|
|
&("struct A;
|
|
|
|
macro_rules! add_method (($T:ty) =>
|
2014-11-14 09:18:10 -08:00
|
|
|
(impl $T { fn thingy(&self) {self;} }));
|
|
|
|
add_method!(A);",
|
2014-07-07 09:53:41 -07:00
|
|
|
vec!(vec!(0)),
|
|
|
|
true),
|
|
|
|
0)
|
|
|
|
}
|
|
|
|
|
2014-07-04 10:57:28 -07:00
|
|
|
// item fn hygiene
|
2015-01-17 23:55:21 +00:00
|
|
|
// expands to fn q(x_1: i32){fn g(x_2: i32){x_2 + x_1};}
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn issue_9383(){
|
2013-11-11 17:35:25 -08:00
|
|
|
run_renaming_test(
|
2015-01-17 23:55:21 +00:00
|
|
|
&("macro_rules! bad_macro (($ex:expr) => (fn g(x: i32){ x + $ex }));
|
|
|
|
fn q(x: i32) { bad_macro!(x); }",
|
2014-06-30 12:42:33 -07:00
|
|
|
vec!(vec!(1),vec!(0)),true),
|
2013-11-11 17:35:25 -08:00
|
|
|
0)
|
2014-06-30 12:42:33 -07:00
|
|
|
}
|
2013-11-11 17:35:25 -08:00
|
|
|
|
2014-11-19 11:18:17 -05:00
|
|
|
// closure arg hygiene (ExprClosure)
|
2015-01-17 23:55:21 +00:00
|
|
|
// expands to fn f(){(|x_1 : i32| {(x_2 + x_1)})(3);}
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn closure_arg_hygiene(){
|
2014-07-04 11:02:17 -07:00
|
|
|
run_renaming_test(
|
2014-11-14 09:18:10 -08:00
|
|
|
&("macro_rules! inject_x (()=>(x));
|
2015-01-17 23:55:21 +00:00
|
|
|
fn f(){(|x : i32| {(inject_x!() + x)})(3);}",
|
2014-07-04 11:02:17 -07:00
|
|
|
vec!(vec!(1)),
|
|
|
|
true),
|
|
|
|
0)
|
|
|
|
}
|
|
|
|
|
2014-07-06 15:10:57 -07:00
|
|
|
// macro_rules in method position. Sadly, unimplemented.
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn macro_in_method_posn(){
|
2014-07-07 15:12:31 -07:00
|
|
|
expand_crate_str(
|
2015-01-17 23:55:21 +00:00
|
|
|
"macro_rules! my_method (() => (fn thirteen(&self) -> i32 {13}));
|
2014-07-07 15:12:31 -07:00
|
|
|
struct A;
|
2014-11-14 09:18:10 -08:00
|
|
|
impl A{ my_method!(); }
|
2014-07-07 15:12:31 -07:00
|
|
|
fn f(){A.thirteen;}".to_string());
|
|
|
|
}
|
|
|
|
|
2014-07-06 15:10:57 -07:00
|
|
|
// another nested macro
|
|
|
|
// expands to impl Entries {fn size_hint(&self_1) {self_1;}
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn item_macro_workaround(){
|
2014-07-06 15:10:57 -07:00
|
|
|
run_renaming_test(
|
|
|
|
&("macro_rules! item { ($i:item) => {$i}}
|
|
|
|
struct Entries;
|
|
|
|
macro_rules! iterator_impl {
|
2014-11-14 09:18:10 -08:00
|
|
|
() => { item!( impl Entries { fn size_hint(&self) { self;}});}}
|
2014-07-06 15:10:57 -07:00
|
|
|
iterator_impl! { }",
|
|
|
|
vec!(vec!(0)), true),
|
|
|
|
0)
|
|
|
|
}
|
|
|
|
|
2013-07-12 18:35:47 -07:00
|
|
|
// run one of the renaming tests
|
2015-01-17 23:33:05 +00:00
|
|
|
fn run_renaming_test(t: &RenamingTest, test_idx: usize) {
|
2013-09-05 09:29:31 -07:00
|
|
|
let invalid_name = token::special_idents::invalid.name;
|
2013-07-12 18:35:47 -07:00
|
|
|
let (teststr, bound_connections, bound_ident_check) = match *t {
|
2014-07-16 22:37:28 +02:00
|
|
|
(ref str,ref conns, bic) => (str.to_string(), conns.clone(), bic)
|
2013-06-29 05:59:08 -07:00
|
|
|
};
|
2014-05-25 03:17:19 -07:00
|
|
|
let cr = expand_crate_str(teststr.to_string());
|
2014-06-26 14:39:54 -07:00
|
|
|
let bindings = crate_bindings(&cr);
|
|
|
|
let varrefs = crate_varrefs(&cr);
|
2013-10-28 19:20:26 +11:00
|
|
|
|
2013-06-29 05:59:08 -07:00
|
|
|
// must be one check clause for each binding:
|
|
|
|
assert_eq!(bindings.len(),bound_connections.len());
|
|
|
|
for (binding_idx,shouldmatch) in bound_connections.iter().enumerate() {
|
2014-10-14 23:05:01 -07:00
|
|
|
let binding_name = mtwt::resolve(bindings[binding_idx]);
|
|
|
|
let binding_marks = mtwt::marksof(bindings[binding_idx].ctxt, invalid_name);
|
2013-06-29 05:59:08 -07:00
|
|
|
// shouldmatch can't name varrefs that don't exist:
|
2015-03-24 16:53:34 -07:00
|
|
|
assert!((shouldmatch.is_empty()) ||
|
2013-06-29 05:59:08 -07:00
|
|
|
(varrefs.len() > *shouldmatch.iter().max().unwrap()));
|
|
|
|
for (idx,varref) in varrefs.iter().enumerate() {
|
2015-02-01 12:44:15 -05:00
|
|
|
let print_hygiene_debug_info = || {
|
2014-07-06 15:06:32 -07:00
|
|
|
// 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,
|
2014-10-09 15:17:22 -04:00
|
|
|
None => panic!("varref with 0 path segments?")
|
2014-07-06 15:06:32 -07:00
|
|
|
};
|
|
|
|
let varref_name = mtwt::resolve(final_varref_ident);
|
|
|
|
let varref_idents : Vec<ast::Ident>
|
|
|
|
= varref.segments.iter().map(|s| s.identifier)
|
|
|
|
.collect();
|
2015-01-07 18:53:58 -08:00
|
|
|
println!("varref #{}: {:?}, resolves to {}",idx, varref_idents, varref_name);
|
2015-07-28 18:07:20 +02:00
|
|
|
println!("varref's first segment's string: \"{}\"", final_varref_ident);
|
2014-07-06 15:06:32 -07:00
|
|
|
println!("binding #{}: {}, resolves to {}",
|
2014-10-14 23:05:01 -07:00
|
|
|
binding_idx, bindings[binding_idx], binding_name);
|
2014-07-06 15:06:32 -07:00
|
|
|
mtwt::with_sctable(|x| mtwt::display_sctable(x));
|
|
|
|
};
|
2013-06-29 05:59:08 -07:00
|
|
|
if shouldmatch.contains(&idx) {
|
|
|
|
// it should be a path of length 1, and it should
|
2013-07-12 18:35:47 -07:00
|
|
|
// be free-identifier=? or bound-identifier=? to the given binding
|
2013-06-29 05:59:08 -07:00
|
|
|
assert_eq!(varref.segments.len(),1);
|
2014-10-14 23:05:01 -07:00
|
|
|
let varref_name = mtwt::resolve(varref.segments[0].identifier);
|
|
|
|
let varref_marks = mtwt::marksof(varref.segments[0]
|
2014-02-25 04:47:19 +08:00
|
|
|
.identifier
|
|
|
|
.ctxt,
|
|
|
|
invalid_name);
|
2014-01-30 14:28:44 -08:00
|
|
|
if !(varref_name==binding_name) {
|
2014-01-09 21:06:55 +11:00
|
|
|
println!("uh oh, should match but doesn't:");
|
2014-07-06 15:06:32 -07:00
|
|
|
print_hygiene_debug_info();
|
2013-06-29 05:59:08 -07:00
|
|
|
}
|
|
|
|
assert_eq!(varref_name,binding_name);
|
2014-01-30 14:28:44 -08:00
|
|
|
if bound_ident_check {
|
2013-09-05 09:29:31 -07:00
|
|
|
// we're checking bound-identifier=?, and the marks
|
|
|
|
// should be the same, too:
|
2013-07-12 18:35:47 -07:00
|
|
|
assert_eq!(varref_marks,binding_marks.clone());
|
|
|
|
}
|
2013-06-29 05:59:08 -07:00
|
|
|
} else {
|
2014-10-14 23:05:01 -07:00
|
|
|
let varref_name = mtwt::resolve(varref.segments[0].identifier);
|
2013-06-29 05:59:08 -07:00
|
|
|
let fail = (varref.segments.len() == 1)
|
2014-07-04 12:18:37 -07:00
|
|
|
&& (varref_name == binding_name);
|
2013-06-29 05:59:08 -07:00
|
|
|
// temp debugging:
|
2014-01-30 14:28:44 -08:00
|
|
|
if fail {
|
2013-09-05 13:28:57 -07:00
|
|
|
println!("failure on test {}",test_idx);
|
|
|
|
println!("text of test case: \"{}\"", teststr);
|
|
|
|
println!("");
|
|
|
|
println!("uh oh, matches but shouldn't:");
|
2014-07-06 15:06:32 -07:00
|
|
|
print_hygiene_debug_info();
|
2013-06-29 05:59:08 -07:00
|
|
|
}
|
|
|
|
assert!(!fail);
|
|
|
|
}
|
|
|
|
}
|
2013-06-25 11:40:51 -07:00
|
|
|
}
|
|
|
|
}
|
2013-05-21 16:57:21 -07:00
|
|
|
|
2015-03-28 20:16:11 -07:00
|
|
|
#[test]
|
|
|
|
fn fmt_in_macro_used_inside_module_macro() {
|
2014-11-14 09:18:10 -08:00
|
|
|
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!();
|
2014-05-25 03:17:19 -07:00
|
|
|
".to_string();
|
2013-07-12 18:35:47 -07:00
|
|
|
let cr = expand_crate_str(crate_str);
|
|
|
|
// find the xx binding
|
2014-06-26 14:39:54 -07:00
|
|
|
let bindings = crate_bindings(&cr);
|
2014-02-28 13:09:09 -08:00
|
|
|
let cxbinds: Vec<&ast::Ident> =
|
2015-09-24 23:05:02 +03:00
|
|
|
bindings.iter().filter(|b| b.name.as_str() == "xx").collect();
|
2015-02-18 14:48:57 -05:00
|
|
|
let cxbinds: &[&ast::Ident] = &cxbinds[..];
|
2015-04-15 22:12:12 -07:00
|
|
|
let cxbind = match (cxbinds.len(), cxbinds.get(0)) {
|
|
|
|
(1, Some(b)) => *b,
|
2014-10-09 15:17:22 -04:00
|
|
|
_ => panic!("expected just one binding for ext_cx")
|
2013-07-12 18:35:47 -07:00
|
|
|
};
|
2014-02-25 04:47:19 +08:00
|
|
|
let resolved_binding = mtwt::resolve(*cxbind);
|
2014-06-26 14:39:54 -07:00
|
|
|
let varrefs = crate_varrefs(&cr);
|
2013-10-28 19:20:26 +11:00
|
|
|
|
2013-07-12 18:35:47 -07:00
|
|
|
// the xx binding should bind all of the xx varrefs:
|
2014-02-14 07:07:09 +02:00
|
|
|
for (idx,v) in varrefs.iter().filter(|p| {
|
2014-01-31 15:26:59 -08:00
|
|
|
p.segments.len() == 1
|
2015-09-24 23:05:02 +03:00
|
|
|
&& p.segments[0].identifier.name.as_str() == "xx"
|
2014-01-31 15:26:59 -08:00
|
|
|
}).enumerate() {
|
2014-10-14 23:05:01 -07:00
|
|
|
if mtwt::resolve(v.segments[0].identifier) != resolved_binding {
|
2014-01-09 21:06:55 +11:00
|
|
|
println!("uh oh, xx binding didn't match xx varref:");
|
2014-10-14 21:59:41 -04:00
|
|
|
println!("this is xx varref \\# {}", idx);
|
|
|
|
println!("binding: {}", cxbind);
|
|
|
|
println!("resolves to: {}", resolved_binding);
|
2014-10-14 23:05:01 -07:00
|
|
|
println!("varref: {}", v.segments[0].identifier);
|
2014-10-14 21:59:41 -04:00
|
|
|
println!("resolves to: {}",
|
2014-10-14 23:05:01 -07:00
|
|
|
mtwt::resolve(v.segments[0].identifier));
|
2014-02-25 04:47:19 +08:00
|
|
|
mtwt::with_sctable(|x| mtwt::display_sctable(x));
|
2013-07-12 18:35:47 -07:00
|
|
|
}
|
2014-10-14 23:05:01 -07:00
|
|
|
assert_eq!(mtwt::resolve(v.segments[0].identifier),
|
2014-02-28 12:54:01 -08:00
|
|
|
resolved_binding);
|
2013-07-12 18:35:47 -07:00
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2013-06-04 14:56:33 -07:00
|
|
|
#[test]
|
|
|
|
fn pat_idents(){
|
2014-05-07 16:33:43 -07:00
|
|
|
let pat = string_to_pat(
|
2014-05-25 03:17:19 -07:00
|
|
|
"(a,Foo{x:c @ (b,9),y:Bar(4,d)})".to_string());
|
2014-07-07 16:35:15 -07:00
|
|
|
let idents = pattern_bindings(&*pat);
|
2014-06-26 14:39:54 -07:00
|
|
|
assert_eq!(idents, strs_to_idents(vec!("a","c","b","d")));
|
2013-06-04 14:56:33 -07:00
|
|
|
}
|
2013-06-07 12:28:03 -07:00
|
|
|
|
2014-06-24 16:33:44 -07:00
|
|
|
// 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.
|
2013-11-11 17:35:25 -08:00
|
|
|
#[test]
|
2014-07-04 12:18:37 -07:00
|
|
|
fn crate_bindings_test(){
|
2015-01-17 23:55:21 +00:00
|
|
|
let the_crate = string_to_crate("fn main (a: i32) -> i32 {|b| {
|
2014-06-24 16:33:44 -07:00
|
|
|
match 34 {None => 3, Some(i) | i => j, Foo{k:z,l:y} => \"banana\"}} }".to_string());
|
2014-06-26 14:39:54 -07:00
|
|
|
let idents = crate_bindings(&the_crate);
|
|
|
|
assert_eq!(idents, strs_to_idents(vec!("a","b","None","i","i","z","y")));
|
2013-11-11 17:35:25 -08:00
|
|
|
}
|
|
|
|
|
2014-07-04 12:18:37 -07:00
|
|
|
// test the IdentRenamer directly
|
|
|
|
#[test]
|
|
|
|
fn ident_renamer_test () {
|
2015-01-17 23:55:21 +00:00
|
|
|
let the_crate = string_to_crate("fn f(x: i32){let x = x; x}".to_string());
|
2014-07-04 12:18:37 -07:00
|
|
|
let f_ident = token::str_to_ident("f");
|
|
|
|
let x_ident = token::str_to_ident("x");
|
2015-01-17 23:55:21 +00:00
|
|
|
let int_ident = token::str_to_ident("i32");
|
2014-07-08 22:28:52 -07:00
|
|
|
let renames = vec!((x_ident,Name(16)));
|
2014-07-04 12:18:37 -07:00
|
|
|
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();
|
2015-02-24 21:15:45 +03:00
|
|
|
assert_eq!(resolved, [f_ident.name,Name(16),int_ident.name,Name(16),Name(16),Name(16)]);
|
2014-07-04 12:18:37 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// test the PatIdentRenamer; only PatIdents get renamed
|
|
|
|
#[test]
|
|
|
|
fn pat_ident_renamer_test () {
|
2015-01-17 23:55:21 +00:00
|
|
|
let the_crate = string_to_crate("fn f(x: i32){let x = x; x}".to_string());
|
2014-07-04 12:18:37 -07:00
|
|
|
let f_ident = token::str_to_ident("f");
|
|
|
|
let x_ident = token::str_to_ident("x");
|
2015-01-17 23:55:21 +00:00
|
|
|
let int_ident = token::str_to_ident("i32");
|
2014-07-08 22:28:52 -07:00
|
|
|
let renames = vec!((x_ident,Name(16)));
|
2014-07-04 12:18:37 -07:00
|
|
|
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;
|
2015-02-24 21:15:45 +03:00
|
|
|
assert_eq!(resolved, [f_ident.name,Name(16),int_ident.name,Name(16),x_name,x_name]);
|
2014-07-04 12:18:37 -07:00
|
|
|
}
|
2013-02-26 10:15:29 -08:00
|
|
|
}
|