366 lines
9.4 KiB
PHP
366 lines
9.4 KiB
PHP
|
<?php
|
||
|
|
namespace Rails\ActiveModel;
|
||
|
|
|
||
|
|
use Closure;
|
||
|
|
use Rails;
|
||
|
|
|
||
|
|
class Collection implements \ArrayAccess, \Iterator
|
||
|
|
{
|
||
|
|
/* ArrayAccess { */
|
||
|
|
protected $members = array();
|
||
|
|
|
||
|
|
public function offsetSet($offset, $value)
|
||
|
|
{
|
||
|
|
if (is_null($offset)) {
|
||
|
|
$this->members[] = $value;
|
||
|
|
} else {
|
||
|
|
$this->members[$offset] = $value;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public function offsetExists($offset)
|
||
|
|
{
|
||
|
|
return isset($this->members[$offset]);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function offsetUnset($offset)
|
||
|
|
{
|
||
|
|
unset($this->members[$offset]);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function offsetGet($offset)
|
||
|
|
{
|
||
|
|
return isset($this->members[$offset]) ? $this->members[$offset] : null;
|
||
|
|
}
|
||
|
|
/* } Iterator {*/
|
||
|
|
protected $position = 0;
|
||
|
|
|
||
|
|
public function rewind()
|
||
|
|
{
|
||
|
|
reset($this->members);
|
||
|
|
$this->position = key($this->members);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function current()
|
||
|
|
{
|
||
|
|
return $this->members[$this->position];
|
||
|
|
}
|
||
|
|
|
||
|
|
public function key()
|
||
|
|
{
|
||
|
|
return key($this->members);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function next()
|
||
|
|
{
|
||
|
|
next($this->members);
|
||
|
|
$this->position = key($this->members);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function valid()
|
||
|
|
{
|
||
|
|
return array_key_exists($this->position, $this->members);
|
||
|
|
}
|
||
|
|
/* } */
|
||
|
|
|
||
|
|
public function __construct(array $members = array())
|
||
|
|
{
|
||
|
|
$this->members = $members;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function merge()
|
||
|
|
{
|
||
|
|
foreach (func_get_args() as $coll) {
|
||
|
|
if ($coll instanceof self)
|
||
|
|
$coll = $coll->members();
|
||
|
|
$this->members = array_merge($this->members, $coll);
|
||
|
|
}
|
||
|
|
return $this;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function members()
|
||
|
|
{
|
||
|
|
return $this->members;
|
||
|
|
}
|
||
|
|
|
||
|
|
# Another way to get members.
|
||
|
|
public function toArray()
|
||
|
|
{
|
||
|
|
return $this->members;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Each (experimental)
|
||
|
|
*
|
||
|
|
* If string is passed, it'll be taken as method name to be called.
|
||
|
|
* Eg. $posts->each('destroy'); - All posts will be destroyed.
|
||
|
|
* In this case, $params for the method may be passed.
|
||
|
|
*
|
||
|
|
* A Closure may also be passed.
|
||
|
|
*/
|
||
|
|
public function each($function, array $params = array())
|
||
|
|
{
|
||
|
|
if (is_string($function)) {
|
||
|
|
foreach ($this->members() as $m) {
|
||
|
|
call_user_func_array(array($m, $function), $params);
|
||
|
|
}
|
||
|
|
} elseif ($function instanceof Closure) {
|
||
|
|
foreach ($this->members() as $idx => $m) {
|
||
|
|
$function($m, $idx);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
throw new Exception\InvalidArgumentException(
|
||
|
|
sprintf('Argument must be an either a string or a Closure, %s passed.', gettype($model))
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public function reduce($var, Closure $block)
|
||
|
|
{
|
||
|
|
foreach ($this->members() as $m) {
|
||
|
|
$var = $block($var, $m);
|
||
|
|
}
|
||
|
|
return $var;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function sort(Closure $criteria)
|
||
|
|
{
|
||
|
|
usort($this->members, $criteria);
|
||
|
|
$this->rewind();
|
||
|
|
return $this;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function unshift($model)
|
||
|
|
{
|
||
|
|
if ($model instanceof Base)
|
||
|
|
$model = array($model);
|
||
|
|
elseif (!$model instanceof self) {
|
||
|
|
throw new Exception\InvalidArgumentException(
|
||
|
|
sprintf('Argument must be an instance of either ActiveRecord\Base or ActiveRecord\Collection, %s passed.', gettype($model))
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
foreach ($model as $m)
|
||
|
|
array_unshift($this->members, $m);
|
||
|
|
|
||
|
|
return $this;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Searches objects for a property with a value and returns object.
|
||
|
|
*/
|
||
|
|
public function search($prop, $value)
|
||
|
|
{
|
||
|
|
foreach ($this->members() as $obj) {
|
||
|
|
if ($obj->$prop == $value)
|
||
|
|
return $obj;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
# Returns a Collection with the models that matches the options.
|
||
|
|
# Eg: $posts->select(array('is_active' => true, 'user_id' => 4));
|
||
|
|
# If Closure passed as $opts, the model that returns == true on the function
|
||
|
|
# will be added.
|
||
|
|
public function select($opts)
|
||
|
|
{
|
||
|
|
$objs = array();
|
||
|
|
|
||
|
|
if (is_array($opts)) {
|
||
|
|
foreach ($this as $obj) {
|
||
|
|
foreach ($opts as $prop => $cond) {
|
||
|
|
if (!$obj->$prop || $obj->$prop != $cond)
|
||
|
|
continue;
|
||
|
|
$objs[] = $obj;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} elseif ($opts instanceof Closure) {
|
||
|
|
foreach ($this->members() as $obj) {
|
||
|
|
$opts($obj) && $objs[] = $obj;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return new self($objs);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Removes members according to attributes.
|
||
|
|
*/
|
||
|
|
public function remove($attrs)
|
||
|
|
{
|
||
|
|
!is_array($attrs) && $attrs = array('id' => $attrs);
|
||
|
|
|
||
|
|
foreach ($this->members() as $k => $m) {
|
||
|
|
foreach ($attrs as $a => $v) {
|
||
|
|
if ($m->getAttribute($a) != $v)
|
||
|
|
continue 2;
|
||
|
|
}
|
||
|
|
unset($this->members[$k]);
|
||
|
|
}
|
||
|
|
$this->members = array_values($this->members);
|
||
|
|
return $this;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function max($criteria)
|
||
|
|
{
|
||
|
|
if (!$this->members)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
$current = key($this->members);
|
||
|
|
if (count($this->members) < 2)
|
||
|
|
return $this->members[$current];
|
||
|
|
|
||
|
|
$max = $this->members[$current];
|
||
|
|
|
||
|
|
if ($criteria instanceof Closure) {
|
||
|
|
$params = $this;
|
||
|
|
foreach($params as $current) {
|
||
|
|
if (!$next = next($params))
|
||
|
|
break;
|
||
|
|
$max = $criteria($max, $next);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
|
||
|
|
}
|
||
|
|
return $max;
|
||
|
|
}
|
||
|
|
|
||
|
|
# Deprecated in favor of none.
|
||
|
|
public function blank()
|
||
|
|
{
|
||
|
|
return empty($this->members);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function none()
|
||
|
|
{
|
||
|
|
return empty($this->members);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function any()
|
||
|
|
{
|
||
|
|
return (bool)$this->members;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* TODO: xml shouldn't be created here.
|
||
|
|
*/
|
||
|
|
public function toXml()
|
||
|
|
{
|
||
|
|
if ($this->blank())
|
||
|
|
return;
|
||
|
|
|
||
|
|
$t = get_class($this->current());
|
||
|
|
$t = $t::t();
|
||
|
|
|
||
|
|
$xml = '<?xml version="1.0" encoding="UTF-8"?>';
|
||
|
|
$xml .= '<' . $t . '>';
|
||
|
|
|
||
|
|
foreach ($this->members() as $obj) {
|
||
|
|
$xml .= $obj->toXml(array('skip_instruct' => true));
|
||
|
|
}
|
||
|
|
|
||
|
|
$xml .= '</' . $t . '>';
|
||
|
|
|
||
|
|
return $xml;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function asJson()
|
||
|
|
{
|
||
|
|
$json = [];
|
||
|
|
foreach ($this->members as $member)
|
||
|
|
$json[] = $member->asJson();
|
||
|
|
return $json;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function toJson()
|
||
|
|
{
|
||
|
|
return json_encode($this->asJson());
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Returns an array of the attributes in the models.
|
||
|
|
* $attrs could be a string of a single attribute we want, and
|
||
|
|
* an indexed array will be returned.
|
||
|
|
* If $attrs is an array of many attributes, an associative array will be returned.
|
||
|
|
*
|
||
|
|
* $collection->attributes(['id', 'createdAt']);
|
||
|
|
* No -> $collection->attributes('id', 'name');
|
||
|
|
*/
|
||
|
|
public function getAttributes($attrs)
|
||
|
|
{
|
||
|
|
$models_attrs = array();
|
||
|
|
|
||
|
|
if (is_string($attrs)) {
|
||
|
|
foreach ($this as $m) {
|
||
|
|
// if (Rails::config()->ar2) {
|
||
|
|
// $models_attrs[] = $m->$attrs();
|
||
|
|
// } else {
|
||
|
|
$models_attrs[] = $m->$attrs;
|
||
|
|
// }
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
foreach ($this->members() as $m) {
|
||
|
|
$model_attrs = [];
|
||
|
|
foreach ($attrs as $attr) {
|
||
|
|
// if (!Rails::config()->ar2) {
|
||
|
|
$model_attrs[$attr] = $m->$attr;
|
||
|
|
// } else {
|
||
|
|
// $model_attrs[$attr] = $m->$attr();
|
||
|
|
// }
|
||
|
|
}
|
||
|
|
$models_attrs[] = $model_attrs;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return $models_attrs;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function size()
|
||
|
|
{
|
||
|
|
return count($this->members);
|
||
|
|
}
|
||
|
|
|
||
|
|
# Removes dupe models based on id or other attribute.
|
||
|
|
public function unique($attr = 'id')
|
||
|
|
{
|
||
|
|
$checked = array();
|
||
|
|
foreach ($this->members() as $k => $obj) {
|
||
|
|
if (in_array($obj->$attr, $checked))
|
||
|
|
unset($this->members[$k]);
|
||
|
|
else
|
||
|
|
$checked[] = $obj->$attr;
|
||
|
|
}
|
||
|
|
return $this;
|
||
|
|
}
|
||
|
|
|
||
|
|
# array_slices the collection.
|
||
|
|
public function slice($offset, $length = null)
|
||
|
|
{
|
||
|
|
$clone = clone $this;
|
||
|
|
$clone->members = array_slice($clone->members, $offset, $length);
|
||
|
|
return $clone;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function deleteIf(Closure $conditions)
|
||
|
|
{
|
||
|
|
$deleted = false;
|
||
|
|
foreach ($this->members() as $k => $m) {
|
||
|
|
if ($conditions($m)) {
|
||
|
|
unset($this[$k]);
|
||
|
|
$deleted = true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if ($deleted)
|
||
|
|
$this->members = array_values($this->members);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function replace($replacement)
|
||
|
|
{
|
||
|
|
if ($replacement instanceof self)
|
||
|
|
$this->members = $replacement->members();
|
||
|
|
elseif (is_array($replacement))
|
||
|
|
$this->members = $replacement;
|
||
|
|
else
|
||
|
|
throw new Exception\InvalidArgumentException(sprintf("%s expects a %s or an array, %s passed", __METHOD__, __CLASS__, gettype($replacement)));
|
||
|
|
}
|
||
|
|
}
|