/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
#include "ProcessLifeManagement_precompiled.h"
#include <platform_impl.h>
#include <IGameFramework.h>
#include "ProcessLifeManagementGem.h"
#include <LyShine/ILyShine.h>
#include <LyShine/Bus/UiTextBus.h>
#include <LyShine/Bus/UiCanvasBus.h>

#include <AzFramework/Input/Devices/Gamepad/InputDeviceGamepad.h>
#include <AzFramework/Input/Devices/Keyboard/InputDeviceKeyboard.h>
#include <AzFramework/Input/Devices/Mouse/InputDeviceMouse.h>
#include <AzFramework/Input/Devices/Touch/InputDeviceTouch.h>

using namespace AzFramework;
using namespace ProcessLifeManagement;

////////////////////////////////////////////////////////////////////////////////////////////////////
ProcessLifeManagementGem::ProcessLifeManagementGem()
    : CryHooksModule()
    , m_pausedCanvasId() // Defaults to AZ::EntityId::InvalidEntityId
{
    ApplicationLifecycleEvents::Bus::Handler::BusConnect();
}

////////////////////////////////////////////////////////////////////////////////////////////////////
ProcessLifeManagementGem::~ProcessLifeManagementGem()
{
    ApplicationLifecycleEvents::Bus::Handler::BusDisconnect();
}

////////////////////////////////////////////////////////////////////////////////////////////////////
void ProcessLifeManagementGem::OnApplicationConstrained(ApplicationLifecycleEvents::Event /*lastEvent*/)
{
    if (m_pausedCanvasId.IsValid())
    {
        // Already paused.
        return;
    }

    if (!(gEnv && gEnv->pGame))
    {
        // Game is not valid
        return;
    }

    // Load/show a modal pause screen.
    if (gEnv && gEnv->pLyShine)
    {
        m_pausedCanvasId = gEnv->pLyShine->LoadCanvas("@assets@/ui/canvases/paused.uicanvas");
    }

    if (!m_pausedCanvasId.IsValid())
    {
        // Unable to load canvas.
        return;
    }

    // Update text to reflect input required to unpause (should be localized).
    const AzFramework::InputDevice* inputDeviceTouch = nullptr;
    AzFramework::InputDeviceRequestBus::EventResult(inputDeviceTouch,
                                                    AzFramework::InputDeviceTouch::Id,
                                                    &AzFramework::InputDeviceRequests::GetInputDevice);
    const char* instructionsText = (inputDeviceTouch && inputDeviceTouch->IsConnected()) ?
                                   "Touch the screen to resume" :
                                   "Press any key or button to resume";

    AZ::Entity* instructionsTextElement = nullptr;
    EBUS_EVENT_ID_RESULT(instructionsTextElement, m_pausedCanvasId, UiCanvasBus, FindElementByName, "InstructionsText");
    if (instructionsTextElement != nullptr && instructionsTextElement->GetId().IsValid())
    {
        EBUS_EVENT_ID(instructionsTextElement->GetId(), UiTextBus, SetText, instructionsText);
    }

    // Pause the game.
    if (gEnv->pGame && gEnv->pGame->GetIGameFramework())
    {
        gEnv->pGame->GetIGameFramework()->PauseGame(true, false);
    }
}

////////////////////////////////////////////////////////////////////////////////////////////////////
void ProcessLifeManagementGem::OnApplicationUnconstrained(ApplicationLifecycleEvents::Event /*lastEvent*/)
{
    if (m_pausedCanvasId.IsValid())
    {
        // Exclusively capture all input.
        InputChannelNotificationBus::Handler::BusConnect();
    }
}

////////////////////////////////////////////////////////////////////////////////////////////////////
void ProcessLifeManagementGem::OnInputChannelEvent(const InputChannel& inputChannel, bool& o_hasBeenConsumed)
{
    // Set this to true so that other listeners will not try and process the event.
    o_hasBeenConsumed = true;

    bool shouldUnpause = false;
    const InputChannelId inputChannelId = inputChannel.GetInputChannelId();
    const InputDeviceId inputDeviceId = inputChannel.GetInputDevice().GetInputDeviceId();
    if (InputDeviceKeyboard::IsKeyboardDevice(inputDeviceId) ||
        InputDeviceTouch::IsTouchDevice(inputDeviceId))
    {
        shouldUnpause = true;
    }
    else if (InputDeviceMouse::IsMouseDevice(inputDeviceId))
    {
        shouldUnpause = AZStd::find(InputDeviceMouse::Button::All.begin(),
                                    InputDeviceMouse::Button::All.end(),
                                    inputChannelId) != InputDeviceMouse::Button::All.end();
    }
    else
    {
        shouldUnpause = AZStd::find(InputDeviceGamepad::Button::All.begin(),
                                    InputDeviceGamepad::Button::All.end(),
                                    inputChannelId) != InputDeviceGamepad::Button::All.end();
    }

    if (shouldUnpause)
    {
        // Unpause the game.
        if (gEnv->pGame && gEnv->pGame->GetIGameFramework())
        {
            gEnv->pGame->GetIGameFramework()->PauseGame(false, false);
        }

        // Stop exclusively capturing input.
        InputChannelNotificationBus::Handler::BusDisconnect();

        // Unload/hide the modal pause screen.
        gEnv->pLyShine->ReleaseCanvas(m_pausedCanvasId, false);
        m_pausedCanvasId.SetInvalid();

        // Set this to false so that other listeners can process the event.
        o_hasBeenConsumed = false;
    }
}

AZ_DECLARE_MODULE_CLASS(ProcessLifeManagement_698948526ada4d13bada933e2d7ee463, ProcessLifeManagement::ProcessLifeManagementGem)