96 lines
3.3 KiB
Rust
96 lines
3.3 KiB
Rust
// Copyright 2012 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.
|
|
|
|
use ast;
|
|
use attr;
|
|
use codemap::{DUMMY_SP, Span, ExpnInfo, NameAndSpan, MacroAttribute};
|
|
use codemap;
|
|
use parse::token::{intern, InternedString, keywords};
|
|
use parse::{token, ParseSess};
|
|
use ptr::P;
|
|
|
|
/// Craft a span that will be ignored by the stability lint's
|
|
/// call to codemap's is_internal check.
|
|
/// The expanded code uses the unstable `#[prelude_import]` attribute.
|
|
fn ignored_span(sess: &ParseSess, sp: Span) -> Span {
|
|
let info = ExpnInfo {
|
|
call_site: DUMMY_SP,
|
|
callee: NameAndSpan {
|
|
format: MacroAttribute(intern("std_inject")),
|
|
span: None,
|
|
allow_internal_unstable: true,
|
|
}
|
|
};
|
|
let expn_id = sess.codemap().record_expansion(info);
|
|
let mut sp = sp;
|
|
sp.expn_id = expn_id;
|
|
return sp;
|
|
}
|
|
|
|
pub fn no_core(krate: &ast::Crate) -> bool {
|
|
attr::contains_name(&krate.attrs, "no_core")
|
|
}
|
|
|
|
pub fn no_std(krate: &ast::Crate) -> bool {
|
|
attr::contains_name(&krate.attrs, "no_std") || no_core(krate)
|
|
}
|
|
|
|
pub fn maybe_inject_crates_ref(sess: &ParseSess,
|
|
mut krate: ast::Crate,
|
|
alt_std_name: Option<String>)
|
|
-> ast::Crate {
|
|
if no_core(&krate) {
|
|
return krate;
|
|
}
|
|
|
|
let name = if no_std(&krate) { "core" } else { "std" };
|
|
let crate_name = token::intern(&alt_std_name.unwrap_or(name.to_string()));
|
|
|
|
krate.module.items.insert(0, P(ast::Item {
|
|
attrs: vec![attr::mk_attr_outer(attr::mk_attr_id(),
|
|
attr::mk_word_item(InternedString::new("macro_use")))],
|
|
vis: ast::Visibility::Inherited,
|
|
node: ast::ItemKind::ExternCrate(Some(crate_name)),
|
|
ident: token::str_to_ident(name),
|
|
id: ast::DUMMY_NODE_ID,
|
|
span: DUMMY_SP,
|
|
}));
|
|
|
|
let span = ignored_span(sess, DUMMY_SP);
|
|
krate.module.items.insert(0, P(ast::Item {
|
|
attrs: vec![ast::Attribute {
|
|
node: ast::Attribute_ {
|
|
style: ast::AttrStyle::Outer,
|
|
value: P(ast::MetaItem {
|
|
node: ast::MetaItemKind::Word(token::intern_and_get_ident("prelude_import")),
|
|
span: span,
|
|
}),
|
|
id: attr::mk_attr_id(),
|
|
is_sugared_doc: false,
|
|
},
|
|
span: span,
|
|
}],
|
|
vis: ast::Visibility::Inherited,
|
|
node: ast::ItemKind::Use(P(codemap::dummy_spanned(ast::ViewPathGlob(ast::Path {
|
|
global: false,
|
|
segments: vec![name, "prelude", "v1"].into_iter().map(|name| ast::PathSegment {
|
|
identifier: token::str_to_ident(name),
|
|
parameters: ast::PathParameters::none(),
|
|
}).collect(),
|
|
span: span,
|
|
})))),
|
|
id: ast::DUMMY_NODE_ID,
|
|
ident: keywords::Invalid.ident(),
|
|
span: span,
|
|
}));
|
|
|
|
krate
|
|
}
|