Firesphere\GraphQLJWT\Helpers\MemberTokenGenerator

Complexity

Total Complexity

10

Size/Duplication

Total Lines	60
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	getErrorMessage()	0	15	6
A	generateResponse()	0	25	4

<?php declare(strict_types=1);

namespace Firesphere\GraphQLJWT\Helpers;

use App\Users\GraphQL\Types\TokenStatusEnum;
use InvalidArgumentException;
use SilverStripe\Core\Extensible;
use SilverStripe\Security\Member;

/**
 * Generates / Validates a MemberTokenType for graphql responses
 *
 * @mixin Extensible
 */
trait MemberTokenGenerator
{
    /**
     * Humanise error message based on status code
     *
     * @param string $status
     * @return string
     * @throws InvalidArgumentException
     */
    public function getErrorMessage(string $status): string
    {
        switch ($status) {
            case TokenStatusEnum::STATUS_EXPIRED:
                return _t('JWT.STATUS_EXPIRED', 'Token is expired, please renew your token with a refreshToken query');
            case TokenStatusEnum::STATUS_DEAD:
                return _t('JWT.STATUS_DEAD', 'Token is expired, but is too old to renew. Please log in again.');
            case TokenStatusEnum::STATUS_INVALID:
                return _t('JWT.STATUS_INVALID', 'Invalid token provided');
            case TokenStatusEnum::STATUS_BAD_LOGIN:
                return _t('JWT.STATUS_BAD_LOGIN', 'Sorry your email and password combination is rejected');
            case TokenStatusEnum::STATUS_OK:
                return _t('JWT.STATUS_OK', 'Token is ok');
            default:
                throw new InvalidArgumentException("Invalid status");
        }
    }

    /**
     * Generate MemberToken response
     *
     * @param string $status Status code
     * @param Member $member
     * @param string $token
     * @return array Response in format required by MemberToken
     */
    protected function generateResponse(string $status, Member $member = null, string $token = null): array
    {
        // Success response
        if ($status === TokenStatusEnum::STATUS_OK) {
            $response = [
                'Valid'   => true,
                'Member'  => $member && $member->exists() ? $member : null,
                'Token'   => $token,
                'Status'  => $status,
                'Code'    => 200,
                'Message' => $this->getErrorMessage($status),
            ];
        } else {
            $response = [
                'Valid'   => false,
                'Member'  => null,
                'Token'   => $token,
                'Status'  => $status,
                'Code'    => 401,
                'Message' => $this->getErrorMessage($status),
            ];
        }

        $this->extend('updateMemberToken', $response);

It seems like extend() must be provided by classes using this trait. How about adding it as abstract method to this trait?

        return $response;
    }
}