docxtor2 0.0.9

data/lib/docxtor2/known/templates.rb

@@ -0,0 +1,7 @@
+module Docxtor2
+  module Known
+    module Templates
+      DEFAULT = File.join(Path::TEMPLATES, 'default')
+    end
+  end
+end

data/lib/docxtor2/object_utils.rb

@@ -0,0 +1,7 @@
+module Docxtor2
+  module ObjectUtils
+    def find_argument(args, klass, default = nil)
+      args.find { |arg| arg.is_a? klass } || default
+    end
+  end
+end

data/lib/docxtor2/package.rb

@@ -0,0 +1,33 @@
+module Docxtor2
+  class Package
+    attr_reader :parts
+
+    def initialize(parts, document)
+      @parts = parts
+      @parts[Known::Parts::DOCUMENT] = document
+    end
+
+    def save(filepath)
+      Zip::ZipOutputStream.open(filepath) do |ostream|
+        write_parts(ostream)
+      end
+    end
+
+    def to_stream
+      ostream = Zip::ZipOutputStream.new("streamed", true)
+      write_parts(ostream)
+      string_io = ostream.close_buffer
+      string_io.rewind
+      string_io
+    end
+
+    private
+
+    def write_parts(ostream)
+      @parts.each do |name, part|
+        ostream.put_next_entry(part.filename)
+        ostream.puts part.content
+      end
+    end
+  end
+end

data/lib/docxtor2/package/document.rb

@@ -0,0 +1,28 @@
+module Docxtor2
+  class Package::Document < Package::Part
+    def initialize(content, filepath = DOCUMENT_XML_PATH)
+      super(filepath, self.class.render(content))
+    end
+
+    def self.render(content)
+      xml = Builder::XmlMarkup.new
+      
+      xml.instruct! :xml, :version => "1.0", :encoding => "UTF-8", :standalone => "yes"
+      xml.w :document, "xmlns:ve" => "http://schemas.openxmlformats.org/markup-compatibility/2006",
+                       "xmlns:o" => "urn:schemas-microsoft-com:office:office",
+                       "xmlns:r" => "http://schemas.openxmlformats.org/officeDocument/2006/relationships",
+                       "xmlns:m" => "http://schemas.openxmlformats.org/officeDocument/2006/math",
+                       "xmlns:v" => "urn:schemas-microsoft-com:vml",
+                       "xmlns:wp" => "http://schemas.openxmlformats.org/drawingml/2006/wordprocessingDrawing",
+                       "xmlns:w10" => "urn:schemas-microsoft-com:office:word",
+                       "xmlns:w" => "http://schemas.openxmlformats.org/wordprocessingml/2006/main",
+                       "xmlns:wne" => "http://schemas.microsoft.com/office/word/2006/wordml" do
+        xml.w :body do
+          xml << content
+        end
+      end
+
+      xml.target!
+    end
+  end
+end

data/lib/docxtor2/package/document/element.rb

@@ -0,0 +1,89 @@
+module Docxtor2
+  class Package::Document::Element < ElementList
+    include BlockEvaluator
+    include ObjectUtils
+
+    def initialize(*args, &block)
+      @params = create_attributes(args)
+      @block = block
+      super()
+    end
+
+    def render(xml)
+      @xml = xml
+      evaluate &@block
+    end
+
+    protected
+
+    def mappings
+      {}
+    end
+
+    def aliases
+      {}
+    end
+
+    def write_element(name, &block)
+      @xml.tag!("w:#{name}") do
+        write_properties(name)
+        evaluate &block
+        write_elements(@xml)
+      end
+    end
+
+    def write_properties(el)
+      @properties = get_properties_for(el)
+      unless @properties.nil?
+        @xml.tag!("w:#{el}Pr") do
+          @properties.each { |k, v| write_property(k, v) }
+        end
+      end
+    end
+
+    def write_property(key, val)
+      if self_closing? val
+        @xml.tag!("w:#{key}")
+      elsif !val.nil?
+        if val.is_a?(Hash) && !val.empty?
+          @xml.tag!("w:#{key}", prefixize(val))
+        elsif !val.to_s.empty?
+          @xml.tag!("w:#{key}", 'w:val' => val)
+        end
+      end
+    end
+
+    private
+
+    def create_attributes(args)
+      hash = find_argument(args, Hash, {})
+      pairs = hash.map do |k, v|
+        aliases.key?(k) ? [aliases[k], v] : [k, v]
+      end
+      Hash[pairs]
+    end
+
+    def self_closing?(val)
+      !!val == val && val
+    end
+
+    def get_properties_for(el)
+      map = mappings[el]
+      unless map.nil?
+        pairs = @params.
+          reject { |k, v| !map.key?(k) }.
+          map { |k, v|
+            element = map[k]
+            element.is_a?(Hash) ?
+              [element[:name], v] :
+              [element, v]
+          }
+        Hash[pairs]
+      end
+    end
+
+    def prefixize(attrs)
+      Hash[attrs.map { |k, v| ["w:#{k}", v] }]
+    end
+  end
+end

