fix: add support for decoration via RequestHandlerMiddlewareAdapter. s…

…ilverstripe#36
charles-ritchie · Feb 9, 2021 · 1e8156a · 1e8156a
1 parent f356413
commit 1e8156a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/Controller.php b/src/Controller.php
@@ -6,6 +6,7 @@
 use SilverStripe\Control\HTTPRequest;
 use SilverStripe\GraphQL\Controller as GraphQLController;
 use SilverStripe\Control\Director;
+use SilverStripe\Control\Middleware\RequestHandlerMiddlewareAdapter;
 use SilverStripe\Core\Injector\Injector;
 use SilverStripe\Core\Injector\InjectorNotFoundException;
 use SilverStripe\Core\Path;
@@ -101,6 +102,12 @@ protected function findAvailableRoutes($schemas = []): array
 
             try {
                 $routeController = Injector::inst()->get($routeClass);
+
+                // Add support for decoration via RequestHandlerMiddlewareAdapter
+                if ($routeController instanceof RequestHandlerMiddlewareAdapter) {
+                    $routeController = $routeController->getRequestHandler();
+                }
+
                 if ($routeController instanceof GraphQLController) {
                     $schemaKey = class_exists(Schema::class)
                         ? $routeController->getSchema()->getSchemaKey()