1 // Copyright 2012 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
14 use ext::hygiene::{Mark, SyntaxContext};
15 use symbol::{Symbol, keywords};
16 use syntax_pos::{DUMMY_SP, Span};
17 use codemap::{ExpnInfo, NameAndSpan, MacroAttribute, dummy_spanned, respan};
19 use tokenstream::TokenStream;
21 /// Craft a span that will be ignored by the stability lint's
22 /// call to codemap's `is_internal` check.
23 /// The expanded code uses the unstable `#[prelude_import]` attribute.
24 fn ignored_span(sp: Span) -> Span {
25 let mark = Mark::fresh(Mark::root());
26 mark.set_expn_info(ExpnInfo {
29 format: MacroAttribute(Symbol::intern("std_inject")),
31 allow_internal_unstable: true,
32 allow_internal_unsafe: false,
35 sp.with_ctxt(SyntaxContext::empty().apply_mark(mark))
38 pub fn injected_crate_name() -> Option<&'static str> {
39 INJECTED_CRATE_NAME.with(|name| name.get())
43 static INJECTED_CRATE_NAME: Cell<Option<&'static str>> = Cell::new(None);
46 pub fn maybe_inject_crates_ref(mut krate: ast::Crate, alt_std_name: Option<&str>) -> ast::Crate {
47 // the first name in this list is the crate name of the crate with the prelude
48 let names: &[&str] = if attr::contains_name(&krate.attrs, "no_core") {
50 } else if attr::contains_name(&krate.attrs, "no_std") {
51 if attr::contains_name(&krate.attrs, "compiler_builtins") {
54 &["core", "compiler_builtins"]
61 krate.module.items.insert(0, P(ast::Item {
62 attrs: vec![attr::mk_attr_outer(DUMMY_SP,
64 attr::mk_word_item(ast::Ident::from_str("macro_use")))],
65 vis: dummy_spanned(ast::VisibilityKind::Inherited),
66 node: ast::ItemKind::ExternCrate(alt_std_name.map(Symbol::intern)),
67 ident: ast::Ident::from_str(name),
68 id: ast::DUMMY_NODE_ID,
74 // the crates have been injected, the assumption is that the first one is the one with
78 INJECTED_CRATE_NAME.with(|opt_name| opt_name.set(Some(name)));
80 let span = ignored_span(DUMMY_SP);
81 krate.module.items.insert(0, P(ast::Item {
82 attrs: vec![ast::Attribute {
83 style: ast::AttrStyle::Outer,
84 path: ast::Path::from_ident(ast::Ident::new(Symbol::intern("prelude_import"), span)),
85 tokens: TokenStream::empty(),
86 id: attr::mk_attr_id(),
87 is_sugared_doc: false,
90 vis: respan(span.shrink_to_lo(), ast::VisibilityKind::Inherited),
91 node: ast::ItemKind::Use(P(ast::UseTree {
93 segments: [name, "prelude", "v1"].into_iter().map(|name| {
94 ast::PathSegment::from_ident(ast::Ident::from_str(name))
98 kind: ast::UseTreeKind::Glob,
101 id: ast::DUMMY_NODE_ID,
102 ident: keywords::Invalid.ident(),