Added option to easily select different modules

graphpql · Mar 16, 2024 · 3d31e9b · 3d31e9b
1 parent 328c03e
commit 3d31e9b
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/ApiPresenter.php b/src/ApiPresenter.php
@@ -6,7 +6,7 @@
 
 use \Nette\Application\Responses\TextResponse;
 
-final class ApiPresenter implements \Nette\Application\IPresenter
+class ApiPresenter implements \Nette\Application\IPresenter
 {
     private \Graphpinator\Graphpinator $graphpinator;
 
@@ -23,10 +23,7 @@ public function __construct(
             $debugMode
                 ? \Graphpinator\ErrorHandlingMode::OUTPUTABLE
                 : \Graphpinator\ErrorHandlingMode::ALL,
-            new \Graphpinator\Module\ModuleSet([
-                //new \Graphpinator\QueryCost\MaxDepthModule(15),
-                new \Graphpinator\PersistedQueries\PersistedQueriesModule($schema, $this->cache),
-            ]),
+            $this->getEnabledModules(),
             new \Graphpinator\Nette\TracyLogger(),
         );
     }
@@ -40,6 +37,14 @@ public function run(\Nette\Application\Request $request) : \Nette\Application\Re
         };
     }
 
+    protected function getEnabledModules() : \Graphpinator\Module\ModuleSet
+    {
+        return new \Graphpinator\Module\ModuleSet([
+            //new \Graphpinator\QueryCost\MaxDepthModule(15),
+            new \Graphpinator\PersistedQueries\PersistedQueriesModule($schema, $this->cache),
+        ]);
+    }
+
     private function createPreflightResponse() : TextResponse
     {
         $this->response->setHeader('Access-Control-Allow-Credentials', 'true');