#include <memory>
#include <vector>

#include "Minecraft.Client/net/minecraft/client/Minecraft.h"
#include "Minecraft.Client/net/minecraft/client/multiplayer/MultiPlayerLocalPlayer.h"
#include "Minecraft.Client/Common/Source Files/Tutorial/Tutorial.h"
#include "ChoiceTask.h"
#include "Minecraft.World/net/minecraft/world/level/material/Material.h"
#include "4J_Input.h"
#include "Minecraft.Client/Common/Source Files/Tutorial/Constraints/InputConstraint.h"
#include "Minecraft.Client/Common/Source Files/Tutorial/Tasks/TutorialTask.h"
#include "Minecraft.Client/Linux/Linux_UIController.h"

ChoiceTask::ChoiceTask(
    Tutorial* tutorial, int descriptionId, int promptId /*= -1*/,
    bool requiresUserInput /*= false*/, int iConfirmMapping /*= 0*/,
    int iCancelMapping /*= 0*/,
    eTutorial_CompletionAction cancelAction /*= e_Tutorial_Completion_None*/)
    : TutorialTask(tutorial, descriptionId, false, nullptr, true, false,
                   false) {
    if (requiresUserInput == true) {
        constraints.push_back(new InputConstraint(iConfirmMapping));
        constraints.push_back(new InputConstraint(iCancelMapping));
    }
    m_iConfirmMapping = iConfirmMapping;
    m_iCancelMapping = iCancelMapping;
    m_bConfirmMappingComplete = false;
    m_bCancelMappingComplete = false;

    m_cancelAction = cancelAction;

    m_promptId = promptId;
    tutorial->addMessage(m_promptId);
}

bool ChoiceTask::isCompleted() {
    Minecraft* pMinecraft = Minecraft::GetInstance();

    if (m_bConfirmMappingComplete || m_bCancelMappingComplete) {
        enableConstraints(false, true);
        return true;
    }

    if (ui.GetMenuDisplayed(tutorial->getPad())) {
        // If a menu is displayed, then we use the handleUIInput to complete the
        // task
    } else {
        // If the player is under water then allow all keypresses so they can
        // jump out
        if (pMinecraft->localplayers[tutorial->getPad()]->isUnderLiquid(
                Material::water))
            return false;

        if (!m_bConfirmMappingComplete &&
            InputManager.GetValue(pMinecraft->player->GetXboxPad(),
                                  m_iConfirmMapping) > 0) {
            m_bConfirmMappingComplete = true;
        }
        if (!m_bCancelMappingComplete &&
            InputManager.GetValue(pMinecraft->player->GetXboxPad(),
                                  m_iCancelMapping) > 0) {
            m_bCancelMappingComplete = true;
        }
    }

    if (m_bConfirmMappingComplete || m_bCancelMappingComplete) {
        enableConstraints(false, true);
    }
    return m_bConfirmMappingComplete || m_bCancelMappingComplete;
}

eTutorial_CompletionAction ChoiceTask::getCompletionAction() {
    if (m_bCancelMappingComplete) {
        return m_cancelAction;
    } else {
        return e_Tutorial_Completion_None;
    }
}

int ChoiceTask::getPromptId() {
    if (m_bShownForMinimumTime)
        return m_promptId;
    else
        return -1;
}

void ChoiceTask::setAsCurrentTask(bool active /*= true*/) {
    enableConstraints(active);
    TutorialTask::setAsCurrentTask(active);
}

void ChoiceTask::handleUIInput(int iAction) {
    if (bHasBeenActivated && m_bShownForMinimumTime) {
        if (iAction == m_iConfirmMapping) {
            m_bConfirmMappingComplete = true;
        } else if (iAction == m_iCancelMapping) {
            m_bCancelMappingComplete = true;
        }
    }
}