lib/wasabi/document.rb in wasabi-5.0.3 vs lib/wasabi/document.rb in wasabi-5.1.0

@@ -101,15 +101,17 @@
 
     def type_namespaces
       @type_namespaces ||= begin
         namespaces = []
 
-        parser.types.each do |type, info|
-          namespaces << [[type], info[:namespace]]
+        parser.types.each do |ns, types|
+          types.each do |type, info|
+            namespaces << [[type], info[:namespace]]
 
-          element_keys(info).each do |field|
-            namespaces << [[type, field], info[:namespace]]
+            element_keys(info).each do |field|
+              namespaces << [[type, field], info[:namespace]]
+            end
           end
         end if document
 
         namespaces
       end
@@ -117,15 +119,17 @@
 
     def type_definitions
       @type_definitions ||= begin
         result = []
 
-        parser.types.each do |type, info|
-          element_keys(info).each do |field|
-            field_type = info[field][:type]
-            tag, namespace = field_type.split(":").reverse
+        parser.types.each do |ns, types|
+          types.each do |type, info|
+            element_keys(info).each do |field|
+              field_type = info[field][:type]
+              tag, namespace = field_type.split(":").reverse
 
-            result << [[type, field], tag] if user_defined(namespace)
+              result << [[type, field], tag] if user_defined(namespace)
+            end
           end
         end if document
 
         result
       end