Spaces:
No application file
No application file
File size: 3,183 Bytes
d2897cd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 |
<?php
declare(strict_types=1);
namespace Mautic\IntegrationsBundle\Auth\Support\Oauth2\Token;
use kamermans\OAuth2\Persistence\TokenPersistenceInterface;
use kamermans\OAuth2\Token\TokenInterface;
use Mautic\IntegrationsBundle\Exception\IntegrationNotSetException;
use Mautic\IntegrationsBundle\Helper\IntegrationsHelper;
use Mautic\PluginBundle\Entity\Integration;
class TokenPersistence implements TokenPersistenceInterface
{
private ?Integration $integration = null;
public function __construct(
private IntegrationsHelper $integrationsHelper
) {
}
/**
* Restore the token data into the give token.
*
* @param TokenInterface|IntegrationToken $token
*
* @return TokenInterface|IntegrationToken Restored token
*/
public function restoreToken(TokenInterface $token): TokenInterface
{
$apiKeys = $this->getIntegration()->getApiKeys();
$apiKeys['expires_at'] ??= null;
return new IntegrationToken(
empty($apiKeys['access_token']) ? null : $apiKeys['access_token'],
empty($apiKeys['refresh_token']) ? null : $apiKeys['refresh_token'],
$apiKeys['expires_at'] ?? null
);
}
/**
* Save the token data.
*
* @param TokenInterface|IntegrationToken $token
*/
public function saveToken(TokenInterface $token): void
{
$integration = $this->getIntegration();
$oldApiKeys = $integration->getApiKeys();
if (null === $oldApiKeys) {
$oldApiKeys = [];
}
$newApiKeys = [
'access_token' => $token->getAccessToken(),
'refresh_token' => $token->getRefreshToken(),
'expires_at' => $token->getExpiresAt(),
];
$extraData = $token instanceof IntegrationToken ? $token->getExtraData() : [];
$newApiKeys = array_merge($oldApiKeys, $extraData, $newApiKeys);
$integration->setApiKeys($newApiKeys);
$this->integrationsHelper->saveIntegrationConfiguration($integration);
}
/**
* Delete the saved token data.
*/
public function deleteToken(): void
{
$integration = $this->getIntegration();
$apiKeys = $integration->getApiKeys();
// Must delete both the access token and the expiration in order for the middleware to refresh
unset($apiKeys['access_token'], $apiKeys['expires_at']);
$integration->setApiKeys($apiKeys);
$this->integrationsHelper->saveIntegrationConfiguration($integration);
}
/**
* Returns true if a token exists (although it may not be valid).
*/
public function hasToken(): bool
{
return !empty($this->getIntegration()->getApiKeys()['access_token']);
}
public function setIntegration(Integration $integration): void
{
$this->integration = $integration;
}
/**
* @throws IntegrationNotSetException
*/
private function getIntegration(): Integration
{
if ($this->integration) {
return $this->integration;
}
throw new IntegrationNotSetException('Integration not set');
}
}
|