Add SymfonyCacheCredential

Author: phamviet

composer.json

@@ -12,7 +12,8 @@
         "symfony/deprecation-contracts": "^2.1 || ^3.0",
         "symfony/http-client": "^4.4.16 || ^5.1.7 || ^6.0 || ^7.0",
         "symfony/http-client-contracts": "^1.1.8 || ^2.0 || ^3.0",
-        "symfony/service-contracts": "^1.0 || ^2.0 || ^3.0"
+        "symfony/service-contracts": "^1.0 || ^2.0 || ^3.0",
+        "symfony/cache-contracts": "^1.0 || ^2.0 || ^3.0"
     },
     "require-dev": {
     },

composer.lock

@@ -59,7 +59,7 @@ protected function fetchToken(array $scopes, array $params): ?TokenInterface
             ]);
             $result = $response->toArray();
         } catch (DecodingExceptionInterface $e) {
-            $this->logger->info('Failed to decode Credentials.', ['exception' => $e]);
+            $this->logger->info('Failed to decode token response.', ['exception' => $e]);
 
             return null;
         } catch (TransportExceptionInterface|HttpExceptionInterface $e) {

src/Client/AadClient.php

@@ -0,0 +1,40 @@
+<?php
+
+namespace Azure\Identity\Credential;
+
+use Azure\Identity\TokenInterface;
+use Psr\Cache\CacheException;
+use Psr\Log\LoggerInterface;
+use Symfony\Contracts\Cache\CacheInterface;
+use Symfony\Contracts\Cache\ItemInterface;
+
+class SymfonyCacheCredential implements TokenCredentialInterface
+{
+    public function __construct(private TokenCredentialInterface $decorated, private CacheInterface $cache, private ?LoggerInterface $logger = null)
+    {
+    }
+
+    public function getToken(array $scopes, array $options = []): ?TokenInterface
+    {
+        $credential = $this->decorated;
+        $closure = \Closure::fromCallable(static function (ItemInterface $item) use ($scopes, $options, $credential) {
+            $token = $credential->getToken($scopes, $options);
+
+            if (null !== $token && null !== $exp = $token->getExpireDate()) {
+                $item->expiresAt($exp);
+            } else {
+                $item->expiresAfter(0);
+            }
+
+            return $token;
+        });
+
+        try {
+            return $this->cache->get('Azure.Credentials.' . sha1(serialize([$scopes, $options, \get_class($this->decorated)])), $closure);
+        } catch (CacheException $e) {
+            $this->logger?->error('Failed to get Azure credentials from cache.', ['exception' => $e]);
+
+            return $credential->getToken($scopes, $options);
+        }
+    }
+}

src/Credential/SymfonyCacheCredential.php

@@ -8,10 +8,7 @@ class Token implements TokenInterface
 
     private $expireDate;
 
-    /**
-     * @throws \Exception
-     */
-    public function __construct(public string $token, public int $expiresIn)
+    public function __construct(public string $accessToken, public int $expiresIn)
     {
         $this->expireDate = (new \DateTimeImmutable())
             ->add(new \DateInterval(sprintf('PT%dS', $expiresIn)))
@@ -24,8 +21,13 @@ public function isExpired(): bool
     }
 
     /**
-     * @throws \Exception
+     * @return \DateTimeImmutable
      */
+    public function getExpireDate(): \DateTimeImmutable
+    {
+        return $this->expireDate;
+    }
+
     public static function fromArray(array $token): TokenInterface
     {
         return new self($token['access_token'], $token['expires_in']);

src/Token.php

@@ -3,10 +3,10 @@
 namespace Azure\Identity;
 
 /**
- * @property-read string $token
  * @property-read string $accessToken
  */
 interface TokenInterface
 {
     public function isExpired(): bool;
+    public function getExpireDate(): \DateTimeImmutable;
 }