Skip to content

Commit

Permalink
Auto merge of #34010 - jseyfried:decorate_expanded, r=nrc
Browse files Browse the repository at this point in the history
Run decorators on expanded AST

Fixes #32950.
r? @nrc
  • Loading branch information
bors committed Jun 8, 2016
2 parents 368f6ae + 635a82e commit ff13155
Show file tree
Hide file tree
Showing 2 changed files with 32 additions and 13 deletions.
27 changes: 14 additions & 13 deletions src/libsyntax/ext/expand.rs
Original file line number Diff line number Diff line change
Expand Up @@ -723,11 +723,7 @@ fn expand_annotatable(a: Annotatable,
-> SmallVector<Annotatable> {
let a = expand_item_multi_modifier(a, fld);

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

let mut new_items: SmallVector<Annotatable> = match a {
let new_items: SmallVector<Annotatable> = match a {
Annotatable::Item(it) => match it.node {
ast::ItemKind::Mac(..) => {
let new_items: SmallVector<P<ast::Item>> = it.and_then(|it| match it.node {
Expand All @@ -745,7 +741,7 @@ fn expand_annotatable(a: Annotatable,
if valid_ident {
fld.cx.mod_push(it.ident);
}
let macro_use = contains_macro_use(fld, &new_attrs[..]);
let macro_use = contains_macro_use(fld, &it.attrs);
let result = with_exts_frame!(fld.cx.syntax_env,
macro_use,
noop_fold_item(it, fld));
Expand All @@ -754,13 +750,7 @@ fn expand_annotatable(a: Annotatable,
}
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()
}
_ => noop_fold_item(it, fld).into_iter().map(|i| Annotatable::Item(i)).collect(),
},

Annotatable::TraitItem(it) => match it.node {
Expand Down Expand Up @@ -789,6 +779,17 @@ fn expand_annotatable(a: Annotatable,
}
};

new_items.into_iter().flat_map(|a| decorate(a, fld)).collect()
}

fn decorate(a: Annotatable, fld: &mut MacroExpander) -> SmallVector<Annotatable> {
let mut decorator_items = SmallVector::zero();
let mut new_attrs = Vec::new();
expand_decorators(a.clone(), fld, &mut decorator_items, &mut new_attrs);
let decorator_items =
decorator_items.into_iter().flat_map(|a| expand_annotatable(a, fld)).collect();

let mut new_items = SmallVector::one(a.fold_attrs(new_attrs));
new_items.push_all(decorator_items);
new_items
}
Expand Down
18 changes: 18 additions & 0 deletions src/test/compile-fail/issue-32950.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
// Copyright 2016 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

#![feature(type_macros, concat_idents, rustc_attrs)]
#![allow(unused)]

#[derive(Debug)] struct FooBar;
#[derive(Debug)] struct Baz<T>(T, concat_idents!(Foo, Bar));

#[rustc_error]
fn main() {} //~ ERROR compilation successful

0 comments on commit ff13155

Please sign in to comment.