#include "../../Minecraft.World/Platform/stdafx.h"

#include <assert.h>
#include "XUI_Teleport.h"
#include "../../Minecraft.Client/Player/ServerPlayer.h"
#include "../../Minecraft.Client/Network/PlayerConnection.h"
#include "../../Minecraft.Client/Network/PlayerList.h"
#include "../../Minecraft.Client/MinecraftServer.h"
#include "../../Minecraft.World/Util/StringHelpers.h"
#include "../../Minecraft.Client/Rendering/EntityRenderers/PlayerRenderer.h"
#include "XUI_InGamePlayerOptions.h"
#include "../../Minecraft.Client/Minecraft.h"
#include "../../Minecraft.Client/Player/MultiPlayerLocalPlayer.h"
#include "../../Minecraft.Client/Network/ClientConnection.h"
#include "../../Minecraft.World/Headers/net.minecraft.network.packet.h"
#include "../../Minecraft.Client/Platform/Xbox/Network/NetworkPlayerXbox.h"
#include "../../Minecraft.Client/Commands/TeleportCommand.h"

//----------------------------------------------------------------------------------
// Performs initialization tasks - retrieves controls.
//----------------------------------------------------------------------------------
HRESULT CScene_Teleport::OnInit(XUIMessageInit* pInitData, BOOL& bHandled) {
    TeleportMenuInitData* initParam =
        (TeleportMenuInitData*)pInitData->pvInitData;

    m_iPad = initParam->iPad;
    m_teleportToPlayer = initParam->teleportToPlayer;

    delete initParam;

    MapChildControls();

    if (m_teleportToPlayer) {
        m_title.SetText(app.GetString(IDS_TELEPORT_TO_PLAYER));
    } else {
        m_title.SetText(app.GetString(IDS_TELEPORT_TO_ME));
    }

    if (app.GetLocalPlayerCount() > 1) {
        app.AdjustSplitscreenScene(m_hObj, &m_OriginalPosition, m_iPad);
    }

    int playerCount = g_NetworkManager.GetPlayerCount();

    m_playersCount = 0;
    for (int i = 0; i < playerCount; ++i) {
        INetworkPlayer* player = g_NetworkManager.GetPlayerByIndex(i);

        if (player != NULL &&
            !(player->IsLocal() && player->GetUserIndex() == m_iPad)) {
            m_players[m_playersCount] = player->GetSmallId();
            ++m_playersCount;
        }
    }

    g_NetworkManager.RegisterPlayerChangedCallback(
        m_iPad, &CScene_Teleport::OnPlayerChanged, this);

    ui.SetTooltips(m_iPad, IDS_TOOLTIPS_SELECT, IDS_TOOLTIPS_BACK);

    CXuiSceneBase::ShowDarkOverlay(m_iPad, TRUE);

    return S_OK;
}

HRESULT CScene_Teleport::OnDestroy() {
    g_NetworkManager.UnRegisterPlayerChangedCallback(
        m_iPad, &CScene_Teleport::OnPlayerChanged, this);
    return S_OK;
}

HRESULT CScene_Teleport::OnKeyDown(XUIMessageInput* pInputData,
                                   BOOL& rfHandled) {
    ui.AnimateKeyPress(pInputData->UserIndex, pInputData->dwKeyCode);

    HRESULT hr = S_OK;
    switch (pInputData->dwKeyCode) {
        case VK_PAD_B:
        case VK_PAD_BACK:
        case VK_ESCAPE:
            CXuiSceneBase::PlayUISFX(eSFX_Back);
            app.NavigateBack(pInputData->UserIndex);
            rfHandled = TRUE;

            break;
    }

    return hr;
}

//----------------------------------------------------------------------------------
// Handler for the button press message.
//----------------------------------------------------------------------------------
HRESULT CScene_Teleport::OnNotifyPressEx(HXUIOBJ hObjPressed,
                                         XUINotifyPress* pNotifyPressData,
                                         BOOL& rfHandled) {
    // This assumes all buttons can only be pressed with the A button
    ui.AnimateKeyPress(pNotifyPressData->UserIndex, VK_PAD_A);

    if (hObjPressed == playersList) {
        INetworkPlayer* selectedPlayer = g_NetworkManager.GetPlayerBySmallId(
            m_players[playersList.GetCurSel()]);
        INetworkPlayer* thisPlayer =
            g_NetworkManager.GetLocalPlayerByUserIndex(m_iPad);

        std::shared_ptr<GameCommandPacket> packet;
        if (m_teleportToPlayer) {
            packet = TeleportCommand::preparePacket(thisPlayer->GetUID(),
                                                    selectedPlayer->GetUID());
        } else {
            packet = TeleportCommand::preparePacket(selectedPlayer->GetUID(),
                                                    thisPlayer->GetUID());
        }
        ClientConnection* conn =
            Minecraft::GetInstance()->getConnection(m_iPad);
        conn->send(packet);
    }
    return S_OK;
}

