From 6a20483b8714640a0d8f2849756a68bb414c2be4 Mon Sep 17 00:00:00 2001
From: Vladimir Razuvaev <vladimir.razuvaev@gmail.com>
Date: Thu, 27 Jul 2017 16:28:56 +0700
Subject: [PATCH] Server: Ability to use thunk for root value and context

---
 src/Server/Helper.php       | 52 ++++++++++++++++++++++++++++++++-----
 src/Server/ServerConfig.php | 12 ++++-----
 2 files changed, 52 insertions(+), 12 deletions(-)

diff --git a/src/Server/Helper.php b/src/Server/Helper.php
index f3205e0..de15317 100644
--- a/src/Server/Helper.php
+++ b/src/Server/Helper.php
@@ -95,14 +95,16 @@ class Helper
             if (!$doc instanceof DocumentNode) {
                 $doc = Parser::parse($doc);
             }
-            if ($op->isReadOnly() && AST::getOperation($doc, $op->operation) !== 'query') {
+
+            $operationType = AST::getOperation($doc, $op->operation);
+            if ($op->isReadOnly() && $operationType !== 'query') {
                 throw new Error("GET supports only query operation");
             }
 
             $validationErrors = DocumentValidator::validate(
                 $config->getSchema(),
                 $doc,
-                $this->resolveValidationRules($config, $op)
+                $this->resolveValidationRules($config, $op, $doc, $operationType)
             );
 
             if (!empty($validationErrors)) {
@@ -114,8 +116,8 @@ class Helper
                     $promiseAdapter,
                     $config->getSchema(),
                     $doc,
-                    $config->getRootValue(),
-                    $config->getContext(),
+                    $this->resolveRootValue($config, $op, $doc, $operationType),
+                    $this->resolveContextValue($config, $op, $doc, $operationType),
                     $op->variables,
                     $op->operation,
                     $config->getDefaultFieldResolver()
@@ -176,15 +178,17 @@ class Helper
     /**
      * @param ServerConfig $config
      * @param OperationParams $params
+     * @param DocumentNode $doc
+     * @param $operationType
      * @return array
      */
-    private function resolveValidationRules(ServerConfig $config, OperationParams $params)
+    private function resolveValidationRules(ServerConfig $config, OperationParams $params, DocumentNode $doc, $operationType)
     {
         // Allow customizing validation rules per operation:
         $validationRules = $config->getValidationRules();
 
         if (is_callable($validationRules)) {
-            $validationRules = $validationRules($params);
+            $validationRules = $validationRules($params, $doc, $operationType);
 
             if (!is_array($validationRules)) {
                 throw new InvariantViolation(sprintf(
@@ -197,6 +201,42 @@ class Helper
         return $validationRules;
     }
 
+    /**
+     * @param ServerConfig $config
+     * @param OperationParams $params
+     * @param DocumentNode $doc
+     * @param $operationType
+     * @return mixed
+     */
+    private function resolveRootValue(ServerConfig $config, OperationParams $params, DocumentNode $doc, $operationType)
+    {
+        $root = $config->getRootValue();
+
+        if (is_callable($root)) {
+            $root = $root($params, $doc, $operationType);
+        }
+
+        return $root;
+    }
+
+    /**
+     * @param ServerConfig $config
+     * @param OperationParams $params
+     * @param DocumentNode $doc
+     * @param $operationType
+     * @return mixed
+     */
+    private function resolveContextValue(ServerConfig $config, OperationParams $params, DocumentNode $doc, $operationType)
+    {
+        $context = $config->getContext();
+
+        if (is_callable($context)) {
+            $context = $context($params, $doc, $operationType);
+        }
+
+        return $context;
+    }
+
     /**
      * Parses HTTP request and returns GraphQL OperationParams contained in this request.
      * For batched requests it returns an array of OperationParams.
diff --git a/src/Server/ServerConfig.php b/src/Server/ServerConfig.php
index b111e07..6661061 100644
--- a/src/Server/ServerConfig.php
+++ b/src/Server/ServerConfig.php
@@ -23,12 +23,12 @@ class ServerConfig
     private $schema;
 
     /**
-     * @var mixed
+     * @var mixed|callable
      */
     private $context;
 
     /**
-     * @var mixed
+     * @var mixed|callable
      */
     private $rootValue;
 
@@ -63,7 +63,7 @@ class ServerConfig
     private $persistentQueryLoader;
 
     /**
-     * @return mixed
+     * @return mixed|callable
      */
     public function getContext()
     {
@@ -71,7 +71,7 @@ class ServerConfig
     }
 
     /**
-     * @param mixed $context
+     * @param mixed|callable $context
      * @return $this
      */
     public function setContext($context)
@@ -81,7 +81,7 @@ class ServerConfig
     }
 
     /**
-     * @param mixed $rootValue
+     * @param mixed|callable $rootValue
      * @return $this
      */
     public function setRootValue($rootValue)
@@ -91,7 +91,7 @@ class ServerConfig
     }
 
     /**
-     * @return mixed
+     * @return mixed|callable
      */
     public function getRootValue()
     {