#include <td/display/state/DebugWorldState.h>

#include <chrono>
#include <iostream>

#include <td/game/World.h>
#include <td/protocol/packet/Packets.h>
#include <td/render/renderer/EntityRenderer.h>
#include <td/render/renderer/PlayerListRenderer.h>
#include <td/render/renderer/TowerRenderer.h>
#include <td/render/renderer/WorldRenderer.h>

#include <server/Server.h>
#include <server/socket/FakeSocket.h>
#include <server/state/LobbyState.h>

#include <client/Client.h>
#include <client/socket/FakeSocket.h>
#include <client/state/GameState.h>
#include <client/state/LoggingState.h>

#include <td/display/Display.h>
#include <td/display/state/DebugWorldState.h>

namespace td {

DebugWorldState::DebugWorldState(Display& a_Display) : DisplayState(a_Display) {
	// server
	m_ServerSocket = std::make_shared<server::FakeSocket>();
	m_Server = std::make_unique<server::Server>(m_ServerSocket);

	// client
	auto clientFakeSocket = client::FakeSocket::Connect(m_ServerSocket);
	m_Client = std::make_unique<client::Client>(clientFakeSocket);

	// TODO: make it better
	m_Client->OnStateChange.Connect([this](client::Client::State& a_State) {
		if (auto gameState = dynamic_cast<client::GameState*>(&a_State)) {
			// render
			auto clientWorld = gameState->GetWorld();
			m_Renderer.AddRenderer<render::WorldRenderer>(m_Camera, clientWorld);
			m_Renderer.AddRenderer<render::EntityRenderer>(m_Camera, clientWorld);
			m_Renderer.AddRenderer<render::TowerRenderer>(m_Camera, clientWorld);

			auto& list = m_Renderer.AddRenderer<render::PlayerListRenderer>(m_Client->GetPlayers());

			list.OnPlayerCreate.Connect([this]() {
				auto newSocket = client::FakeSocket::Connect(m_ServerSocket);
				auto newClient = std::make_unique<client::Client>(newSocket);
				newClient->ChangeState<client::LoggingState>("Bot");
				m_FakeClients.push_back(std::move(newClient));
			});

			list.OnPlayerKick.Connect([this](PlayerID a_Player) {
				auto it = std::find_if(m_FakeClients.begin(), m_FakeClients.end(), [a_Player](auto& clientPtr){
					if (!clientPtr->GetId().has_value())
						return false;
					return clientPtr->GetId().value() == a_Player;
				});
				m_FakeClients.erase(it);
			});

			// update state
			m_ClientState = gameState;
		}
	});

	m_Client->ChangeState<client::LoggingState>("Player0");

	// camera
	m_Camera.SetCamPos({77, 7, 13});
	m_Camera.UpdatePerspective(m_StateMachine.GetAspectRatio());
}

void DebugWorldState::Update(float a_Delta) {
	m_Server->Update(a_Delta);
	m_Client->Update(a_Delta);
	if (m_ClientState)
		m_Renderer.Render(m_ClientState->GetCurrentLerp());
}

void DebugWorldState::OnAspectRatioChange(float a_Ratio) {
	m_Camera.UpdatePerspective(a_Ratio);
}

void DebugWorldState::OnKeyDown(SDL_Keycode a_Key) {
	// temporary tests
	switch (a_Key) {
		case SDLK_A:
			m_Client->SendPacket(td::protocol::packets::SpawnTroopPacket(td::EntityType::Zombie, 1));
			break;

		case SDLK_Z:
			m_Client->SendPacket(td::protocol::packets::PlaceTowerPacket(td::TowerType::Archer, td::TowerCoords(77, 13)));
			break;

		default:
			break;
	}
}

DebugWorldState::~DebugWorldState() {}

}  // namespace td