Merge pull request #18 from plentymarkets/feat/session-token-handler

Credentials handler

src/AmazonPHP/SellingPartner/Configuration.php

@@ -8,55 +8,39 @@
 use Plenty\AmazonPHP\SellingPartner\Exception\InvalidArgumentException;
 use Plenty\AmazonPHP\SellingPartner\STSClient\Credentials;
 
-final class Configuration
+class Configuration
 {
-    private /** [COMPAT] string */ $lwaClientID;
-
-    private /** [COMPAT] string */ $lwaClientSecret;
-
-    private /** [COMPAT] string */ $accessKey;
-
-    private /** [COMPAT] string */ $secretKey;
-
-    private /** [COMPAT] string */ $userAgent;
-
-    private /** [COMPAT] ?string */ $securityToken;
-
-    private /** [COMPAT] string */ $tmpFolderPath;
-
-    private /** [COMPAT] LoggerConfiguration */ $loggerConfiguration;
-
-    private /** [COMPAT] Extensions */ $extensions;
+    /** @var CredentialsHandler  */
+    private CredentialsHandler $credentialsHandler;
 
     public function __construct(
         string $lwaClientID,
         string $lwaClientSecret,
-        string $accessKey,
-        string $secretKey,
-        string $securityToken = null,
+        CredentialsHandler $credentialsHandler,
         Extensions $extensions = null,
         LoggerConfiguration $loggerConfiguration = null
     ) {
+        $this->credentialsHandler = $credentialsHandler;
+
         $this->lwaClientID = $lwaClientID;
         $this->lwaClientSecret = $lwaClientSecret;
-        $this->accessKey = $accessKey;
-        $this->secretKey = $secretKey;
+
         // https://github.com/amzn/selling-partner-api-docs/blob/main/guides/en-US/developer-guide/SellingPartnerApiDeveloperGuide.md#include-a-user-agent-header-in-all-requests
         $this->userAgent = 'Library amazon-php/sp-api-php (language=PHP ' . \phpversion() . '; Platform=' . \php_uname('s') . ' ' . \php_uname('r') . ' ' . \php_uname('m') . ')';
         $this->tmpFolderPath = \sys_get_temp_dir();
         $this->loggerConfiguration = $loggerConfiguration ? $loggerConfiguration : new LoggerConfiguration();
         $this->extensions = $extensions ? $extensions : new Extensions();
-        $this->securityToken = $securityToken;
     }
 
     public static function forIAMUser(string $clientId, string $clientSecret, string $accessKey, string $secretKey) : self
     {
         return new self($clientId, $clientSecret, $accessKey, $secretKey, null);
     }
 
-    public static function forIAMRole(string $clientId, string $clientSecret, Credentials $credentials) : self
+    public static function forIAMRole(array $secrets, STSClient $stsClient, string $roleArn) : self
     {
-        return new self($clientId, $clientSecret, $credentials->accessKeyId(), $credentials->secretAccessKey(), $credentials->sessionToken());
+        $credentialsHandler = new CredentialsHandler($stsClient, $secrets['accessKeyID'], $secrets['secretAccessKey'], $roleArn);
+        return new self($secrets['clientId'], $secrets['clientSecret'], $credentialsHandler);
     }
 
     public function updateIAMRoleCredentials(Credentials $credentials) : self
@@ -80,7 +64,7 @@ public function lwaClientSecret() : string
 
     public function securityToken() : ?string
     {
-        return $this->securityToken;
+        return $this->credentialsHandler->getSecurityToken();
     }
 
     public function apiURL(string $awsRegion) : string
@@ -123,12 +107,12 @@ public function apiHost(string $awsRegion) : string
 
     public function accessKey() : string
     {
-        return $this->accessKey;
+        return $this->credentialsHandler->getAccessKey();
     }
 
     public function secretKey() : string
     {
-        return $this->secretKey;
+        return $this->credentialsHandler->getSecretKey();
     }
 
     public function userAgent() : string

src/AmazonPHP/SellingPartner/CredentialsHandler.php

@@ -0,0 +1,91 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Plenty\AmazonPHP\SellingPartner;
+
+
+use Plenty\AmazonPHP\SellingPartner\STSClient\Credentials;
+
+class CredentialsHandler {
+    /** @var Credentials  */
+    private Credentials $credentials;
+
+    /** @var string  */
+    private string $accessKeyId;
+
+    /** @var string  */
+    private string $secretAccessKey;
+
+    /** @var STSClient  */
+    private STSClient $sts;
+
+    /** @var string  */
+    private string $roleArn;
+
+    /**
+     * @param STSClient $sts
+     * @param string $accessKeyId
+     * @param string $secretAccessKey
+     * @param string $roleArn
+     */
+    public function __construct(STSClient $sts, string $accessKeyId, string $secretAccessKey, string $roleArn)
+    {
+        $this->sts = $sts;
+        $this->accessKeyId = $accessKeyId;
+        $this->secretAccessKey = $secretAccessKey;
+        $this->roleArn = $roleArn;
+    }
+
+    /**
+     * @return string
+     */
+    public function getAccessKey(): string
+    {
+        $this->refreshCredentials();
+        return $this->credentials->accessKeyId();
+    }
+
+    /**
+     * @return string
+     */
+    public function getSecretKey(): string
+    {
+        $this->refreshCredentials();
+        return $this->credentials->secretAccessKey();
+    }
+
+    /**
+     * @return string
+     */
+    public function getSecurityToken(): string
+    {
+        $this->refreshCredentials();
+        return $this->credentials->sessionToken();
+    }
+
+    /**
+     * @return void
+     * @throws Exception\ApiException
+     * @throws \JsonException
+     * @throws \Psr\Http\Client\ClientExceptionInterface
+     */
+    private function refreshCredentials(): void {
+        if($this->areCredentialsValid()) {
+            return;
+        }
+
+        $this->credentials = $this->sts->assumeRole(
+            $this->accessKeyId,
+            $this->secretAccessKey,
+            $this->roleArn
+        );
+    }
+
+    private function areCredentialsValid(): bool {
+        if(empty($this->credentials)) {
+            return false;
+        }
+        return time() < $this->credentials->expiration() - 120; //don't go to the end of life span
+    }
+}