docx_templater 0.0.7 → 0.1.1

data/lib/docx/argument_combiner.rb

@@ -0,0 +1,25 @@
+module Docx
+  class ArgumentCombiner
+    attr_reader :attributes
+    def initialize(*args)
+      @attributes = {}
+      args.flatten!
+      # Prefixes the model name or custom prefix. Makes it so we don't having naming clashes when used with records from multiple m
+      args.each do |arg|
+        if arg.is_a?(Hash) && arg.has_key?(:data) && arg.has_key?(:prefix)
+          template_attributes = (arg[:data].respond_to?(:template_attributes) && :template_attributes) || :attributes
+          arg[:data].send(template_attributes).each_key do |key|
+            @attributes["#{arg[:prefix]}_#{key.to_s}".to_sym] = arg[:data].send(template_attributes)[key]
+          end
+        elsif arg.is_a?(Hash)
+          @attributes.merge!(arg)
+        else
+          template_attributes = (arg.respond_to?(:template_attributes) && :template_attributes) || :attributes
+          arg.send(template_attributes).each_key do |key|
+            @attributes["#{arg.class.name.underscore}_#{key.to_s}".to_sym] = arg.send(template_attributes)[key]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/docx/document_replacer.rb

@@ -0,0 +1,33 @@
+require 'rexml/document'
+require 'docx/placeholder_observer'
+
+module Docx
+	class DocumentReplacer
+
+    attr_reader :doc, :observer
+
+    def initialize(str, data_provider)
+      @doc = REXML::Document.new(str)
+      @observer = Docx::PlaceholderObserver.new(data_provider)
+      walk_node(@doc.root)
+      @observer.end_of_document
+    end
+
+    def replaced
+      @doc.to_s
+    end
+
+    private
+
+    def walk_node(node)
+      if node.is_a?(REXML::Element)
+        node.children.each do |n|
+          walk_node(n)
+        end
+      elsif node.is_a?(REXML::Text)
+        observer.next_node(node)
+      end
+    end
+
+	end
+end

data/lib/docx/nodes_to_fix.rb

@@ -0,0 +1,38 @@
+module Docx
+  class NodesToFix
+    attr_accessor :node_list, :current_node, :value
+    def initialize
+      forget
+    end
+
+    def forget
+      @current_node = nil
+      @node_list = []
+      @value = ''
+    end
+
+    def remember(node, index)
+      new_node = current_node.nil? || current_node != node
+      if new_node
+        @current_node = node
+        @node_list << {:node => node, :range => index..index}
+      else
+        @node_list.last[:range] = (node_list.last[:range].min)..index
+      end
+    end
+
+    def fix
+      @node_list.each do |obj|
+        node = obj[:node]
+        range = obj[:range]
+        new_val = node.value
+        new_val[range] = value.to_s || ''
+        node.value = new_val
+        if new_val =~ /^\s+/ && node.parent
+          node.parent.add_attribute('xml:space', 'preserve')
+        end
+        self.value = nil
+      end
+    end
+  end
+end

data/lib/docx/placeholder_observer.rb

