---

yaml
---
r: 13800
b: refs/heads/try
c: 1586cce
h: refs/heads/master
v: v3

Author: brson

[refs]

@@ -2,5 +2,5 @@
 refs/heads/master: 61b1875c16de39c166b0f4d54bba19f9c6777d1a
 refs/heads/snap-stage1: e33de59e47c5076a89eadeb38f4934f58a3618a6
 refs/heads/snap-stage3: 4a81779abd786ff22d71434c6d9a5917ea4cdfff
-refs/heads/try: e946792a78590df3972fc4a9827d5e94d333d055
+refs/heads/try: 1586cce0ea1082f4716b50520ed9cb40941f7fad
 refs/tags/release-0.1: 1f5c5126e96c79d22cb7862f75304136e204f105

branches/try/src/rustdoc/fold.rs

@@ -16,22 +16,25 @@ type fold_crate<T> = fn~(fold: fold<T>, doc: doc::cratedoc) -> doc::cratedoc;
 type fold_mod<T> = fn~(fold: fold<T>, doc: doc::moddoc) -> doc::moddoc;
 type fold_fn<T> = fn~(fold: fold<T>, doc: doc::fndoc) -> doc::fndoc;
 type fold_const<T> = fn~(fold: fold<T>, doc: doc::constdoc) -> doc::constdoc;
+type fold_enum<T> = fn~(fold: fold<T>, doc: doc::enumdoc) -> doc::enumdoc;
 type fold_modlist<T> = fn~(fold: fold<T>, list: doc::modlist) -> doc::modlist;
 type fold_fnlist<T> = fn~(fold: fold<T>, list: doc::fnlist) -> doc::fnlist;
 type fold_constlist<T> = fn~(
-    fold: fold<T>,
-    list: doc::constlist
-) -> doc::constlist;
+    fold: fold<T>, list: doc::constlist) -> doc::constlist;
+type fold_enumlist<T> = fn~(
+    fold: fold<T>, list: doc::enumlist) -> doc::enumlist;
 
 type t<T> = {
     ctxt: T,
     fold_crate: fold_crate<T>,
     fold_mod: fold_mod<T>,
     fold_fn: fold_fn<T>,
     fold_const: fold_const<T>,
+    fold_enum: fold_enum<T>,
     fold_modlist: fold_modlist<T>,
     fold_fnlist: fold_fnlist<T>,
-    fold_constlist: fold_constlist<T>
+    fold_constlist: fold_constlist<T>,
+    fold_enumlist: fold_enumlist<T>
 };
 
 
@@ -43,19 +46,23 @@ fn mk_fold<T:copy>(
     fold_mod: fold_mod<T>,
     fold_fn: fold_fn<T>,
     fold_const: fold_const<T>,
+    fold_enum: fold_enum<T>,
     fold_modlist: fold_modlist<T>,
     fold_fnlist: fold_fnlist<T>,
-    fold_constlist: fold_constlist<T>
+    fold_constlist: fold_constlist<T>,
+    fold_enumlist: fold_enumlist<T>
 ) -> fold<T> {
     fold({
         ctxt: ctxt,
         fold_crate: fold_crate,
         fold_mod: fold_mod,
         fold_fn: fold_fn,
         fold_const: fold_const,
+        fold_enum: fold_enum,
         fold_modlist: fold_modlist,
         fold_fnlist: fold_fnlist,
-        fold_constlist: fold_constlist
+        fold_constlist: fold_constlist,
+        fold_enumlist: fold_enumlist
     })
 }
 
@@ -66,9 +73,11 @@ fn default_seq_fold<T:copy>(ctxt: T) -> fold<T> {
         {|f, d| default_seq_fold_mod(f, d)},
         {|f, d| default_seq_fold_fn(f, d)},
         {|f, d| default_seq_fold_const(f, d)},
+        {|f, d| default_seq_fold_enum(f, d)},
         {|f, d| default_seq_fold_modlist(f, d)},
         {|f, d| default_seq_fold_fnlist(f, d)},
-        {|f, d| default_seq_fold_constlist(f, d)}
+        {|f, d| default_seq_fold_constlist(f, d)},
+        {|f, d| default_seq_fold_enumlist(f, d)}
     )
 }
 
@@ -107,6 +116,13 @@ fn default_seq_fold_const<T>(
     doc
 }
 
+fn default_seq_fold_enum<T>(
+    _fold: fold<T>,
+    doc: doc::enumdoc
+) -> doc::enumdoc {
+    doc
+}
+
 fn default_seq_fold_modlist<T>(
     fold: fold<T>,
     list: doc::modlist
@@ -134,6 +150,15 @@ fn default_seq_fold_constlist<T>(
     })
 }
 
+fn default_seq_fold_enumlist<T>(
+    fold: fold<T>,
+    list: doc::enumlist
+) -> doc::enumlist {
+    doc::enumlist(vec::map(*list) {|doc|
+        fold.fold_enum(fold, doc)
+    })
+}
+
 #[test]
 fn default_fold_should_produce_same_doc() {
     let source = "mod a { fn b() { } mod c { fn d() { } } }";
@@ -152,4 +177,14 @@ fn default_fold_should_produce_same_consts() {
     let fld = default_seq_fold(());
     let folded = fld.fold_crate(fld, doc);
     assert doc == folded;
+}
+
+#[test]
+fn default_fold_should_produce_same_enums() {
+    let source = "enum a { b }";
+    let ast = parse::from_str(source);
+    let doc = extract::extract(ast, "");
+    let fld = default_seq_fold(());
+    let folded = fld.fold_crate(fld, doc);
+    assert doc == folded;
 }