1:   2:   3:   4:   5:   6:   7:   8:   9:  10:  11:  12:  13:  14:  15:  16:  17:  18:  19:  20:  21:  22:  23:  24:  25:  26:  27:  28:  29:  30:  31:  32:  33:  34:  35:  36:  37:  38:  39:  40:  41:  42:  43:  44:  45:  46:  47:  48:  49:  50:  51:  52:  53:  54:  55:  56:  57:  58:  59:  60:  61:  62:  63:  64:  65:  66:  67:  68:  69:  70:  71:  72:  73:  74:  75:  76:  77:  78:  79:  80:  81:  82:  83:  84:  85:  86:  87:  88:  89:  90:  91:  92:  93:  94:  95:  96:  97:  98:  99: 100: 

<?php

/**
 * This file is part of the LeanMapperQuery extension
 * for the Lean Mapper library (http://leanmapper.com)
 * Copyright (c) 2013 Michal Bohuslávek
 */

namespace LeanMapperQuery;

use LeanMapper;
use LeanMapper\Filtering;
use LeanMapper\Fluent;
use LeanMapper\Reflection\Property;
use LeanMapper\Relationship;
use LeanMapperQuery\Caller;
use LeanMapperQuery\IQuery;
use LeanMapperQuery\Exception\InvalidArgumentException;
use LeanMapperQuery\Exception\InvalidMethodCallException;
use LeanMapperQuery\Exception\InvalidRelationshipException;
use LeanMapperQuery\Exception\InvalidStateException;
use LeanMapperQuery\Exception\MemberAccessException;

/**
 * @author Michal Bohuslávek
 */
class Entity extends LeanMapper\Entity
{
    /** @var array */
    protected static $magicMethodsPrefixes = array();


    protected function queryProperty($field, IQuery $query)
    {
        if ($this->isDetached()) {
            throw new InvalidStateException('Cannot query detached entity.');
        }
        $property = $this->getCurrentReflection()->getEntityProperty($field);
        if ($property === NULL) {
            throw new MemberAccessException("Cannot access undefined property '$field' in entity " . get_called_class() . '.');
        }
        if (!$property->hasRelationship()) {
            throw new InvalidArgumentException("Property '{$property->getName()}' in entity ". get_called_class() . " has no relationship.");
        }
        $class = $property->getType();
        $filters = $this->createImplicitFilters($class, new Caller($this, $property))->getFilters();
        $mapper = $this->mapper;
        $filters[] = function (Fluent $fluent) use ($mapper, $query) {
            $query->applyQuery($fluent, $mapper);
        };

        $relationship = $property->getRelationship();
        if ($relationship instanceof Relationship\BelongsToMany) {
            $targetTable = $relationship->getTargetTable();
            $referencingColumn = $relationship->getColumnReferencingSourceTable();
            $rows = $this->row->referencing($targetTable, $referencingColumn, new Filtering($filters));

        } elseif ($relationship instanceof Relationship\HasMany) {
            $relationshipTable = $relationship->getRelationshipTable();
            $sourceReferencingColumn = $relationship->getColumnReferencingSourceTable();
            $targetReferencingColumn = $relationship->getColumnReferencingTargetTable();
            $targetTable = $relationship->getTargetTable();
            $rows = array();
            foreach ($this->row->referencing($relationshipTable, $sourceReferencingColumn) as $relationship) {
                $row = $relationship->referenced($targetTable, $targetReferencingColumn, new Filtering($filters));
                $row !== NULL && $rows[] = $row;
            }
        } else {
            throw new InvalidRelationshipException('Only BelongsToMany and HasMany relationships are supported when querying entity property. ' . get_class($relationship) . ' given.');
        }
        $entities = array();
        foreach ($rows as $row) {
            $entity = $this->entityFactory->createEntity($class, $row);
            $entity->makeAlive($this->entityFactory);
            $entities[] = $entity;
        }
        return $entities;
    }

    public function __call($name, array $arguments)
    {
        if (preg_match('#^('.implode('|', static::$magicMethodsPrefixes).')(.+)$#', $name, $matches)) {
            if (count($arguments) !== 1) {
                throw new InvalidMethodCallException(get_called_class() . "::$name expects exactly 1 argument. " . count($arguments) . ' given.');
            }
            list($query) = $arguments;
            if (!$query instanceof IQuery) {
                throw new InvalidArgumentException('Argument 1 passed to ' . get_called_class() . "::$name must implement interface LeanMapperQuery\\IQuery. " . gettype($query) . ' given.');
            }
            list(, $method, $field) = $matches;
            $field = lcfirst($field);
            return $this->$method($field, $query);

        } else {
            return parent::__call($name, $arguments);
        }
    }

}