9d7b130041
For the benefit of the pretty printer we want to keep track of how string literals in the ast were originally represented in the source code. This commit changes parser functions so they don't extract strings from the token stream without at least also returning what style of string literal it was. This is stored in the resulting ast node for string literals, obviously, for the package id in `extern mod = r"package id"` view items, for the inline asm in `asm!()` invocations. For `asm!()`'s other arguments or for `extern "Rust" fn()` items, I just the style of string, because it seemed disproportionally cumbersome to thread that information through the string processing that happens with those string literals, given the limited advantage raw string literals would provide in these positions. The other syntax extensions don't seem to store passed string literals in the ast, so they also discard the style of strings they parse.
60 lines
1.8 KiB
Rust
60 lines
1.8 KiB
Rust
// Copyright 2012-2013 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.
|
|
|
|
/*
|
|
* The compiler code necessary to support the env! extension. Eventually this
|
|
* should all get sucked into either the compiler syntax extension plugin
|
|
* interface.
|
|
*/
|
|
|
|
use ast;
|
|
use codemap::Span;
|
|
use ext::base::*;
|
|
use ext::base;
|
|
use ext::build::AstBuilder;
|
|
|
|
use std::os;
|
|
|
|
pub fn expand_option_env(cx: @ExtCtxt, sp: Span, tts: &[ast::token_tree])
|
|
-> base::MacResult {
|
|
let var = get_single_str_from_tts(cx, sp, tts, "option_env!");
|
|
|
|
let e = match os::getenv(var) {
|
|
None => quote_expr!(cx, ::std::option::None::<&'static str>),
|
|
Some(s) => quote_expr!(cx, ::std::option::Some($s))
|
|
};
|
|
MRExpr(e)
|
|
}
|
|
|
|
pub fn expand_env(cx: @ExtCtxt, sp: Span, tts: &[ast::token_tree])
|
|
-> base::MacResult {
|
|
let exprs = get_exprs_from_tts(cx, sp, tts);
|
|
|
|
if exprs.len() == 0 {
|
|
cx.span_fatal(sp, "env! takes 1 or 2 arguments");
|
|
}
|
|
|
|
let (var, _var_str_style) = expr_to_str(cx, exprs[0], "expected string literal");
|
|
let msg = match exprs.len() {
|
|
1 => format!("Environment variable {} not defined", var).to_managed(),
|
|
2 => {
|
|
let (s, _style) = expr_to_str(cx, exprs[1], "expected string literal");
|
|
s
|
|
}
|
|
_ => cx.span_fatal(sp, "env! takes 1 or 2 arguments")
|
|
};
|
|
|
|
let e = match os::getenv(var) {
|
|
None => cx.span_fatal(sp, msg),
|
|
Some(s) => cx.expr_str(sp, s.to_managed())
|
|
};
|
|
MRExpr(e)
|
|
}
|