ArrayListTrait.php

<?php
namespace Seboettg\Collection;
 
 
trait ArrayListTrait
{
    protected $array;
 
    public function clear()
    {
        $this->array = [];
        return $this;
    }
 
    public function get($key)
    {
        return isset($this->array[$key]) ? $this->array[$key] : null;
    }
 
    public function current()
    {
        return current($this->array);
    }
 
    public function next()
    {
        return next($this->array);
    }
 
    public function prev()
    {
        return prev($this->array);
    }
 
    public function set($key, $element)
    {
        $this->array[$key] = $element;
        return $this;
    }
 
    public function setArray(array $array)
    {
        return $this->replace($array);
    }
 
    public function append($element)
    {
        $this->array[] = $element;
        return $this;
    }
 
    public function add($key, $element)
    {
 
        if (!array_key_exists($key, $this->array)) {
            $this->array[$key] = $element;
        } elseif (is_array($this->array[$key])) {
            $this->array[$key][] = $element;
        } else {
            $this->array[$key] = [$this->array[$key], $element];
        }
 
        return $this;
    }
 
    public function remove($key)
    {
        unset($this->array[$key]);
        return $this;
    }
 
    public function hasKey($key)
    {
        return array_key_exists($key, $this->array);
    }
 
    public function hasValue($value)
    {
        $result = array_search($value, $this->array, true);
        return ($result !== false);
    }
 
    public function replace(array $data)
    {
        $this->array = $data;
        return $this;
    }
 
    public function getIterator()
    {
        return new \ArrayIterator($this->array);
    }
 
    public function offsetGet($offset)
    {
        return isset($this->array[$offset]) ? $this->array[$offset] : null;
    }
 
    public function offsetSet($offset, $value)
    {
        $this->array[$offset] = $value;
    }
 
    public function offsetExists($offset)
    {
        return isset($this->array[$offset]);
    }
 
    public function offsetUnset($offset)
    {
        unset($this->array[$offset]);
    }
 
    public function toArray()
    {
        return $this->array;
    }
 
    public function count()
    {
        return count($this->array);
    }
 
    public function shuffle() {
        shuffle($this->array);
        return $this;
    }
 
    public function filter(\Closure $closure)
    {
        return new ArrayList(array_filter($this->array, $closure));
    }
 
    public function filterByKeys(array $keys)
    {
        return new ArrayList(
            array_filter($this->array, function($key) use ($keys) {
                return array_search($key, $keys) !== false;
            }, ARRAY_FILTER_USE_KEY)
        );
    }
 
    public function map(\closure $mapFunction)
    {
        return new ArrayList(array_map($mapFunction, $this->array));
    }
 
    public function flatten()
    {
        $flattenedArray = [];
        array_walk_recursive($this->array, function($item) use(&$flattenedArray) {
            $flattenedArray[] = $item;
        });
        return new ArrayList($flattenedArray);
    }
}