added row mapping support via setRowMapping() callback and mapping config option

Author: dg

src/Bridges/DatabaseDI/DatabaseExtension.php

@@ -12,6 +12,7 @@
 use Nette\Schema\Expect;
 use Tracy;
 use function is_array, is_string;
+use Nette\Database\Table\ActiveRow;
 
 
 /**
@@ -38,6 +39,10 @@ public function getConfigSchema(): Nette\Schema\Schema
 				'explain' => Expect::bool(true),
 				'reflection' => Expect::string(), // BC
 				'conventions' => Expect::string('discovered'), // Nette\Database\Conventions\DiscoveredConventions
+				'mapping' => Expect::structure([
+					'convention' => Expect::string(''),
+					'tables' => Expect::arrayOf('string', 'string'),
+				])->before(fn($v) => is_string($v) ? ['convention' => $v] : $v),
 				'autowired' => Expect::bool(),
 			]),
 		)->before(fn($val) => is_array(reset($val)) || reset($val) === null
@@ -115,6 +120,15 @@ private function setupDatabase(\stdClass $config, string $name): void
 			$explorer->addSetup('setConventions', [Nette\DI\Helpers::filterArguments([$config->conventions])[0]]);
 		}
 
+		if ($config->mapping->convention || $config->mapping->tables) {
+			$explorer->addSetup('setRowMapping', [
+				new Statement([self::class, 'createRowMapping'], [
+					$config->mapping->convention,
+					(array) $config->mapping->tables,
+				]),
+			]);
+		}
+
 		$builder->addAlias($this->prefix("$name.connection"), $this->prefix($name));
 		$builder->addAlias($this->prefix("$name.context"), $this->prefix($name));
 		$builder->addAlias($this->prefix("$name.explorer"), $this->prefix($name));
@@ -124,4 +138,23 @@ private function setupDatabase(\stdClass $config, string $name): void
 			$builder->addAlias("nette.database.$name.context", $this->prefix("$name.explorer"));
 		}
 	}
+
+
+	public static function createRowMapping(string $convention, array $tables): \Closure
+	{
+		return static function (string $table) use ($convention, $tables): string {
+			if (isset($tables[$table])) {
+				return $tables[$table];
+			}
+
+			if ($convention !== '') {
+				$class = str_replace('*', str_replace(' ', '', ucwords(strtr($table, '_', ' '))), $convention);
+				if (class_exists($class)) {
+					return $class;
+				}
+			}
+
+			return ActiveRow::class;
+		};
+	}
 }

src/Database/Explorer.php

@@ -7,22 +7,31 @@
 
 namespace Nette\Database;
 
-use JetBrains\PhpStorm\Language;
-use Nette;
-use Nette\Caching\Cache;
-use Nette\Utils\Arrays;
-use function func_get_args, str_replace, ucfirst;
+use Closure;
 
 
 /**
  * Manages database connection and executes SQL queries.
  */
 class Explorer extends Database
 {
+	/** @var ?(Closure(string $table): class-string<Table\ActiveRow>) */
+	private ?Closure $rowMapping = null;
+
+
+	public function setRowMapping(?Closure $factory): void
+	{
+		$this->rowMapping = $factory;
+	}
+
+
 	/** @internal */
 	public function createActiveRow(Table\Selection $selection, array $row): Table\ActiveRow
 	{
-		return new Table\ActiveRow($row, $selection);
+		$class = $this->rowMapping
+			? ($this->rowMapping)($selection->getName())
+			: Table\ActiveRow::class;
+		return new $class($row, $selection);
 	}