cwyckoff-babel_icious 0.0.4.5 → 0.0.4.6

data/lib/babel_icious.rb

@@ -4,6 +4,7 @@ require 'xml'
 $LOAD_PATH.unshift(File.expand_path(File.dirname(__FILE__))) unless $LOAD_PATH.include?(File.expand_path(File.dirname(__FILE__)))
 
 require "babel_icious/core_ext/enumerable"
+require "babel_icious/core_ext/libxml_node"
 require "babel_icious/target_mapper"
 require "babel_icious/map_factory"
 require "babel_icious/path_translator"

data/lib/babel_icious/core_ext/libxml_node.rb

@@ -0,0 +1,28 @@
+require 'xml/libxml'
+
+module BabeliciousNodeHacks
+
+  def concatenate_children(glue)
+    res = self.children.inject('') {|a,b| a << "#{b.content}#{glue}"}
+    res.chop
+  end
+  
+  def child_content(child)
+    child_arr(child).first.content
+  end
+
+  def child_name(child)
+    child_arr(child).first.name
+  end
+  
+  private
+  
+  def child_arr(child)
+    child_arr = self.find(child)
+    raise "Cannot retrieve content or name for child node #{child}.  Current node has more than one child." if child_arr.size > 1
+    child_arr
+  end
+
+end
+
+class XML::Node; include BabeliciousNodeHacks; end

data/lib/babel_icious/mapper.rb

@@ -14,6 +14,10 @@ module Babelicious
         yield self
       end
 
+      def customize(&block)
+        current_target_mapper.register_customized(&block)
+      end
+
       def direction(dir={})
         current_target_mapper.direction = @direction = dir
       end
@@ -41,8 +45,9 @@ module Babelicious
         current_target_mapper.register_condition(:when, nil, &block)
       end
 
-      def unless(condition)
-        current_target_mapper.register_condition(:unless, condition)
+      def unless(condition=nil, &block)
+        current_target_mapper.register_condition(:unless, condition, &block)
+        self
       end
       
       private

data/lib/babel_icious/mappers/base_map.rb

@@ -13,6 +13,10 @@ module Babelicious
     def register_condition(condition_key, condition, &block)
       map_condition.register(condition_key, condition, &block)
     end
+
+    def register_customized(&block)
+      @customized_map = block
+    end
     
     protected
 

data/lib/babel_icious/mappers/hash_map.rb

@@ -48,43 +48,12 @@ module Babelicious
     end 
     
     def map_source_value(source_value)
-      if(@opts[:concatenate])
-        HashMappingStrategies::Concatenate.map(source_value, @opts[:concatenate])
+      if(@customized_map)
+        @customized_map.call(source_value)
       else 
         source_value
       end 
     end
 
   end
-  
-  module HashMappingStrategies
-
-    class Concatenate
-
-      class << self
-        
-        def map(source_value, concat)
-          if(source_value.kind_of?(Hash))
-            concat_values_from_hash(source_value).join(concat)
-          elsif(source_value.kind_of?(Array)) 
-            source_value.join(concat)
-          else 
-            source_value
-          end 
-        end
-
-        private
-        
-        def concat_values_from_hash(source_value)
-          source_value.each do |key, value|
-            unless value.is_a?(Array)
-              return concat_values_from_hash(value)
-            else 
-              return value
-            end
-          end
-        end
-      end
-    end
-  end
 end

data/lib/babel_icious/mappers/xml_map.rb

@@ -18,12 +18,15 @@ module Babelicious
     
     def initialize(path_translator, opts={})
       @path_translator, @opts = path_translator, opts
-      @xml_value_mapper = XmlValueMapper.new(@path_translator, @opts)
     end
     
     def value_from(source)
       source.find("/#{@path_translator.full_path}").each do |node|
-        return @xml_value_mapper.map(node)
+        if(node.children.size > 1)
+          return node
+        else
+          return node.content
+        end
       end
     end
 
@@ -74,74 +77,4 @@ module Babelicious
     end
   end
 
-  
-  class XmlValueMapper
-
-    def initialize(path_translator, opts={})
-      @path_translator, @opts = path_translator, opts
-    end
-    
-    def map(node)
-      if(node.children.size > 1)
-        content = {}
-         map_child(node, content)
-      else
-        return node.content
-      end
-    end
-
-    private
-    
-    def map_child(node, content)
-      node.each_element do |child|
-        if(content[child.name])
-          update_content_key(content, child)
-        else
-          create_content_key(content, child)
-        end 
-      end
-      {node.name => content}
-    end
-    
-    private
-    
-    def content_value_is_array?(content, child) 
-      content[child.name].is_a?(Array)
-    end 
-    
-    def create_content_key(content, child)
-      unless final_node?(child)
-        content[child.name] = {child.child.name => child.child.content}
-      else 
-        set_value_in_array(content, child)
-      end
-    end
-    
-    def final_node?(child)
-      !child.children? || child.child.name == "text"
-    end
-    
-    def set_value_in_array(content, child)
-      content[child.name] = [] unless content_value_is_array?(content, child)
-      if(child.children?)
-        if (child.parent.find(child.name)).to_a.size == 1
-          content[child.name] = child.child.content
-        else 
-          content[child.name] << child.child.content
-        end
-      else
-        content[child.name] = ""
-      end 
-    end
-
-    def update_content_key(content, child)
-      unless final_node?(child)
-        content[child.name] = [content[child.name]] unless content_value_is_array?(content, child) 
-        content[child.name] << {child.child.name => child.child.content}
-      else 
-        set_value_in_array(content, child)
-      end
-    end
-  end
-
 end

data/lib/babel_icious/target_mapper.rb

@@ -32,6 +32,10 @@ module Babelicious
       @mappings.last[1].register_condition(condition_key, condition, &block)
     end
     
+    def register_customized(&block)
+      @mappings.last[1].register_customized(&block)
+    end
+    
     def register_mapping(opts={})
       raise TargetMapperError, "Both :from and :to keys must be set (e.g., {:from => \"foo/bar\", :to => \"bar/foo\")" unless (opts[:from] && opts[:to])
       target = MapFactory.target(@direction, opts)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: cwyckoff-babel_icious
 version: !ruby/object:Gem::Version 
-  version: 0.0.4.5
+  version: 0.0.4.6
 platform: ruby
 authors: 
 - Chris Wyckoff
@@ -42,6 +42,7 @@ files:
 - lib/babel_icious/mapper.rb
 - lib/babel_icious/path_translator.rb
 - lib/babel_icious/target_mapper.rb
+- lib/babel_icious/core_ext/libxml_node.rb
 - lib/babel_icious/core_ext/enumerable.rb
 - README.rdoc
 - MIT-LICENSE