SimpleSAML\Module\authYubiKey\Controller\Yubikey

Complexity

Total Complexity

8

Size/Duplication

Total Lines	73
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	main()	0	32	6
A	setYubikey()	0	3	1

<?php

declare(strict_types=1);

namespace SimpleSAML\Module\authYubiKey\Controller;

use SimpleSAML\Configuration;
use SimpleSAML\Error;
use SimpleSAML\Module\authYubiKey\Auth\Source;
use SimpleSAML\Session;
use SimpleSAML\XHTML\Template;
use Symfony\Component\HttpFoundation\Request;

/**
 * Controller class for the authyubikey module.
 *
 * This class serves the different views available in the module.
 *
 * @package simplesamlphp/simplesamlphp-module-authyubikey
 */
class Yubikey
{
    /**
     * @var \SimpleSAML\Module\authYubiKey\Auth\Source\YubiKey|string
     * @psalm-var \SimpleSAML\Module\authYubiKey\Auth\Source\YubiKey|class-string
     */
    protected $yubikey = Source\YubiKey::class;


    /**
     * Controller constructor.
     *
     * It initializes the global configuration and session for the controllers implemented here.
     *
     * @param \SimpleSAML\Configuration $config The configuration to use by the controllers.
     * @param \SimpleSAML\Session $session The session to use by the controllers.
     *
     * @throws \Exception
     */
    public function __construct(
        protected Configuration $config,
        protected Session $session,
    ) {
    }


    /**
     * Inject the \SimpleSAML\Module\authYubiKey\Auth\Source\YubiKey dependency.
     *
     * @param \SimpleSAML\Module\authYubiKey\Auth\Source\YubiKey $yubikey
     */
    public function setYubikey(Source\YubiKey $yubikey): void
    {
        $this->yubikey = $yubikey;
    }


    /**
     * @param \Symfony\Component\HttpFoundation\Request $request The current request.
     * @return \SimpleSAML\XHTML\Template
     */
    public function main(Request $request): Template
    {
        if (!$request->query->has('AuthState')) {
            throw new Error\BadRequest('Missing AuthState parameter.');
        }
        $stateId = $request->query->all()['AuthState'];

        // attempt to log in
        $t = new Template($this->config, 'authYubiKey:yubikeylogin.twig');

        $errorCode = null;
        $otp = $request->request->has('otp') ? $request->request->all()['otp'] : null;
        if ($otp !== null) {
            // attempt to log in

            /** @psalm-var string $errorCode */
            $errorCode = $this->yubikey::handleLogin($stateId, $otp);
            $errorCodes = (new Error\ErrorCodes())->getAllMessages();

            if (array_key_exists($errorCode, $errorCodes['title'])) {
                $t->data['errorTitle'] = $errorCodes['title'][$errorCode];
            }

            if (array_key_exists($errorCode, $errorCodes['descr'])) {
                $t->data['errorDesc'] = $errorCodes['descr'][$errorCode];
            }
        }

        $t->data['errorCode'] = $errorCode;
        $t->data['stateParams'] = ['AuthState' => $stateId];

        return $t;
    }
}