fixes #547 - ensures Grant Type identifier is passed to tokencontroller

bshaffer · May 9, 2015 · 49bc005 · 49bc005 · martyncrow · May 11, 2015
1 parent 1f91815
commit 49bc005
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/OAuth2/Server.php b/src/OAuth2/Server.php
@@ -348,17 +348,17 @@ public function getAccessTokenData(RequestInterface $request, ResponseInterface
         return $value;
     }
 
-    public function addGrantType(GrantTypeInterface $grantType, $key = null)
+    public function addGrantType(GrantTypeInterface $grantType, $identifier = null)
     {
-        if (is_string($key)) {
-            $this->grantTypes[$key] = $grantType;
-        } else {
-            $this->grantTypes[$grantType->getQuerystringIdentifier()] = $grantType;
+        if (!is_string($identifier)) {
+            $identifier = $grantType->getQuerystringIdentifier();
         }
 
+        $this->grantTypes[$identifier] = $grantType;
+
         // persist added grant type down to TokenController
         if (!is_null($this->tokenController)) {
-            $this->getTokenController()->addGrantType($grantType);
+            $this->getTokenController()->addGrantType($grantType, $identifier);
         }
     }