Collection further implemented

Stdlib/Collection/Collection.php

@@ -15,6 +15,8 @@
  */
 namespace phpOMS\Stdlib\Collection;
 
+use phpOMS\Utils\ArrayUtils;
+
 /**
  * Multimap utils.
  *
@@ -28,7 +30,6 @@ namespace phpOMS\Stdlib\Collection;
  */
 class Collection implements \Countable, \ArrayAccess, \Iterator, \JsonSerializable
 {
-    private $count = 0;
     private $collection = [];
 
     public function __construct(array $data)
@@ -81,15 +82,18 @@ class Collection implements \Countable, \ArrayAccess, \Iterator, \JsonSerializab
         return count($this->collection);
     }
 
-    public function chunk(int size)
+    public function chunk(int $size) : Collection
     {
-        return new self(array_chunk($this->collection, size));
+        return new self(array_chunk($this->collection, $size));
     }
 
-    public function collapse()
+    public function collapse() : Collection
     {
         $return = [];
-        return new self(array_walk_recursive($this->collection, function($a) use (&$return) { $return[] = $a; });)
+
+        return new self(array_walk_recursive($this->collection, function ($a) use (&$return) {
+            $return[] = $a;
+        }));
     }
 
     public function combine(array $values) : Collection
@@ -117,7 +121,7 @@ class Collection implements \Countable, \ArrayAccess, \Iterator, \JsonSerializab
         foreach ($this->collection as $key => $value) {
             if (is_string($find) && ((is_string($value) && $find === $value) || (is_array($value) && in_array($find, $value)))) {
                 return true;
-            } elseif($find instanceof \Collection) {
+            } elseif ($find instanceof Collection) {
                 $result = $find($value, $key);
 
                 if ($result) {
@@ -159,11 +163,15 @@ class Collection implements \Countable, \ArrayAccess, \Iterator, \JsonSerializab
         return $diff;
     }
 
-    public function every(int $n)
+    public function every(int $n) : Collection
     {
+        $values = array_values($this->collection);
+        $keys   = array_keys($this->collection);
+        $count  = count($values);
+
         $new = [];
-        for ($i = 0; $i < $this->count(); $i += $n) {
+        for ($i = 0; $i < $count; $i += $n) {
-            $new[] = $this->get($i);
+            $new[$keys[$i]] = $values[$i];
         }
 
         return new self($new);
@@ -182,7 +190,7 @@ class Collection implements \Countable, \ArrayAccess, \Iterator, \JsonSerializab
         return null;
     }
 
-    public function except($filter)
+    public function except($filter) : Collection
     {
         if (!is_array($filter)) {
             $filter = [$filter];
@@ -199,116 +207,276 @@ class Collection implements \Countable, \ArrayAccess, \Iterator, \JsonSerializab
         return new self($new);
     }
 
-    public function filter()
+    public function filter($filter) : Collection
     {
+        $new = [];
+        foreach ($this->collection as $key => $value) {
+            if ($filter($key, $value)) {
+                $new[$key] = $value;
+            }
         }
 
-    public function first()
+        return new self($new);
-    {
-        return reset($this->collection);
     }
 
-    public function last()
+    public function first(\Closure $filter = null)
     {
-        $end = end($this->collection);
+        foreach ($this->collection as $key => $value) {
-        reset($this->collection);
+            if (!isset($filter) || $filter($key, $value)) {
-
+                return $value;
-        return $end;
+            }
         }
 
-    public function sort()
+        return null;
-    {
     }
 
-    public function reverse()
+    public function flatten() : Collection
     {
+        return new self(ArrayUtils::arrayFlatten($this->collection));
     }
 
-    public function map()
+    public function flip() : Collection
     {
+        return new self(array_flip($this->collection));
     }
 
-    public function flatten()
+    public function groupBy($filter) : Collection
     {
+        $new = [];
+        foreach ($this->collection as $key => $value) {
+            if (is_string($filter)) {
+                $group = $filter;
+            } elseif ($filter instanceof \Closure) {
+                $group = $filter($value, $key);
+            } else {
+                throw new \Exception();
             }
 
-    public function flip()
+            $new[$value[$group]][] = $value;
-    {
         }
 
-    public function remove()
+        return new self($new);
+    }
+
+    public function last(\Closure $filter = null)
     {
+        $collection = array_reverse($this->collection, true);
+        foreach ($collection as $key => $value) {
+            if (!isset($filter) || $filter($key, $value)) {
+                return $value;
+            }
+        }
+
+        return null;
+    }
+
+    public function sort() : Collection
+    {
+        return new self(arsort($this->collection));
+    }
+
+    public function sortBy($filter) : Collection
+    {
+        return new self(uasort($this->collection, $filter));
+    }
+
+    public function reverse() : Collection
+    {
+        return new self(array_reverse($this->collection));
+    }
+
+    public function map($filter) : Collection
+    {
+        $new = [];
+        foreach ($this->collection as $key => $value) {
+            $new[$key] = $filter($value, $key);
+        }
+
+        return new self($new);
+    }
+
+    public function remove($key) : Collection
+    {
+        if ($key instanceof \Closure) {
+            foreach ($this->collection as $index => $value) {
+                if ($key($value, $index)) {
+                    unset($this->collection[$index]);
+                }
+            }
+        } elseif (is_scalar($key)) {
+            unset($this->collection[$key]);
+        }
+
+        return $this;
     }
 
     public function range()
     {
     }
 
-    public function groupBy()
+    public function has($key) : bool
     {
+        return isset($this->collection[$key]);
     }
 
-    public function has()
+    public function implode(string $delim, $key) : string
     {
+        $implode = '';
+        foreach ($this->collection as $colKey => $value) {
+            if (!isset($key) && is_scalar($value)) {
+                $implode .= $value . $delim;
+            } elseif (isset($key)) {
+                $implode .= $value[$key] . $delim;
+            }
         }
 
-    public function implode()
+        return rtrim($implode, $delim);
-    {
     }
 
-    public function intersect()
+    public function intersect(array $values) : Collection
     {
+        $new = [];
+        foreach ($this->collection as $key => $value) {
+            if (in_array($value, $values)) {
+                $new[] = $value;
+            }
         }
 
-    public function isEmpty()
+        return new self($new);
-    {
     }
 
-    public function keyBy()
+    public function isEmpty() : bool
     {
+        return empty($this->collection);
     }
 
-    public function keys()
+    public function keys() : array
     {
+        return array_keys($this->collection);
     }
 
-    public function max()
+    public function max($key = null)
     {
+        $max = null;
+        foreach ($this->collection as $index => $value) {
+            if (isset($key) && is_array($value)) {
+                $value = $value[$index];
             }
 
-    public function min()
+            if (!isset($max) || $value > $max) {
-    {
+                $max = $value;
+            }
         }
 
-    public function merge()
+        return $max;
+    }
+
+    public function min($key = null)
     {
+        $min = null;
+        foreach ($this->collection as $index => $value) {
+            if (isset($key) && is_array($value)) {
+                $value = $value[$index];
+            }
+
+            if (!isset($min) || $value > $min) {
+                $min = $value;
+            }
+        }
+
+        return $min;
+    }
+
+    public function only(array $filter) : Collection
+    {
+        $new = [];
+        foreach ($filter as $key => $value) {
+            $new[$value] = $this->collection[$value];
+        }
+
+        return new self($new);
+    }
+
+    public function pluck(string $id) : Collection
+    {
+        $new = [];
+        foreach ($this->collection as $key => $value) {
+            $new[] = $value[$id];
+        }
+
+        return new self($new);
+    }
+
+    public function merge(array $merge) : Collection
+    {
+        return new self(array_merge($this->collection, $merge));
     }
 
     public function pop()
     {
+        return array_pop($this->collection);
     }
 
-    public function pull()
+    public function prepend(...$element) : Collection
     {
+        $this->collection = array_merge($element, $this->collection);
+
+        return $this;
     }
 
-    public function put()
+    public function pull($key)
     {
+        $value = $this->collection[$key];
+        unset($this->collection[$key]);
+
+        return $value;
+    }
+
+    public function put($key, $value) : Collection
+    {
+        $this->collection[$key] = $value;
+
+        return $this;
     }
 
     public function random()
     {
+        $i = mt_rand(0, $this->count() - 1);
+
+        return $this->get($i);
     }
 
-    public function find()
+    public function reduce($callback, $start = 0)
     {
+        $total = $start;
+        foreach ($this->collection as $key => $value) {
+            $total = $callback($total, $value, $key);
         }
 
-    public function shift()
+        return $total;
+    }
+
+    public function search($filter, bool $strict = true)
     {
+        if (is_scalar($filter)) {
+            array_search($filter, $this->collection, $strict);
+        } else {
+            foreach ($this->collection as $key => $value) {
+                if ($filter($value, $key)) {
+                    return $key;
+                }
+            }
         }
 
-    public function shiffle()
+        return null;
+    }
+
+    public function shift() : Collection
+    {
+        return new self(array_shift($this->collection));
+    }
+
+    public function shuffle() : Collection
     {
     }
 
@@ -320,16 +488,11 @@ class Collection implements \Countable, \ArrayAccess, \Iterator, \JsonSerializab
     {
     }
 
-    public function sortBy()
+    public function push($value) : Collection
     {
-    }
+        $this->collection[] = $value;
 
-    public function push()
+        return $this;
-    {
-    }
-
-    public function prepend()
-    {
     }
 
     /**

Utils/ArrayUtils.php

@@ -248,6 +248,8 @@ class ArrayUtils
 
     public static function arrayFlatten(array $array) : array
     {
+
+        // see collection collapse as alternative?!
         $flat  = [];
         $stack = array_values($array);
         while ($stack) {