Skip to content

Commit

Permalink
auto merge of #8188 : huonw/rust/cfg-macro, r=pcwalton
Browse files Browse the repository at this point in the history
Example:

    if cfg!(test) {
       calculation_to_run_only_when_testing();
    }

Closes #8130.
  • Loading branch information
bors committed Aug 2, 2013
2 parents dbde42e + e995d99 commit 2460170
Show file tree
Hide file tree
Showing 4 changed files with 83 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/libsyntax/ext/base.rs
Original file line number Diff line number Diff line change
Expand Up @@ -199,6 +199,8 @@ pub fn syntax_expander_table() -> SyntaxEnv {
ext::source_util::expand_mod));
syntax_expanders.insert(intern(&"asm"),
builtin_normal_tt(ext::asm::expand_asm));
syntax_expanders.insert(intern(&"cfg"),
builtin_normal_tt(ext::cfg::expand_cfg));
syntax_expanders.insert(
intern(&"trace_macros"),
builtin_normal_tt(ext::trace_macros::expand_trace_macros));
Expand Down
45 changes: 45 additions & 0 deletions src/libsyntax/ext/cfg.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
// Copyright 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 cfg! extension, which
expands to a literal `true` or `false` based on whether the given cfgs
match the current compilation environment.
*/

use ast;
use codemap::span;
use ext::base::*;
use ext::base;
use ext::build::AstBuilder;
use attr;
use attr::*;
use parse;
use parse::token;
use parse::attr::parser_attr;

pub fn expand_cfg(cx: @ExtCtxt, sp: span, tts: &[ast::token_tree]) -> base::MacResult {
let p = parse::new_parser_from_tts(cx.parse_sess(), cx.cfg(), tts.to_owned());

let mut cfgs = ~[];
// parse `cfg!(meta_item, meta_item(x,y), meta_item="foo", ...)`
while *p.token != token::EOF {
cfgs.push(p.parse_meta_item());
if p.eat(&token::EOF) { break } // trailing comma is optional,.
p.expect(&token::COMMA);
}

// test_cfg searches for meta items looking like `cfg(foo, ...)`
let in_cfg = &[cx.meta_list(sp, @"cfg", cfgs)];

let matches_cfg = attr::test_cfg(cx.cfg(), in_cfg.iter().transform(|&x| x));
let e = cx.expr_bool(sp, matches_cfg);
MRExpr(e)
}
1 change: 1 addition & 0 deletions src/libsyntax/syntax.rs
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,7 @@ pub mod ext {
}


pub mod cfg;
pub mod fmt;
pub mod env;
pub mod bytes;
Expand Down
35 changes: 35 additions & 0 deletions src/test/run-pass/syntax-extension-cfg.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
// Copyright 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.

// xfail-fast compile-flags doesn't work with fast-check
// compile-flags: --cfg foo --cfg bar(baz) --cfg qux="foo"

fn main() {
// check
if ! cfg!(foo) { fail!() }
if cfg!(not(foo)) { fail!() }

if ! cfg!(bar(baz)) { fail!() }
if cfg!(not(bar(baz))) { fail!() }

if ! cfg!(qux="foo") { fail!() }
if cfg!(not(qux="foo")) { fail!() }

if ! cfg!(foo, bar(baz), qux="foo") { fail!() }
if cfg!(not(foo, bar(baz), qux="foo")) { fail!() }

if cfg!(not_a_cfg) { fail!() }
if cfg!(not_a_cfg, foo, bar(baz), qux="foo") { fail!() }

if ! cfg!(not(not_a_cfg)) { fail!() }
if ! cfg!(not(not_a_cfg), foo, bar(baz), qux="foo") { fail!() }

if cfg!(trailing_comma, ) { fail!() }
}

0 comments on commit 2460170

Please sign in to comment.