diff options
| -rw-r--r-- | src/libsyntax/ext/base.rs | 2 | ||||
| -rw-r--r-- | src/libsyntax/ext/bytes.rs | 28 | ||||
| -rw-r--r-- | src/libsyntax/syntax.rc | 1 | ||||
| -rw-r--r-- | src/test/run-pass/syntax-extension-bytes.rs | 16 |
4 files changed, 47 insertions, 0 deletions
diff --git a/src/libsyntax/ext/base.rs b/src/libsyntax/ext/base.rs index 51e127ab60f..a97c69ba4ff 100644 --- a/src/libsyntax/ext/base.rs +++ b/src/libsyntax/ext/base.rs @@ -139,6 +139,8 @@ pub fn syntax_expander_table() -> SyntaxEnv { @SE(ItemDecorator(ext::auto_encode::expand_auto_decode))); syntax_expanders.insert(@~"env", builtin_normal_tt(ext::env::expand_syntax_ext)); + syntax_expanders.insert(@~"bytes", + builtin_normal_tt(ext::bytes::expand_syntax_ext)); syntax_expanders.insert(@~"concat_idents", builtin_normal_tt( ext::concat_idents::expand_syntax_ext)); diff --git a/src/libsyntax/ext/bytes.rs b/src/libsyntax/ext/bytes.rs new file mode 100644 index 00000000000..03020f26a93 --- /dev/null +++ b/src/libsyntax/ext/bytes.rs @@ -0,0 +1,28 @@ +// 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 bytes! extension. */ + +use ast; +use codemap::span; +use ext::base::*; +use ext::base; +use ext::build::{mk_u8, mk_slice_vec_e}; + +pub fn expand_syntax_ext(cx: @ext_ctxt, sp: span, tts: &[ast::token_tree]) + -> base::MacResult { + let var = get_single_str_from_tts(cx, sp, tts, "bytes!"); + let mut bytes = ~[]; + for var.each |byte| { + bytes.push(mk_u8(cx, sp, byte)); + } + let e = mk_slice_vec_e(cx, sp, bytes); + MRExpr(e) +} diff --git a/src/libsyntax/syntax.rc b/src/libsyntax/syntax.rc index 8deca72779e..565a8a18c6f 100644 --- a/src/libsyntax/syntax.rc +++ b/src/libsyntax/syntax.rc @@ -78,6 +78,7 @@ pub mod ext { pub mod fmt; pub mod env; + pub mod bytes; pub mod concat_idents; pub mod log_syntax; pub mod auto_encode; diff --git a/src/test/run-pass/syntax-extension-bytes.rs b/src/test/run-pass/syntax-extension-bytes.rs new file mode 100644 index 00000000000..9249267355d --- /dev/null +++ b/src/test/run-pass/syntax-extension-bytes.rs @@ -0,0 +1,16 @@ +// 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. + +fn main() { + let vec = bytes!("abc"); + assert_eq!(vec[0], 97); + assert_eq!(vec[1], 98); + assert_eq!(vec[2], 99); +} |
