calco 0.1.0

data/lib/calco/elements/if.rb

@@ -0,0 +1,26 @@
+require_relative 'element'
+
+module Calco
+
+  class If < Element
+
+    def initialize condition, _then, _else
+
+      super()
+
+      @condition, @_then, @_else = condition, Constant.wrap(_then), Constant.wrap(_else)
+
+    end
+
+    def generate row
+
+      _then = generate_operand(@_then, row)
+      _else = generate_operand(@_else, row)
+
+      "IF(#{@condition.generate(row)};#{_then};#{_else})"
+
+    end
+
+  end
+
+end

data/lib/calco/elements/operation.rb

@@ -0,0 +1,34 @@
+require_relative 'element'
+
+module Calco
+
+  class Operation < Element
+
+    include Operators
+
+    def initialize operator, operand1, operand2
+      @operator, @operand1, @operand2 = operator, Constant.wrap(operand1), Constant.wrap(operand2)
+    end
+
+    def generate row
+
+      return "#{@engine.operator(@operator)}#{generate_operand(@operand1, row)}" unless @operand2
+      
+      operand1 = generate_operand(@operand1, row)
+      operand2 = generate_operand(@operand2, row)
+
+      "#{operand1}#{@engine.operator(@operator)}#{operand2}"
+
+    end
+
+    def as_operand row
+      "(#{generate(row)})"
+    end
+    
+    def -@
+      Operation.new('-', self, nil)
+    end
+    
+  end
+
+end

data/lib/calco/elements/operator.rb

@@ -0,0 +1,17 @@
+module Calco
+
+  module Operators
+
+    %w[+ - / * == != < > >= <=].each do |op|
+
+      op_str = op == '==' ? '=' : op
+
+      define_method(op) do |arg|
+        Operation.new(op_str, self, arg)
+      end
+
+    end
+
+  end
+
+end

data/lib/calco/elements/or.rb

@@ -0,0 +1,26 @@
+require_relative 'element'
+
+module Calco
+
+  class Or < Element
+
+    def initialize condition1, condition2
+
+      super()
+
+      @condition1, @condition2 = Constant.wrap(condition1), Constant.wrap(condition2)
+
+    end
+
+    def generate row
+
+      condition1 = generate_operand(@condition1, row)
+      condition2 = generate_operand(@condition2, row)
+
+      "OR(#{condition1};#{condition2})"
+
+    end
+
+  end
+
+end

data/lib/calco/elements/value_extractor.rb

@@ -0,0 +1,42 @@
+require_relative 'element'
+
+module Calco
+
+  class ValueExtractor < Element
+
+    def initialize variable, reference_type
+      @variable, @reference_type = variable, reference_type
+      @variable.reference_type = reference_type
+    end
+
+    def column= column_name
+      @variable.column = column_name
+    end
+
+    def value_name
+      @variable.name
+    end
+
+    def value
+      @variable.value
+    end
+
+    def absolute_row
+      @variable.absolute_row
+    end
+
+    def reference_type
+      @variable.reference_type
+    end
+
+    def generate row
+      
+      return nil if @variable.absolute_row and @variable.absolute_row != row
+
+      @engine.value(self)
+      
+    end
+
+  end
+
+end

data/lib/calco/elements/variable.rb

@@ -0,0 +1,35 @@
+require_relative 'element'
+require_relative 'aggregator'
+
+module Calco
+
+  class Variable < Element
+
+    include Operators
+
+    attr_accessor :value, :name
+
+    def initialize name, value
+
+      super()
+
+      @name, @value = name, value
+
+    end
+
+    def [] range
+      
+      raise ArgumentError, "Expected Range got #{range.class}" unless range.is_a?(Range)
+      raise ArgumentError, "Invalid start of range (must be > 0, was #{range.first})" unless range.first > 0
+      
+      Aggregator.new(self, range)
+      
+    end
+    
+    def generate row
+      @engine.column_reference(self, row)
+    end
+
+  end
+
+end

data/lib/calco/engines/calculator_builtin_functions.rb

