csvp 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8cb41b2450b7bea60d05bdf52f1a13637d737f17
-  data.tar.gz: 38ca9ef1a7d02348cb936b27f6f537a2bcd9d23a
+  metadata.gz: a766e9348e13295b12e40d73354c12515ff00dd3
+  data.tar.gz: 96a31740723ee4b08f1f3e07172b516ea57e8606
 SHA512:
-  metadata.gz: 415435b52116bbe0e727457fa0e70307ff88be4c7276ef28959fd57884aa403c2c4764a5276900e09653f1985af2aaa83a99bc9072cfc2cef98c105cdb16ca9a
-  data.tar.gz: e89e3355c1b3bce8438ff5e36fd197c34f06786874da122300b735ed81bc2e556484af5729f0cd47963e2c40fbcd76ef8172764f91af05ce0cb1abacefbdd3e0
+  metadata.gz: 6c615a14d1214a5463e578c46e8a1dda086572f8f6269989c93fcb70915a7881c7d0ceb4fd539a94e083bf1abc11328f857c499584c22864d34e4a43d05ac977
+  data.tar.gz: 33e6899cd903fc0142ef00717c3d0d52a118411d665805c557756acd0f1d02a74d8c187e1492dd32ecf1488c98abd6efcdeafdf64336d2fb02d975877043dd36

data/lib/csvp.rb

@@ -1,8 +1,9 @@
+require 'csv'
+
 # Prints Enumerable in CSV format on the best effort basis.
 #
-# @param enum [Enumerable]
-# @param separator [String]
-# @param quote [String]
+# @param enum [Enumerable] Enumberable to print.
+# @param options [Hash] Options described in Ruby CSV module.
 #
 # Example:
 #   ?> csvp [{a: 1, b: 2}, {a: 3, b: 4} ]
@@ -11,12 +12,9 @@
 #   3,4
 #
 # This implementation attempts to follow https://tools.ietf.org/html/rfc4180
-
-def csvp(enum, separator: ',', quote: "")
+def csvp(enum, options = {})
   return if enum.nil?
 
-  add_quotes = proc { |str| "#{quote}#{str.gsub(quote, quote*2)}#{quote}" }
-
   # Case rules encapsulated as procs.
   open_struct = proc { |obj| defined?(OpenStruct) && OpenStruct === obj }
   active_record_base = proc { |obj| defined?(ActiveRecord::Base) && ActiveRecord::Base === obj }
@@ -35,35 +33,34 @@ def csvp(enum, separator: ',', quote: "")
 
   # Try to determine the header row.
   first = enum.first
-
   columns = case first
     when Hash then first.keys
     when Struct then first.members
     when open_struct then first.instance_variable_get("@table").keys
     when active_record_base then first.attributes.keys
   end
-  puts columns.map(&:to_s).map(&add_quotes).join(separator) if columns
 
-  # Print out the rest of the CSV.
-  enum.each do |row|
-    values = case row
-      when Hash then row.values
-      when open_struct then row.instance_variable_get("@table").values
-      when active_record_base then row.attributes.values
-      when Enumerable then row
-      else [row]
-    end
+  CSV($stdout, options) do |csv|
+    csv << columns if columns
 
-    puts values.map(&:to_s).map(&add_quotes).join(separator)
+    enum.each do |row|
+      csv << case row
+        when Hash then row.values
+        when open_struct then row.instance_variable_get("@table").values
+        when active_record_base then row.attributes.values
+        when Enumerable then row
+        else [row]
+      end
+    end
   end
 
   nil
 end
 
 def tsvp(enum)
-  csvp enum, separator: "\t"
+  csvp enum, col_sep: "\t"
 end
 
 def qcsvp(enum)
-  csvp enum, quote: "\""
+  csvp enum, quote_char: "\"", force_quotes: true
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: csvp
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Martin Scavnicky
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-06-05 00:00:00.000000000 Z
+date: 2016-06-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec