<?php

namespace TwoFAS\TwoFAS\Http\Middleware;

use TwoFAS\TwoFAS\Exceptions\User_Not_Found_Exception;
use TwoFAS\TwoFAS\Helpers\Flash;
use TwoFAS\TwoFAS\Http\JSON_Response;
use TwoFAS\TwoFAS\Http\Redirection_Response;
use TwoFAS\TwoFAS\Http\Request;
use TwoFAS\TwoFAS\Http\Route;
use TwoFAS\TwoFAS\Http\View_Response;
use TwoFAS\TwoFAS\Notifications\Notification_Repository;
use TwoFAS\TwoFAS\Storage\User_Storage;

class Check_Second_Factor_Enabled extends Middleware {

	/**
	 * @var Request
	 */
	private $request;

	/**
	 * @var Flash
	 */
	private $flash;

	/**
	 * @var Notification_Repository
	 */
	private $notification_repository;

	/**
	 * @var User_Storage
	 */
	private $user_storage;

	/**
	 * @param Request                 $request
	 * @param Flash                   $flash
	 * @param Notification_Repository $notification_repository
	 * @param User_Storage            $user_storage
	 */
	public function __construct(
		Request $request,
		Flash $flash,
		Notification_Repository $notification_repository,
		User_Storage $user_storage
	) {
		$this->request                 = $request;
		$this->flash                   = $flash;
		$this->notification_repository = $notification_repository;
		$this->user_storage            = $user_storage;
	}

	/**
	 * @return JSON_Response|Redirection_Response|View_Response
	 *
	 * @throws User_Not_Found_Exception
	 */
	public function handle() {
		if ( ! $this->user_storage->is_2fa_enabled() ) {
			if ( $this->request->is_ajax() ) {
				return $this->json( array(
					'error' => $this->notification_repository->get_by_key( 'second-factor-status-disabled' ),
				), 403 );
			}

			$this->flash->add_message( 'error', 'second-factor-status-disabled' );

			return $this->redirection( Route::SUBMENU_CHANNEL );
		}

		return $this->next->handle();
	}
}