<?php

namespace TwoFAS\TwoFAS\Authentication;

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 TwoFAS\TwoFAS\Http\Session;
use WP_Error;
use WP_User;

/**
 * This class ends a login process.
 * It closes an authentication, destroys a session and resets a step token.
 * It also returns a final response.
 */
final class Login_Cleaner extends Login_Handler {

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

	/**
	 * @var Authenticator
	 */
	private $authenticator;

	/**
	 * @var Session
	 */
	private $session;

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

	/**
	 * @param null|WP_Error|WP_User $user
	 *
	 * @return bool
	 */
	public function supports( $user ) {
		if ( ! $this->login_support->has_final_response() ) {
			return false;
		}

		if ( $this->is_wp_user( $user ) ) {
			return true;
		}

		try {
			$this->get_wp_user();

			return true;
		} catch ( User_Not_Found_Exception $e ) {
			return false;
		}
	}

	/**
	 * @param null|WP_Error|WP_User $user
	 *
	 * @return bool|JSON_Response|Redirection_Response|View_Response
	 */
	protected function handle( $user ) {
		$this->step_token->reset();
		$this->session->destroy();
		$this->authenticator->close_authentication();

		return $this->login_support->get_final_response();
	}
}