spreet 0.0.3 → 0.0.4

data/lib/spreet/document.rb

@@ -0,0 +1,54 @@
+module Spreet
+
+  class Document
+    attr_reader :sheets
+    @@handlers = {}
+    @@associations = {}
+    
+    def initialize(option={})
+      @sheets = Sheets.new(self)
+    end
+
+    def write(file, options={})
+      handler = self.class.extract_handler(file, options.delete(:format))
+      handler.write(self, file, options)
+    end
+
+    class << self
+    
+      def register_handler(klass, name, options={})
+        if klass.respond_to?(:read) or klass.respond_to?(:write)
+          if name.is_a?(Symbol)
+            @@handlers[name] = klass # options.merge(:class=>klass)
+          elsif
+            raise ArgumentError.new("Name is invalid. Symbol expected, #{name.class.name} got.")
+          end
+        else
+          raise ArgumentError.new("Handler do not support :read or :write method.")
+        end
+      end
+      
+      def read(file, options={})
+        handler = extract_handler(file, options.delete(:format))
+        return handler.read(file, options)
+      end
+
+      def extract_handler(file, handler_name=nil)
+        file_path = Pathname.new(file)
+        extension = file_path.extname.to_s[1..-1]
+        if !handler_name and extension.size > 0
+          handler_name = extension.to_sym
+        end
+        if @@handlers[handler_name]
+          return @@handlers[handler_name]
+        else
+          raise ArgumentError.new("No corresponding handler (#{handler_name.inspect}). Available: #{@@handlers.keys.collect{|k| k.inspect}.join(', ')}.")
+        end
+      end
+
+    end
+    
+  end  
+
+
+end

data/lib/spreet/handlers.rb

@@ -1,25 +1,9 @@
 module Spreet
-
-  # Default handler
-  class Handler
-
-    def self.read(file, options={})
-      raise NotImplementedError.new
-    end
-
-    def self.write(spreet, file, options={})
-      raise NotImplementedError.new
-    end
-    
+  module Handlers
+    autoload :Base,         'spreet/handlers/base'
+    autoload :CSV,          'spreet/handlers/csv'
+    autoload :ExcelCSV,     'spreet/handlers/excel_csv'
+    autoload :OpenDocument, 'spreet/handlers/open_document'
   end
-
 end
 
-require 'spreet/handlers/csv'
-require 'spreet/handlers/open_document'
-
-Spreet::Document.register_handler Spreet::Handlers::CSV, :csv
-Spreet::Document.register_handler Spreet::Handlers::ExcelCSV, :xcsv
-# Spreet::Document.register_handler Spreet::Handlers::HTML, :html
-Spreet::Document.register_handler Spreet::Handlers::OpenDocument, :ods
-# Spreet::Document.register_handler Spreet::Handlers::PDF, :pdf

data/lib/spreet/handlers/base.rb

@@ -0,0 +1,19 @@
+module Spreet
+  module Handlers
+
+    # Default handler
+    class Base
+      
+      def self.read(file, options={})
+        raise NotImplementedError.new
+      end
+      
+      def self.write(spreet, file, options={})
+        raise NotImplementedError.new
+      end
+      
+    end
+
+    
+  end
+end

data/lib/spreet/handlers/csv.rb

@@ -1,21 +1,15 @@
-# encoding: utf-8
-require 'fastercsv'
 require 'csv'
-require 'iconv'
 
 module Spreet
-  # Universal CSV support
-  CSV = (::CSV.const_defined?(:Reader) ? ::FasterCSV : ::CSV).freeze
-
   module Handlers
 
-    class CSV < Spreet::Handler
+    class CSV < Spreet::Handlers::Base
 
       # Read a CSV file and create its Spreet document
       def self.read(file, options={})
         spreet = Spreet::Document.new
         sheet = spreet.sheets.add
-        Spreet::CSV.foreach(file) do |row|
+        ::CSV.foreach(file) do |row|
           sheet.row *row
         end
         return spreet
@@ -25,7 +19,7 @@ module Spreet
       # Write a Spreet to a CSV file
       def self.write(spreet, file, options={})
         sheet = spreet.sheets[options[:sheet]||0]
-        Spreet::CSV.open(file, "wb") do |csv|
+        ::CSV.open(file, "wb") do |csv|
           sheet.each_row do |row|
             csv << row.collect{|c| c.text}
           end
@@ -34,35 +28,6 @@ module Spreet
 
     end
 
