super_find.php | searchcode

/models/behaviors/super_find.php

http://github.com/jrbasso/super_find
PHP | 277 lines | 216 code | 11 blank | 50 comment | 46 complexity | c5536ff152160444a0c91ec4f42e96d0 MD5 | raw file

<?php

/**

 * Super Find Behavior

 *

 * Licensed under The MIT License

 * Redistributions of files must retain the above copyright notice.

 *

 * @link          http://github.com/jrbasso/super_find

 * @package       super_find

 * @subpackage    super_find.models.behaviors

 * @since         SuperFind v0.1

 * @license       MIT License (http://www.opensource.org/licenses/mit-license.php)

 */

class SuperFindBehavior extends ModelBehavior {



/**

 * Makes the find with the possibility of using conditions of all relationships

 *

 * @access public

 * @param object $Model Pointer to model

 * @param array $conditions SQL conditions array, or type of find operation (all / first / count /

 *              neighbors / list / threaded)

 * @param mixed $fields Either a single string of a field name, or an array of field names, or

 *               options for matching

 * @param string $order SQL ORDER BY conditions (e.g. "price DESC" or "name ASC")

 * @param integer $recursive The number of levels deep to fetch associated records

 * @return array Array of records

 * @access public

 */

	function superFind(&$Model, $conditions = null, $fields = array(), $order = null, $recursive = null) {

		if (!is_string($conditions)) {

			$type = 'first';

			$query = array_merge(compact('conditions', 'fields', 'order', 'recursive'), array('limit' => 1));

		} else {

			list($type, $query) = array($conditions, $fields);

		}



		$relations = array('belongsTo', 'hasOne', 'hasMany', 'hasAndBelongsToMany');

		$originalRelations = array();

		foreach ($relations as $relation) {

			$originalRelations[$relation] = $Model->$relation;

			$extraFinds[$relation] = array();

		}

		if (isset($query['conditions'])) {

			if (!is_array($query['conditions'])) {

				$query['conditions'] = (array)$query['conditions'];

			}

			foreach ($query['conditions'] as $key => $value) {

				$originalKey = $key;

				$check =& $value;

				if (is_string($key)) {

					$check =& $key;

				}

				if (!preg_match('/^\w+\.\w+(\.\w+)*/', $check, $matches)) {

					continue;

				}

				$modelsName = explode('.', $matches[0]);

				array_pop($matches); // Remove the field name

				if ($modelsName[0] === $Model->alias) {

					if (count($modelsName) > 1) {

						$check = substr($check, strlen($modelsName[0]) + 1);

						unset($modelsName[0]);

					} else {

						continue;

					}

				}

				$mainModel = reset($modelsName);

				if (count($modelsName) < 3 && ($mainModel === $Model->alias || isset($Model->belongsTo[$mainModel]) || isset($Model->hasOne[$mainModel]))) {

					unset($query['conditions'][$originalKey]);

					if ($check === $key) {

						$query['conditions'][$check] = $value;

					} else {

						$query['conditions'][] = $check;

					}

					if ($mainModel !== $Model->alias) {

						if ($Model->Behaviors->attached('Containable')) {

							$query['contain'][] = $mainModel;

							if (isset($query['recursive'])) {

								unset($query['recursive']);

							}

						} else {

							if (isset($query['recursive'])) {

								if ($query['recursive'] < 0) {

									$query['recursive'] = 0;

								} else {

									$query['recursive'] = 0;

								}

							}

						}

					}

					continue;

				}

				if (isset($Model->hasMany[$mainModel], $Model->$mainModel)) {

					$extraFinds['hasMany'][$mainModel][$key] = $value;

					unset($query['conditions'][$originalKey]);

				} elseif (isset($Model->hasAndBelongsToMany[$mainModel], $Model->$mainModel)) {

					$extraFinds['hasAndBelongsToMany'][$mainModel][$key] = $value;

					unset($query['conditions'][$originalKey]);

				} else {

					if (count($modelsName) > 2) {

						if (isset($Model->belongsTo[$mainModel], $Model->$mainModel)) {

							$extraFinds['belongsTo'][$mainModel][$key] = $value;

						} elseif (isset($Model->hasOne[$mainModel], $Model->$mainModel)) {

							$extraFinds['hasOne'][$mainModel][$key] = $value;

						}

						unset($query['conditions'][$originalKey]);

					}

				}

			}

			$this->_subFindBelongsTo($extraFinds['belongsTo'], $Model, $query);

			$this->_subFindHasMany($extraFinds['hasMany'], $Model, $query);

			$this->_subFindHasAndBelongsToMany($extraFinds['hasAndBelongsToMany'], $Model, $query);

		}



		$return = $Model->find($type, $query);

		foreach ($relations as $relation) {

			$Model->$relation = $originalRelations[$relation];

		}



		return $return;

	}



/**

 * Make sub finds to belongsTo assoc

 *

 * @access protected

 * @return void

 */

