mirror of
https://github.com/elyby/accounts.git
synced 2024-11-15 17:56:30 +05:30
64 lines
2.1 KiB
PHP
64 lines
2.1 KiB
PHP
<?php
|
|
namespace api\components\OAuth2\Storage;
|
|
|
|
use api\components\OAuth2\Entities\RefreshTokenEntity;
|
|
use common\components\Redis\Key;
|
|
use common\components\Redis\Set;
|
|
use common\models\OauthSession;
|
|
use ErrorException;
|
|
use League\OAuth2\Server\Entity\RefreshTokenEntity as OriginalRefreshTokenEntity;
|
|
use League\OAuth2\Server\Storage\AbstractStorage;
|
|
use League\OAuth2\Server\Storage\RefreshTokenInterface;
|
|
use Yii;
|
|
use yii\helpers\Json;
|
|
|
|
class RefreshTokenStorage extends AbstractStorage implements RefreshTokenInterface {
|
|
|
|
public $dataTable = 'oauth_refresh_tokens';
|
|
|
|
public function get($token) {
|
|
$result = Json::decode((new Key($this->dataTable, $token))->getValue());
|
|
if ($result === null) {
|
|
return null;
|
|
}
|
|
|
|
$entity = new RefreshTokenEntity($this->server);
|
|
$entity->setId($result['id']);
|
|
$entity->setAccessTokenId($result['access_token_id']);
|
|
$entity->setSessionId($result['session_id']);
|
|
|
|
return $entity;
|
|
}
|
|
|
|
public function create($token, $expireTime, $accessToken) {
|
|
$sessionId = $this->server->getAccessTokenStorage()->get($accessToken)->getSession()->getId();
|
|
$payload = Json::encode([
|
|
'id' => $token,
|
|
'access_token_id' => $accessToken,
|
|
'session_id' => $sessionId,
|
|
]);
|
|
|
|
$this->key($token)->setValue($payload);
|
|
$this->sessionHash($sessionId)->add($token);
|
|
}
|
|
|
|
public function delete(OriginalRefreshTokenEntity $token) {
|
|
if (!$token instanceof RefreshTokenEntity) {
|
|
throw new ErrorException('Token must be instance of ' . RefreshTokenEntity::class);
|
|
}
|
|
|
|
$this->key($token->getId())->delete();
|
|
$this->sessionHash($token->getSessionId())->remove($token->getId());
|
|
}
|
|
|
|
public function sessionHash(string $sessionId) : Set {
|
|
$tableName = Yii::$app->db->getSchema()->getRawTableName(OauthSession::tableName());
|
|
return new Set($tableName, $sessionId, 'refresh_tokens');
|
|
}
|
|
|
|
private function key(string $token) : Key {
|
|
return new Key($this->dataTable, $token);
|
|
}
|
|
|
|
}
|