-    class ExcelCSV < Spreet::Handler
-
-      # Read a CSV file and create its Spreet document
-      def self.read(file, options={})
-        spreet = Spreet::Document.new
-        sheet = spreet.sheets.add
-        options = {:col_sep=>';'}.merge(options)
-        ic = Iconv.new('utf-8', 'cp1252')
-        Spreet::CSV.foreach(file, options) do |row|
-          sheet.row *(row.collect{|v| ic.iconv(v.to_s)})
-        end
-        return spreet
-      end
-
-
-      # Write a Spreet to a CSV file
-      def self.write(spreet, file, options={})
-        sheet = spreet.sheets[options[:sheet]||0]
-        options = {:col_sep=>';'}.merge(options)
-        ic = Iconv.new('cp1252', 'utf-8')
-        Spreet::CSV.open(file, "wb", options) do |csv|
-          sheet.each_row do |row|
-            csv << row.collect{|c| ic.iconv(c.text)}
-          end
-        end
-      end
-
-    end
-
   end
 end
 

data/lib/spreet/handlers/excel_csv.rb

@@ -0,0 +1,36 @@
+require 'csv'
+
+module Spreet
+  module Handlers
+
+    class ExcelCSV < Spreet::Handlers::Base
+
+      # Read a CSV file and create its Spreet document
+      def self.read(file, options={})
+        spreet = Spreet::Document.new
+        sheet = spreet.sheets.add
+        options = {:col_sep=>';', :encoding => "CP1252"}.merge(options)
+        ::CSV.foreach(file, options) do |row|
+          sheet.row *(row.map{|v| v.to_s.encode('utf-8')}) # collect{|v| v.to_s.encode('cp1252')}
+        end
+        return spreet
+      end
+
+
+      # Write a Spreet to a CSV file
+      def self.write(spreet, file, options={})
+        sheet = spreet.sheets[options[:sheet]||0]
+        options = {:col_sep=>';', :encoding => "CP1252"}.merge(options)
+        ::CSV.open(file, "wb", options) do |csv|
+          sheet.each_row do |row|
+            csv << row.map(&:text)
+          end
+        end
+      end
+
+    end
+
+  end
+end
+
+

data/lib/spreet/handlers/open_document.rb

@@ -1,10 +1,13 @@
 # encoding: utf-8
-require 'zip/zip'
+require 'zip'
 require 'libxml'
+require 'money'
+require 'time'
+require 'duration'
 
 module Spreet
   module Handlers