@@ -0,0 +1,79 @@
+require 'rexml/document'
+require 'docx/nodes_to_fix'
+module Docx
+  class PlaceholderObserver
+    attr_reader :data_provider
+    def initialize(data_provider)
+      @data_provider = data_provider
+      @buffer = ''
+      @state = :waiting_for_opening
+      @nodes_to_fix = NodesToFix.new
+      @fixes_to_make = []
+    end
+
+    def next_node(node)
+      node.value.split(//).each_with_index do |c,index|
+        next_char(node,index,c)
+      end
+    end
+
+    def end_of_document
+      make_fixes
+    end
+
+    private
+
+    attr_accessor :state, :buffer, :nodes_to_fix
+
+    def next_char(node, index, c)
+      send(state, node, index, c)
+    end
+
+    def waiting_for_opening(node, index, c)
+      if c == '|'
+        add_char_to_buffer(node,index,c)
+        if buffer == '||'
+          self.state = :capturing_placeholder
+        end
+      else
+        truncate_buffer
+      end
+    end
+
+    def capturing_placeholder(node, index, c)
+      add_char_to_buffer(node,index,c)
+      if buffer[-2..-1] == '||'
+        key = buffer[2..-3]
+        if data_provider.has_key?(key.to_sym)
+          new_value = data_provider[key.to_sym]
+          save_fix_for_later(new_value)
+        end
+        self.state = :waiting_for_opening
+        truncate_buffer
+      end
+    end
+
+    def add_char_to_buffer(node, index, c)
+      @nodes_to_fix.remember(node,index)
+      @buffer << c
+    end
+
+    def truncate_buffer
+      @buffer = ''
+      @nodes_to_fix.forget
+    end
+
+    def save_fix_for_later(val)
+      @nodes_to_fix.value = val
+      @fixes_to_make << @nodes_to_fix
+      @nodes_to_fix = NodesToFix.new
+    end
+
+    def make_fixes
+      @fixes_to_make.reverse.each do |nodes_to_fix|
+        nodes_to_fix.fix
+      end
+      @fixes_to_make = []
+    end
+  end
+end

data/lib/docx_templater.rb

@@ -1,5 +1,7 @@
 require 'zip/zipfilesystem'
 require 'htmlentities'
+require 'docx/argument_combiner'
+require 'docx/document_replacer'
 
 # Use .docx as reusable templates
 # 
@@ -32,53 +34,20 @@ class DocxTemplater
   end
 
   def generate_tags_for(*args)
-    attributes = {}
-    args.flatten!
-    # Prefixes the model name or custom prefix. Makes it so we don't having naming clashes when used with records from multiple m
-    args.each do |arg|
-      if arg.is_a?(Hash) && arg.has_key?(:data) && arg.has_key?(:prefix)
-        template_attributes = (arg[:data].respond_to?(:template_attributes) && :template_attributes) || :attributes
-        arg[:data].send(template_attributes).each_key do |key|
-          attributes["#{arg[:prefix]}_#{key.to_s}".to_sym] = arg[:data].send(template_attributes)[key]
-        end
-      elsif arg.is_a?(Hash)
-        attributes.merge!(arg)
-      else
-        template_attributes = (arg.respond_to?(:template_attributes) && :template_attributes) || :attributes
-        arg.send(template_attributes).each_key do |key|
-          attributes["#{arg.class.name.underscore}_#{key.to_s}".to_sym] = arg.send(template_attributes)[key]
-        end
-      end
-    end
-    attributes
+    Docx::ArgumentCombiner.new(*args).attributes
   end
 
-  private
-  def all_tags_regex
-    /\|\|\<*.+?\>*\|\|/
-  end
-  
-  def malformed_tag_regex
-    /(?<=>)\w{3,}(?=<)/
-  end
-  
-  def well_formed_tag_regex
-    /(?<=\|\|)\w{3,}(?=\|\|)/
-  end
-  
-  def just_label_regex
-    /(?<=>)(\w{3,})/
-  end
-  
   def entry_requires_replacement?(entry)
     entry.ftype != :directory && entry.name =~ /document|header|footer/
   end
   
   def get_entry_content(entry, data_provider)
+    file_string = entry.get_input_stream.read
     if entry_requires_replacement?(entry)
-      replace_entry_content(entry.get_input_stream.read, data_provider)
+      replacer = Docx::DocumentReplacer.new(file_string, data_provider)
+      replacer.replaced
     else
-      entry.get_input_stream.read
+      file_string
     end
   end
   
@@ -86,29 +55,4 @@ class DocxTemplater
     output.put_next_entry(entry.name)
     output.write get_entry_content(entry, data_provider) if entry.ftype != :directory
   end
-  
-  def replace_entry_content(str, data_provider)
-    possible_tags = str.scan(all_tags_regex)
-    # Loops through what looks like are tags. Anything with ||name|| even if they are not in the available tags list
-    possible_tags.each do |tag|
-      #extracts just the tag name
-      tag_name = malformed_tag_regex.match(tag)
-      tag_name ||= well_formed_tag_regex.match(tag)
-      tag_name ||= ''
-      # This will handle instances where someone edits just part of a tag and Word wraps that part in more XML
-      words = tag.scan(just_label_regex).flatten!
-      if words.respond_to?(:size) && words.size > 1
-        #Then the tag was split by word
-        tag_name = words.join('')
-      end
-      tag_name = tag_name.to_s.to_sym
-      # if in the available tag list, replace with the new value
-      if data_provider.has_key?(tag_name)
-        encoder = HTMLEntities.new
-        content = encoder.encode("#{data_provider[tag_name]}")
-        str.gsub!(tag, content)
-      end
-    end
-    str
-  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: docx_templater
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.1.1
   prerelease: 
 platform: ruby
 authors:
@@ -52,6 +52,10 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/docx_templater.rb
+- lib/docx/argument_combiner.rb
+- lib/docx/document_replacer.rb
+- lib/docx/nodes_to_fix.rb
+- lib/docx/placeholder_observer.rb
 homepage: http://rubygems.org/gems/docx_templater
 licenses: []
 post_install_message: 
@@ -72,7 +76,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.24
+rubygems_version: 1.8.23
 signing_key: 
 specification_version: 3
 summary: Uses a .docx as a template and replaces 'tags' within || with other content