rubyXL 1.0.4

data/lib/tests/test6.rb

@@ -0,0 +1,50 @@
+require '../rubyXL'
+require 'rubygems'
+# require File.expand_path(File.join(File.dirname(__FILE__),'workbook'))
+# require File.expand_path(File.join(File.dirname(__FILE__),'parser'))
+# require File.expand_path(File.join(File.dirname(__FILE__),'color'))
+# require File.expand_path(File.join(File.dirname(__FILE__),'cell'))
+
+module RubyXL
+wb = Workbook.new([],nil)
+wb.worksheets = [Worksheet.new(wb,'Sheet1')]
+cell = wb.worksheets[0].add_cell(0,0,'1.00.0')
+# cell.change_font_italics(false)
+# p cell.is_italicized(wb)
+# cell.change_font_bold(false)
+# p cell.is_bolded(wb)
+# cell.change_font_underline(false)
+# p cell.is_underlined(wb)
+# p cell.font_name(wb)
+# p cell.font_size(wb)
+# p cell.font_color(wb)
+# p cell.fill_color(wb)
+
+wb.worksheets[0].add_cell(2,5,'$1,000.00')
+wb.worksheets[0].add_cell(3,3,'6/14/11')
+wb.worksheets[0].add_cell(4,0,1)
+wb.worksheets[0].add_cell(4,1,2)
+wb.worksheets[0].add_cell(4,2,3)
+wb.worksheets[0].add_cell(4,3,4)
+wb.worksheets[0].add_cell(4,4,0,'AVERAGE(A5:D5)')
+
+cell.change_font_color('ff0000')
+
+wb.write('/Users/vbhagwat/Desktop/test2/Output/nums.xlsx')
+
+
+wb2 = Parser.parse('/Users/vbhagwat/Desktop/5-1_5-20.xlsx')
+# wb2.worksheets[0].merge_cells(0,1,0,2)
+# wb2.worksheets[0].merge_cells(0,0,0,1)
+#.change_font_size(30)
+wb2.write('/Users/vbhagwat/Desktop/test2/Output/nums2.xlsx')
+
+wb3 = Parser.parse('/Users/vbhagwat/Documents/excelTestFiles/paneWorkbook2/paneWorkbook2.xlsx')
+
+
+ 
+# p Color.find(8)
+# c = Color::ColorProperties
+# p c.has_value?({:hex=>'#000000', :name=>"black"})
+# p Color.find('black')
+end

data/lib/tests/test7.rb

@@ -0,0 +1,48 @@
+require '../rubyXL'
+require 'rubygems'
+require 'rubyXL'
+# require File.expand_path(File.join(File.dirname(__FILE__),'workbook'))
+
+module RubyXL
+wb = Workbook.new([],nil)
+wb.worksheets = [Worksheet.new(wb,'Sheet1')]
+
+cell = wb.worksheets[0].add_cell(0,0,'A1')
+cell2 = wb.worksheets[0].add_cell(0,1,'B1')
+cell3 = wb.worksheets[0].add_cell(0,5,'F1')
+
+wb.worksheets[0].sheet_data[0].each do |c|
+  unless c.nil?
+    c.change_font_bold(true)
+    c.change_font_underline(true)
+  end
+end
+
+cell.change_horizontal_alignment('center')
+wb.worksheets[0].change_row_horizontal_alignment(0,'justify')
+wb.worksheets[0].change_row_vertical_alignment(0,'center')
+
+wb.worksheets[0].change_row_fill(0,'FF0000')
+cell2.change_fill('0000FF')
+
+cell4 = wb.worksheets[0].add_cell(0,3,'D1')
+puts '
+
+
+
+'
+p wb.fills
+puts '
+
+
+
+
+'
+cell4.change_fill('A43502')
+
+cell5 = wb.worksheets[0].add_cell(1,0,'A2')
+wb.worksheets[0].change_row_fill(1,'00FF00')
+cell5.change_fill('FF0000')
+
+wb.write('/Users/vbhagwat/Desktop/test2/Output/test7.xlsx')
+end

data/lib/tests/test8.rb

@@ -0,0 +1,12 @@
+require '../rubyXL'
+require 'rubygems'
+# require File.expand_path(File.join(File.dirname(__FILE__),'workbook'))
+# require File.expand_path(File.join(File.dirname(__FILE__),'parser'))
+
+module RubyXL
+
+wb = Parser.parse('/Users/vbhagwat/Desktop/macros.xlsm', false)
+wb.write('/Users/vbhagwat/Desktop/test2/Output/macros.xlsm')
+puts "completed writing /Users/vbhagwat/Desktop/test2/Output/macros.xlsm"
+
+end

