<?php
+/**
+ * @copyright Copyright (C) 2020, Friendica
+ *
+ * @license GNU AGPL version 3 or any later version
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <https://www.gnu.org/licenses/>.
+ *
+ */
namespace Friendica;
protected $totalCount = 0;
/**
- * @param BaseModel[] $models
- * @param int|null $totalCount
+ * @param BaseEntity[] $entities
+ * @param int|null $totalCount
+ */
+ public function __construct(array $entities = [], int $totalCount = null)
+ {
+ parent::__construct($entities);
+
+ $this->totalCount = $totalCount ?? count($entities);
+ }
+
+ /**
+ * @inheritDoc
+ */
+ public function offsetSet($offset, $value)
+ {
+ if (is_null($offset)) {
+ $this->totalCount++;
+ }
+
+ parent::offsetSet($offset, $value);
+ }
+
+ /**
+ * @inheritDoc
*/
- public function __construct(array $models = [], int $totalCount = null)
+ public function offsetUnset($offset)
{
- parent::__construct($models);
+ if ($this->offsetExists($offset)) {
+ $this->totalCount--;
+ }
- $this->models = $models;
- $this->totalCount = $totalCount ?? count($models);
+ parent::offsetUnset($offset);
}
/**
{
return $this->totalCount;
}
+
+ /**
+ * Return the values from a single field in the collection
+ *
+ * @param string $column
+ * @param int|null $index_key
+ * @return array
+ * @see array_column()
+ */
+ public function column($column, $index_key = null)
+ {
+ return array_column($this->getArrayCopy(), $column, $index_key);
+ }
+
+ /**
+ * Apply a callback function on all elements in the collection and returns a new collection with the updated elements
+ *
+ * @param callable $callback
+ * @return BaseCollection
+ * @see array_map()
+ */
+ public function map(callable $callback)
+ {
+ return new static(array_map($callback, $this->getArrayCopy()), $this->getTotalCount());
+ }
+
+ /**
+ * Filters the collection based on a callback that returns a boolean whether the current item should be kept.
+ *
+ * @param callable|null $callback
+ * @param int $flag
+ * @return BaseCollection
+ * @see array_filter()
+ */
+ public function filter(callable $callback = null, int $flag = 0)
+ {
+ return new static(array_filter($this->getArrayCopy(), $callback, $flag));
+ }
}