void CScene_Teleport::OnPlayerChanged(void* callbackParam,
                                      INetworkPlayer* pPlayer, bool leaving) {
    CScene_Teleport* scene = (CScene_Teleport*)callbackParam;
    bool playerFound = false;

    for (int i = 0; i < scene->m_playersCount; ++i) {
        if (playerFound) {
            scene->m_players[i - 1] = scene->m_players[i];
        } else if (scene->m_players[i] == pPlayer->GetSmallId()) {
            if (scene->playersList.GetCurSel() ==
                scene->playersList.GetItemCount() - 1) {
                scene->playersList.SetCurSel(scene->playersList.GetItemCount() -
                                             2);
            }
            // Player removed
            playerFound = true;
        }
    }

    if (playerFound) {
        --scene->m_playersCount;
        scene->playersList.DeleteItems(scene->playersList.GetItemCount() - 1,
                                       1);
    }

    if (!playerFound) {
        // Player added
        scene->m_players[scene->m_playersCount] = pPlayer->GetSmallId();
        ++scene->m_playersCount;
        scene->playersList.InsertItems(scene->playersList.GetItemCount(), 1);
    }
}

HRESULT CScene_Teleport::OnGetSourceDataText(
    XUIMessageGetSourceText* pGetSourceTextData, BOOL& bHandled) {
    if (pGetSourceTextData->bItemData) {
        if (pGetSourceTextData->iItem < m_playersCount) {
            INetworkPlayer* player = g_NetworkManager.GetPlayerBySmallId(
                m_players[pGetSourceTextData->iItem]);
            if (player != NULL) {
#ifndef _CONTENT_PACKAGE
                if (app.DebugSettingsOn() &&
                    (app.GetGameSettingsDebugMask() &
                     (1L << eDebugSetting_DebugLeaderboards))) {
                    pGetSourceTextData->szText = L"WWWWWWWWWWWWWWWW";
                } else
#endif
                {
                    pGetSourceTextData->szText = player->GetOnlineName();
                }
            } else {
                pGetSourceTextData->szText = L"";
            }

            HRESULT hr;
            HXUIOBJ hButton, hVisual, hPlayerIcon, hVoiceIcon;
            hButton = playersList.GetItemControl(pGetSourceTextData->iItem);
            hr = XuiControlGetVisual(hButton, &hVisual);

            // Set the players icon
            hr = XuiElementGetChildById(hVisual, L"IconGroup", &hPlayerIcon);
            short colourIndex =
                app.GetPlayerColour(m_players[pGetSourceTextData->iItem]);
            int playFrame = 0;
            switch (colourIndex) {
                case 1:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P1", &playFrame);
                    break;
                case 2:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P2", &playFrame);
                    break;
                case 3:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P3", &playFrame);
                    break;
                case 4:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P4", &playFrame);
                    break;
                case 5:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P5", &playFrame);
                    break;
                case 6:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P6", &playFrame);
                    break;
                case 7:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P7", &playFrame);
                    break;
                case 8:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P8", &playFrame);
                    break;
                case 9:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P9", &playFrame);
                    break;
                case 10:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P10", &playFrame);
                    break;
                case 11:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P11", &playFrame);
                    break;
                case 12:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P12", &playFrame);
                    break;
                case 13:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P13", &playFrame);
                    break;
                case 14:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P14", &playFrame);
                    break;
                case 15:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P15", &playFrame);
                    break;
                case 0:
                default:
                    XuiElementFindNamedFrame(hPlayerIcon, L"P0", &playFrame);
                    break;
            };
            if (playFrame < 0) playFrame = 0;
            XuiElementPlayTimeline(hPlayerIcon, playFrame, playFrame, playFrame,
                                   FALSE, FALSE);

            // Set the voice icon
            hr = XuiElementGetChildById(hVisual, L"VoiceGroup", &hVoiceIcon);

            playFrame = -1;
            if (player != NULL && player->HasVoice()) {
                if (player->IsMutedByLocalUser(m_iPad)) {
                    // Muted image
                    XuiElementFindNamedFrame(hVoiceIcon, L"Muted", &playFrame);
                } else if (player->IsTalking()) {
                    // Talking image
                    XuiElementFindNamedFrame(hVoiceIcon, L"Speaking",
                                             &playFrame);
                } else {
                    // Not talking image
                    XuiElementFindNamedFrame(hVoiceIcon, L"NotSpeaking",
                                             &playFrame);
                }
            }

            if (playFrame < 0) {
                XuiElementFindNamedFrame(hVoiceIcon, L"Normal", &playFrame);
            }
            XuiElementPlayTimeline(hVoiceIcon, playFrame, playFrame, playFrame,
                                   FALSE, FALSE);

            pGetSourceTextData->bDisplay = TRUE;

            bHandled = TRUE;
        }
    }
    return S_OK;
}

HRESULT CScene_Teleport::OnGetSourceDataImage(
    XUIMessageGetSourceImage* pGetSourceImageData, BOOL& bHandled) {
    if (pGetSourceImageData->bItemData) {
        if (pGetSourceImageData->iItem < m_playersCount) {
            bHandled = TRUE;
        }
    }
    return S_OK;
}

HRESULT CScene_Teleport::OnGetItemCountAll(
    XUIMessageGetItemCount* pGetItemCountData, BOOL& bHandled) {
    pGetItemCountData->cItems = m_playersCount;
    bHandled = TRUE;
    return S_OK;
}

HRESULT CScene_Teleport::OnCustomMessage_Splitscreenplayer(bool bJoining,
                                                           BOOL& bHandled) {
    bHandled = true;
    return app.AdjustSplitscreenScene_PlayerChanged(m_hObj, &m_OriginalPosition,
                                                    m_iPad, bJoining);
}