Namespacify all elements of an XPath, not just the first

Author: chewi

lib/roxml/xml/references.rb

@@ -1,3 +1,5 @@
+require "rexml/xpath_parser"
+
 module ROXML
   class RequiredElementMissing < Exception # :nodoc:
   end
@@ -50,12 +52,24 @@ def conventionize(what)
     end
 
     def namespacify(what)
-      if what.to_s.present? && !what.to_s.include?(':') && opts.namespace != false
-        [opts.namespace, @instance.class.roxml_namespace, @default_namespace].each do |namespace|
-          return opts.namespace == '*' ? (what == '*' ? "*" : "*[local-name()='#{what}']") : "#{namespace}:#{what}" if namespace
+      if what.to_s.present? && opts.namespace != false && ns = [opts.namespace, @instance.class.roxml_namespace, @default_namespace].compact.map(&:to_s).first
+        parser = REXML::Parsers::XPathParser.new
+        parsed = parser.parse what
+
+        parsed.each_cons(4).with_index.each do |a,i|
+          if a[0..2] == [:child, :qname, ""]
+            if ns == "*"
+              parsed[i+1,3] = [:any, :predicate, [:eq, [:function, "local-name", []], [:literal, a[3]]]] if a[3] != "*"
+            else
+              a[2].replace ns
+            end
+          end
         end
+
+        parser.abbreviate parsed
+      else
+        what
       end
-      what
     end
 
     def apply_blocks(val)

spec/xml/namespace_spec.rb

@@ -125,6 +125,18 @@ class VApp
         @instance.default_namespace_with_namespacey_from.should == 'namespacey node'
       end
     end
+
+    context "with an namespacey XPath :from" do
+      it "should use the given namespace" do
+        @instance.default_namespace_with_namespacey_xpath_from.should == 'namespacey xpath node'
+      end
+    end
+
+    context "with an XPath :from" do
+      it "should use the default namespace" do
+        @instance.default_namespace_with_xpath_from.should == 'default namespace with xpath node'
+      end
+    end
   end
 
   context "with a default namespace declared" do
@@ -139,6 +151,8 @@ class DefaultNamespaceyObject
       xml_reader :default_namespace_with_namespaceless_from_and_explicit_namespace, :from => 'with_namespaceless_from', :namespace => 'explicit'
       xml_reader :default_namespace_with_namespaceless_from_and_namespace_false, :from => 'with_namespaceless_from', :namespace => false
       xml_reader :default_namespace_with_namespacey_from, :from => 'namespacey:with_namespacey_from'
+      xml_reader :default_namespace_with_namespacey_xpath_from, :from => 'xpath/namespacey:default_namespace'
+      xml_reader :default_namespace_with_xpath_from, :from => 'xpath/default_namespace'
 
       # These are handled in the "roxml namespacey declaration" shared spec
       # xml_reader :default_namespace_with_namespacey_from_and_namespace_false, :from => 'namespacey:with_namespaceless_from', :namespace => false
@@ -155,6 +169,8 @@ class DefaultNamespaceyObject
           <default_declared:with_namespaceless_from>default namespace node</default_declared:with_namespaceless_from>
           <explicit:default_and_explicit_namespace>explicit namespace node</explicit:default_and_explicit_namespace>
           <default_namespace_with_namespace_false>namespaceless node</default_namespace_with_namespace_false>
+          <default_declared:xpath><namespacey:default_namespace>namespacey xpath node</namespacey:default_namespace></default_declared:xpath>
+          <default_declared:xpath><default_declared:default_namespace>default namespace with xpath node</default_declared:default_namespace></default_declared:xpath>
         </book>
       })
     end
@@ -172,6 +188,8 @@ class XmlDefaultNamespaceyObject
       xml_reader :default_namespace_with_namespaceless_from_and_explicit_namespace, :from => 'with_namespaceless_from', :namespace => 'explicit'
       xml_reader :default_namespace_with_namespaceless_from_and_namespace_false, :from => 'with_namespaceless_from', :namespace => false
       xml_reader :default_namespace_with_namespacey_from, :from => 'namespacey:with_namespacey_from'
+      xml_reader :default_namespace_with_namespacey_xpath_from, :from => 'xpath/namespacey:default_namespace'
+      xml_reader :default_namespace_with_xpath_from, :from => 'xpath/default_namespace'
 
       # These are handled in the "roxml namespacey declaration" shared spec
       # xml_reader :default_namespace_with_namespacey_from_and_namespace_false, :from => 'namespacey:with_namespaceless_from', :namespace => false
@@ -188,6 +206,8 @@ class XmlDefaultNamespaceyObject
           <with_namespaceless_from>default namespace node</with_namespaceless_from>
           <explicit:default_and_explicit_namespace>explicit namespace node</explicit:default_and_explicit_namespace>
           <default_namespace_with_namespace_false xmlns="">namespaceless node</default_namespace_with_namespace_false>
+          <xpath><namespacey:default_namespace>namespacey xpath node</namespacey:default_namespace></xpath>
+          <xpath><default_namespace>default namespace with xpath node</default_namespace></xpath>
         </book>
       })
     end