data/lib/tests/test9.rb

@@ -0,0 +1,60 @@
+require 'rubygems'
+require 'nokogiri'
+require File.expand_path(File.join(File.dirname(__FILE__),'Hash'))
+
+module RubyXL
+testWb = Nokogiri::XML.parse(File.read('/Users/vbhagwat/Desktop/testWorkbook.xml'))
+testStyles = Nokogiri::XML.parse(File.read('/Users/vbhagwat/Desktop/testStyles.xml'))
+
+puts "testWb.css('workbook definedNames')[0]"
+p testWb.css('workbook definedNames')[0]
+puts '/css'
+
+
+
+
+Hash.xml_node_to_hash(test_wb.css('workbook definedNames')[0])
+
+puts '
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+'
+h= Hash.xml_node_to_hash(test_styles.css('styleSheet cellXfs')[0])
+
+puts '
+
+
+
+
+
+'
+p h
+# 
+# :cellXfs=>{
+#   :xf=>[{
+#     :attributes=>{
+#       :xfId=>0, :fontId=>0, :numFmtId=>0, :borderId=>0, :fillId=>0}}, 
+#     {:attributes=>{
+#       :applyFill=>0, :applyFont=>0, :applyAlignment=>1, :xfId=>0, :fontId=>0, 
+#       :numFmtId=>0, :applyNumberFormat=>0, :borderId=>0, :applyBorder=>0, :fillId=>0}, 
+#       :alignment=>{:attributes=>{:horizontal=>"center"}}}], 
+#       
+#   :attributes=>{:count=>2}}
+end

data/lib/workbook.rb

