minimalist_ods 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 463a57f017a1b0a8761d32fc7a52d9a191a03750725489eb8e12b9e0a0d4cdca
-  data.tar.gz: f3afeb53140f7e7ded4e96b786ec889adc55cbbd8644e20dbd00653d763cac62
+  metadata.gz: b24d9ff3c01e1adeac8c948f79e999e6af9a89c5b6976362e22011ad89d0c57a
+  data.tar.gz: cd9d65128da3b457baab953a3a845b6cc2f32c2cb9be5053790dbb94556d72a3
 SHA512:
-  metadata.gz: 6954e7a37852a4c3b72b3422667755a557818dcc1aa80f07b448e60fe82d0e6e2ae42c418ff28f5b0c1f59e29275928c6ef47660734fc5955557d930c4351d56
-  data.tar.gz: 45be33c3a46aa0b9f2ce7bfbba3dc237a2bb14fe8ae4cc3cdd125f68a5819c61c15f9817fd97ab352174d6937035db8c4bbbbff3be1975f4fec235400b7ab66e
+  metadata.gz: cb1fa28505949a88f5a9d5b91c38083af58ef89b9120469f1e1c987e710db037f5a1687d9022d116609aceee38409d9551da9ca39fd368fe1e91c87b1652775f
+  data.tar.gz: 4a7287bb2f07909192d4f2914cb1bc787aaf39c9446dca3a44ca8af3fec8a13a517b6401b76b672f59ec468928c3b7b6b454607565bfc24ace70b2b99b57ea8b

data/lib/minimalist_ods.rb

@@ -2,6 +2,7 @@
 
 # A Ruby Minimalist ODS
 require 'rubygems'
+require 'stringio'
 require 'zip'
 require 'date'
 
@@ -94,43 +95,9 @@ class MinimalistODS
     </table:table-cell>
   XML
 
-  TABLE_OPEN = 1
-  TABLE_CLOSED = 0
-
-  attr_reader :zip, :save_as, :creator, :buffer
-
-  def initialize(save_as, creator = 'minimalist-ods')
-    @save_as = save_as
-    @creator = creator
-    init_zip!
-    init_mimetype!
-    init_meta!
-    init_manifest!
-    init_content!
-  end
-
-  def init_zip!
-    @zip = Zip::File.open(save_as, Zip::File::CREATE)
-  end
-
-  def init_mimetype!
-    write_to_zip('mimetype', MIMETYPE)
-  end
-
-  def init_meta!
-    meta = META_TEMPLATE.gsub(':CREATOR', creator).gsub(':TIME', Time.now.strftime('%Y-%m-%dT%H:%M:%S'))
-    write_to_zip('meta.xml', meta)
-  end
-
-  def init_manifest!
-    write_to_zip('META-INF/manifest.xml', MANIFEST_TEMPLATE)
-  end
-
-  def init_content!
-    @buffer = @zip.get_output_stream('content.xml')
-    @status = TABLE_CLOSED
-    buffer.write(CONTENT_HEADER)
-  end
+  TABLE_OPEN = 0
+  TABLE_CLOSED = 1
+  FILE_CLOSED = 2
 
   class MinimalistOODSError < StandardError
   end
@@ -153,16 +120,40 @@ class MinimalistODS
     end
   end
 
+  class FileClosed < MinimalistOODSError
+    def initialize
+      super('The file is now closed')
+    end
+  end
+
+  class FileNotClosed < MinimalistOODSError
+    def initialize
+      super('The file is still opened')
+    end
+  end
+
   class InvalidParameter < MinimalistOODSError
   end
 
+  attr_reader :zip_buffer, :content_buffer, :save_as, :save_to_disk, :creator
+
+  def initialize(save_as = nil, creator = 'minimalist-ods')
+    @save_as = save_as
+    @creator = creator
+    @save_to_disk = !(save_as.instance_of? StringIO)
+
+    init_content
+  end
+
   def open_table(table_name, cols_number)
+    raise FileClosed if @status == FILE_CLOSED
     raise TableAlreadyOpened if @status == TABLE_OPEN
     raise InvalidParameter, "Got invalid value `#{cols_number}' for table size" if cols_number.nil? || !cols_number.is_a?(Integer) || !cols_number.positive?
+
     @cols_number = cols_number
 
     table_header = TABLE_TEMPLATE.gsub(':NAME', table_name).gsub(':COL_NUMBER', cols_number.to_s)
-    buffer.write(table_header)
+    content_buffer.write(table_header)
     @status = TABLE_OPEN
   end
 
@@ -170,29 +161,62 @@ class MinimalistODS
     raise InvalidRowLength.new(@cols_number, row.size) if row.size != @cols_number
 
     cells = row.map { |cell| cell_to_xml(cell) }.join
-    buffer.write(ROW_TEMPLATE.gsub(':CELLS', cells))
+    content_buffer.write(ROW_TEMPLATE.gsub(':CELLS', cells))
   end
 
-
   def close_table
     raise TableNotOpened if @status == TABLE_CLOSED
 
-    buffer.write('</table:table>')
+    content_buffer.write('</table:table>')
     @status = TABLE_CLOSED
   end
 
   def close_file
-    buffer.write(CONTENT_FOOTER)
-    buffer.close
-    zip.close
+    raise TableAlreadyOpened if @status == TABLE_OPEN
+
+    content_buffer.write(CONTENT_FOOTER)
+    @status = FILE_CLOSED
+    create_zip
+    write_file!
   end
 
   private
 
-  def write_to_zip(file_name, content)
-    stream = zip.get_output_stream(file_name)
-    stream.write(content)
-    stream.close
+  def init_content
+    @content_buffer = StringIO.new
+    content_buffer.write(CONTENT_HEADER)
+    @status = TABLE_CLOSED
+  end
+
+  def format_meta
+    META_TEMPLATE.gsub(':CREATOR', creator).gsub(':TIME', Time.now.strftime('%Y-%m-%dT%H:%M:%S'))
+  end
+
+  def create_zip
+    @zip_buffer = Zip::OutputStream.write_buffer do |zip_handler|
+      zip_handler.put_next_entry('mimetype')
+      zip_handler.write(MIMETYPE)
+
+      zip_handler.put_next_entry('meta.xml')
+      zip_handler.write(format_meta)
+
+      zip_handler.put_next_entry('META-INF/manifest.xml')
+      zip_handler.write(MANIFEST_TEMPLATE)
+
+      zip_handler.put_next_entry('content.xml')
+      zip_handler.write(content_buffer.string)
+    end
+  end
+
+  def write_file!
+    if save_to_disk
+      File.open(save_as, 'wb') do |file|
+        file.write(zip_buffer.string)
+      end
+    else
+      save_as.write(zip_buffer.string)
+      save_as.rewind
+    end
   end
 
   def cell_to_xml(cell)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: minimalist_ods
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Gilberto Vargas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-07-01 00:00:00.000000000 Z
+date: 2024-08-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rubyzip