add object support to pluck

audreyt · Jun 12, 2012 · b797bd3 · b797bd3
1 parent 7347cb0
commit b797bd3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/prelude.ls b/prelude.ls
@@ -50,7 +50,9 @@ exports.find = find = (f, xs) -->
   void
 
 exports.pluck = pluck = (prop, xs) -->
-  [x[prop] for x in xs when x[prop]?]
+  if typeof! xs is \Object
+  then {[key, x[prop]] for key, x of xs when x[prop]?}
+  else [x[prop] for x in xs when x[prop]?]
 
 exports.head = head = (xs) -> 
   return void if not xs.length

diff --git a/test.ls b/test.ls
@@ -55,7 +55,9 @@ eq 2, find (==2), {a:1, b:2}
 
 # pluck
 eq 'one,two' "#{ pluck \num [num: \one; num: \two] }"
-
+plucked = pluck \num {a: {num: 1}, b: {num: 2}}
+eq 1 plucked.a
+eq 2 plucked.b
 
 list = [1 2 3 4 5]
 string = 'abcde'