@@ -0,0 +1,32 @@
+require 'date'
+require 'time'
+
+require 'calco'
+
+module Calco
+
+  module CalculatorBuiltinFunctions
+    
+    def YEAR date
+      date.year
+    end
+    
+    def TODAY
+      Date.today
+    end
+    
+    def DATEVALUE text
+      Date.parse(text)
+    end
+    
+    def TIMEVALUE text
+      Time.parse(text)
+    end
+    
+    def LEFT text, n
+      text[0, n]
+    end
+    
+  end
+
+end

data/lib/calco/engines/csv_engine.rb

@@ -0,0 +1,80 @@
+require 'csv'
+require 'calco'
+
+module Calco
+
+  class CSVEngine < Calco::DefaultEngine
+
+    def initialize col_sep = ',', quote_char = '"'
+      @col_sep, @quote_char = col_sep, quote_char
+    end
+
+    def empty_row
+      @out_stream.write CSV.generate_line([])
+    end
+
+    def write_row sheet, row_id
+
+      row_id += 1 if row_id > 0 and sheet.has_titles?
+
+      cells = sheet.row(row_id)
+
+      @out_stream.write CSV.generate_line(cells, :col_sep => @col_sep, :quote_char => @quote_char)
+
+    end
+
+    def generate_cell row_number, column, cell_style, column_style, column_type
+
+      if column.is_a?(Calco::Formula)
+
+        res = super
+
+        if column_type =~ /^[$]/
+          res = "=DOLLAR(#{res})"
+        elsif column_type == '%'
+          res = "=(#{res})%"
+        else
+          res = "=#{res}"
+        end
+
+      elsif column.respond_to?(:value)
+
+        value = column.value
+
+        style = cell_style ? "#{cell_style.generate(row_number)}" : ''
+
+        if value.is_a?(Time)
+          res = "=TIMEVALUE(\"#{value.strftime("%H:%M:%S")}\")#{style}"
+        elsif value.is_a?(Date)
+          res = "=DATEVALUE(\"#{value.strftime("%Y-%m-%d")}\")#{style}"
+        elsif value.is_a?(Numeric)
+
+          if column_type =~ /^[$]/
+            res = "=DOLLAR(#{value}#{style})"
+          elsif column_type == '%'
+            res = "=(#{value}*100)%"
+          else
+            res = "#{value}#{style}"
+          end
+
+        else
+          res = "#{value}#{style}"
+        end
+
+      end
+
+      res
+
+    end
+
+    def style statement, row
+      "+STYLE(#{statement.generate(row)})"
+    end
+
+    def current
+      'CURRENT()'
+    end
+
+  end
+
+end

data/lib/calco/engines/default_engine.rb

