Add setup callback function

Author: Michael

src/BaseRepository.php

@@ -11,6 +11,7 @@ class BaseRepository extends EntityRepository
 	protected $namedParamCounter = 0;
 	protected $joins = array();
 	protected $disableJoins = false;
+	protected $setupFunction = null;
 
 	public function disableJoins(bool $disableJoins)
 	{
@@ -31,7 +32,13 @@ public function countRows(string $column, array $filters = array())
 		$query = $queryBuilder->getQuery();
 
 		return $query->getSingleScalarResult();
+	}
+
+	public function setup(callable $callback)
+	{
+		$this->setupFunction = $callback;
 
+		return $this;
 	}
 
 	public function findFiltered(array $filters = array(), $order = array(), $limit = null, $offset = 0)
@@ -66,6 +73,10 @@ public function buildQuery(array $filters = array(), $order = array(), $limit =
 				$this->alias
 			));
 
+		// Got a setup function?
+		if(is_callable($this->setupFunction))
+			$queryBuilder = call_user_func($this->setupFunction, $this->alias, $queryBuilder);
+
 		// Defaults options
 		$opt = array_merge(array(
 			'disable_joins' => false