-    class OpenDocument < Spreet::Handler
+    class OpenDocument < Spreet::Handlers::Base
       DATE_REGEXP = /\%./
       DATE_ELEMENTS = {
         "m" => "<number:month number:style=\"long\"/>",
@@ -62,7 +65,7 @@ module Spreet
 
       def self.read(file, options={})
         spreet = nil
-        Zip::ZipFile.open(file) do |zile|
+        Zip::File.open(file) do |zile|
           # Check mime_type
           entry = zile.find_entry "mimetype"
           if entry.nil?
@@ -135,7 +138,7 @@ module Spreet
                         elsif value_type == :currency
                           value = cell.attributes.get_attribute_ns(XMLNS_OFFICE, "value").value
                           currency = cell.attributes.get_attribute_ns(XMLNS_OFFICE, "currency").value
-                          sheet[x,y] = Money.new(value, currency)
+                          sheet[x,y] = Money.new(value.to_f, currency)
                         elsif value_type == :date
                           value = cell.attributes.get_attribute_ns(XMLNS_OFFICE, "date-value").value
                           if value.match(/\d{1,8}-\d{1,2}-\d{1,2}/)
@@ -202,9 +205,9 @@ module Spreet
         xml_escape << ".force_encoding('US-ASCII')" if xml_escape.respond_to?(:force_encoding)
         mime_type = MIME_ODS
         # name = #{table.model.name}.model_name.human.gsub(/[^a-z0-9]/i,'_')
-        Zip::ZipOutputStream.open(file) do |zile|
+        Zip::OutputStream.open(file) do |zile|
           # MimeType in first place
-          zile.put_next_entry('mimetype', nil, nil, Zip::ZipEntry::STORED)
+          zile.put_next_entry('mimetype', nil, nil, Zip::Entry::STORED)
           zile << mime_type
           
           # Manifest

data/lib/spreet/sheet.rb

@@ -0,0 +1,125 @@
+module Spreet
+
+  class Sheet
+    attr_reader :document, :name, :columns
+    attr_accessor :current_row
+
+    def initialize(document, name=nil)
+      @document = document
+      self.name = name
+      raise ArgumentError.new("Must be a Document") unless document.is_a? Document
+      @current_row = 0
+      @cells = {} # BigArray::Cells.new
+      @bound = compute_bound
+    end
+
+    def name=(value)
+      unless value
+        value = (@document.sheets.count > 0 ? @document.sheets[-1].name.succ : "Sheet 1")
+      end 
+      raise ArgumentError.new("Name of sheet must be given") if value.to_s.strip.size.zero?
+      if @document.sheets[value]
+        raise ArgumentError.new("Name of sheet must be unique")
+      end
+      @name = value
+    end
+
+    def next_row(increment = 1)
+      @current_row += increment
+    end
+    
+    def previous_row(increment = 1)
+      @current_row -= increment
+    end
+
+    def [](*args)
+      coord = Coordinates.new(*args)
+      @cells[coord.to_i] ||= Cell.new(self, coord)
+      return @cells[coord.to_i]
+    end
+
+    def []=(*args)
+      value = args.delete_at(-1)
+      cell = self[*args]
+      cell.value = value
+      @updated = true
+    end
+
+    def row(*args)
+      options = {}
+      options = args.delete_at(-1) if args[-1].is_a? Hash
+      row = options[:row] || @current_row
+      args.each_index do |index|
+        self[index, row] = args[index]
+      end
+      next_row
+    end
+
+    def rows(index)
+      row = []
+      for i in 0..bound.x
+        row[i] = self[i, index]
+      end
+      return row
+    end
+
+    def each_row(&block)
+      for j in 0..bound.y
+        yield rows(j)
+      end
+    end
+
+    # Find or build cell
+    def cell(*args)
+      return c
+    end
+    
+    def bound
+      if @updated
+        compute_bound
+      else
+        @bound
+      end
+    end
+
+    def remove!(coordinates)
+      raise ArgumentError.new("Must be a Coordinates") unless document.is_a?(Coordinates)
+      @cells.delete(coordinates.to_i)
+      @updated = true
+    end
+
+    # Moves the sheet to an other position in the list of sheets
+    def move_to(position)
+      @document.sheets.move_at(self, position)
+    end
+
+    # Moves the sheet higher in the list of sheets
+    def move_higher(increment=1)
+      @document.sheets.move(self, increment)
+    end
+
+    # Moves the sheet lower in the list of sheets
+    def move_lower(increment=1)
+      @document.sheets.move(self, -increment)
+    end
+
+    private
+
+    def compute_bound
+      bound = Coordinates.new(0,0)
+      for index, cell in @cells
+      # for cell in @cells.compact
+        unless cell.empty?
+          bound.x = cell.coordinates.x if cell.coordinates.x > bound.x
+          bound.y = cell.coordinates.y if cell.coordinates.y > bound.y
+        end
+      end
+      @updated = false
+      @bound = bound
+      return @bound
+    end
+
+  end
+  
+
+end

data/lib/spreet/sheets.rb

@@ -0,0 +1,63 @@
+module Spreet
+
+  class Sheets
+
+    def initialize(document)
+      raise ArgumentError.new("Must be a Document") unless document.is_a?(Document)
+      @document = document
+      @array = []
+    end
+
+    def count
+      @array.size
+    end
+
+    def index(name_or_sheet)
+      if name_or_sheet.is_a? String
+        @array.each_index do |i|
+          return i if @array[i].name == name_or_sheet
+        end
+      elsif name_or_sheet.is_a? Integer
+        return (@array[name_or_sheet].nil? ? nil : name_or_sheet)
+      else
+        return @array.index(name_or_sheet)
+      end
+    end
+
+    def add(name=nil, position=-1)
+      sheet = Sheet.new(@document, name)
+      @array.insert(position, sheet)
+      return sheet
+    end
+
+    def [](sheet)
+      sheet = index(sheet)
+      return (sheet.is_a?(Integer) ? @array[sheet] : nil)
+    end
+
+    def remove(sheet)
+      @array.delete_at(index(sheet))
+    end
+
+    def move(sheet, shift=0)
+      position = index(sheet) + shift
+      position = 0 if position < 0
+      position = self.count-1 if position >= self.count
+      move_at(sheet, position)
+    end
+
+    def move_at(sheet, position=-1)
+      if i = index(sheet)
+        @array.insert(position, @array.delete_at(i))
+      end
+    end
+
+    def each(&block)
+      for item in @array
+        yield item
+      end
+    end
+
+  end
+  
+end