RoutingExtension: added support for custom single-route classes. (#162)

Author: vlastavesely

src/Bridges/ApplicationDI/RoutingExtension.php

@@ -21,6 +21,7 @@ class RoutingExtension extends Nette\DI\CompilerExtension
 	public $defaults = [
 		'debugger' => NULL,
 		'routes' => [], // of [mask => action]
+		'routeClass' => NULL,
 		'cache' => FALSE,
 	];
 
@@ -44,8 +45,9 @@ public function loadConfiguration()
 			->setClass(Nette\Application\IRouter::class)
 			->setFactory(Nette\Application\Routers\RouteList::class);
 
+		$routeClass = $config['routeClass'] ?: 'Nette\Application\Routers\Route';
 		foreach ($config['routes'] as $mask => $action) {
-			$router->addSetup('$service[] = new Nette\Application\Routers\Route(?, ?);', [$mask, $action]);
+			$router->addSetup('$service[] = new ' . $routeClass . '(?, ?)', [$mask, $action]);
 		}
 
 		if ($this->name === 'routing') {

tests/Bridges.DI/RoutingExtension.basic.phpt

@@ -14,6 +14,10 @@ use Tester\Assert;
 require __DIR__ . '/../bootstrap.php';
 
 
+class Route extends Nette\Application\Routers\Route
+{}
+
+
 test(function () {
 	$loader = new DI\Config\Loader;
 	$config = $loader->load(Tester\FileMock::create('
@@ -25,13 +29,39 @@ test(function () {
 
 	$compiler = new DI\Compiler;
 	$compiler->addExtension('routing', new RoutingExtension(FALSE));
-	$code = $compiler->addConfig($config)->compile();
+	$code = $compiler->addConfig($config)->setClassName('Container1')->compile();
 	eval($code);
 
-	$container = new Container;
+	$container = new Container1;
 	$router = $container->getService('router');
 	Assert::type(Nette\Application\Routers\RouteList::class, $router);
 	Assert::count(2, $router);
 	Assert::same('index.php', $router[0]->getMask());
 	Assert::same('item/<id>', $router[1]->getMask());
+
+	Assert::type(Nette\Application\Routers\RouteList::class, $router);
+	Assert::type(Nette\Application\Routers\Route::class, $router[0]);
+});
+
+
+test(function () {
+	$loader = new DI\Config\Loader;
+	$config = $loader->load(Tester\FileMock::create('
+	routing:
+		routeClass:
+			Route
+		routes:
+			item/<id>: Homepage:detail
+	', 'neon'));
+
+	$compiler = new DI\Compiler;
+	$compiler->addExtension('routing', new RoutingExtension(FALSE));
+	$code = $compiler->addConfig($config)->setClassName('Container2')->compile();
+	eval($code);
+
+	$container = new Container2;
+	$router = $container->getService('router');
+
+	Assert::type(Nette\Application\Routers\RouteList::class, $router);
+	Assert::type(Route::class, $router[0]);
 });