table_view 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 476e9d19235a9aa447fd2ef34c4674ea8973ab28f0a060f234ad995f9cd44b78
-  data.tar.gz: ba50dcdee078e4430a3bf79054430203c82b829754a94fe1c099a56d5b59fce5
+  metadata.gz: 3300ffe7a8230c8e8829c3d89e29dfc1a20b3a1c168a0bf6286386ddcd910dcc
+  data.tar.gz: 04a7f4fa63083e8af2069f580802c777b3e2584bb78cb55995832264124db911
 SHA512:
-  metadata.gz: 611592ed47d624624ee67598836a7a89dca3de47e5e7abea507c5c192bb2e57c44730658fdde80b3f96e6928c9484f781fda20a10222816382b7ddf141484d18
-  data.tar.gz: fe40a86cc404326c3925ea6b3fcedca89dd6ce7686cdfa15909fbaa35bc624a777499f09169abf7aa72f8cd43225eefecec2fdbe51c7047fee817b74776ebfd0
+  metadata.gz: efe9227370ec63a54e3e1d9d426c85c94c0a78d29ec546ec6ce98237e34974cea33b9483afabbe0f5847371027b28b3b5e0b380ba15aafb44b09b8eeacc1dffc
+  data.tar.gz: 17239ff091a4d52bf19f3746a835695f9f2ee7d6d7d9aa2093a4c61d3a49e2d1b4d1d96b67cd7746755e7d4dca9f7e7937ba51c3974b1a2856a65a7a342f7cab

data/.gitignore

@@ -7,6 +7,7 @@
 /spec/reports/
 /tmp/
 
+/spec/fixtures/**/*.got.*
 Gemfile.lock
 
 # rspec failure tracking

data/lib/table_view/base.rb

@@ -1,25 +1,11 @@
 # frozen_string_literal: true
 
-require "csv"
+require_relative "formats/array"
+require_relative "formats/csv"
 
 module TableView
   class Base
-    def as_array
-      array = []
-      array << headers
-      each_row do |row|
-        array << row
-      end
-      array
-    end
-
-    def write_csv(io)
-      csv = CSV.new(io)
-      csv << headers
-      each_row do |row|
-        csv << row
-      end
-      csv.close
-    end
+    include TableView::Formats::Array
+    include TableView::Formats::CSV
   end
 end

data/lib/table_view/formats/array.rb

@@ -0,0 +1,14 @@
+module TableView
+  module Formats
+    module Array
+      def as_array
+        array = []
+        array << headers
+        each_row do |row|
+          array << row
+        end
+        array
+      end
+    end
+  end
+end

data/lib/table_view/formats/csv.rb

@@ -0,0 +1,35 @@
+require "csv"
+
+module TableView
+  module Formats
+    module CSV
+      def as_csv
+        StringIO.open do |io|
+          write_csv_to_io(io)
+          io.string
+        end
+      end
+
+      def write_csv(filepath_or_io)
+        if filepath_or_io.is_a?(String)
+          File.open(filepath_or_io, "wb") do |f|
+            write_csv_to_io(f)
+          end
+        else
+          write_csv_to_io(filepath_or_io)
+        end
+      end
+
+      protected
+
+      def write_csv_to_io(io)
+        csv = ::CSV.new(io)
+        csv << headers
+        each_row do |row|
+          csv << row
+        end
+        csv.close
+      end
+    end
+  end
+end

data/lib/table_view/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module TableView
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: table_view
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Tim Diggins
@@ -87,6 +87,8 @@ files:
 - bin/setup
 - lib/table_view.rb
 - lib/table_view/base.rb
+- lib/table_view/formats/array.rb
+- lib/table_view/formats/csv.rb
 - lib/table_view/version.rb
 - table_view.gemspec
 homepage: https://github.com/timdiggins/table_view