---

yaml
---
r: 13802
b: refs/heads/try
c: 0b9f2a8
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: b7b8455e540a4acca4c602f248534e264b45664d
+refs/heads/try: 0b9f2a82154a9a8df9691242a8be8a0a0bdbcd9e
 refs/tags/release-0.1: 1f5c5126e96c79d22cb7862f75304136e204f105

branches/try/src/rustdoc/fold.rs

@@ -97,7 +97,8 @@ fn default_seq_fold_mod<T>(
     ~{
         mods: fold.fold_modlist(fold, doc.mods),
         fns: fold.fold_fnlist(fold, doc.fns),
-        consts: fold.fold_constlist(fold, doc.consts)
+        consts: fold.fold_constlist(fold, doc.consts),
+        enums: fold.fold_enumlist(fold, doc.enums)
         with *doc
     }
 }

branches/try/src/rustdoc/tystr_pass.rs

@@ -17,7 +17,8 @@ fn run(
 ) -> doc::cratedoc {
     let fold = fold::fold({
         fold_fn: fold_fn,
-        fold_const: fold_const
+        fold_const: fold_const,
+        fold_enum: fold_enum
         with *fold::default_seq_fold(srv)
     });
     fold.fold_crate(fold, doc)
@@ -179,4 +180,45 @@ fn should_add_const_types() {
     let doc = extract::from_srv(srv, "");
     let doc = run(srv, doc);
     assert doc.topmod.consts[0].ty == some("bool");
-}
+}
+
+fn fold_enum(
+    fold: fold::fold<astsrv::srv>,
+    doc: doc::enumdoc
+) -> doc::enumdoc {
+    let srv = fold.ctxt;
+
+    ~{
+        variants: vec::map(doc.variants) {|variant|
+            let sig = astsrv::exec(srv) {|ctxt|
+                alt ctxt.map.get(doc.id) {
+                  ast_map::node_item(@{
+                    node: ast::item_enum(ast_variants, _), _
+                  }) {
+                    let ast_variant = option::get(
+                        vec::find(ast_variants) {|v|
+                            v.node.name == variant.name
+                        });
+
+                    pprust::variant_to_str(ast_variant)
+                  }
+                }
+            };
+
+            ~{
+                sig: some(sig)
+                with *variant
+            }
+        }
+        with *doc
+    }
+}
+
+#[test]
+fn should_add_variant_sigs() {
+    let source = "enum a { b(int) }";
+    let srv = astsrv::mk_srv_from_str(source);
+    let doc = extract::from_srv(srv, "");
+    let doc = run(srv, doc);
+    assert doc.topmod.enums[0].variants[0].sig == some("b(int)");
+}