prototype for passing the Client ID

googleapis · Dec 7, 2024 · 2c83660 · 2c83660
1 parent 876f14a
commit 2c83660
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 10 deletions.
diff --git a/src/Credentials/UserRefreshCredentials.php b/src/Credentials/UserRefreshCredentials.php
@@ -141,11 +141,12 @@ public function __construct(
      *     @type string $id_token
      * }
      */
-    public function fetchAuthToken(?callable $httpHandler = null, array $headers = [])
+    public function fetchAuthToken(?callable $httpHandler = null, array $headers = [], ?string $clientId = null)
     {
         return $this->auth->fetchAuthToken(
             $httpHandler,
-            $this->applyTokenEndpointMetrics($headers, $this->isIdTokenRequest ? 'it' : 'at')
+            $this->applyTokenEndpointMetrics($headers, $this->isIdTokenRequest ? 'it' : 'at'),
+            $clientId
         );
     }
 

diff --git a/src/FetchAuthTokenCache.php b/src/FetchAuthTokenCache.php
@@ -80,13 +80,13 @@ public function getFetcher()
      * @return array<mixed> the response
      * @throws \Exception
      */
-    public function fetchAuthToken(?callable $httpHandler = null)
+    public function fetchAuthToken(?callable $httpHandler = null, ?string $clientId = null)
     {
         if ($cached = $this->fetchAuthTokenFromCache()) {
             return $cached;
         }
 
-        $auth_token = $this->fetcher->fetchAuthToken($httpHandler);
+        $auth_token = $this->fetcher->fetchAuthToken($httpHandler, [], $clientId);
 
         $this->saveAuthTokenInCache($auth_token);
 
@@ -235,7 +235,8 @@ public function getUniverseDomain(): string
     public function updateMetadata(
         $metadata,
         $authUri = null,
-        ?callable $httpHandler = null
+        ?callable $httpHandler = null,
+        ?string $clientId = null,
     ) {
         if (!$this->fetcher instanceof UpdateMetadataInterface) {
             throw new \RuntimeException(
@@ -263,7 +264,8 @@ public function updateMetadata(
         $newMetadata = $this->fetcher->updateMetadata(
             $metadata,
             $authUri,
-            $httpHandler
+            $httpHandler,
+            $clientId
         );
 
         if (!$cached && $token = $this->fetcher->getLastReceivedToken()) {

diff --git a/src/OAuth2.php b/src/OAuth2.php
@@ -669,13 +669,13 @@ public function generateCredentialsRequest(?callable $httpHandler = null, array
      *        endpoint request.
      * @return array<mixed> the response
      */
-    public function fetchAuthToken(?callable $httpHandler = null, array $headers = [])
+    public function fetchAuthToken(?callable $httpHandler = null, array $headers = [], string $clientId = null)
     {
         if (is_null($httpHandler)) {
             $httpHandler = HttpHandlerFactory::build(HttpClientCache::getHttpClient());
         }
 
-        $response = $httpHandler($this->generateCredentialsRequest($httpHandler, $headers));
+        $response = $httpHandler($this->generateCredentialsRequest($httpHandler, $headers), ['clientId' => $clientId ]);
         $credentials = $this->parseTokenResponse($response);
         $this->updateToken($credentials);
         if (isset($credentials['scope'])) {

diff --git a/src/UpdateMetadataTrait.php b/src/UpdateMetadataTrait.php
@@ -50,7 +50,8 @@ public function getUpdateMetadataFunc()
     public function updateMetadata(
         $metadata,
         $authUri = null,
-        ?callable $httpHandler = null
+        ?callable $httpHandler = null,
+        ?string $clientId = null,
     ) {
         $metadata_copy = $metadata;
 
@@ -63,7 +64,7 @@ public function updateMetadata(
             // Auth metadata has already been set
             return $metadata_copy;
         }
-        $result = $this->fetchAuthToken($httpHandler);
+        $result = $this->fetchAuthToken($httpHandler, [], $clientId);
         if (isset($result['access_token'])) {
             $metadata_copy[self::AUTH_METADATA_KEY] = ['Bearer ' . $result['access_token']];
         } elseif (isset($result['id_token'])) {