@@ -0,0 +1,140 @@
+require 'calco'
+
+module Calco
+
+  class DefaultEngine
+
+    # output can be a String (as a file name) or an output stream
+    # examples:
+    #     engine.save doc, "my_doc.txt"  do |spreadsheet|
+    #        ...
+    #     end
+    #
+    #     engine.save doc, $stdout  do |spreadsheet|
+    #        ...
+    #     end
+    def save doc, output, &data_iterator
+    
+      if output.respond_to?(:write)
+        @out_stream = output
+      else
+        @out_stream = open(output, "w")
+      end
+    
+      data_iterator.call(doc)
+      
+    end
+
+    def empty_row
+    end
+
+    def write_row sheet, row_id
+    
+      cells = sheet.row(row_id)
+      
+      cells.each_index do |i|
+
+        cell = cells[i]
+
+        @out_stream.write "#{COLUMNS[i]}#{row_id}: #{cell}\n"
+
+      end
+      
+      @out_stream.write "\n"
+      
+    end
+
+    def generate row_number, columns, cell_styles, column_styles, column_types
+
+      cells = []
+
+      columns.each_with_index do |column, i|
+        cells << generate_cell(row_number, column, cell_styles[i], column_styles[i], column_types[i])
+      end
+
+      cells
+
+    end
+
+    def generate_cell row_number, column, cell_style, column_style, column_type
+
+      cell = '' unless column
+      cell = column.generate(row_number) if column
+
+      if cell_style
+        cell = cell.to_s + cell_style.generate(row_number)
+      end
+
+      cell
+
+    end
+
+    def reference_types
+      {
+          :current => proc { |element, r, c| current },
+          :normal => proc { |element, r, c| "#{c}#{r}" },
+          :absolute => proc { |element, r, c| "$#{c}$#{element.absolute_row}" },
+          :absolute_row => proc { |element, r, c| "#{c}$#{element.absolute_row}" },
+          :absolute_column => proc { |element, r, c| "$#{c}#{r}" }
+      }
+    end
+
+    def column_reference element, row
+      reference_types[element.reference_type].call(element, row, element.column)
+    end
+
+    def range_reference element, row, range
+    
+      from = reference_types[element.reference_type].call(element, range.first, element.column)
+      
+      if range.grouping_range?
+        to = row - 1
+      elsif range.last == -1
+        to = DefaultEngine.row_infinity
+      else
+        to = range.last
+      end
+      
+      to = reference_types[element.reference_type].call(element, to, element.column)
+      
+      "#{from}:#{to}"
+      
+    end
+
+    def value element
+      if ! element.respond_to?(:value)
+        element.inspect
+      elsif element.value.is_a?(Time)
+        res = element.value.strftime("%H:%M:%S")
+      elsif element.value.is_a?(Date)
+        res = element.value.strftime("%Y-%m-%d")
+      else
+        element.value.inspect
+      end
+    end
+
+    def style statement, row
+      "; apply_style(#{statement.generate(row)})"
+    end
+
+    def operator op
+      op
+    end
+
+    def current
+      'self'
+    end
+
+    def self.row_infinity
+      @@row_infinity
+    end
+    
+    def self.row_infinity= value
+      @@row_infinity = value
+    end
+
+    @@row_infinity = 1048576
+
+  end
+
+end

data/lib/calco/engines/office_engine.rb

