pluckit 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2f60041a9e7e6bf150d5f7a8d9cdaa72b795650b
-  data.tar.gz: fd72fe2ed30100f8e5a07d1fe531d41153b91bb9
+  metadata.gz: 42c6c3073a9907cc8cfc8dfe0fe43e4fe7a60fb5
+  data.tar.gz: 2c978a6bbd3d96e3c040af65cf85e4266f84041c
 SHA512:
-  metadata.gz: d61d2afa5f0b1308cf2c47c88a317b16174f81cb54cdd87e2104290c85789ad062c0447a77f1c2220389e82aa1df026396534caa3879cf702f5dcdce3a8d42f9
-  data.tar.gz: 78982ef586f5c3b8b3b202715c25a73d09d8be9c6df6fe9ee47cc71778941ab49d18ee7c0922453c036f492995e0ae3b5909bff4bd7d7b1c09beb3843d6e6c31
+  metadata.gz: 5b18dcf896e35913022655730e469702cacad1c64e2aaffed5f1312edc223a1c735b027cf49ac7a7d19bed7737ca88fbf18120e61bb6bfa27f979051c499c66d
+  data.tar.gz: 4b31395eaaec78888732784450b796885353e3b0b200cac26c3fc5ce510d3a90ab6bedaeb418a67f3699e390065ec94219f97c0c5609bd2a9de5cfc1739f9a70

data/lib/pluckit.rb

@@ -1,19 +1,28 @@
 module PluckIt
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 
   class << self
 
-    def pluck v, handle
+    def pluck v, *handles
+      if handles.count > 1
+        handles.map {|h| pluck_single v, h }
+      else
+        pluck_single v, handles.first
+      end
+    end
+
+
+    private
+
+    def pluck_single v, handle
       if v.is_a? Hash
         v[handle]
       elsif ([Symbol, String].include? handle.class) and v.respond_to? handle
         v.send handle
-      elsif v.is_a? Array
-        if handle.is_a? Regexp
-          v.grep handle
-        else
-          v[handle]
-        end
+      elsif handle.is_a? Regexp and v.respond_to? :grep
+        v.grep handle
+      elsif v.respond_to? :[]
+        v[handle]
       else
         raise ArgumentError.new "invalid handle: #{handle}, for value #{v}"
       end
@@ -24,27 +33,27 @@ end
 
 
 class Array
-  def pluck(handle)
+  def pluck *handles
     each_with_object(clone.clear) do |val, res|
-      res << PluckIt.pluck(val, handle)
+      res << PluckIt.pluck(val, *handles)
     end
   end
 end
 
 
 class Hash
-  def pluck(handle)
+  def pluck *handles
     each_with_object(clone.clear) do |(key, val), res|
-      res[key] = PluckIt.pluck(val, handle)
+      res[key] = PluckIt.pluck(val, *handles)
     end
   end
 end
 
 
 class Set
-  def pluck(handle)
+  def pluck *handles
     each_with_object(clone.clear) do |val, res|
-      res.add PluckIt.pluck(val, handle)
+      res.add PluckIt.pluck(val, *handles)
     end
   end
 end

data/test/test_array.rb

@@ -21,6 +21,15 @@ class PluckItArrayTest < Minitest::Test
       [ 3, 6, 9 ],
       data.pluck(:last)
     )
+
+    assert_equal(
+      [
+        [ 1, 2 ],
+        [ 4, 5 ],
+        [ 7, 8 ],
+      ],
+      data.pluck(0, 1)
+    )
   end
 
 

data/test/test_hash.rb

@@ -31,6 +31,19 @@ class PluckItHashTest < Minitest::Test
         c: { x: 3, y: 6 },
       }.pluck(:x)
     )
+
+    assert_equal(
+      {
+        a: [ 1, 2 ],
+        b: [ 2, 4 ],
+        c: [ 3, 6 ],
+      },
+      {
+        a: { x: 1, y: 2 },
+        b: { x: 2, y: 4 },
+        c: { x: 3, y: 6 },
+      }.pluck(:x, :y)
+    )
   end
 
 

data/test/test_pluck.rb

@@ -40,6 +40,11 @@ class PluckItTest < Minitest::Test
         /a/
       )
     )
+
+    assert_equal(
+      [ 1, 2 ],
+      PluckIt.pluck(data, 0, 1)
+    )
   end
 
 
@@ -62,11 +67,17 @@ class PluckItTest < Minitest::Test
     assert_nil(
       PluckIt.pluck(data, :count)
     )
+
+    assert_equal(
+      [ 1, 2 ],
+      PluckIt.pluck(data, :a, :b)
+    )
   end
 
 
   class ABC
     def foo() 123 end
+    def bar() 456 end
   end
 
   def test_obj
@@ -79,6 +90,11 @@ class PluckItTest < Minitest::Test
       self.class.const_get(:ABC),
       PluckIt.pluck(ABC.new, :class)
     )
+
+    assert_equal(
+      [ 123, 456 ],
+      PluckIt.pluck(ABC.new, :foo, :bar)
+    )
   end
 
 
@@ -94,6 +110,11 @@ class PluckItTest < Minitest::Test
       3,
       PluckIt.pluck(data, :count)
     )
+
+    assert_equal(
+      [ 1, 3 ],
+      PluckIt.pluck(data, :first, :count)
+    )
   end
 
 

data/test/test_set.rb

@@ -21,6 +21,15 @@ class PluckItSetTest < Minitest::Test
       Set.new([ 3, 6, 9 ]),
       data.pluck(:last)
     )
+
+    assert_equal(
+      Set.new([
+        [ 1, 3 ],
+        [ 4, 6 ],
+        [ 7, 9 ],
+      ]),
+      data.pluck(:first, :max)
+    )
   end
 
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pluckit
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Daniel Pepper
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-10-08 00:00:00.000000000 Z
+date: 2017-11-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: minitest
@@ -69,7 +69,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.13
+rubygems_version: 2.6.11
 signing_key: 
 specification_version: 4
 summary: PluckIt