DataProvider.php

<?php

namespace sammaye\mongoyii;

use Yii;
use CActiveDataProvider;

use sammaye\mongoyii\Document;
use sammaye\mongoyii\Query;

/**
 * EMongoDataProvider
 *
 * A data Provider helper for interacting with the EMongoCursor
 */
class DataProvider extends CActiveDataProvider
{
	/**
	 * The primary ActiveRecord class name. The {@link getData()} method
	 * will return a list of objects of this class.
	 * @var string
	 */
	public $modelClass;

	/**
	 * The AR finder instance (eg <code>Post::model()</code>).
	 * This property can be set by passing the finder instance as the first parameter
	 * to the constructor. For example, <code>Post::model()->published()</code>.
	 * @var Model
	 */
	public $model;

	/**
	 * The name of key attribute for {@link modelClass}. If not set,
	 * it means the primary key of the corresponding database table will be used.
	 * @var string
	 */
	public $keyAttribute = '_id';

	/**
	 * @var array The criteria array
	 */
	private $_criteria;

	/**
	 * The internal MongoDB cursor as a MongoCursor instance
	 * @var \MongoCursor
	 */
	private $_cursor;

	/**
	 * @var Sort
	 */
	private $_sort;

	/**
	 * Creates the EMongoDataProvider instance
	 * @param string|Document $modelClass
	 * @param array $config
	 */
	public function __construct($modelClass, $config = array())
	{
		if(is_string($modelClass)){
			$this->modelClass = $modelClass;
			$this->model = Document::model($this->modelClass);
		}elseif($modelClass instanceof Document){
			$this->modelClass = get_class($modelClass);
			$this->model = $modelClass;
		}
		$this->setId($this->modelClass);
		foreach($config as $key => $value){
			$this->$key = $value;
		}
		
		if(!$this->getCriteria()){
			$this->setCriteria(new Query);
		}
	}

	/**
	 * @see CActiveDataProvider::getCriteria()
	 * @return array|Query
	 */
	public function getCriteria()
	{
		return $this->_criteria;
	}

	/**
	 * @see CActiveDataProvider::setCriteria()
	 * @param array|Query $value
	 * @throws Exception
	 */
	public function setCriteria($value)
	{
		if(is_array($value)){
			$this->_criteria = new Query($value);
		}elseif($value instanceof Query){
			$this->_criteria = $value;
		}else{
			throw new Exception(
				'Criteria for the mongoyii\DataProvider must be either an array or a mongoyii\Query object'
			);
		}
	}

	/**
	 * @see CActiveDataProvider::fetchData()
	 * @return array
	 */
	public function fetchData()
	{
		$criteria = $this->getCriteria();

		if(($pagination = $this->getPagination()) !== false){
			$pagination->setItemCount($this->getTotalItemCount());
			$criteria->limit = $pagination->getLimit();
			$criteria->skip = $pagination->getOffset();
		}

		if(($sort = $this->getSort()) !== false){
			$sort = $sort->getOrderBy();
			if(count($sort) > 0){
				$criteria->sort = $sort;
			}
		}

		$c = $this->model->find($criteria);
		return iterator_to_array(
			$c, 
			false
		);
	}

	/**
	 * @see CActiveDataProvider::fetchKeys()
	 * @return array
	 */
	public function fetchKeys()
	{
		$keys = array();
		foreach($this->getData() as $i => $data){
			$key = $this->keyAttribute === null ? $data->{$data->primaryKey()} : $data->{$this->keyAttribute};
			$keys[$i] = is_array($key) ? implode(',', $key) : $key;
		}
		return $keys;
	}

	/**
	 * @see CActiveDataProvider::calculateTotalItemCount()
	 * @return int
	 */
	public function calculateTotalItemCount()
	{
		$criteria = $this->getCriteria();
		return $this->model->count($criteria->getCondition());
	}

	public function setSort($value)
	{
		if(is_array($value))
		{
			if(isset($value['class']))
			{
				$sort=$this->getSort($value['class']);
				unset($value['class']);
			}
			else
				$sort=$this->getSort();

			foreach($value as $k=>$v)
				$sort->$k=$v;
		}
		else
			$this->_sort=$value;
	}

	/**
	 * Returns the sort object. We don't use the newer getSort function because it does not have the same functionality
	 * between 1.1.10 and 1.1.13, the functionality we need is actually in 1.1.13 only
	 * @param string $className
	 * @return \CSort|Sort|false - the sorting object. If this is false, it means the sorting is disabled.
	 */
	public function getSort($className = 'sammaye\mongoyii\Sort')
	{
		if($this->_sort === null){
			$this->_sort = new $className;
			if(($id = $this->getId()) != ''){
				$this->_sort->sortVar = $id . '_sort';
			}
			$this->_sort->modelClass = $this->modelClass;
		}
		return $this->_sort;
	}
}