kalanis\OAuth2\Storage\NDB\AccessTokenStorage

Complexity

Total Complexity

12

Size/Duplication

Total Lines	95
Duplicated Lines	0 %

Test Coverage

Coverage

97.44%

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	getScopeTable()	0	3	1
A	__construct()	0	4	1
A	remove()	0	3	1
A	getTable()	0	3	1
A	getValidAccessToken()	0	21	4
A	store()	0	26	4

<?php

namespace kalanis\OAuth2\Storage\NDB;


use DateTime;
use kalanis\OAuth2\Exceptions\InvalidScopeException;
use kalanis\OAuth2\Storage\AccessTokens\AccessToken;
use kalanis\OAuth2\Storage\AccessTokens\IAccessToken;
use kalanis\OAuth2\Storage\AccessTokens\IAccessTokenStorage;
use Nette\Database\Explorer;
use Nette\Database\SqlLiteral;
use Nette\Database\Table\ActiveRow;
use Nette\Database\Table\Selection;
use PDOException;


/**
 * AccessTokenStorage
 * @package kalanis\OAuth2\Storage\NDB
 */
class AccessTokenStorage implements IAccessTokenStorage
{

    public function __construct(
        private readonly Explorer $context,
    )
    {
    }

    /**
     * Store access token
     * @throws InvalidScopeException
     */
    public function store(IAccessToken $accessToken): void
    {
        $connection = $this->context->getConnection();
        $connection->beginTransaction();
        $this->getTable()->insert([
            'access_token' => $accessToken->getAccessToken(),
            'client_id' => $accessToken->getClientId(),
            'user_id' => $accessToken->getUserId(),
            'expires_at' => $accessToken->getExpires(),
        ]);

        try {
            foreach ($accessToken->getScope() as $scope) {
                $this->getScopeTable()->insert([
                    'access_token' => $accessToken->getAccessToken(),
                    'scope_name' => $scope,
                ]);
            }
        } catch (PDOException $e) {
            // MySQL error 1452 - Cannot add or update a child row: a foreign key constraint fails
            if (in_array(1452, (array) $e->errorInfo)) {
                throw new InvalidScopeException;
            }
            throw $e;
        }
        $connection->commit();
    }

    /**
     * Get authorization code table
     * @return Selection<ActiveRow>
     */
    protected function getTable(): Selection
    {
        return $this->context->table('oauth_access_token');
    }

    /**
     * Get scope table
     * @return Selection<ActiveRow>
     */
    protected function getScopeTable(): Selection
    {
        return $this->context->table('oauth_access_token_scope');
    }

    /**
     * Remove access token
     * @param string $token
     */
    public function remove(string $token): void
    {
        $this->getTable()->where(['access_token' => $token])->delete();
    }

    /**
     * Get valid access token
     * @param string $accessToken
     * @throws \Exception
     * @return IAccessToken|null
     */
    public function getValidAccessToken(string $accessToken): ?IAccessToken
    {
        $row = $this->getTable()
            ->where(['access_token' => $accessToken])
            ->where(new SqlLiteral('TIMEDIFF(expires_at, NOW()) >= 0'))
            ->fetch();

        if (!$row) {
            return null;
        }

        $scopes = $this->getScopeTable()
            ->where(['access_token' => $accessToken])
            ->fetchPairs('scope_name');

        return new AccessToken(
            strval($row['access_token']),
            new DateTime(strval($row['expires_at'])),
            is_numeric($row['client_id'])? intval($row['client_id']) : strval($row['client_id']),
            is_null($row['user_id']) ? null : strval($row['user_id']),
            array_map('strval', array_keys($scopes))
        );
    }
}