Add support for the take().

Author: javadev

lodash-plugin/src/main/java/com/github/underscore/lodash/$.java

@@ -96,6 +96,14 @@ public Chain<List<T>> remove(final Predicate<T> pred) {
             return new Chain<List<T>>($.remove(value(), pred));
         }
 
+        public Chain<List<T>> take() {
+            return new Chain<List<T>>($.take(value()));
+        }
+
+        public Chain<List<T>> take(final Integer n) {
+            return new Chain<List<T>>($.take(value(), n));
+        }
+
         public Chain<List<T>> xor(final List<T> list) {
             return new Chain<List<T>>($.xor(value(), list));
         }
@@ -260,6 +268,22 @@ public List<T> remove(final Predicate<T> pred) {
         return remove((List<T>) getIterable(), pred);
     }
 
+    public static <T> List<T> take(final Iterable<T> iterable) {
+        return first(newArrayList(iterable), 1);
+    }
+
+    public List<T> take() {
+        return take(getIterable());
+    }
+
+    public static <T> List<T> take(final Iterable<T> iterable, final Integer n) {
+        return first(newArrayList(iterable), n);
+    }
+
+    public List<T> take(final Integer n) {
+        return take(getIterable(), n);
+    }
+
     public static <T> List<T> xor(final List<T> ... lists) {
         int index = -1;
         int length = lists.length;

lodash-plugin/src/test/java/com/github/underscore/lodash/LodashTest.java

@@ -287,6 +287,31 @@ public Boolean apply(final Integer n) {
         assertEquals("[2, 4]", evens.toString());
     }
 
+/*
+_.take([1, 2, 3]);
+// → [1]
+
+_.take([1, 2, 3], 2);
+// → [1, 2]
+
+_.take([1, 2, 3], 5);
+// → [1, 2, 3]
+
+_.take([1, 2, 3], 0);
+// → []
+*/
+    @Test
+    public void take() {
+        assertEquals("[1]", $.take(asList(1, 2, 3)).toString());
+        assertEquals("[1]", new $(asList(1, 2, 3)).take().toString());
+        assertEquals("[1]", $.chain(asList(1, 2, 3)).take().value().toString());
+        assertEquals("[1, 2]", $.take(asList(1, 2, 3), 2).toString());
+        assertEquals("[1, 2]", new $(asList(1, 2, 3)).take(2).toString());
+        assertEquals("[1, 2]", $.chain(asList(1, 2, 3)).take(2).value().toString());
+        assertEquals("[1, 2, 3]", $.take(asList(1, 2, 3), 5).toString());
+        assertEquals("[]", $.take(asList(1, 2, 3), 0).toString());
+    }
+
 /*
 _.xor([1, 2], [4, 2]);
 // → [1, 4]

src/main/java/com/github/underscore/$.java

@@ -585,7 +585,7 @@ public static <E> E first(final E[] array) {
     }
 
     public static <E> List<E> first(final List<E> list, final int n) {
-        return list.subList(0, n);
+        return list.subList(0, Math.min(n, list.size()));
     }
 
     public T first() {