data/lib/docxtor2/package/document/heading.rb

@@ -0,0 +1,11 @@
+module Docxtor2
+  class Package::Document::Heading < Package::Document::Paragraph
+    H1 = 1
+    H2 = 2
+
+    def initialize(nesting = H1, *args, &block)
+      super(*args, &block)
+      @params[:style] ||= nesting
+    end
+  end
+end

data/lib/docxtor2/package/document/page_break.rb

@@ -0,0 +1,16 @@
+module Docxtor2
+  class Package::Document::PageBreak < Package::Document::Element
+    def initialize(*args, &block)
+      super(*args, &block)
+    end
+
+    def render(xml)
+      super(xml)
+      write_element(:p) do
+        write_element(:r) do
+          @xml.w :br, 'w:type' => 'page'
+        end
+      end
+    end
+  end
+end

data/lib/docxtor2/package/document/paragraph.rb

@@ -0,0 +1,90 @@
+module Docxtor2
+  class Package::Document::Paragraph < Package::Document::Element
+    def initialize(*args, &block)
+      super(*args, &block)
+      Known::Mappings::PARAGRAPH_COMPLEX.each do |name, element|
+        @params[name] ||= {}
+      end
+      @params[:space] ||= 'default'
+      @contents = create_contents(args)
+    end
+
+    def render(xml)
+      super(xml)
+      write_element(:p) do
+        write_element(:r) do
+          write_contents
+        end
+      end
+    end
+
+    Known::Mappings::PARAGRAPH_SIMPLE.each do |name, element|
+      define_method(name) { |val| @params[name] = val }
+    end
+    Known::Mappings::RUN.each do |name, element|
+      define_method(name) { @params[name] = true }
+    end
+    Known::Mappings::PARAGRAPH_COMPLEX.each do |name, element|
+      define_method(name) { |attrs| @params[name].merge!(attrs) }
+    end
+
+    def line_break
+      @contents << :br
+    end
+
+    def preserve_whitespace
+      @params[:space] = 'preserve'
+    end
+
+    def write(text)
+      @contents << text
+    end
+
+    protected
+
+    def mappings
+      super.merge({
+        :p => Known::Mappings::PARAGRAPH,
+        :r => Known::Mappings::RUN
+      })
+    end
+
+    def aliases
+      super.merge(Known::Aliases::PARAGRAPH)
+    end
+
+    private
+
+    def write_contents
+      @contents.each { |c| write_content(c) }
+    end
+
+    def write_content(content)
+      content == :br ?
+        write_line_break :
+        write_text(content)
+    end
+
+    def write_line_break
+      @xml.w :br
+    end
+
+    def write_text(text)
+      @xml.w :t, 'xml:space' => @params[:space] do
+        @xml.text! text
+      end
+    end
+
+    def create_contents(args)
+      str = find_argument(args, String)
+      str.nil? ? [] : [str]
+    end
+
+    alias_method :b, :bold
+    alias_method :i, :italic
+    alias_method :u, :underline
+    alias_method :br, :line_break
+    alias_method :w, :write
+    alias_method :ind, :indent
+  end
+end

data/lib/docxtor2/package/document/table_of_contents.rb

