add RawQueryFunctions

Author: MartkCz

src/Query/RawQueryFactory.php

@@ -10,10 +10,12 @@ final class RawQueryFactory
 {
 
 	private QueryMetadataExtractor $queryMetadataExtractor;
+	private RawQueryFunctions $functions;
 
 	public function __construct(QueryMetadataExtractor $queryMetadataExtractor)
 	{
 		$this->queryMetadataExtractor = $queryMetadataExtractor;
+		$this->functions = new RawQueryFunctions($queryMetadataExtractor);
 	}
 
 	/**
@@ -34,6 +36,9 @@ public function create(string $sql, array $aliases): string
 	 */
 	private function replace(array $matches, array $aliases): string
 	{
+		if ($matches[1] === 'fn') {
+			return $this->replaceFunctions($matches, $aliases);
+		}
 		if (!isset($aliases[$matches[1]])) {
 			throw new InvalidArgumentException(sprintf('Alias %s not set', $matches[1]));
 		}
@@ -44,14 +49,33 @@ private function replace(array $matches, array $aliases): string
 			$this->queryMetadataExtractor->getTableName($aliases[$matches[1]]);
 	}
 
+	/**
+	 * @param mixed[] $matches
+	 * @param string[] $aliases
+	 */
+	private function replaceFunctions(array $matches, array $aliases): string
+	{
+		if (count($matches) !== 4) {
+			throw new InvalidArgumentException('Raw query function must have method');
+		}
+
+		$name = $matches[2];
+		$args = array_map('trim', explode(',', $matches[3]));
+
+		return $this->functions->call($aliases, $name, ...$args);
+	}
+
 	/**
 	 * @param string[] $aliases
 	 */
 	private function buildRegex(array $aliases): string
 	{
+		$aliases[] = 'fn';
 		$group = implode('|', array_map(fn (string $alias) => preg_quote($alias, '#'), $aliases));
 
-		return '#%(' . $group . ')(?:\.(\w+))?#';
+		$args = '(?:\((.*?)\))?';
+
+		return '#%(' . $group . ')(?:\.(\w+)' . $args . ')?#';
 	}
 
 }

src/Query/RawQueryFunctions.php

@@ -0,0 +1,72 @@
+<?php declare(strict_types = 1);
+
+namespace Utilitte\Doctrine\Query;
+
+use InvalidArgumentException;
+use Utilitte\Doctrine\QueryMetadataExtractor;
+
+final class RawQueryFunctions
+{
+
+	private QueryMetadataExtractor $queryMetadataExtractor;
+
+	public function __construct(QueryMetadataExtractor $queryMetadataExtractor)
+	{
+		$this->queryMetadataExtractor = $queryMetadataExtractor;
+	}
+
+	public function call(array $aliases, string $name, string ...$args)
+	{
+		if ($name === 'call' || !method_exists($this, $name)) {
+			throw new InvalidArgumentException(sprintf('Raw query function %s not exists', $name));
+		}
+
+		return $this->$name($aliases, ...$args);
+	}
+
+	public function join(array $aliases, string $field, ?string $alias = null, ?string $sourceAlias = null): string
+	{
+		[$table, $column] = explode('.' , $field);
+		$alias = $alias ? ' ' . $alias : '';
+
+		$metadata = $this->queryMetadataExtractor->getClassMetadata($aliases[$table]);
+		$tableName = $this->queryMetadataExtractor->getTableName($metadata->getAssociationMapping($column)['targetEntity']);
+
+		return sprintf(
+			'%s%s ON %s%s = %s',
+			$tableName,
+			$alias,
+			$sourceAlias ? $sourceAlias . '.' : '',
+			$metadata->getSingleAssociationJoinColumnName($column),
+			$metadata->getSingleAssociationReferencedJoinColumnName($column)
+		);
+	}
+
+	public function joinTable(array $aliases, string $field): string
+	{
+		[$table, $column] = explode('.' , $field);
+
+		$metadata = $this->queryMetadataExtractor->getClassMetadata($aliases[$table]);
+
+		return $this->queryMetadataExtractor->getTableName($metadata->getAssociationMapping($column)['targetEntity']);
+	}
+
+	public function joinSource(array $aliases, string $field): string
+	{
+		[$table, $column] = explode('.' , $field);
+
+		$metadata = $this->queryMetadataExtractor->getClassMetadata($aliases[$table]);
+
+		return $metadata->getSingleAssociationJoinColumnName($column);
+	}
+
+	public function joinTarget(array $aliases, string $field): string
+	{
+		[$table, $column] = explode('.' , $field);
+
+		$metadata = $this->queryMetadataExtractor->getClassMetadata($aliases[$table]);
+
+		return $metadata->getSingleAssociationReferencedJoinColumnName($column);
+	}
+
+}