	function _subFindBelongsTo($data, &$Model, &$query) {

		foreach ($data as $modelName => $extraConditions) {

			$pk = $modelName . '.' . $Model->$modelName->primaryKey;

			$removeBehavior = false;

			if (!$Model->$modelName->Behaviors->attached('SuperFind')) {

				$removeBehavior = true;

				$Model->$modelName->Behaviors->attach('SuperFind.SuperFind');

			}

			$data = $Model->$modelName->superFind('all', array(

				'fields' => array($pk),

				'conditions' => $extraConditions,

				'recursive' => -1

			));

			if ($removeBehavior) {

				$Model->$modelName->Behaviors->detach('SuperFind.SuperFind');

			}

			if (empty($data)) {

				$query['conditions'] = '1 = 0';

				break;

			}

			$masterModelIds = array_unique(Set::extract('{n}.' . $pk, $data));



			$fk = $Model->belongsTo[$modelName]['foreignKey'];

			if (isset($query['conditions'][$fk])) {

				$query['conditions'][$fk] = array_intersect((array)$query['conditions'][$fk], $masterModelIds);

			} else {

				$query['conditions'][$fk] = $masterModelIds;

			}

			$otherModelIds = array_unique(Set::extract('{n}.' . $pk, $data));

			$this->_addCondition($Model, 'belongsTo', $modelName, $pk, $otherModelIds);

		}

	}



/**

 * Make sub finds to hasMany assoc

 *

 * @access protected

 * @return void

 */

	function _subFindHasMany($data, &$Model, &$query) {

		foreach ($data as $modelName => $extraConditions) {

			$fk = $Model->hasMany[$modelName]['foreignKey'];

			$pk = $modelName . '.' . $Model->$modelName->primaryKey;

			$removeBehavior = false;

			if (!$Model->$modelName->Behaviors->attached('SuperFind')) {

				$removeBehavior = true;

				$Model->$modelName->Behaviors->attach('SuperFind.SuperFind');

			}

			$data = $Model->$modelName->superFind('all', array(

				'fields' => array($fk, $pk),

				'conditions' => $extraConditions,

				'recursive' => -1

			));

			if ($removeBehavior) {

				$Model->$modelName->Behaviors->detach('SuperFind.SuperFind');

			}

			$masterModelIds = array_unique(Set::extract('{n}.' . $modelName . '.' . $fk, $data));

			if (empty($masterModelIds)) {

				$query['conditions'] = '1 = 0';

				break;

			}

			$selfPk = $Model->alias . '.' . $Model->primaryKey;

			if (isset($query['conditions'][$selfPk])) {

				$query['conditions'][$selfPk] = array_intersect((array)$query['conditions'][$selfPk], $masterModelIds);

			} else {

				$query['conditions'][$selfPk] = $masterModelIds;

			}

			$otherModelIds = array_unique(Set::extract('{n}.' . $pk, $data));

			$this->_addCondition($Model, 'hasMany', $modelName, $pk, $otherModelIds);

		}

	}



/**

 * Make sub finds to hasAndBelongsToMany assoc

 *

 * @access protected

 * @return void

 */

	function _subFindHasAndBelongsToMany($data, &$Model, &$query) {

		foreach ($data as $modelName => $extraConditions) {

			$pk = $modelName . '.' . $Model->$modelName->primaryKey;

			$removeBehavior = false;

			if (!$Model->$modelName->Behaviors->attached('SuperFind')) {

				$removeBehavior = true;

				$Model->$modelName->Behaviors->attach('SuperFind.SuperFind');

			}

			$data = $Model->$modelName->superFind('all', array(

				'fields' => array($pk),

				'conditions' => $extraConditions,

				'recursive' => -1

			));

			if ($removeBehavior) {

				$Model->$modelName->Behaviors->detach('SuperFind.SuperFind');

			}

			if (empty($data)) {

				$query['conditions'] = '1 = 0';

				break;

			}

			$otherModelIds = array_unique(Set::extract('{n}.' . $pk, $data));



			$relationModel = new Model(array(

				'table' => $Model->hasAndBelongsToMany[$modelName]['joinTable'],

				'ds' => $Model->useDbConfig,

				'name' => 'Relation'

			));

			$relationModel->Behaviors->attach('SuperFind.SuperFind');

			$data = $relationModel->superFind('all', array(

				'fields' => array($Model->hasAndBelongsToMany[$modelName]['foreignKey']),

				'conditions' => array($Model->hasAndBelongsToMany[$modelName]['associationForeignKey'] => $otherModelIds)

			));

			unset($relationModel);

			if (empty($data)) {

				$query['conditions'] = '1 = 0';

				break;

			}

			$masterModelIds = array_unique(Set::extract('{n}.Relation.' . $Model->hasAndBelongsToMany[$modelName]['foreignKey'], $data));



			$selfPk = $Model->alias . '.' . $Model->primaryKey;

			if (isset($query['conditions'][$selfPk])) {

				$query['conditions'][$selfPk] = array_intersect((array)$query['conditions'][$selfPk], $masterModelIds);

			} else {

				$query['conditions'][$selfPk] = $masterModelIds;

			}

			$this->_addCondition($Model, 'hasAndBelongsToMany', $modelName, $pk, $otherModelIds);

		}

	}

/**

 * Add a conditions in model

 *

 * @access protected

 * @return void

 */

	function _addCondition(&$Model, $type, $otherModel, $field, $value) {

		$relation =& $Model->$type;

		if (!empty($relation[$otherModel]['conditions'])) {

			$relation[$otherModel]['conditions'] = array($relation[$otherModel]['conditions']);

		}

		if (isset($relation[$otherModel]['conditions'][$field])) {

			if (!is_array($relation[$otherModel]['conditions'][$field])) {

				$relation[$otherModel]['conditions'][$field] = array($relation[$otherModel]['conditions'][$field]);

			}

		} else {

			$relation[$otherModel]['conditions'][$field] = array();

		}

		$relation[$otherModel]['conditions'][$field] = array_merge($relation[$otherModel]['conditions'][$field], $value);

	}

}



?>