silverstripe-framework/core/model/SQLMap.php

<?php

/**
 * @package sapphire
 * @subpackage model
 */

/**
 * This is a class used to represent key->value pairs generated from database queries.
 * The query isn't actually executed until you need it.
 * @package sapphire
 * @subpackage model
 */
class SQLMap extends Object implements IteratorAggregate {
	/**
	 * The query used to generate the map.
	 * @var SQLQuery
	 */
	protected $query;
	
	/**
	 * Construct a SQLMap.
	 * @param SQLQuery $query The query to generate this map. THis isn't executed until it's needed.
	 */
	public function __construct(SQLQuery $query) {
		if(!$query) {
			user_error('SQLMap constructed with null query.', E_USER_ERROR);
		}
		
		$this->query = $query;
	}
	
	/**
	 * Get the name of an item.
	 * @param string|int $id The id of the item.
	 * @return string
	 */
	public function getItem($id) {
		if($id) {
			$baseTable = reset($this->query->from);
			$this->query->where[] = "$baseTable.ID = $id";
			$record = $this->query->execute()->first();
			if($record) {
				$className = $record['ClassName'];
				$obj = new $className($record);
				return $obj->Title;
			}
		}
	}
	
	public function getIterator() {
		$this->genItems();
		return new SQLMap_Iterator($this->items->getIterator());
	}
	
	/**
	 * Get the items in this class.
	 * @return DataObjectSet
	 */
	public function getItems() {
		$this->genItems();
		return $this->items;
	}
	
	/**
	 * Generate the items in this map. This is used by
	 * getItems() if the items have not been generated already.
	 */
	protected function genItems() {
		if(!isset($this->items)) {
			$this->items = new DataObjectSet();
			$items = $this->query->execute();	
			
			foreach($items as $item) {
				$className = isset($item['RecordClassName'])  ? $item['RecordClassName'] :  $item['ClassName'];

				if(!$className) {
					user_error('SQLMap query could not retrieve className', E_USER_ERROR);
				}
				
				$this->items->push(new $className($item));
			}
		}
	}
}

class SQLMap_Iterator extends Object implements Iterator {
	protected $items;
	
	function __construct(Iterator $items) {
		$this->items = $items;
	}

	
	/*
	 * Iterator functions - necessary for foreach to work
	 */
	public function rewind() {
		return $this->items->rewind() ? $this->items->rewind()->Title : null;
	}
	
	public function current() {
		return $this->items->current()->Title;
	}
	
	public function key() {
		return $this->items->current()->ID;
	}
	
	public function next() {
		$next = $this->items->next();
		return isset($next->Title) ? $next->Title : null;
	}
	
	public function valid() {
	 	return $this->items->valid();
	}
}

?>