Entity: getCountOf* method

src/LeanMapperWorkflow/Entity.php

@@ -2,7 +2,47 @@
 
 namespace LeanMapperWorkflow;
 
+use LeanMapper\Relationship\HasMany;
+use LeanMapper\Filtering;
+use LeanMapper\Fluent;
+
 
 abstract class Entity extends \LeanMapper\Entity
 {
+
+	/**
+	 * @param string $name
+	 * @param array  $arguments
+	 */
+	public function __call($name, array $arguments)
+	{
+		if (substr($name, 0, 10) === 'getCountOf')
+		{
+			$property = $this->getCurrentReflection()->getEntityProperty(lcfirst(substr($name, 10)));
+
+			if ($property->containsCollection())
+			{
+				$relationship = $property->getRelationship();
+				$table = $relationship instanceof HasMany ? $relationship->getRelationshipTable() : $relationship->getTargetTable();
+				$column = $relationship->getColumnReferencingSourceTable();
+				$filters = array('counter');
+				$propertyFilters = $property->getFilters();
+				if ($propertyFilters !== NULL) {
+					$filters = array_merge($filters, $propertyFilters);
+				}
+
+				$filtering = new Filtering(function(Fluent $statement) use ($column) {
+					$statement->removeClause('select')->select('COUNT(*) [count], %n', $column)
+						->groupBy($column);
+				});
+
+				$rows = $this->row->referencing($table, $column, $filtering);
+
+				return empty($rows) ? 0 : $rows[0]->count;
+			}
+		}
+
+		return parent::__call($name, $arguments);
+	}
+
 }