Multi-Tenant-Storage

Author: SatansFosterSon

src/Oauth2CredentialManagers/CacheStore.php

@@ -40,9 +40,18 @@ public function getRefreshToken(): string
         return $this->data('refresh_token');
     }
 
-    public function getTenantId(): string
+    public function getTenants(): ?array
     {
-        return $this->data('tenant_id');
+        return $this->data('tenants');
+    } 
+
+    public function getTenantId(int $tenant =0): string
+    {
+        if(!isset($this->data('tenants')[$tenant]))
+        {
+            throw new \Exception("No such tenant exists");
+        }
+        return $this->data('tenants')[$tenant]['Id'];
     }
 
     public function getExpires(): int
@@ -88,14 +97,14 @@ public function refresh(): void
         $this->store($newAccessToken);
     }
 
-    public function store(AccessTokenInterface $token, string $tenantId = null): void
+    public function store(AccessTokenInterface $token, array $tenants = null): void
     {
         $this->cache->forever($this->cacheKey, [
             'token'         => $token->getToken(),
             'refresh_token' => $token->getRefreshToken(),
             'id_token'      => $token->getValues()['id_token'],
             'expires'       => $token->getExpires(),
-            'tenant_id'     => $tenantId ?? $this->getTenantId()
+            'tenants'       => $tenants ?? $this->getTenants()
         ]);
     }
 

src/Oauth2CredentialManagers/FileStore.php

@@ -42,9 +42,18 @@ public function getRefreshToken(): string
         return $this->data('refresh_token');
     }
 
-    public function getTenantId(): string
+    public function getTenants(): ?array
     {
-        return $this->data('tenant_id');
+        return $this->data('tenants');
+    } 
+
+    public function getTenantId(int $tenant =0): string
+    {
+        if(!isset($this->data('tenants')[$tenant]))
+        {
+            throw new \Exception("No such tenant exists");
+        }
+        return $this->data('tenants')[$tenant]['Id'];
     }
 
     public function getExpires(): int
@@ -90,14 +99,14 @@ public function refresh(): void
         $this->store($newAccessToken);
     }
 
-    public function store(AccessTokenInterface $token, string $tenantId = null): void
+    public function store(AccessTokenInterface $token, array $tenants = null): void
     {
         $ret = $this->disk->put($this->filePath, json_encode([
             'token'         => $token->getToken(),
             'refresh_token' => $token->getRefreshToken(),
             'id_token'      => $token->getValues()['id_token'],
             'expires'       => $token->getExpires(),
-            'tenant_id'     => $tenantId ?? $this->getTenantId()
+            'tenants'       => $tenants ?? $this->getTenants()
         ]), 'private');
 
         if ($ret === false) {

src/OauthCredentialManager.php

@@ -21,10 +21,15 @@ public function getRefreshToken(): string;
      */
     public function getAuthorizationUrl(): string;
 
+    /**
+     * Get all tenants available
+    **/
+    public function getTenants(): ?array;
+
     /**
      * Get the current tenant ID
      */
-    public function getTenantId(): string;
+    public function getTenantId(int $tenant =0): string;
 
     /**
      * Get the time the current access token expires (unix timestamp)
@@ -59,13 +64,13 @@ public function refresh(): void;
      *   'refresh_token' => $token->getRefreshToken(),
      *   'id_token'      => $token->getValues()['id_token'],
      *   'expires'       => $token->getExpires(),
-     *   'tenant_id'     => $tenantId ?? $this->getTenantId(),
+     *   'tenants'       => $tenants ?? $this->getTenants(),
      * ]
      *
      * @param AccessTokenInterface $token
-     * @param string|null          $tenantId
+     * @param Array|null          $tenants
      */
-    public function store(AccessTokenInterface $token, string $tenantId = null): void;
+    public function store(AccessTokenInterface $token, array $tenants = null): void;
 
     /**
      * Get the current authenticated users details according to the id token
@@ -87,9 +92,9 @@ public function getUser(): ?array;
      *   'refresh_token' => 'string',
      *   'id_token'      => 'string',
      *   'expires'       => 000000,
-     *   'tenant_id'     => 'string',
+     *   'tenants'     => 'array',
      * ]
      */
     public function getData(): array;
 
-}
+}