@@ -0,0 +1,263 @@
+require 'date'
+
+require 'zip'
+require 'tmpdir'
+require 'pathname'
+require 'tempfile'
+require 'rexml/document'
+
+require 'calco'
+
+module Calco
+
+  class OfficeEngine < DefaultEngine
+
+    def initialize ods_template, first_row_is_header = true
+      @ods_template = ods_template
+      @first_row_is_header = first_row_is_header
+    end
+    
+    # output is a String (as a file name)
+    def save doc, to_filename, &data_iterator
+
+      content_xml_file = Tempfile.new('office-gen')
+      result_xml_file = Tempfile.new('office-gen')
+      
+      Zip::File.open(@ods_template) do |zipfile|
+        content = zipfile.read("content.xml")
+        open(content_xml_file, "w") {|out| out.write content}
+      end
+
+      write_result_content doc, content_xml_file, result_xml_file, @first_row_is_header, &data_iterator
+
+      FileUtils.cp(@ods_template, to_filename)
+
+      Zip::File.open(to_filename) do |zipfile|
+
+        zipfile.get_output_stream("content.xml") do |os|
+
+          File.open(result_xml_file).each_line do |line|
+            os.puts line
+          end
+
+        end
+
+      end
+
+    end
+
+    def empty_row
+      @out_stream.write '<table:table-row/>'
+    end
+    
+    def write_row sheet, row_id
+
+      return if row_id == 0 and sheet.has_titles?
+      
+      row_id += 1 # office sheet indexes start at 1
+      
+      cells = sheet.row(row_id)
+
+      @out_stream.write '<table:table-row>'
+
+      cells.each_index do |i|
+
+        cell = cells[i]
+
+        @out_stream.write cell
+
+      end
+
+      @out_stream.write '</table:table-row>'
+
+    end
+
+    def generate_cell row_number, column, cell_style, column_style, column_type
+
+      return '<table:table-cell/>' unless column
+      return '<table:table-cell/>' if column.absolute_row and column.absolute_row != row_number
+
+      cell = column.generate(row_number)
+
+      if cell_style
+        cell = cell.to_s + cell_style.generate(row_number)
+      end
+
+      if column_style
+        column_style = %[table:style-name="#{column_style}"]
+      else
+        column_style = ''
+      end
+
+      if column_type
+
+        if column_type == '%'
+          column_type = %[office:value-type="percentage"]
+        elsif column_type =~ /\$([A-Z]{3})/
+          column_type = %[office:value-type="currency" office:currency="#{$1}"]
+        else
+          column_type = %[office:value-type="#{column_type}"]
+        end
+
+      end
+
+      if column.is_a?(Formula)
+
+        column_type = 'office:value-type="float"' unless column_type
+
+        %[<table:table-cell #{column_style} #{column_type} table:formula="of:=#{cell}" />]
+
+      elsif column.respond_to?(:value)
+
+        if column.value.nil?
+          
+          "<table:table-cell/>"
+          
+        elsif column.value.is_a?(Numeric)
+
+          column_type = 'office:value-type="float"' unless column_type
+
+          %[<table:table-cell #{column_style} #{column_type} office:value="#{cell}"/>]
+
+        elsif column.value.is_a?(Date)
+
+          column_type = 'office:value-type="date"' unless column_type
+
+          %[<table:table-cell #{column_style} #{column_type} office:date-value="#{cell.to_s}"/>]
+
+        else
+
+          column_type = 'office:value-type="string"' unless column_type
+
+          %[
+            <table:table-cell #{column_style} #{column_type}>
+              <text:p><![CDATA[#{cell}]]></text:p>
+            </table:table-cell>
+          ]
+
+        end
+
+      end
+
+    end
+
+    def column_reference element, row
+      element.is_a?(Current) ? "#{super}" : "[.#{super}]"
+    end
+
+    def value element
+    
+      if ! element.respond_to?(:value)
+        s = element.to_s
+      else
+        s = element.value.to_s
+      end
+      
+      s.gsub!('"', '&quot;') if s =~ /"/
+      s
+      
+    end
+
+    def style statement, row
+      "+ORG.OPENOFFICE.STYLE(#{statement.generate(row)})"
+    end
+
+    def operator op
+
+      if op == '!='
+        '&lt;&gt;'
+      elsif op == '<'
+        '&lt;'
+      elsif op == '>'
+        '&gt;'
+      else
+        op
+      end
+
+    end
+
+    def current
+      'ORG.OPENOFFICE.CURRENT()'
+    end
+
+    private
+
+    # returns the parent table and removes template/example rows, also returns
+    # the first template/example row (to find cell styles for instance)
+    def retrieve_template_row doc, first_row_is_header
+
+      root = doc.root
+
+      count = 0
+      template_row = nil
+
+      table = root.elements['//table:table']
+      table.each_element('table:table-row') do |row|
+
+        if first_row_is_header && count == 0
+          # keep the header row
+        else
+
+          table.delete_element(row)
+
+          template_row = row unless template_row
+
+        end
+
+        count += 1
+
+      end
+
+      raise "Cannot find template row in #{@ods_template}" unless template_row
+
+      return table, template_row
+
+    end
+
+    def create_temporary xml, to_filename
+
+      to = Pathname.new(to_filename)
+
+      temp_file = Tempfile.new('office-gen', to.dirname.to_s)
+
+      File.open(temp_file, 'w') { |stream| stream.puts xml }
+
+      temp_file
+
+    end
+
+    def write_result_content doc, content_xml_file, result_xml_file, first_row_is_header, &data_iterator
+
+      file = File.new(content_xml_file)
+
+      xml = REXML::Document.new(file)
+
+      table, template_row = retrieve_template_row(xml, first_row_is_header)
+
+      table.add_text "%%%Insert data here%%%\n"
+
+      temp_file = create_temporary(xml, result_xml_file)
+
+      File.open(result_xml_file, 'w') do |stream|
+
+        @out_stream = stream
+
+        File.open(temp_file, 'r').each do |line|
+
+          if line =~ /(.*)%%%Insert data here%%%(.*)/
+            @out_stream.write $1
+            data_iterator.call(doc)
+            @out_stream.write $2
+          else
+            @out_stream.write line
+          end
+
+        end
+
+      end
+
+    end
+
+  end
+
+end