@@ -0,0 +1,90 @@
+module Docxtor2
+  class Package::Document::TableOfContents < Package::Document::Element
+    def initialize(text, &block)
+      super({ :style => Known::Styles::TOC }, &block)
+      @text = text
+    end
+
+    # TODO: Add support for extended styling & properties
+
+    def render(xml)
+      super(xml)
+
+      @xml.w :sdt, "xmlns:w" => "http://schemas.openxmlformats.org/wordprocessingml/2006/main" do
+
+        write_properties
+
+        @xml.w :sdtContent do
+
+          write_heading
+
+          @xml.w :p do
+            @xml.w :pPr do
+              @xml.w :pStyle, "w:val" => Known::Styles::TOC_PARAGRAPH
+              @xml.w :tabs do
+                @xml.w :tab, "w:val" => "right", "w:leader" => "dot", "w:pos" => 9350
+              end
+              @xml.w :rPr do
+                @xml.w :noProof
+              end
+            end
+            @xml.w :r do
+              @xml.w :fldChar, "w:fldCharType" => "begin", "w:dirty" => true
+            end
+
+            write_instruct
+
+            @xml.w :r do
+              @xml.w :fldChar, "w:fldCharType" => "separate"
+            end
+          end
+
+          @xml.w :p do
+            @xml.w :r do
+              # @xml.w :rPr do
+              #   @xml.w :b
+              #   @xml.w :bCs
+              #   @xml.w :noProof
+              # end
+              @xml.w :fldChar, "w:fldCharType" => "end"
+            end
+          end
+
+        end
+      end
+    end
+
+    private
+
+    def write_properties
+      @xml.w :sdtPr do
+        @xml.w :docPartObj do
+          @xml.w :docPartGallery, "w:val" => "Table of Contents"
+          @xml.w :docPartUnique
+        end
+      end
+    end
+
+    def write_heading
+      @xml.w :p do
+        @xml.w :pPr do
+          @xml.w :pStyle, "w:val" => @params[:style]
+        end
+        @xml.w :r do
+          @xml.w :t do
+            @xml.text! @text
+          end
+        end
+      end
+    end
+
+    def write_instruct
+      @xml.w :r do
+        @xml.w :instrText, "xml:space" => "preserve" do
+          @xml.text! " TOC \\o '1-3' \\h \\z \\u "
+        end
+      end
+    end
+
+  end
+end

data/lib/docxtor2/package/part.rb

@@ -0,0 +1,10 @@
+module Docxtor2
+  class Package::Part
+    attr_reader :filename, :content
+
+    def initialize(filename, content)
+      @filename = filename
+      @content = content
+    end
+  end
+end

data/lib/docxtor2/template_parser.rb

@@ -0,0 +1,42 @@
+module Docxtor2
+  class TemplateParser
+
+    class << self
+      def parse(template)
+        instance = new(template)
+        instance.parts
+      end
+    end
+
+    def initialize(template)
+      @parts = parse(template)
+    end
+
+    def parts
+      @parts
+    end
+
+    private
+
+    def parse(template)
+      Dir.chdir(template) do
+        Hash[create_parts]
+      end
+    end
+
+    def create_parts
+      Dir[SEARCH_PATTERN].
+        delete_if { |file| File.directory?(file) }.
+        map { |file| create_part(file) }
+    end
+
+    def create_part(file)
+      content = File.read(file)
+      part = Package::Part.new(file, content)
+      key = File.basename(file, '.xml')
+
+      [key, part]
+    end
+
+  end
+end

data/lib/docxtor2/version.rb

@@ -0,0 +1,3 @@
+module Docxtor2
+  VERSION = "0.0.9"
+end

data/spec/docxtor2/content_builder_spec.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+module Docxtor2
+  describe ContentBuilder do
+    include WordprocessingMLMatchers
+
+    subject {
+      ContentBuilder.build do
+        table_of_contents 'Contents'
+        p :style => 3 do
+          write "content 1"
+          line_break
+          write "content 2"
+        end
+      end
+    }
+
+    it 'renders given DSL instructions' do
+      subject.should contain_paragraph_text
+      subject.should contain_element_style(:p)
+      subject.should contain_table_of_contents
+    end
+  end
+end