Auto merge of #103 - emilio:const-enum-unnamed, r=fitzgen

codegen: Generate constants names for unnamed enums in classes.

Fixes #84.

cc @upsuper, sorry for not doing it before, I totally forgot about it.

r? @fitzgen or @nox

4 files changed, 61 insertions, 5 deletions

diff --git a/src/codegen/mod.rs b/src/codegen/mod.rs
index 5688065f..55312c31 100644
--- a/src/codegen/mod.rs
+++ b/src/codegen/mod.rs
@@ -16,6 +16,7 @@ use ir::layout::Layout;
 use ir::annotations::FieldAccessorKind;
 
 use std::ops;
+use std::borrow::Cow;
 use std::mem;
 use std::collections::BTreeSet;
 use std::collections::HashSet;
@@ -1232,6 +1233,9 @@ impl CodeGenerator for Enum {
             result.push(constant);
         }
 
+        // Used to mangle the constants we generate in the unnamed-enum case.
+        let mut parent_canonical_name = None;
+
         // A map where we keep a value -> variant relation.
         let mut seen_values = HashMap::<_, String>::new();
         let enum_ty = item.expect_type();
@@ -1264,11 +1268,21 @@ impl CodeGenerator for Enum {
                     if enum_ty.name().is_none() {
                         // NB: if we want to do this for other kind of nested
                         // enums we can probably mangle the name.
-                        if item.is_toplevel(ctx) {
-                            add_constant(enum_ty, &name, &variant_name,
-                                         &variant_name, enum_rust_ty.clone(),
-                                         result);
-                        }
+                        let mangled_name = if item.is_toplevel(ctx) {
+                            variant_name.clone()
+                        } else {
+                            if parent_canonical_name.is_none() {
+                                parent_canonical_name = Some(item.parent_id().canonical_name(ctx));
+                            }
+
+                            Cow::Owned(
+                                format!("{}_{}", parent_canonical_name.as_ref().unwrap(),
+                                                 variant_name))
+                        };
+
+                        add_constant(enum_ty, &name, &mangled_name,
+                                     &variant_name, enum_rust_ty.clone(),
+                                     result);
                     }
 
                     entry.insert(variant_name.into_owned());
diff --git a/tests/expectations/anon_enum.rs b/tests/expectations/anon_enum.rs
index b06585b1..17212c12 100644
--- a/tests/expectations/anon_enum.rs
+++ b/tests/expectations/anon_enum.rs
@@ -10,6 +10,8 @@ pub struct Test {
     pub foo: ::std::os::raw::c_int,
     pub bar: f32,
 }
+pub const Test_T_NONE: Test__bindgen_ty_bindgen_id_6 =
+    Test__bindgen_ty_bindgen_id_6::T_NONE;
 #[repr(u32)]
 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
 pub enum Test__bindgen_ty_bindgen_id_6 { T_NONE = 0, }
diff --git a/tests/expectations/const_enum_unnamed.rs b/tests/expectations/const_enum_unnamed.rs
new file mode 100644
index 00000000..e16dc405
--- /dev/null
+++ b/tests/expectations/const_enum_unnamed.rs
@@ -0,0 +1,31 @@
+/* automatically generated by rust-bindgen */
+
+
+#![allow(non_snake_case)]
+
+
+pub const FOO_BAR: _bindgen_ty_bindgen_id_1 =
+    _bindgen_ty_bindgen_id_1::FOO_BAR;
+pub const FOO_BAZ: _bindgen_ty_bindgen_id_1 =
+    _bindgen_ty_bindgen_id_1::FOO_BAZ;
+#[repr(u32)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
+pub enum _bindgen_ty_bindgen_id_1 { FOO_BAR = 0, FOO_BAZ = 1, }
+#[repr(C)]
+#[derive(Debug, Copy)]
+pub struct Foo {
+    pub _address: u8,
+}
+pub const Foo_FOO_BAR: Foo__bindgen_ty_bindgen_id_5 =
+    Foo__bindgen_ty_bindgen_id_5::FOO_BAR;
+#[repr(u32)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
+pub enum Foo__bindgen_ty_bindgen_id_5 { FOO_BAR = 10, }
+#[test]
+fn bindgen_test_layout_Foo() {
+    assert_eq!(::std::mem::size_of::<Foo>() , 1usize);
+    assert_eq!(::std::mem::align_of::<Foo>() , 1usize);
+}
+impl Clone for Foo {
+    fn clone(&self) -> Self { *self }
+}
diff --git a/tests/headers/const_enum_unnamed.hpp b/tests/headers/const_enum_unnamed.hpp
new file mode 100644
index 00000000..eb139434
--- /dev/null
+++ b/tests/headers/const_enum_unnamed.hpp
@@ -0,0 +1,9 @@
+
+enum {
+  FOO_BAR,
+  FOO_BAZ,
+};
+
+class Foo {
+  enum { FOO_BAR = 10 };
+};