You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
531 lines
12 KiB
531 lines
12 KiB
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Yansongda\Supports;
|
|
|
|
use ArrayAccess;
|
|
use ArrayIterator;
|
|
use Countable;
|
|
use IteratorAggregate;
|
|
use JsonSerializable;
|
|
use Yansongda\Supports\Traits\Accessable;
|
|
use Yansongda\Supports\Traits\Arrayable;
|
|
use Yansongda\Supports\Traits\Serializable;
|
|
|
|
class Collection implements ArrayAccess, Countable, IteratorAggregate, JsonSerializable
|
|
{
|
|
use Serializable;
|
|
use Accessable;
|
|
use Arrayable;
|
|
|
|
/**
|
|
* The collection data.
|
|
*/
|
|
protected array $items = [];
|
|
|
|
/**
|
|
* set data.
|
|
*
|
|
* @param mixed $items
|
|
*/
|
|
public function __construct($items = [])
|
|
{
|
|
foreach ($this->getArrayableItems($items) as $key => $value) {
|
|
$this->set($key, $value);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Wrap the given value in a collection if applicable.
|
|
*
|
|
* @param mixed $value
|
|
*/
|
|
public static function wrap($value): self
|
|
{
|
|
return $value instanceof self ? new static($value) : new static(Arr::wrap($value));
|
|
}
|
|
|
|
/**
|
|
* Get the underlying items from the given collection if applicable.
|
|
*
|
|
* @param array|static $value
|
|
*/
|
|
public static function unwrap($value): array
|
|
{
|
|
return $value instanceof self ? $value->all() : $value;
|
|
}
|
|
|
|
/**
|
|
* Return all items.
|
|
*/
|
|
public function all(): array
|
|
{
|
|
return $this->items;
|
|
}
|
|
|
|
/**
|
|
* Return specific items.
|
|
*/
|
|
public function only(array $keys): array
|
|
{
|
|
$return = [];
|
|
|
|
foreach ($keys as $key) {
|
|
$value = $this->get($key);
|
|
|
|
if (!is_null($value)) {
|
|
$return[$key] = $value;
|
|
}
|
|
}
|
|
|
|
return $return;
|
|
}
|
|
|
|
/**
|
|
* Get all items except for those with the specified keys.
|
|
*
|
|
* @param mixed $keys
|
|
*/
|
|
public function except($keys): self
|
|
{
|
|
$keys = is_array($keys) ? $keys : func_get_args();
|
|
|
|
return new static(Arr::except($this->items, $keys));
|
|
}
|
|
|
|
/**
|
|
* Run a filter over each of the items.
|
|
*/
|
|
public function filter(callable $callback = null): self
|
|
{
|
|
if ($callback) {
|
|
return new static(Arr::where($this->items, $callback));
|
|
}
|
|
|
|
return new static(array_filter($this->items));
|
|
}
|
|
|
|
/**
|
|
* Merge the collection with the given items.
|
|
*
|
|
* @param mixed $items
|
|
*/
|
|
public function merge($items): self
|
|
{
|
|
return new static(array_merge($this->items, $this->getArrayableItems($items)));
|
|
}
|
|
|
|
/**
|
|
* To determine Whether the specified element exists.
|
|
*
|
|
* @param string|int $key
|
|
*/
|
|
public function has($key): bool
|
|
{
|
|
return !is_null(Arr::get($this->items, $key));
|
|
}
|
|
|
|
/**
|
|
* Retrieve the first item.
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function first()
|
|
{
|
|
return reset($this->items);
|
|
}
|
|
|
|
/**
|
|
* Retrieve the last item.
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function last()
|
|
{
|
|
$end = end($this->items);
|
|
|
|
reset($this->items);
|
|
|
|
return $end;
|
|
}
|
|
|
|
/**
|
|
* add the item value.
|
|
*
|
|
* @param string|int|null $key
|
|
* @param mixed $value
|
|
*/
|
|
public function add($key, $value): void
|
|
{
|
|
Arr::set($this->items, $key, $value);
|
|
}
|
|
|
|
/**
|
|
* Set the item value.
|
|
*
|
|
* @param string|int|null $key
|
|
* @param mixed $value
|
|
*/
|
|
public function set($key, $value): void
|
|
{
|
|
Arr::set($this->items, $key, $value);
|
|
}
|
|
|
|
/**
|
|
* Retrieve item from Collection.
|
|
*
|
|
* @param string|int|null $key
|
|
* @param mixed $default
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function get($key = null, $default = null)
|
|
{
|
|
return Arr::get($this->items, $key, $default);
|
|
}
|
|
|
|
/**
|
|
* Remove item form Collection.
|
|
*
|
|
* @param string|int $key
|
|
*/
|
|
public function forget($key): void
|
|
{
|
|
Arr::forget($this->items, $key);
|
|
}
|
|
|
|
/**
|
|
* Get a flattened array of the items in the collection.
|
|
*
|
|
* @param float|int $depth
|
|
*/
|
|
public function flatten($depth = INF): self
|
|
{
|
|
return new static(Arr::flatten($this->items, $depth));
|
|
}
|
|
|
|
/**
|
|
* Run a map over each of the items.
|
|
*/
|
|
public function map(callable $callback): self
|
|
{
|
|
$keys = array_keys($this->items);
|
|
$items = array_map($callback, $this->items, $keys);
|
|
|
|
return new static(array_combine($keys, $items));
|
|
}
|
|
|
|
/**
|
|
* Get and remove the last item from the collection.
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function pop()
|
|
{
|
|
return array_pop($this->items);
|
|
}
|
|
|
|
/**
|
|
* Push an item onto the beginning of the collection.
|
|
*
|
|
* @param mixed|null $key
|
|
* @param mixed $value
|
|
*/
|
|
public function prepend($value, $key = null): self
|
|
{
|
|
$this->items = Arr::prepend($this->items, $value, $key);
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* Push an item onto the end of the collection.
|
|
*
|
|
* @param mixed $value
|
|
*/
|
|
public function push($value): self
|
|
{
|
|
$this->offsetSet(null, $value);
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* Get and remove an item from the collection.
|
|
*
|
|
* @param mixed $default
|
|
* @param mixed $key
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function pull($key, $default = null)
|
|
{
|
|
return Arr::pull($this->items, $key, $default);
|
|
}
|
|
|
|
/**
|
|
* Put an item in the collection by key.
|
|
*
|
|
* @param mixed $key
|
|
* @param mixed $value
|
|
*/
|
|
public function put($key, $value): self
|
|
{
|
|
$this->offsetSet($key, $value);
|
|
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* Get one or a specified number of items randomly from the collection.
|
|
*
|
|
* @throws \InvalidArgumentException
|
|
*/
|
|
public function random(?int $number = null): self
|
|
{
|
|
return new static(Arr::random($this->items, $number ?? 1));
|
|
}
|
|
|
|
/**
|
|
* Reduce the collection to a single value.
|
|
*
|
|
* @param mixed|null $initial
|
|
*/
|
|
public function reduce(callable $callback, $initial = null)
|
|
{
|
|
return array_reduce($this->items, $callback, $initial);
|
|
}
|
|
|
|
/**
|
|
* Reset the keys on the underlying array.
|
|
*/
|
|
public function values(): self
|
|
{
|
|
return new static(array_values($this->items));
|
|
}
|
|
|
|
/**
|
|
* Determine if all items in the collection pass the given test.
|
|
*
|
|
* @param callable|string $key
|
|
*/
|
|
public function every($key): bool
|
|
{
|
|
$callback = $this->valueRetriever($key);
|
|
|
|
foreach ($this->items as $k => $v) {
|
|
if (!$callback($v, $k)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* Chunk the underlying collection array.
|
|
*/
|
|
public function chunk(int $size): self
|
|
{
|
|
if ($size <= 0) {
|
|
return new static();
|
|
}
|
|
$chunks = [];
|
|
foreach (array_chunk($this->items, $size, true) as $chunk) {
|
|
$chunks[] = new static($chunk);
|
|
}
|
|
|
|
return new static($chunks);
|
|
}
|
|
|
|
/**
|
|
* Sort through each item with a callback.
|
|
*/
|
|
public function sort(callable $callback = null): self
|
|
{
|
|
$items = $this->items;
|
|
$callback ? uasort($items, $callback) : asort($items);
|
|
|
|
return new static($items);
|
|
}
|
|
|
|
/**
|
|
* Sort the collection using the given callback.
|
|
*
|
|
* @param callable|string $callback
|
|
*/
|
|
public function sortBy($callback, int $options = SORT_REGULAR, bool $descending = false): self
|
|
{
|
|
$results = [];
|
|
$callback = $this->valueRetriever($callback);
|
|
// First we will loop through the items and get the comparator from a callback
|
|
// function which we were given. Then, we will sort the returned values and
|
|
// and grab the corresponding values for the sorted keys from this array.
|
|
foreach ($this->items as $key => $value) {
|
|
$results[$key] = $callback($value, $key);
|
|
}
|
|
$descending ? arsort($results, $options) : asort($results, $options);
|
|
// Once we have sorted all of the keys in the array, we will loop through them
|
|
// and grab the corresponding model so we can set the underlying items list
|
|
// to the sorted version. Then we'll just return the collection instance.
|
|
foreach (array_keys($results) as $key) {
|
|
$results[$key] = $this->items[$key];
|
|
}
|
|
|
|
return new static($results);
|
|
}
|
|
|
|
/**
|
|
* Sort the collection in descending order using the given callback.
|
|
*
|
|
* @param callable|string $callback
|
|
*/
|
|
public function sortByDesc($callback, int $options = SORT_REGULAR): self
|
|
{
|
|
return $this->sortBy($callback, $options, true);
|
|
}
|
|
|
|
/**
|
|
* Sort the collection keys.
|
|
*/
|
|
public function sortKeys(int $options = SORT_REGULAR, bool $descending = false): self
|
|
{
|
|
$items = $this->items;
|
|
$descending ? krsort($items, $options) : ksort($items, $options);
|
|
|
|
return new static($items);
|
|
}
|
|
|
|
/**
|
|
* Sort the collection keys in descending order.
|
|
*/
|
|
public function sortKeysDesc(int $options = SORT_REGULAR): self
|
|
{
|
|
return $this->sortKeys($options, true);
|
|
}
|
|
|
|
public function query(int $encodingType = PHP_QUERY_RFC1738): string
|
|
{
|
|
return Arr::query($this->all(), $encodingType);
|
|
}
|
|
|
|
public function toString(string $separator = '&'): string
|
|
{
|
|
return Arr::toString($this->all(), $separator);
|
|
}
|
|
|
|
/**
|
|
* Build to array.
|
|
*/
|
|
public function toArray(): array
|
|
{
|
|
return $this->all();
|
|
}
|
|
|
|
/**
|
|
* Build to json.
|
|
*/
|
|
public function toJson(int $option = JSON_UNESCAPED_UNICODE): string
|
|
{
|
|
return json_encode($this->all(), $option);
|
|
}
|
|
|
|
/**
|
|
* (PHP 5 >= 5.0.0)<br/>
|
|
* Retrieve an external iterator.
|
|
*
|
|
* @see http://php.net/manual/en/iteratoraggregate.getiterator.php
|
|
*
|
|
* @return ArrayIterator An instance of an object implementing <b>Iterator</b> or
|
|
* <b>ArrayIterator</b>
|
|
*/
|
|
#[\ReturnTypeWillChange]
|
|
public function getIterator()
|
|
{
|
|
return new ArrayIterator($this->items);
|
|
}
|
|
|
|
/**
|
|
* (PHP 5 >= 5.1.0)<br/>
|
|
* Count elements of an object.
|
|
*
|
|
* @see http://php.net/manual/en/countable.count.php
|
|
*
|
|
* @return int The custom count as an integer.
|
|
* </p>
|
|
* <p>
|
|
* The return value is cast to an integer
|
|
*/
|
|
#[\ReturnTypeWillChange]
|
|
public function count()
|
|
{
|
|
return count($this->items);
|
|
}
|
|
|
|
/**
|
|
* (PHP 5 >= 5.0.0)<br/>
|
|
* Offset to unset.
|
|
*
|
|
* @see http://php.net/manual/en/arrayaccess.offsetunset.php
|
|
*
|
|
* @param mixed $offset <p>
|
|
* The offset to unset.
|
|
* </p>
|
|
*/
|
|
#[\ReturnTypeWillChange]
|
|
public function offsetUnset($offset)
|
|
{
|
|
if ($this->offsetExists($offset)) {
|
|
$this->forget($offset);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Determine if the given value is callable, but not a string.
|
|
*
|
|
* @param mixed $value
|
|
*/
|
|
protected function useAsCallable($value): bool
|
|
{
|
|
return !is_string($value) && is_callable($value);
|
|
}
|
|
|
|
/**
|
|
* Get a value retrieving callback.
|
|
*
|
|
* @param mixed $value
|
|
*/
|
|
protected function valueRetriever($value): callable
|
|
{
|
|
if ($this->useAsCallable($value)) {
|
|
return $value;
|
|
}
|
|
|
|
return function ($item) use ($value) {
|
|
return data_get($item, $value);
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Results array of items from Collection or Arrayable.
|
|
*
|
|
* @param mixed $items
|
|
*/
|
|
protected function getArrayableItems($items): array
|
|
{
|
|
if (is_array($items)) {
|
|
return $items;
|
|
}
|
|
|
|
if ($items instanceof self) {
|
|
return $items->all();
|
|
}
|
|
|
|
if ($items instanceof JsonSerializable) {
|
|
return $items->jsonSerialize();
|
|
}
|
|
|
|
return (array) $items;
|
|
}
|
|
}
|
|
|