csvp 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bd64e586de8ea735f1d88ad7c881bb4d6c637a4b
-  data.tar.gz: 3034ebd73917f9a3f21b6a33e1b911672bbdac55
+  metadata.gz: 8cb41b2450b7bea60d05bdf52f1a13637d737f17
+  data.tar.gz: 38ca9ef1a7d02348cb936b27f6f537a2bcd9d23a
 SHA512:
-  metadata.gz: 8877057a52f562f88d75b89b900dccb12e6a8dfaef081a09687930f47c2aad371a8a11dc3f96f0f6d22405c59f8e4f613e94af0ceb3b93c96e84dd634673897f
-  data.tar.gz: 62c7420dfb8f263ce38ac7085600a6a64c76f3863be6b2c2d16b607245f631b530f1c94a9b39a01e6cd9725419e3582626438fbbbb46c7b737bf1f1b60912160
+  metadata.gz: 415435b52116bbe0e727457fa0e70307ff88be4c7276ef28959fd57884aa403c2c4764a5276900e09653f1985af2aaa83a99bc9072cfc2cef98c105cdb16ca9a
+  data.tar.gz: e89e3355c1b3bce8438ff5e36fd197c34f06786874da122300b735ed81bc2e556484af5729f0cd47963e2c40fbcd76ef8172764f91af05ce0cb1abacefbdd3e0

data/lib/csvp.rb

@@ -17,29 +17,39 @@ def csvp(enum, separator: ',', quote: "")
 
   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 }
+  active_record_relation = proc { |obj| defined?(ActiveRecord::Relation) && ActiveRecord::Relation === obj }
+  not_enumerable = proc { |obj| !obj.is_a?(Enumerable) }
+
   # Vectorize scalar values into Enumerables to avoid multiple codepaths.
   enum = case enum
     when Struct then enum.to_h
-    when OpenStruct then enum.to_h
-    when -> (obj) { defined?(ActiveRecord::Base) && ActiveRecord::Base === obj } then enum.attributes
-    when -> (obj) { defined?(ActiveRecord::Relation) && ActiveRecord::Relation === obj } then enum.to_a
-    when -> (obj) { !obj.is_a?(Enumerable) } then [enum]
+    when open_struct then enum.to_h
+    when active_record_base then enum.attributes
+    when active_record_relation then enum.to_a
+    when not_enumerable then [enum]
     else enum
   end
 
-  columns = case enum.first
-    when Hash then enum.first.keys
-    when Struct then enum.first.members
-    when OpenStruct then enum.first.instance_variable_get("@table").keys
-    when -> (obj) { defined?(ActiveRecord::Base) && ActiveRecord::Base === obj } then enum.first.attributes.keys
+  # 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 OpenStruct then row.instance_variable_get("@table").values
-      when -> (obj) { defined?(ActiveRecord::Base) && ActiveRecord::Base === obj } then row.attributes.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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: csvp
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Martin Scavnicky
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-06-03 00:00:00.000000000 Z
+date: 2016-06-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -94,7 +94,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.1'
-description: Print enumerables as CSV effortlessly.
+description: Effortlessly print enumerables as CSV.
 email: martin.scavnicky@gmail.com
 executables: []
 extensions: []