Implement bytes! syntax extension

3 files changed, 31 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;