<?php

namespace Redmix0901\Oauth2Sso\Http\Middleware;

use Closure;
use League\OAuth2\Client\Provider\Exception\IdentityProviderException;
use Illuminate\Http\JsonResponse;
use Illuminate\Http\Response;
use Illuminate\Contracts\Config\Repository as Config;
use Illuminate\Support\Facades\Cookie;
use Illuminate\Support\Facades\Event;
use Illuminate\Support\Facades\Request;
use Redmix0901\Oauth2Sso\SingleSignOn;

class OAuth2SsoMiddleware
{
    /**
     *@var string
     */
    const ACTION_REDIRECT_WITH_ALL = 'redirect_with_all';

    /**
     *@var string
     */
    const ACTION_REDIRECT_IF_LOGIN = 'redirect_if_login';

    /**
     *@var string
     */
    const ACTION_CREATE_COOKIE = 'cookie';

    /**
     *@var \Redmix0901\Oauth2Sso\SingleSignOn
     */
    protected $singleSignOn;

    /**
     *
     * @var Illuminate\Contracts\Config\Repository
     */
    protected $config;

    /**
     * OAuth2SsoMiddleware constructor.
     *
     * @param Repository $config
     * @param SingleSignOn $singleSignOn
     */
    public function __construct(SingleSignOn $singleSignOn, Config $config)
    {
        $this->config = $config;
        $this->singleSignOn = $singleSignOn;
    }

    /**
     * Handle an incoming request.
     *
     * @param  \Illuminate\Http\Request  $request
     * @param  \Closure  $next
     * @param  string[]  ...$action
     * @return mixed
     *
     * @throws \IdentityProviderException
     */
    public function handle($request, Closure $next, ...$action)
    {
        if (empty($action)) {
            $action = [null];
        }

        /**
         *
         * @var \League\OAuth2\Client\Token\AccessToken $accessToken
         */
        $accessToken = $this->singleSignOn->getAccessTokenLocal();

        /**
         * Không có $accessToken tồn tại.
         */
        if (!$accessToken) {
            return $this->redirectTo($request, $next, $action);
        }

        try {

            /**
             * Xem $accessToken hết hạn sử dụng chưa.
             * Nếu hết hạn sẽ lấy $accessToken và
             * refresh token mới bằng refresh token hiện tại.
             */
            $accessToken = $this->singleSignOn->refreshTokenIfExpired($accessToken);

            /**
             * Kiểm tra bằng cách lấy resource owner bằng $accessToken.
             */
            $resourceOwner = $this->singleSignOn->getUserByToken($accessToken);

        } catch (IdentityProviderException $e) {

            /**
             * Xóa $accessToken trên session nếu có lỗi.
             */
            $this->singleSignOn->deleteAccessTokenLocal();

            return $this->redirectTo($request, $next, $action);
        }

        $user = $resourceOwner->toArray();

        if (isset($user['message']) && $user['message'] == 'Unauthenticated.') {
            /**
             * Xóa $accessToken trên session nếu có lỗi.
             */
            $this->singleSignOn->deleteAccessTokenLocal();

            return $this->redirectTo($request, $next, $action);
        }

        $request->attributes->add(['oauth2_user' => $resourceOwner]);

        return $this->sendReponse($request, $next, $action);
    }

    /**
     * Trả về reponse sau khi tạo mới token.
     *
     * Token sẽ được gắn trên cookie nếu có yêu cầu và hợp lệ.
     *
     * @param  \Illuminate\Http\Request  $request
     * @param  \Illuminate\Http\Response  $response
     * @param  string[]  ...$action
     * @return \Illuminate\Http\Response
     */
    protected function sendReponse($request, $next, $action)
    {
        if ($this->shouldReceiveFreshAccessToken($request, $next, $action)) {

            $config = $this->config->get('session');

            /**
             *
             * @var \League\OAuth2\Client\Token\AccessToken $accessToken
             */
            $accessToken = $this->singleSignOn->getAccessTokenLocal();

            return $next($request)->withCookie(
                cookie(
                    SingleSignOn::cookie(),
                    $accessToken->getToken(),
                    $config['lifetime']
                )
            );
        } elseif (Request::hasCookie(SingleSignOn::cookie())) {
            Cookie::queue(Cookie::forget(SingleSignOn::cookie()));
        }

        return $next($request);
    }

    /**
     * Quyết định xem có được gắn token vào cookie hay không.
     *
     * @param  \Illuminate\Http\Request  $request
     * @param  \Illuminate\Http\Response  $response
     * @param  string[]  ...$action
     * @return bool
     */
    protected function shouldReceiveFreshAccessToken($request, $next, $action)
    {
        /**
         *
         * @var \League\OAuth2\Client\Token\AccessToken $accessToken
         */
        $accessToken = $this->singleSignOn->getAccessTokenLocal();

        return in_array(self::ACTION_CREATE_COOKIE, $action)
                    && $request->isMethod('GET')
                    && !empty($accessToken)
                    && !$accessToken->hasExpired();
    }

    /**
     * Kiểm tra token có tồn tại trên cookie chưa.
     *
     * @param  \Illuminate\Http\Response  $response
     * @return bool
     */
    protected function alreadyContainsToken($response)
    {
        foreach ($response->headers->getCookies() as $cookie) {
            if ($cookie->getName() === SingleSignOn::cookie()) {
                return true;
            }
        }

        return false;
    }

    /**
     * Redirect về Auth nếu có yêu cầu.
     *
     * Hoặc tiếp tục request với tạo token trên cookie.
     *
     * @param  \Illuminate\Http\Request  $request
     * @param  \Illuminate\Http\Response  $response
     * @param  string[]  ...$action
     * @return redirect
     */
    protected function redirectTo($request, $next, $action)
    {
        /**
         * Nếu có ACTION_REDIRECT_WITH_ALL trong $action thì sẽ redirect về Server Auth,
         *
         * kể cả đã hoặc chưa đăng nhập Server Auth.
         */
        if (in_array(self::ACTION_REDIRECT_WITH_ALL, $action)) {
            return $this->singleSignOn->getAuthRedirect();
        }

        /**
         * Nếu có ACTION_REDIRECT_IF_LOGIN trong $action thì sẽ redirect về Server Auth,
         *
         * khi đã đăng nhập Server Auth.
         */
        if (in_array(self::ACTION_REDIRECT_IF_LOGIN, $action)) {
            if ($this->singleSignOn->checkCookie()) {
                return $this->singleSignOn->getAuthRedirect();
            }
        }

        return $this->sendReponse($request, $next, $action);
    }
}