<?php

namespace TwoFAS\TwoFAS\Authentication;

use TwoFAS\TwoFAS\Exceptions\Response\Errors;
use TwoFAS\TwoFAS\Exceptions\User_Not_Found_Exception;
use TwoFAS\TwoFAS\Http\JSON_Response;
use TwoFAS\TwoFAS\Http\Redirection_Response;
use TwoFAS\TwoFAS\Http\View_Response;
use WP_Error;
use WP_User;

/**
 * This class is responsible for generating a new step token.
 */
final class Step_Token_Manager extends Login_Handler {

	/**
	 * @var Step_Token
	 */
	private $step_token;

	/**
	 * @param Login_Support $login_support
	 * @param Step_Token    $step_token
	 */
	public function __construct( Login_Support $login_support, Step_Token $step_token ) {
		parent::__construct( $login_support );
		$this->step_token = $step_token;
	}

	/**
	 * @param null|WP_Error|WP_User $user
	 *
	 * @return bool
	 */
	public function supports( $user ) {
		return $this->is_wp_user( $user );
	}

	/**
	 * @param null|WP_Error|WP_User $user
	 *
	 * @return bool|JSON_Response|Redirection_Response|View_Response
	 */
	protected function handle( $user ) {
		try {
			$this->step_token->generate();
		} catch ( User_Not_Found_Exception $e ) {
			return $this->json_error( Errors::USER_NOT_FOUND_ERROR, 404 );
		}

		return $this->fallback( $user );
	}
}