rustc: Parse and typecheck repeating vector expressions; e.g. [ 0, ..512 ]

7 files changed, 90 insertions, 32 deletions

diff --git a/src/libsyntax/ast.rs b/src/libsyntax/ast.rs
index 34144a9f031..cc06049bd45 100644
--- a/src/libsyntax/ast.rs
+++ b/src/libsyntax/ast.rs
@@ -350,7 +350,10 @@ enum expr_ {
     // A struct literal expression.
     //
     // XXX: Add functional record update.
-    expr_struct(@path, ~[field])
+    expr_struct(@path, ~[field]),
+
+    // A vector literal constructed from one repeated element.
+    expr_repeat(@expr /* element */, @expr /* count */, mutability)
 }
 
 #[auto_serialize]
diff --git a/src/libsyntax/fold.rs b/src/libsyntax/fold.rs
index 5fa9c584bc4..521d9ab946a 100644
--- a/src/libsyntax/fold.rs
+++ b/src/libsyntax/fold.rs
@@ -402,6 +402,8 @@ fn noop_fold_expr(e: expr_, fld: ast_fold) -> expr_ {
           expr_vec(exprs, mutt) {
             expr_vec(fld.map_exprs(|x| fld.fold_expr(x), exprs), mutt)
           }
+          expr_repeat(expr, count, mutt) =>
+            expr_repeat(fld.fold_expr(expr), fld.fold_expr(count), mutt),
           expr_rec(fields, maybe_expr) {
             expr_rec(vec::map(fields, fold_field),
                      option::map(maybe_expr, |x| fld.fold_expr(x)))
diff --git a/src/libsyntax/parse/lexer.rs b/src/libsyntax/parse/lexer.rs
index 9583be3461d..74dbea41d82 100644
--- a/src/libsyntax/parse/lexer.rs
+++ b/src/libsyntax/parse/lexer.rs
@@ -483,6 +483,10 @@ fn next_token_inner(rdr: string_reader) -> token::token {
       ',' { bump(rdr); return token::COMMA; }
       '.' {
         bump(rdr);
+        if rdr.curr == '.' && nextch(rdr) != '.' {
+            bump(rdr);
+            return token::DOTDOT;
+        }
         if rdr.curr == '.' && nextch(rdr) == '.' {
             bump(rdr);
             bump(rdr);
diff --git a/src/libsyntax/parse/parser.rs b/src/libsyntax/parse/parser.rs
index 508a9f8f979..c76a82a5c9b 100644
--- a/src/libsyntax/parse/parser.rs
+++ b/src/libsyntax/parse/parser.rs
@@ -27,33 +27,34 @@ import ast::{_mod, add, alt_check, alt_exhaustive, arg, arm, attribute,
              expr_call, expr_cast, expr_copy, expr_do_body,
              expr_fail, expr_field, expr_fn, expr_fn_block, expr_if,
              expr_index, expr_lit, expr_log, expr_loop,
-             expr_loop_body, expr_mac, expr_move, expr_path,
-             expr_rec, expr_ret, expr_swap, expr_struct, expr_tup, expr_unary,
-             expr_unary_move, expr_vec, expr_vstore, expr_while, extern_fn,
-             field, fn_decl, foreign_item, foreign_item_fn, foreign_mod,
-             ident, impure_fn, infer, inherited, init_assign, init_move,
-             initializer, instance_var, item, item_, item_class, item_const,
-             item_enum, item_fn, item_foreign_mod, item_impl, item_mac,
-             item_mod, item_trait, item_ty, lit, lit_, lit_bool, lit_float,
-             lit_int, lit_int_unsuffixed, lit_nil, lit_str, lit_uint, local,
-             m_const, m_imm, m_mutbl, mac_, mac_aq, mac_ellipsis, mac_invoc,
-             mac_invoc_tt, mac_var, matcher, match_nonterminal, match_seq,
-             match_tok, method, mode, mt, mul, mutability, neg, noreturn, not,
-             pat, pat_box, pat_enum, pat_ident, pat_lit, pat_range, pat_rec,
-             pat_tup, pat_uniq, pat_wild, path, private, proto, proto_any,
-             proto_bare, proto_block, proto_box, proto_uniq, provided, public,
-             pure_fn, purity, re_anon, re_named, region, rem, required,
-             ret_style, return_val, self_ty, shl, shr, stmt, stmt_decl,
-             stmt_expr, stmt_semi, subtract, sty_box, sty_by_ref, sty_region,
-             sty_uniq, sty_value, token_tree, trait_method, trait_ref,
-             tt_delim, tt_seq, tt_tok, tt_nonterminal, ty, ty_, ty_bot,
-             ty_box, ty_field, ty_fn, ty_infer, ty_mac, ty_method, ty_nil,
-             ty_param, ty_path, ty_ptr, ty_rec, ty_rptr, ty_tup, ty_u32,
-             ty_uniq, ty_vec, ty_fixed_length, unchecked_blk, uniq,
-             unsafe_blk, unsafe_fn, variant, view_item, view_item_,
-             view_item_export, view_item_import, view_item_use, view_path,
-             view_path_glob, view_path_list, view_path_simple, visibility,
-             vstore, vstore_box, vstore_fixed, vstore_slice, vstore_uniq};
+             expr_loop_body, expr_mac, expr_move, expr_path, expr_rec,
+             expr_repeat, expr_ret, expr_swap, expr_struct, expr_tup,
+             expr_unary, expr_unary_move, expr_vec, expr_vstore, expr_while,
+             extern_fn, field, fn_decl, foreign_item, foreign_item_fn,
+             foreign_mod, ident, impure_fn, infer, inherited, init_assign,
+             init_move, initializer, instance_var, item, item_, item_class,
+             item_const, item_enum, item_fn, item_foreign_mod, item_impl,
+             item_mac, item_mod, item_trait, item_ty, lit, lit_, lit_bool,
+             lit_float, lit_int, lit_int_unsuffixed, lit_nil, lit_str,
+             lit_uint, local, m_const, m_imm, m_mutbl, mac_, mac_aq,
+             mac_ellipsis, mac_invoc, mac_invoc_tt, mac_var, matcher,
+             match_nonterminal, match_seq, match_tok, method, mode, mt, mul,
+             mutability, neg, noreturn, not, pat, pat_box, pat_enum,
+             pat_ident, pat_lit, pat_range, pat_rec, pat_tup, pat_uniq,
+             pat_wild, path, private, proto, proto_any, proto_bare,
+             proto_block, proto_box, proto_uniq, provided, public, pure_fn,
+             purity, re_anon, re_named, region, rem, required, ret_style,
+             return_val, self_ty, shl, shr, stmt, stmt_decl, stmt_expr,
+             stmt_semi, subtract, sty_box, sty_by_ref, sty_region, sty_uniq,
+             sty_value, token_tree, trait_method, trait_ref, tt_delim, tt_seq,
+             tt_tok, tt_nonterminal, ty, ty_, ty_bot, ty_box, ty_field, ty_fn,
+             ty_infer, ty_mac, ty_method, ty_nil, ty_param, ty_path, ty_ptr,
+             ty_rec, ty_rptr, ty_tup, ty_u32, ty_uniq, ty_vec,
+             ty_fixed_length, unchecked_blk, uniq, unsafe_blk, unsafe_fn,
+             variant, view_item, view_item_, view_item_export,
+             view_item_import, view_item_use, view_path, view_path_glob,
+             view_path_list, view_path_simple, visibility, vstore, vstore_box,
+             vstore_fixed, vstore_slice, vstore_uniq};
 
 export file_type;
 export parser;
@@ -365,6 +366,7 @@ class parser {
     // Parses something like "&x"
     fn parse_region() -> @region {
         self.expect(token::BINOP(token::AND));
+
         alt copy self.token {
           token::IDENT(sid, _) {
             self.bump();
@@ -812,11 +814,36 @@ class parser {
         } else if self.token == token::LBRACKET {
             self.bump();
             let mutbl = self.parse_mutability();
-            let es = self.parse_seq_to_end(
-                token::RBRACKET, seq_sep_trailing_allowed(token::COMMA),
-                |p| p.parse_expr());
+            if self.token == token::RBRACKET {
+                // Empty vector.
+                self.bump();
+                ex = expr_vec(~[], mutbl);
+            } else {
+                // Nonempty vector.
+                let first_expr = self.parse_expr();
+                if self.token == token::COMMA &&
+                        self.look_ahead(1) == token::DOTDOT {
+                    // Repeating vector syntax: [ 0, ..512 ]
+                    self.bump();
+                    self.bump();
+                    let count = self.parse_expr();
+                    self.expect(token::RBRACKET);
+                    ex = expr_repeat(first_expr, count, mutbl);
+                } else if self.token == token::COMMA {
+                    // Vector with two or more elements.
+                    self.bump();
+                    let remaining_exprs =
+                        self.parse_seq_to_end(token::RBRACKET,
+                            seq_sep_trailing_allowed(token::COMMA),
+                            |p| p.parse_expr());
+                    ex = expr_vec(~[first_expr] + remaining_exprs, mutbl);
+                } else {
+                    // Vector with one element.
+                    self.expect(token::RBRACKET);
+                    ex = expr_vec(~[first_expr], mutbl);
+                }
+            }
             hi = self.span.hi;
-            ex = expr_vec(es, mutbl);
         } else if self.token == token::ELLIPSIS {
             self.bump();
             return pexpr(self.mk_mac_expr(lo, self.span.hi, mac_ellipsis));
diff --git a/src/libsyntax/parse/token.rs b/src/libsyntax/parse/token.rs
index 3dbcef66732..1228926e6e4 100644
--- a/src/libsyntax/parse/token.rs
+++ b/src/libsyntax/parse/token.rs
@@ -49,6 +49,7 @@ enum token {
     /* Structural symbols */
     AT,
     DOT,
+    DOTDOT,
     ELLIPSIS,
     COMMA,
     SEMI,
@@ -134,6 +135,7 @@ fn to_str(in: interner<@~str>, t: token) -> ~str {
       /* Structural symbols */
       AT { ~"@" }
       DOT { ~"." }
+      DOTDOT { ~".." }
       ELLIPSIS { ~"..." }
       COMMA { ~"," }
       SEMI { ~";" }
diff --git a/src/libsyntax/print/pprust.rs b/src/libsyntax/print/pprust.rs
index c4b643de8d7..9eccf688233 100644
--- a/src/libsyntax/print/pprust.rs
+++ b/src/libsyntax/print/pprust.rs
@@ -948,6 +948,22 @@ fn print_expr(s: ps, &&expr: @ast::expr) {
         word(s.s, ~"]");
         end(s);
       }
+
+      ast::expr_repeat(element, count, mutbl) => {
+        ibox(s, indent_unit);
+        word(s.s, ~"[");
+        if mutbl == ast::m_mutbl {
+            word(s.s, ~"mut");
+            nbsp(s);
+        }
+        print_expr(s, element);
+        word(s.s, ~",");
+        word(s.s, ~"..");
+        print_expr(s, count);
+        word(s.s, ~"]");
+        end(s);
+      }
+
       ast::expr_rec(fields, wth) {
         word(s.s, ~"{");
         commasep_cmnt(s, consistent, fields, print_field, get_span);
diff --git a/src/libsyntax/visit.rs b/src/libsyntax/visit.rs
index 61cbba5fc1f..6eb468efd82 100644
--- a/src/libsyntax/visit.rs
+++ b/src/libsyntax/visit.rs
@@ -364,6 +364,10 @@ fn visit_expr<E>(ex: @expr, e: E, v: vt<E>) {
     alt ex.node {
       expr_vstore(x, _) { v.visit_expr(x, e, v); }
       expr_vec(es, _) { visit_exprs(es, e, v); }
+      expr_repeat(element, count, _) => {
+        v.visit_expr(element, e, v);
+        v.visit_expr(count, e, v);
+      }
       expr_rec(flds, base) {
         for flds.each |f| { v.visit_expr(f.node.expr, e, v); }
         visit_expr_opt(base, e, v);