From be162da9614624f8fd5f5d03909bb99c6ab2c647 Mon Sep 17 00:00:00 2001
From: Patrick Walton <pcwalton@mimiga.net>
Date: Sun, 16 Sep 2012 09:27:56 -0700
Subject: [PATCH] libcore: Log newtyped enums correctly. Closes #3506.

---
 src/libcore/repr.rs | 31 +++++++++++++++++++++----------
 1 file changed, 21 insertions(+), 10 deletions(-)

diff --git a/src/libcore/repr.rs b/src/libcore/repr.rs
index a0d6b013407..9c12c5430ca 100644
--- a/src/libcore/repr.rs
+++ b/src/libcore/repr.rs
@@ -500,7 +500,8 @@ fn test_repr2() {
 enum EnumVisitState {
     PreVariant,     // We're before the variant we're interested in.
     InVariant,      // We're inside the variant we're interested in.
-    PostVariant     // We're after the variant we're interested in.
+    PostVariant,    // We're after the variant we're interested in.
+    Newtype         // This is a newtyped enum.
 }
 
 impl EnumVisitState : cmp::Eq {
@@ -859,13 +860,14 @@ impl ReprPrinterWrapper : TyVisitor {
 
     // Enums
 
-    fn visit_enter_enum(_n_variants: uint, sz: uint, align: uint) -> bool {
+    fn visit_enter_enum(n_variants: uint, sz: uint, align: uint) -> bool {
         unsafe {
             self.printer.align(align);
 
             // Write in the location of the end of this enum.
             let end_ptr = transmute(self.printer.ptr as uint + sz);
-            let new_state = EnumState { end_ptr: end_ptr, state: PreVariant };
+            let state = if n_variants == 1 { Newtype } else { PreVariant };
+            let new_state = EnumState { end_ptr: end_ptr, state: state };
             self.printer.enum_stack.push(new_state);
 
             true
@@ -889,6 +891,9 @@ impl ReprPrinterWrapper : TyVisitor {
                         stack.set_elt(stack.len() - 1, enum_state);
                     }
                 }
+                Newtype => {
+                    self.printer.writer.write_str(name);
+                }
                 InVariant | PostVariant => {}
             }
             true
@@ -896,14 +901,17 @@ impl ReprPrinterWrapper : TyVisitor {
     }
 
     fn visit_enum_variant_field(i: uint, inner: *TyDesc) -> bool {
-        if self.printer.enum_stack.last().state == InVariant {
-            if i == 0 {
-                self.printer.writer.write_char('(');
-            } else {
-                self.printer.writer.write_str(", ");
-            }
+        match self.printer.enum_stack.last().state {
+            InVariant | Newtype => {
+                if i == 0 {
+                    self.printer.writer.write_char('(');
+                } else {
+                    self.printer.writer.write_str(", ");
+                }
 
-            intrinsic::visit_tydesc(inner, self as @TyVisitor);
+                intrinsic::visit_tydesc(inner, self as @TyVisitor);
+            }
+            PreVariant | PostVariant => {}
         }
         true
     }
@@ -920,6 +928,9 @@ impl ReprPrinterWrapper : TyVisitor {
                 enum_state.state = PostVariant;
                 stack.set_elt(stack.len() - 1, enum_state);
             }
+            Newtype => {
+                if n_fields >= 1 { self.printer.writer.write_char(')'); }
+            }
             PreVariant | PostVariant => {}
         }
         true