items = array_values($items); parent::__construct(); } /** * Return the class of items in this list, by looking at the first item inside it. */ public function dataClass() { if(count($this->items) > 0) return get_class($this->items[0]); } /** * Return the number of items in this list * * @return int */ public function count() { return count($this->items); } /** * Returns true if this list has items * * @return bool */ public function exists() { return !empty($this->items); } /** * Returns an Iterator for this ArrayList. * This function allows you to use ArrayList in foreach loops * * @return ArrayIterator */ public function getIterator() { foreach($this->items as $i => $item) { if(is_array($item)) $this->items[$i] = new ArrayData($item); } return new ArrayIterator($this->items); } /** * Return an array of the actual items that this ArrayList contains. * * @return array */ public function toArray() { return $this->items; } /** * Walks the list using the specified callback * * @param callable $callback * @return DataList */ public function each($callback) { foreach($this as $item) { $callback($item); } } public function debug() { $val = "

" . $this->class . "

"; return $val; } /** * Return this list as an array and every object it as an sub array as well * * @return array */ public function toNestedArray() { $result = array(); foreach ($this->items as $item) { if (is_object($item)) { if (method_exists($item, 'toMap')) { $result[] = $item->toMap(); } else { $result[] = (array) $item; } } else { $result[] = $item; } } return $result; } /** * Get a sub-range of this dataobjectset as an array * * @param int $offset * @param int $length * @return ArrayList */ public function limit($length, $offset = 0) { if(!$length) { $length = count($this->items); } $list = clone $this; $list->items = array_slice($this->items, $offset, $length); return $list; } /** * Add this $item into this list * * @param mixed $item */ public function add($item) { $this->push($item); } /** * Remove this item from this list * * @param mixed $item */ public function remove($item) { $renumberKeys = false; foreach ($this->items as $key => $value) { if ($item === $value) { $renumberKeys = true; unset($this->items[$key]); } } if($renumberKeys) $this->items = array_values($this->items); } /** * Replaces an item in this list with another item. * * @param array|object $item * @param array|object $with * @return void; */ public function replace($item, $with) { foreach ($this->items as $key => $candidate) { if ($candidate === $item) { $this->items[$key] = $with; return; } } } /** * Merges with another array or list by pushing all the items in it onto the * end of this list. * * @param array|object $with */ public function merge($with) { foreach ($with as $item) $this->push($item); } /** * Removes items from this list which have a duplicate value for a certain * field. This is especially useful when combining lists. * * @param string $field */ public function removeDuplicates($field = 'ID') { $seen = array(); $renumberKeys = false; foreach ($this->items as $key => $item) { $value = $this->extractValue($item, $field); if (array_key_exists($value, $seen)) { $renumberKeys = true; unset($this->items[$key]); } $seen[$value] = true; } if($renumberKeys) $this->items = array_values($this->items); } /** * Pushes an item onto the end of this list. * * @param array|object $item */ public function push($item) { $this->items[] = $item; } /** * Pops the last element off the end of the list and returns it. * * @return array|object */ public function pop() { return array_pop($this->items); } /** * Add an item onto the beginning of the list. * * @param array|object $item */ public function unshift($item) { array_unshift($this->items, $item); } /** * Shifts the item off the beginning of the list and returns it. * * @return array|object */ public function shift() { return array_shift($this->items); } /** * Returns the first item in the list * * @return mixed */ public function first() { return reset($this->items); } /** * Returns the last item in the list * * @return mixed */ public function last() { return end($this->items); } /** * Returns a map of this list * * @param string $keyfield - the 'key' field of the result array * @param string $titlefield - the value field of the result array * @return array */ public function map($keyfield = 'ID', $titlefield = 'Title') { $map = array(); foreach ($this->items as $item) { $map[$this->extractValue($item, $keyfield)] = $this->extractValue( $item, $titlefield ); } return $map; } /** * Find the first item of this list where the given key = value * * @param string $key * @param string $value * @return mixed */ public function find($key, $value) { foreach ($this->items as $item) { if ($this->extractValue($item, $key) == $value) { return $item; } } } /** * Returns an array of a single field value for all items in the list. * * @param string $colName * @return array */ public function column($colName = 'ID') { $result = array(); foreach ($this->items as $item) { $result[] = $this->extractValue($item, $colName); } return $result; } /** * You can always sort a ArrayList * * @param string $by * @return bool */ public function canSortBy($by) { return true; } /** * Reverses an {@link ArrayList} * * @return ArrayList */ public function reverse() { $list = clone $this; $list->items = array_reverse($this->items); return $list; } /** * Parses a specified column into a sort field and direction * * @param string $column String to parse containing the column name * @param string $direction Optional Additional argument which may contain the direction * @return array Sort specification in the form array("Column", SORT_ASC). */ protected function parseSortColumn($column, $direction = null) { // Substitute the direction for the column if column is a numeric index if($direction && (empty($column) || is_numeric($column))) { $column = $direction; $direction = null; } // Parse column specification, considering possible ansi sql quoting // Note that table prefix is allowed, but discarded if(preg_match('/^("?(?[^"\s]+)"?\\.)?"?(?[^"\s]+)"?(\s+(?((asc)|(desc))(ending)?))?$/i', $column, $match)) { $column = $match['column']; if(empty($direction) && !empty($match['direction'])) { $direction = $match['direction']; } } else { throw new InvalidArgumentException("Invalid sort() column"); } // Parse sort direction specification if(empty($direction) || preg_match('/^asc(ending)?$/i', $direction)) { $direction = SORT_ASC; } elseif(preg_match('/^desc(ending)?$/i', $direction)) { $direction = SORT_DESC; } else { throw new InvalidArgumentException("Invalid sort() direction"); } return array($column, $direction); } /** * Sorts this list by one or more fields. You can either pass in a single * field name and direction, or a map of field names to sort directions. * * Note that columns may be double quoted as per ANSI sql standard * * @return DataList * @see SS_List::sort() * @example $list->sort('Name'); // default ASC sorting * @example $list->sort('Name DESC'); // DESC sorting * @example $list->sort('Name', 'ASC'); * @example $list->sort(array('Name'=>'ASC,'Age'=>'DESC')); */ public function sort() { $args = func_get_args(); if(count($args)==0){ return $this; } if(count($args)>2){ throw new InvalidArgumentException('This method takes zero, one or two arguments'); } // One argument and it's a string if(count($args)==1 && is_string($args[0])){ list($column, $direction) = $this->parseSortColumn($args[0]); $columnsToSort[$column] = $direction; } else if(count($args)==2) { list($column, $direction) = $this->parseSortColumn($args[0], $args[1]); $columnsToSort[$column] = $direction; } else if(is_array($args[0])) { foreach($args[0] as $key => $value) { list($column, $direction) = $this->parseSortColumn($key, $value); $columnsToSort[$column] = $direction; } } else { throw new InvalidArgumentException("Bad arguments passed to sort()"); } // Store the original keys of the items as a sort fallback, so we can preserve the original order in the event // that array_multisort is unable to work out a sort order for them. This also prevents array_multisort trying // to inspect object properties which can result in errors with circular dependencies $originalKeys = array_keys($this->items); // This the main sorting algorithm that supports infinite sorting params $multisortArgs = array(); $values = array(); $firstRun = true; foreach($columnsToSort as $column => $direction) { // The reason these are added to columns is of the references, otherwise when the foreach // is done, all $values and $direction look the same $values[$column] = array(); $sortDirection[$column] = $direction; // We need to subtract every value into a temporary array for sorting foreach($this->items as $index => $item) { $values[$column][] = strtolower($this->extractValue($item, $column)); } // PHP 5.3 requires below arguments to be reference when using array_multisort together // with call_user_func_array // First argument is the 'value' array to be sorted $multisortArgs[] = &$values[$column]; // First argument is the direction to be sorted, $multisortArgs[] = &$sortDirection[$column]; if ($firstRun) { $sortArg = SORT_REGULAR; $multisortArgs[] = &$sortArg; } $firstRun = false; } $multisortArgs[] = &$originalKeys; $list = clone $this; // As the last argument we pass in a reference to the items that all the sorting will be applied upon $multisortArgs[] = &$list->items; call_user_func_array('array_multisort', $multisortArgs); return $list; } /** * Returns true if the given column can be used to filter the records. * * It works by checking the fields available in the first record of the list. */ public function canFilterBy($by) { $firstRecord = $this->first(); if ($firstRecord === false) { return false; } return is_array($firstRecord) ? array_key_exists($by, $firstRecord) : property_exists($by, $firstRecord); } /** * Filter the list to include items with these charactaristics * * @return ArrayList * @see SS_List::filter() * @example $list->filter('Name', 'bob'); // only bob in the list * @example $list->filter('Name', array('aziz', 'bob'); // aziz and bob in list * @example $list->filter(array('Name'=>'bob, 'Age'=>21)); // bob with the Age 21 in list * @example $list->filter(array('Name'=>'bob, 'Age'=>array(21, 43))); // bob with the Age 21 or 43 * @example $list->filter(array('Name'=>array('aziz','bob'), 'Age'=>array(21, 43))); * // aziz with the age 21 or 43 and bob with the Age 21 or 43 */ public function filter() { $keepUs = call_user_func_array(array($this, 'normaliseFilterArgs'), func_get_args()); $itemsToKeep = array(); foreach($this->items as $item){ $keepItem = true; foreach ($keepUs as $column => $value) { if ((is_array($value) && !in_array($this->extractValue($item, $column), $value)) || (!is_array($value) && $this->extractValue($item, $column) != $value) ) { $keepItem = false; break; } } if($keepItem) { $itemsToKeep[] = $item; } } $list = clone $this; $list->items = $itemsToKeep; return $list; } /** * Return a copy of this list which contains items matching any of these charactaristics. * * @example // only bob in the list * $list = $list->filterAny('Name', 'bob'); * @example // azis or bob in the list * $list = $list->filterAny('Name', array('aziz', 'bob'); * @example // bob or anyone aged 21 in the list * $list = $list->filterAny(array('Name'=>'bob, 'Age'=>21)); * @example // bob or anyone aged 21 or 43 in the list * $list = $list->filterAny(array('Name'=>'bob, 'Age'=>array(21, 43))); * @example // all bobs, phils or anyone aged 21 or 43 in the list * $list = $list->filterAny(array('Name'=>array('bob','phil'), 'Age'=>array(21, 43))); * * @param string|array See {@link filter()} * @return DataList */ public function filterAny() { $keepUs = call_user_func_array(array($this, 'normaliseFilterArgs'), func_get_args()); $itemsToKeep = array(); foreach ($this->items as $item) { foreach ($keepUs as $column => $value) { $extractedValue = $this->extractValue($item, $column); $matches = is_array($value) ? in_array($extractedValue, $value) : $extractedValue == $value; if ($matches) { $itemsToKeep[] = $item; break; } } } $list = clone $this; $list->items = array_unique($itemsToKeep, SORT_REGULAR); return $list; } /** * Take the "standard" arguments that the filter/exclude functions take and return a single array with * 'colum' => 'value' * * @param $column array|string The column name to filter OR an assosicative array of column => value * @param $value array|string|null The values to filter the $column against * * @return array The normalised keyed array */ protected function normaliseFilterArgs($column, $value = null) { if(count(func_get_args())>2){ throw new InvalidArgumentException('filter takes one array or two arguments'); } if(count(func_get_args()) == 1 && !is_array(func_get_arg(0))){ throw new InvalidArgumentException('filter takes one array or two arguments'); } $keepUs = array(); if(count(func_get_args())==2){ $keepUs[func_get_arg(0)] = func_get_arg(1); } if(count(func_get_args())==1 && is_array(func_get_arg(0))){ foreach(func_get_arg(0) as $column => $value) { $keepUs[$column] = $value; } } return $keepUs; } /** * Filter this list to only contain the given Primary IDs * * @param array $ids Array of integers, will be automatically cast/escaped. * @return ArrayList */ public function byIDs($ids) { $ids = array_map('intval', $ids); // sanitize return $this->filter('ID', $ids); } public function byID($id) { $firstElement = $this->filter("ID", $id)->first(); if ($firstElement === false) { return null; } return $firstElement; } /** * @see SS_Filterable::filterByCallback() * * @example $list = $list->filterByCallback(function($item, $list) { return $item->Age == 9; }) * @param callable $callback * @return ArrayList */ public function filterByCallback($callback) { if(!is_callable($callback)) { throw new LogicException(sprintf( "SS_Filterable::filterByCallback() passed callback must be callable, '%s' given", gettype($callback) )); } $output = static::create(); foreach($this as $item) { if(call_user_func($callback, $item, $this)) $output->push($item); } return $output; } /** * Exclude the list to not contain items with these charactaristics * * @return ArrayList * @see SS_List::exclude() * @example $list->exclude('Name', 'bob'); // exclude bob from list * @example $list->exclude('Name', array('aziz', 'bob'); // exclude aziz and bob from list * @example $list->exclude(array('Name'=>'bob, 'Age'=>21)); // exclude bob that has Age 21 * @example $list->exclude(array('Name'=>'bob, 'Age'=>array(21, 43))); // exclude bob with Age 21 or 43 * @example $list->exclude(array('Name'=>array('bob','phil'), 'Age'=>array(21, 43))); * // bob age 21 or 43, phil age 21 or 43 would be excluded */ public function exclude() { $removeUs = call_user_func_array(array($this, 'normaliseFilterArgs'), func_get_args()); $hitsRequiredToRemove = count($removeUs); $matches = array(); foreach($removeUs as $column => $excludeValue) { foreach($this->items as $key => $item){ if(!is_array($excludeValue) && $this->extractValue($item, $column) == $excludeValue) { $matches[$key]=isset($matches[$key])?$matches[$key]+1:1; } elseif(is_array($excludeValue) && in_array($this->extractValue($item, $column), $excludeValue)) { $matches[$key]=isset($matches[$key])?$matches[$key]+1:1; } } } $keysToRemove = array_keys($matches,$hitsRequiredToRemove); $itemsToKeep = array(); foreach($this->items as $key => $value) { if(!in_array($key, $keysToRemove)) { $itemsToKeep[] = $value; } } $list = clone $this; $list->items = $itemsToKeep; return $list; } protected function shouldExclude($item, $args) { } /** * Returns whether an item with $key exists * * @param mixed $key * @return bool */ public function offsetExists($offset) { return array_key_exists($offset, $this->items); } /** * Returns item stored in list with index $key * * @param mixed $key * @return DataObject */ public function offsetGet($offset) { if ($this->offsetExists($offset)) return $this->items[$offset]; } /** * Set an item with the key in $key * * @param mixed $key * @param mixed $value */ public function offsetSet($offset, $value) { if($offset == null) { $this->items[] = $value; } else { $this->items[$offset] = $value; } } /** * Unset an item with the key in $key * * @param mixed $key */ public function offsetUnset($offset) { unset($this->items[$offset]); } /** * Extracts a value from an item in the list, where the item is either an * object or array. * * @param array|object $item * @param string $key * @return mixed */ protected function extractValue($item, $key) { if (is_object($item)) { return $item->$key; } else { if (array_key_exists($key, $item)) return $item[$key]; } } }