#include "controls/devices/game_controller.h"

#include <cstddef>
#include <vector>

#ifdef USE_SDL3
#include <SDL3/SDL_error.h>
#include <SDL3/SDL_events.h>
#include <SDL3/SDL_gamepad.h>
#else
#include <SDL.h>

#include "utils/sdl2_backports.h"
#endif

#include "controls/controller_motion.h"
#include "controls/devices/joystick.h"
#include "utils/log.hpp"
#include "utils/sdl_ptrs.h"
#include "utils/stubs.h"

namespace devilution {

std::vector<GameController> GameController::controllers_;

void GameController::UnlockTriggerState()
{
	trigger_left_state_ = ControllerButton_NONE;
	trigger_right_state_ = ControllerButton_NONE;
}

ControllerButton GameController::ToControllerButton(const SDL_Event &event)
{
	switch (event.type) {
#ifdef USE_SDL3
	case SDL_EVENT_GAMEPAD_AXIS_MOTION:
#else
	case SDL_CONTROLLERAXISMOTION:
#endif
	{
		const auto &axis =
#ifdef USE_SDL3
		    event.gaxis;
#else
		    event.caxis;
#endif
		switch (axis.axis) {
#ifdef USE_SDL3
		case SDL_GAMEPAD_AXIS_LEFT_TRIGGER:
#else
		case SDL_CONTROLLER_AXIS_TRIGGERLEFT:
#endif
			if (axis.value < 8192 && trigger_left_is_down_) { // 25% pressed
				trigger_left_is_down_ = false;
				trigger_left_state_ = ControllerButton_AXIS_TRIGGERLEFT;
			}
			if (axis.value > 16384 && !trigger_left_is_down_) { // 50% pressed
				trigger_left_is_down_ = true;
				trigger_left_state_ = ControllerButton_AXIS_TRIGGERLEFT;
			}
			return trigger_left_state_;
#ifdef USE_SDL3
		case SDL_GAMEPAD_AXIS_RIGHT_TRIGGER:
#else
		case SDL_CONTROLLER_AXIS_TRIGGERRIGHT:
#endif
			if (axis.value < 8192 && trigger_right_is_down_) { // 25% pressed
				trigger_right_is_down_ = false;
				trigger_right_state_ = ControllerButton_AXIS_TRIGGERRIGHT;
			}
			if (axis.value > 16384 && !trigger_right_is_down_) { // 50% pressed
				trigger_right_is_down_ = true;
				trigger_right_state_ = ControllerButton_AXIS_TRIGGERRIGHT;
			}
			return trigger_right_state_;
		}
	} break;
#ifdef USE_SDL3
	case SDL_EVENT_GAMEPAD_BUTTON_DOWN:
	case SDL_EVENT_GAMEPAD_BUTTON_UP:
		switch (event.gbutton.button)
#else
	case SDL_CONTROLLERBUTTONDOWN:
	case SDL_CONTROLLERBUTTONUP:
		switch (event.cbutton.button)
#endif
		{
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_SOUTH:
#else
		case SDL_CONTROLLER_BUTTON_A:
#endif
			return ControllerButton_BUTTON_A;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_EAST:
#else
		case SDL_CONTROLLER_BUTTON_B:
#endif
			return ControllerButton_BUTTON_B;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_WEST:
#else
		case SDL_CONTROLLER_BUTTON_X:
#endif
			return ControllerButton_BUTTON_X;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_NORTH:
#else
		case SDL_CONTROLLER_BUTTON_Y:
#endif
			return ControllerButton_BUTTON_Y;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_LEFT_STICK:
#else
		case SDL_CONTROLLER_BUTTON_LEFTSTICK:
#endif
			return ControllerButton_BUTTON_LEFTSTICK;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_RIGHT_STICK:
#else
		case SDL_CONTROLLER_BUTTON_RIGHTSTICK:
#endif
			return ControllerButton_BUTTON_RIGHTSTICK;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_LEFT_SHOULDER:
#else
		case SDL_CONTROLLER_BUTTON_LEFTSHOULDER:
#endif
			return ControllerButton_BUTTON_LEFTSHOULDER;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_RIGHT_SHOULDER:
#else
		case SDL_CONTROLLER_BUTTON_RIGHTSHOULDER:
#endif
			return ControllerButton_BUTTON_RIGHTSHOULDER;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_START:
#else
		case SDL_CONTROLLER_BUTTON_START:
#endif
			return ControllerButton_BUTTON_START;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_BACK:
#else
		case SDL_CONTROLLER_BUTTON_BACK:
#endif
			return ControllerButton_BUTTON_BACK;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_DPAD_UP:
#else
		case SDL_CONTROLLER_BUTTON_DPAD_UP:
#endif
			return ControllerButton_BUTTON_DPAD_UP;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_DPAD_DOWN:
#else
		case SDL_CONTROLLER_BUTTON_DPAD_DOWN:
#endif
			return ControllerButton_BUTTON_DPAD_DOWN;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_DPAD_LEFT:
#else
		case SDL_CONTROLLER_BUTTON_DPAD_LEFT:
#endif
			return ControllerButton_BUTTON_DPAD_LEFT;
#ifdef USE_SDL3
		case SDL_GAMEPAD_BUTTON_DPAD_RIGHT:
#else
		case SDL_CONTROLLER_BUTTON_DPAD_RIGHT:
#endif
			return ControllerButton_BUTTON_DPAD_RIGHT;
		default:
			break;
		}
	default:
		break;
	}
	return ControllerButton_NONE;
}

#ifdef USE_SDL3
SDL_GamepadButton GameController::ToSdlGameControllerButton(ControllerButton button)
#else
SDL_GameControllerButton GameController::ToSdlGameControllerButton(ControllerButton button)
#endif
{
	if (button == ControllerButton_AXIS_TRIGGERLEFT || button == ControllerButton_AXIS_TRIGGERRIGHT)
		UNIMPLEMENTED();
	switch (button) {
	case ControllerButton_BUTTON_A:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_SOUTH;
#else
		return SDL_CONTROLLER_BUTTON_A;
#endif
	case ControllerButton_BUTTON_B:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_EAST;
#else
		return SDL_CONTROLLER_BUTTON_B;
#endif
	case ControllerButton_BUTTON_X:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_WEST;
#else
		return SDL_CONTROLLER_BUTTON_X;
#endif
	case ControllerButton_BUTTON_Y:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_NORTH;
#else
		return SDL_CONTROLLER_BUTTON_Y;
#endif
	case ControllerButton_BUTTON_BACK:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_BACK;
#else
		return SDL_CONTROLLER_BUTTON_BACK;
#endif
	case ControllerButton_BUTTON_START:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_START;
#else
		return SDL_CONTROLLER_BUTTON_START;
#endif
	case ControllerButton_BUTTON_LEFTSTICK:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_LEFT_STICK;
#else
		return SDL_CONTROLLER_BUTTON_LEFTSTICK;
#endif
	case ControllerButton_BUTTON_RIGHTSTICK:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_RIGHT_STICK;
#else
		return SDL_CONTROLLER_BUTTON_RIGHTSTICK;
#endif
	case ControllerButton_BUTTON_LEFTSHOULDER:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_LEFT_SHOULDER;
#else
		return SDL_CONTROLLER_BUTTON_LEFTSHOULDER;
#endif
	case ControllerButton_BUTTON_RIGHTSHOULDER:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_RIGHT_SHOULDER;
#else
		return SDL_CONTROLLER_BUTTON_RIGHTSHOULDER;
#endif
	case ControllerButton_BUTTON_DPAD_UP:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_DPAD_UP;
#else
		return SDL_CONTROLLER_BUTTON_DPAD_UP;
#endif
	case ControllerButton_BUTTON_DPAD_DOWN:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_DPAD_DOWN;
#else
		return SDL_CONTROLLER_BUTTON_DPAD_DOWN;
#endif
	case ControllerButton_BUTTON_DPAD_LEFT:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_DPAD_LEFT;
#else
		return SDL_CONTROLLER_BUTTON_DPAD_LEFT;
#endif
	case ControllerButton_BUTTON_DPAD_RIGHT:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_DPAD_RIGHT;
#else
		return SDL_CONTROLLER_BUTTON_DPAD_RIGHT;
#endif
	default:
#ifdef USE_SDL3
		return SDL_GAMEPAD_BUTTON_INVALID;
#else
		return SDL_CONTROLLER_BUTTON_INVALID;
#endif
	}
}

bool GameController::IsPressed(ControllerButton button) const
{
	if (button == ControllerButton_AXIS_TRIGGERLEFT)
		return trigger_left_is_down_;
	if (button == ControllerButton_AXIS_TRIGGERRIGHT)
		return trigger_right_is_down_;

#ifdef USE_SDL3
	const SDL_GamepadButton gcButton = ToSdlGameControllerButton(button);
	return SDL_GamepadHasButton(sdl_game_controller_, gcButton) && SDL_GetGamepadButton(sdl_game_controller_, gcButton);
#else
	const SDL_GameControllerButton gcButton = ToSdlGameControllerButton(button);
	return SDL_GameControllerHasButton(sdl_game_controller_, gcButton) && SDL_GameControllerGetButton(sdl_game_controller_, gcButton) != 0;
#endif
}

bool GameController::ProcessAxisMotion(const SDL_Event &event)
{
#ifdef USE_SDL3
	if (event.type != SDL_EVENT_GAMEPAD_AXIS_MOTION) return false;
	const auto &axis = event.gaxis;
#else
	if (event.type != SDL_CONTROLLERAXISMOTION) return false;
	const auto &axis = event.caxis;
#endif
	switch (axis.axis) {
#ifdef USE_SDL3
	case SDL_GAMEPAD_AXIS_LEFTX:
#else
	case SDL_CONTROLLER_AXIS_LEFTX:
#endif
		leftStickXUnscaled = static_cast<float>(axis.value);
		leftStickNeedsScaling = true;
		break;
#ifdef USE_SDL3
	case SDL_GAMEPAD_AXIS_LEFTY:
#else
	case SDL_CONTROLLER_AXIS_LEFTY:
#endif
		leftStickYUnscaled = static_cast<float>(-axis.value);
		leftStickNeedsScaling = true;
		break;
#ifdef USE_SDL3
	case SDL_GAMEPAD_AXIS_RIGHTX:
#else
	case SDL_CONTROLLER_AXIS_RIGHTX:
#endif
		rightStickXUnscaled = static_cast<float>(axis.value);
		rightStickNeedsScaling = true;
		break;
#ifdef USE_SDL3
	case SDL_GAMEPAD_AXIS_RIGHTY:
#else
	case SDL_CONTROLLER_AXIS_RIGHTY:
#endif
		rightStickYUnscaled = static_cast<float>(-axis.value);
		rightStickNeedsScaling = true;
		break;
	default:
		return false;
	}
	return true;
}

#ifdef USE_SDL3
void GameController::Add(SDL_JoystickID joystickId)
#else
void GameController::Add(int joystickIndex)
#endif
{
	GameController result;
#ifdef USE_SDL3
	Log("Opening game controller for joystick with ID {}", joystickId);
	result.sdl_game_controller_ = SDL_OpenGamepad(joystickId);
#else
	Log("Opening game controller for joystick at index {}", joystickIndex);
	result.sdl_game_controller_ = SDL_GameControllerOpen(joystickIndex);
#endif

	if (result.sdl_game_controller_ == nullptr) {
		Log("{}", SDL_GetError());
		SDL_ClearError();
		return;
	}
	controllers_.push_back(result);

#ifdef USE_SDL3
	result.instance_id_ = joystickId;
	const SDLUniquePtr<char> mapping { SDL_GetGamepadMappingForID(joystickId) };
#else
	SDL_Joystick *const sdlJoystick = SDL_GameControllerGetJoystick(result.sdl_game_controller_);
	result.instance_id_ = SDL_JoystickInstanceID(sdlJoystick);
	const SDL_JoystickGUID guid = SDL_JoystickGetGUID(sdlJoystick);
	const SDLUniquePtr<char> mapping { SDL_GameControllerMappingForGUID(guid) };
#endif
	if (mapping) {
		Log("Opened game controller with mapping:\n{}", mapping.get());
	}
}

void GameController::Remove(SDL_JoystickID instanceId)
{
	Log("Removing game controller with instance id {}", instanceId);
	for (std::size_t i = 0; i < controllers_.size(); ++i) {
		const GameController &controller = controllers_[i];
		if (controller.instance_id_ != instanceId)
			continue;
		controllers_.erase(controllers_.begin() + i);
		return;
	}
	Log("Game controller not found with instance id: {}", instanceId);
}

GameController *GameController::Get(SDL_JoystickID instanceId)
{
	for (auto &controller : controllers_) {
		if (controller.instance_id_ == instanceId)
			return &controller;
	}
	return nullptr;
}

GameController *GameController::Get(const SDL_Event &event)
{
	switch (event.type) {
#ifdef USE_SDL3
	case SDL_EVENT_GAMEPAD_AXIS_MOTION:
		return Get(event.gaxis.which);
	case SDL_EVENT_GAMEPAD_BUTTON_DOWN:
	case SDL_EVENT_GAMEPAD_BUTTON_UP:
		return Get(event.gbutton.which);
#else
	case SDL_CONTROLLERAXISMOTION:
		return Get(event.caxis.which);
	case SDL_CONTROLLERBUTTONDOWN:
	case SDL_CONTROLLERBUTTONUP:
		return Get(event.cbutton.which);
#endif
	default:
		return nullptr;
	}
}

const std::vector<GameController> &GameController::All()
{
	return controllers_;
}

bool GameController::IsPressedOnAnyController(ControllerButton button, SDL_JoystickID *which)
{
	for (auto &controller : controllers_)
		if (controller.IsPressed(button)) {
			if (which != nullptr)
				*which = controller.instance_id_;

			return true;
		}
	return false;
}

GamepadLayout GameController::getLayout(const SDL_Event &event)
{
#if defined(DEVILUTIONX_GAMEPAD_TYPE)
	return GamepadLayout::
	    DEVILUTIONX_GAMEPAD_TYPE;
#elif USE_SDL3
	switch (SDL_GetGamepadTypeForID(event.gdevice.which)) {
	case SDL_GAMEPAD_TYPE_XBOX360:
	case SDL_GAMEPAD_TYPE_XBOXONE:
		return GamepadLayout::Xbox;
	case SDL_GAMEPAD_TYPE_PS3:
	case SDL_GAMEPAD_TYPE_PS4:
	case SDL_GAMEPAD_TYPE_PS5:
		return GamepadLayout::PlayStation;
	case SDL_GAMEPAD_TYPE_NINTENDO_SWITCH_PRO:
	case SDL_GAMEPAD_TYPE_NINTENDO_SWITCH_JOYCON_LEFT:
	case SDL_GAMEPAD_TYPE_NINTENDO_SWITCH_JOYCON_RIGHT:
	case SDL_GAMEPAD_TYPE_NINTENDO_SWITCH_JOYCON_PAIR:
		return GamepadLayout::Nintendo;
	default:
		return GamepadLayout::Generic;
	}
#else
#if SDL_VERSION_ATLEAST(2, 0, 12)
	const int index = event.cdevice.which;
	const SDL_GameControllerType gamepadType = SDL_GameControllerTypeForIndex(index);
	switch (gamepadType) {
	case SDL_CONTROLLER_TYPE_NINTENDO_SWITCH_PRO:
#if SDL_VERSION_ATLEAST(2, 24, 0)
	case SDL_CONTROLLER_TYPE_NINTENDO_SWITCH_JOYCON_LEFT:
	case SDL_CONTROLLER_TYPE_NINTENDO_SWITCH_JOYCON_RIGHT:
	case SDL_CONTROLLER_TYPE_NINTENDO_SWITCH_JOYCON_PAIR:
#endif
		return GamepadLayout::Nintendo;
	case SDL_CONTROLLER_TYPE_PS3:
	case SDL_CONTROLLER_TYPE_PS4:
#if SDL_VERSION_ATLEAST(2, 0, 14)
	case SDL_CONTROLLER_TYPE_PS5:
#endif
		return GamepadLayout::PlayStation;
	case SDL_CONTROLLER_TYPE_XBOXONE:
	case SDL_CONTROLLER_TYPE_XBOX360:
#if SDL_VERSION_ATLEAST(2, 0, 16)
	case SDL_CONTROLLER_TYPE_GOOGLE_STADIA:
	case SDL_CONTROLLER_TYPE_AMAZON_LUNA:
#if SDL_VERSION_ATLEAST(2, 24, 0)
	case SDL_CONTROLLER_TYPE_NVIDIA_SHIELD:
#endif
#endif
		return GamepadLayout::Xbox;
#if SDL_VERSION_ATLEAST(2, 0, 14)
	case SDL_CONTROLLER_TYPE_VIRTUAL:
#endif
	case SDL_CONTROLLER_TYPE_UNKNOWN:
#if SDL_VERSION_ATLEAST(2, 30, 0)
	case SDL_CONTROLLER_TYPE_MAX:
#endif
		break;
	}
#endif
	return GamepadLayout::Generic;
#endif // !defined(DEVILUTIONX_GAMEPAD_TYPE)
}

} // namespace devilution