@@ -0,0 +1,336 @@
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','content_types_writer'))
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','root_rels_writer'))
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','app_writer'))
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','core_writer'))
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','theme_writer'))
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','workbook_rels_writer'))
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','workbook_writer'))
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','styles_writer'))
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','shared_strings_writer'))
+require File.expand_path(File.join(File.dirname(__FILE__),'writer','worksheet_writer'))
+require 'zip'
+
+module RubyXL
+  class Workbook
+    attr_accessor :worksheets, :filepath, :creator, :modifier, :created_at,
+      :modified_at, :company, :application, :appversion, :num_fmts, :fonts, :fills,
+      :borders, :cell_xfs, :cell_style_xfs, :cell_styles, :shared_strings, :calc_chain,
+      :num_strings, :size, :date1904, :external_links, :style_corrector, :drawings,
+      :worksheet_rels, :printer_settings, :macros, :colors, :shared_strings_XML
+
+    APPLICATION = 'Microsoft Macintosh Excel'
+    APPVERSION  = '12.0000'
+    SHEET_NAME = 'Sheet1'
+    def initialize(worksheets=[], filepath=nil, creator=nil, modifier=nil, created_at=nil,
+                   company='', application=APPLICATION,
+                   appversion=APPVERSION, date_1904=0)
+      if worksheets.nil? || worksheets.empty?
+        @worksheets       = [Worksheet.new(self,SHEET_NAME)]
+      else
+        @worksheets       = worksheets
+      end
+      @filepath           = filepath
+      @creator            = creator
+      @modifier           = modifier
+      @company            = company
+      @application        = application
+      @appversion         = appversion
+      @num_fmts           = nil
+      @fonts              = nil
+      @fills              = nil
+      @borders            = nil
+      @cell_xfs           = nil
+      @cell_style_xfs     = nil
+      @cell_styles        = nil
+      @shared_strings     = nil
+      @calc_chain         = nil #unnecessary?
+      @num_strings        = 0 #num strings total
+      @size               = 0 #num strings in shared_strings array
+      @date_1904          = date_1904
+      @external_links     = nil
+      @style_corrector    = nil
+      @drawings           = nil
+      @worksheet_rels     = nil
+      @printer_settings   = nil
+      @macros             = nil
+      @colors             = nil
+      @shared_strings_XML = nil
+
+      begin
+        @created_at       = DateTime.parse(created_at).strftime('%Y-%m-%dT%TZ')
+      rescue
+        t = Time.now
+        @created_at       = t.strftime('%Y-%m-%dT%TZ')
+      end
+      @modified_at        = @created_at
+
+      fill_styles()
+      fill_shared_strings()
+    end
+
+    # allows easier access to worksheets
+    def [](worksheet)
+      return worksheets[worksheet]
+    end
+
+    #filepath of xlsx file (including file itself)
+    def write(filepath=@filepath)
+      validate_before_write
+      dirpath = ''
+      extension = 'xls'
+      if(filepath =~ /((.|\s)*)\.xls(x|m)$/)
+        dirpath = $1.to_s()
+        extension += $3.to_s
+      end
+      filename = ''
+      if(filepath =~ /\/((.|\s)*)\/((.|\s)*)\.xls(x|m)$/)
+        filename = $3.to_s()
+      end
+
+      #creates zip file, writes each type of file to zip folder
+      #zips package and renames it to xlsx.
+      zippath = File.join(dirpath, filename + '.zip')
+      File.unlink(zippath) if File.exists?(zippath)
+      FileUtils.mkdir_p(File.join(dirpath,zippath))
+      Zip::ZipFile.open(zippath, Zip::ZipFile::CREATE) do |zipfile|
+        writer = Writer::ContentTypesWriter.new(dirpath,self)
+        zipfile.get_output_stream('[Content_Types].xml') {|f| f.puts(writer.write())}
+
+        writer = Writer::RootRelsWriter.new(dirpath,self)
+        zipfile.get_output_stream(File.join('_rels','.rels')) {|f| f.puts(writer.write())}
+
+        writer = Writer::AppWriter.new(dirpath,self)
+        zipfile.get_output_stream(File.join('docProps','app.xml')) {|f| f.puts(writer.write())}
+
+        writer = Writer::CoreWriter.new(dirpath,self)
+        zipfile.get_output_stream(File.join('docProps','core.xml')) {|f| f.puts(writer.write())}
+
+        writer = Writer::ThemeWriter.new(dirpath,self)
+        zipfile.get_output_stream(File.join('xl','theme','theme1.xml')) {|f| f.puts(writer.write())}
+
+        writer = Writer::WorkbookRelsWriter.new(dirpath,self)
+        zipfile.get_output_stream(File.join('xl','_rels','workbook.xml.rels')) {|f| f.puts(writer.write())}
+
+        writer = Writer::WorkbookWriter.new(dirpath,self)
+        zipfile.get_output_stream(File.join('xl','workbook.xml')) {|f| f.puts(writer.write())}
+
+        writer = Writer::StylesWriter.new(dirpath,self)
+        zipfile.get_output_stream(File.join('xl','styles.xml')) {|f| f.puts(writer.write())}
+
+        unless @shared_strings.nil?
+          writer = Writer::SharedStringsWriter.new(dirpath,self)
+          zipfile.get_output_stream(File.join('xl','sharedStrings.xml')) {|f| f.puts(writer.write())}
+        end
+
+        #preserves external links (exactly, no modification allowed)
+        unless @external_links.nil?
+          #-1 because of rels
+          1.upto(@external_links.size-1) do |i|
+            zipfile.get_output_stream(
+              File.join('xl','externalLinks',"externalLink#{i}.xml")) {|f|
+                f.puts(@external_links[i])
+              }
+          end
+          @external_links['rels'].each_index do |i|
+            unless @external_links['rels'][i].nil?
+              zipfile.get_output_stream(
+                File.join('xl','externalLinks','_rels',"externalLink#{i}.xml.rels")) {|f|
+                  f.puts(@external_links['rels'][i])
+                }
+            end
+          end
+        end
+
+        #preserves drawings (exactly, no modification allowed)
+        unless @drawings.nil?
+          1.upto(@drawings.size) do |i|
+            zipfile.get_output_stream(
+            File.join('xl','drawings',"vmlDrawing#{i}.vml")) {|f|
+              f.puts(@drawings[i])
+            }
+          end
+        end
+
+        unless @printer_settings.nil?
+          1.upto(@printer_settings.size) do |i|
+            zipfile.get_output_stream(
+            File.join('xl','printerSettings',"printerSettings#{i}.bin")) {|f|
+              f.puts(@printer_settings[i])
+            }
+          end
+        end
+
+        unless @worksheet_rels.nil?
+          1.upto(@worksheet_rels.size) do |i|
+            zipfile.get_output_stream(
+            File.join('xl','worksheets','_rels',"sheet#{i}.xml.rels")) {|f|
+              f.puts(@worksheet_rels[i])
+            }
+          end
+        end
+
+        unless @macros.nil?
+          zipfile.get_output_stream(File.join('xl','vbaProject.bin')) {|f| f.puts(@macros)}
+        end
+
+        @worksheets.each_with_index do |sheet,i|
+          writer = Writer::WorksheetWriter.new(dirpath,self,i)
+          zipfile.get_output_stream(File.join('xl','worksheets',"sheet#{i+1}.xml")) {|f| f.puts(writer.write())}
+        end
+      end
+
+      FileUtils.cp(zippath,File.join(dirpath,filename+".#{extension}"))
+      FileUtils.cp(File.join(dirpath,filename+".#{extension}"),filepath)
+      if File.exist?(filepath)
+        FileUtils.rm_rf(dirpath)
+      end
+    end
+
+    #gets style object from style array given index
+    def get_style(style_index)
+      if !@cell_xfs[:xf].is_a?Array
+        @cell_xfs[:xf] = [@cell_xfs[:xf]]
+      end
+
+      xf_obj = @cell_xfs[:xf]
+      if xf_obj.is_a?Array
+        xf_obj = xf_obj[Integer(style_index)]
+      end
+      xf_obj
+    end
+
+    #gets attributes of above style object
+    #necessary because can take the form of hash or array,
+    #based on odd behavior of Nokogiri
+    def get_style_attributes(xf_obj)
+      if xf_obj.is_a?Array
+        xf = xf_obj[1]
+      else
+        xf = xf_obj[:attributes]
+      end
+    end
+
+    def get_fill_color(xf_attributes)
+      if @fills[xf_attributes[:fillId]].nil? || @fills[xf_attributes[:fillId]][:fill].nil? || @fills[xf_attributes[:fillId]][:fill][:patternFill].nil? || @fills[xf_attributes[:fillId]][:fill][:patternFill][:fgColor].nil?
+        'ffffff' #white
+      else
+        @fills[xf_attributes[:fillId]][:fill][:patternFill][:fgColor][:attributes][:rgb]
+      end
+    end
+
+
+    private
+
+    # Do not change. Excel requires that some of these styles be default,
+    # and will simply assume that the 0 and 1 indexed fonts are the default values.
+    def fill_styles()
+      @fonts = {
+                 '0' => {
+                          :font => {
+                                     :sz   => { :attributes => { :val => 10 } },
+                                     :name => { :attributes => { :val => "Verdana" } }
+                                   },
+                          :count=>1
+                        },
+                 '1' => {
+                          :font => {
+                                     :sz   => { :attributes => { :val => 8 } },
+                                     :name => { :attributes => { :val => "Verdana" } }
+                                   },
+                          :count=>0
+                        }
+               }
+
+      @fills = {
+                 '0' => {
+                          :fill => {
+                                     :patternFill => { :attributes => { :patternType => "none" } }
+                                   },
+                  :count=>1} ,
+                 '1' => {
+                          :fill => {
+                                     :patternFill => { :attributes => { :patternType => "gray125" } }
+                                   },
+                          :count=>0
+                        }
+               }
+
+      @borders = {
+                   '0' => {
+                            :border => {
+                                         :left => { },
+                                         :right => { },
+                                         :top => { },
+                                         :bottom => { },
+                                         :diagonal => { }
+                                       },
+                            :count => 1 #count = how many styles reference it
+                          }
+                 }
+
+      @cell_style_xfs = {
+                        :attributes => {
+                                         :count => 1
+                                       },
+                        :xf => {
+                                 :attributes => { :numFmtId => 0, :fontId => 0, :fillId => 0, :borderId => 0 }
+                               }
+                      }
+      @cell_xfs = {
+                        :attributes => {
+                                         :count => 1
+                                       },
+                        :xf => {
+                                 :attributes => { :numFmtId => 0, :fontId => 0, :fillId => 0, :borderId => 0, :xfId => 0 }
+                               }
+                      }
+      @cell_styles = {
+                      :cellStyle => {
+                                      :attributes => { :builtinId=>0, :name=>"Normal", :xfId=>0 }
+                                    },
+                      :attributes => { :count => 1 }
+                    }
+    end
+
+
+    #fills shared strings hash, contains each unique string
+    def fill_shared_strings()
+      if @shared_strings.nil?
+        string_hash = {}
+        string_index = 0
+        @num_strings = 0
+        #fill hash for shared strings
+        @worksheets.each do |sheet|
+          unless sheet.nil?
+            sheet.sheet_data.each do |row|
+              row.each do |cell|
+                unless cell.nil? || cell.value.nil?
+                  #if string not already seen, add it to hash
+                  if cell.datatype == 's'
+                    if string_hash[cell.value.to_s].nil?
+                      string_hash[string_index]=cell.value.to_s
+                      string_hash[cell.value.to_s]=string_index
+                      string_index += 1
+                    end
+                    @num_strings += 1
+                  end
+                end
+              end
+            end
+          end
+        end
+
+        if string_hash[0].nil?
+          @shared_strings = nil
+        else
+          @shared_strings = string_hash
+          @size = string_index
+        end
+      end
+    end
+
+    def validate_before_write
+      ## TODO CHECK IF STYLE IS OK if not raise
+    end
+  end
+end