Making sure that any custom controllers are included too.

pwweb · Nov 4, 2020 · ed62026 · ed62026
1 parent bd312c7
commit ed62026
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/Generators/Scaffold/RoutesGenerator.php b/src/Generators/Scaffold/RoutesGenerator.php
@@ -260,6 +260,7 @@ private function buildText(array $routes, int $indent = 0, string $parent = '')
                         '$ITERATION_CUSTOM_NAME$' => $custom['name'],
                         '$INDENT$' => infy_tabs($tabs),
                     ];
+                    $this->classNames[$custom['controller'].'Controller'] = $custom['controller'].'Controller';
                     $templateString .= get_artomator_template('scaffold.routes.prefixed.custom');
                     $templateString = fill_template($vars, $templateString);
                 }