/** * @inheritDoc IElementType::modifyElementsQuery() * * @param DbCommand $query * @param ElementCriteriaModel $criteria * * @return mixed */ public function modifyElementsQuery(DbCommand $query, ElementCriteriaModel $criteria) { $query->addSelect('assetfiles.sourceId, assetfiles.folderId, assetfiles.filename, assetfiles.kind, assetfiles.width, assetfiles.height, assetfiles.size, assetfiles.dateModified, assetfolders.path as folderPath')->join('assetfiles assetfiles', 'assetfiles.id = elements.id')->join('assetfolders assetfolders', 'assetfolders.id = assetfiles.folderId'); if (!empty($criteria->source)) { $query->join('assetsources assetsources', 'assetfiles.sourceId = assetsources.id'); } if ($criteria->sourceId) { $query->andWhere(DbHelper::parseParam('assetfiles.sourceId', $criteria->sourceId, $query->params)); } if ($criteria->source) { $query->andWhere(DbHelper::parseParam('assetsources.handle', $criteria->source, $query->params)); } if ($criteria->folderId) { if ($criteria->includeSubfolders) { $folders = craft()->assets->getAllDescendantFolders(craft()->assets->getFolderById($criteria->folderId)); $query->andWhere(DbHelper::parseParam('assetfiles.folderId', array_keys($folders), $query->params)); } else { $query->andWhere(DbHelper::parseParam('assetfiles.folderId', $criteria->folderId, $query->params)); } } if ($criteria->filename) { $query->andWhere(DbHelper::parseParam('assetfiles.filename', $criteria->filename, $query->params)); } if ($criteria->kind) { if (is_array($criteria->kind)) { $query->andWhere(DbHelper::parseParam('assetfiles.kind', array_merge(array('or'), $criteria->kind), $query->params)); } else { $query->andWhere(DbHelper::parseParam('assetfiles.kind', $criteria->kind, $query->params)); } } if ($criteria->width) { $query->andWhere(DbHelper::parseParam('assetfiles.width', $criteria->width, $query->params)); } if ($criteria->height) { $query->andWhere(DbHelper::parseParam('assetfiles.height', $criteria->height, $query->params)); } if ($criteria->size) { $query->andWhere(DbHelper::parseParam('assetfiles.size', $criteria->size, $query->params)); } // Clear out existing onPopulateElements handlers $criteria->detachEventHandler('onPopulateElements', array($this, 'eagerLoadTransforms')); // Are we eager-loading any transforms? if ($criteria->withTransforms) { $criteria->attachEventHandler('onPopulateElements', array($this, 'eagerLoadTransforms')); } }