Collection.php

Go to the documentation of this file.

<?php
 
namespace Elgg\Collections;
 
use Elgg\Exceptions\InvalidArgumentException;
use Elgg\Exceptions\OutOfBoundsException;
 
class Collection implements CollectionInterface,
                            \ArrayAccess,
                            \SeekableIterator,
                            \Countable {
 
    protected $items = [];
 
    protected $item_class;
    
    protected $position;
 
    public function __construct($items = [], $item_class = null) {
        if ($item_class) {
            if (!is_subclass_of($item_class, CollectionItemInterface::class)) {
                throw new InvalidArgumentException('Item class must implement ' . CollectionItemInterface::class);
            }
 
            $this->item_class = $item_class;
        }
 
        foreach ($items as $item) {
            $this->add($item);
        }
    }
 
    protected function assertValidItem($item) {
        $class = $this->item_class ?? CollectionItemInterface::class;
 
        if (!$item instanceof $class) {
            throw new InvalidArgumentException('Collection ' . __CLASS__ . ' only accepts instances of ' . $class);
        }
    }
 
    public function all() {
        return $this->items;
    }
 
    #[\ReturnTypeWillChange]
    public function count() {
        return count($this->items);
    }
 
    public function add($item) {
        $this->assertValidItem($item);
 
        $this->items[$item->getID()] = $item;
 
        return $this;
    }
 
    public function get($id) {
        return elgg_extract($id, $this->items);
    }
 
    public function has($id) {
        return array_key_exists($id, $this->items);
    }
 
    public function remove($id) {
        unset($this->items[$id]);
    }
 
    public function fill($items) {
        $this->items = [];
        foreach ($items as $item) {
            $this->add($item);
        }
 
        return $this;
    }
 
    public function merge($items) {
        foreach ($items as $item) {
            $this->add($item);
        }
 
        return $this;
    }
 
    public function filter(?callable $callback = null) {
        if ($callback) {
            $items = array_filter($this->items, $callback);
        } else {
            $items = array_values($this->items);
        }
 
        return new static($items, $this->item_class);
    }
 
    public function sort(?callable $callback = null) {
        if (!$callback) {
            $callback = function (CollectionItemInterface $f1, CollectionItemInterface $f2) {
                $p1 = $f1->getPriority() ?: 500;
                $p2 = $f2->getPriority() ?: 500;
                if ($p1 === $p2) {
                    return 0;
                }
 
                return $p1 < $p2 ? -1 : 1;
            };
        }
 
        uasort($this->items, $callback);
 
        return $this;
    }
 
    public function walk(callable $callback) {
        foreach ($this->items as $id => $item) {
            call_user_func($callback, $item, $id);
        }
 
        return $this;
    }
 
    public function map(callable $callback) {
        $map = [];
 
        $items = $this->filter()->all();
        foreach ($items as $id => &$item) {
            $map[$id] = call_user_func($callback, $item, $id);
        }
 
        return $map;
    }
 
    #[\ReturnTypeWillChange]
    public function offsetExists($offset) {
        return $this->has($offset);
    }
 
    #[\ReturnTypeWillChange]
    public function offsetGet($offset) {
        return $this->get($offset);
    }
 
    #[\ReturnTypeWillChange]
    public function offsetSet($offset, $value) {
        $this->assertValidItem($value);
 
        $key = $value->getID();
        $this->items[$key] = $value;
    }
 
    #[\ReturnTypeWillChange]
    public function offsetUnset($offset) {
        if (isset($this->items[$offset]) && isset($this->position)) {
            // handle unset during iteration
            $this->position--;
        }
        
        unset($this->items[$offset]);
    }
 
    #[\ReturnTypeWillChange]
    public function current() {
        $keys = array_keys($this->items);
        
        return $this->get($keys[$this->position]);
    }
 
    #[\ReturnTypeWillChange]
    public function next() {
        $this->position++;
    }
 
    #[\ReturnTypeWillChange]
    public function key() {
        $keys = array_keys($this->items);
        
        return $keys[$this->position];
    }
 
    #[\ReturnTypeWillChange]
    public function valid() {
        $keys = array_keys($this->items);
        
        return isset($keys[$this->position]);
    }
 
    #[\ReturnTypeWillChange]
    public function rewind() {
        $this->position = 0;
    }
 
    #[\ReturnTypeWillChange]
    public function seek($offset) {
        $keys = array_keys($this->items);
 
        if (!isset($keys[$offset])) {
            throw new OutOfBoundsException();
        }
 
        $this->position = $offset;
    }
}