// Copyright 2017 Citra Emulator Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#include <memory>
#include <thread>
#include "common/param_package.h"
#include "input_common/analog_from_button.h"
#include "input_common/gcadapter/gc_adapter.h"
#include "input_common/gcadapter/gc_poller.h"
#include "input_common/keyboard.h"
#include "input_common/main.h"
#include "input_common/motion_emu.h"
#include "input_common/motion_from_button.h"
#include "input_common/touch_from_button.h"
#include "input_common/udp/client.h"
#include "input_common/udp/udp.h"
#ifdef HAVE_SDL2
#include "input_common/sdl/sdl.h"
#endif

namespace InputCommon {

struct InputSubsystem::Impl {
    void Initialize() {
        gcadapter = std::make_shared<GCAdapter::Adapter>();
        gcbuttons = std::make_shared<GCButtonFactory>(gcadapter);
        Input::RegisterFactory<Input::ButtonDevice>("gcpad", gcbuttons);
        gcanalog = std::make_shared<GCAnalogFactory>(gcadapter);
        Input::RegisterFactory<Input::AnalogDevice>("gcpad", gcanalog);

        keyboard = std::make_shared<Keyboard>();
        Input::RegisterFactory<Input::ButtonDevice>("keyboard", keyboard);
        Input::RegisterFactory<Input::AnalogDevice>("analog_from_button",
                                                    std::make_shared<AnalogFromButton>());
        Input::RegisterFactory<Input::MotionDevice>("keyboard",
                                                    std::make_shared<MotionFromButton>());
        motion_emu = std::make_shared<MotionEmu>();
        Input::RegisterFactory<Input::MotionDevice>("motion_emu", motion_emu);
        Input::RegisterFactory<Input::TouchDevice>("touch_from_button",
                                                   std::make_shared<TouchFromButtonFactory>());

#ifdef HAVE_SDL2
        sdl = SDL::Init();
#endif

        udp = std::make_shared<InputCommon::CemuhookUDP::Client>();
        udpmotion = std::make_shared<UDPMotionFactory>(udp);
        Input::RegisterFactory<Input::MotionDevice>("cemuhookudp", udpmotion);
        udptouch = std::make_shared<UDPTouchFactory>(udp);
        Input::RegisterFactory<Input::TouchDevice>("cemuhookudp", udptouch);
    }

    void Shutdown() {
        Input::UnregisterFactory<Input::ButtonDevice>("keyboard");
        Input::UnregisterFactory<Input::MotionDevice>("keyboard");
        keyboard.reset();
        Input::UnregisterFactory<Input::AnalogDevice>("analog_from_button");
        Input::UnregisterFactory<Input::MotionDevice>("motion_emu");
        motion_emu.reset();
        Input::UnregisterFactory<Input::TouchDevice>("touch_from_button");
#ifdef HAVE_SDL2
        sdl.reset();
#endif
        Input::UnregisterFactory<Input::ButtonDevice>("gcpad");
        Input::UnregisterFactory<Input::AnalogDevice>("gcpad");

        gcbuttons.reset();
        gcanalog.reset();

        Input::UnregisterFactory<Input::MotionDevice>("cemuhookudp");
        Input::UnregisterFactory<Input::TouchDevice>("cemuhookudp");

        udpmotion.reset();
        udptouch.reset();
    }

    [[nodiscard]] std::vector<Common::ParamPackage> GetInputDevices() const {
        std::vector<Common::ParamPackage> devices = {
            Common::ParamPackage{{"display", "Any"}, {"class", "any"}},
            Common::ParamPackage{{"display", "Keyboard/Mouse"}, {"class", "key"}},
        };
#ifdef HAVE_SDL2
        auto sdl_devices = sdl->GetInputDevices();
        devices.insert(devices.end(), sdl_devices.begin(), sdl_devices.end());
#endif
        auto udp_devices = udp->GetInputDevices();
        devices.insert(devices.end(), udp_devices.begin(), udp_devices.end());
        auto gcpad_devices = gcadapter->GetInputDevices();
        devices.insert(devices.end(), gcpad_devices.begin(), gcpad_devices.end());
        return devices;
    }

    [[nodiscard]] AnalogMapping GetAnalogMappingForDevice(
        const Common::ParamPackage& params) const {
        if (!params.Has("class") || params.Get("class", "") == "any") {
            return {};
        }
        if (params.Get("class", "") == "key") {
            // TODO consider returning the SDL key codes for the default keybindings
            return {};
        }
        if (params.Get("class", "") == "gcpad") {
            return gcadapter->GetAnalogMappingForDevice(params);
        }
#ifdef HAVE_SDL2
        if (params.Get("class", "") == "sdl") {
            return sdl->GetAnalogMappingForDevice(params);
        }
#endif
        return {};
    }

    [[nodiscard]] ButtonMapping GetButtonMappingForDevice(
        const Common::ParamPackage& params) const {
        if (!params.Has("class") || params.Get("class", "") == "any") {
            return {};
        }
        if (params.Get("class", "") == "key") {
            // TODO consider returning the SDL key codes for the default keybindings
            return {};
        }
        if (params.Get("class", "") == "gcpad") {
            return gcadapter->GetButtonMappingForDevice(params);
        }
#ifdef HAVE_SDL2
        if (params.Get("class", "") == "sdl") {
            return sdl->GetButtonMappingForDevice(params);
        }
#endif
        return {};
    }

    [[nodiscard]] MotionMapping GetMotionMappingForDevice(
        const Common::ParamPackage& params) const {
        if (!params.Has("class") || params.Get("class", "") == "any") {
            return {};
        }
        if (params.Get("class", "") == "cemuhookudp") {
            // TODO return the correct motion device
            return {};
        }
        return {};
    }

    std::shared_ptr<Keyboard> keyboard;
    std::shared_ptr<MotionEmu> motion_emu;
#ifdef HAVE_SDL2
    std::unique_ptr<SDL::State> sdl;
#endif
    std::shared_ptr<GCButtonFactory> gcbuttons;
    std::shared_ptr<GCAnalogFactory> gcanalog;
    std::shared_ptr<UDPMotionFactory> udpmotion;
    std::shared_ptr<UDPTouchFactory> udptouch;
    std::shared_ptr<CemuhookUDP::Client> udp;
    std::shared_ptr<GCAdapter::Adapter> gcadapter;
};

InputSubsystem::InputSubsystem() : impl{std::make_unique<Impl>()} {}

InputSubsystem::~InputSubsystem() = default;

void InputSubsystem::Initialize() {
    impl->Initialize();
}

void InputSubsystem::Shutdown() {
    impl->Shutdown();
}

Keyboard* InputSubsystem::GetKeyboard() {
    return impl->keyboard.get();
}

const Keyboard* InputSubsystem::GetKeyboard() const {
    return impl->keyboard.get();
}

MotionEmu* InputSubsystem::GetMotionEmu() {
    return impl->motion_emu.get();
}

const MotionEmu* InputSubsystem::GetMotionEmu() const {
    return impl->motion_emu.get();
}

std::vector<Common::ParamPackage> InputSubsystem::GetInputDevices() const {
    return impl->GetInputDevices();
}

AnalogMapping InputSubsystem::GetAnalogMappingForDevice(const Common::ParamPackage& device) const {
    return impl->GetAnalogMappingForDevice(device);
}

ButtonMapping InputSubsystem::GetButtonMappingForDevice(const Common::ParamPackage& device) const {
    return impl->GetButtonMappingForDevice(device);
}

MotionMapping InputSubsystem::GetMotionMappingForDevice(const Common::ParamPackage& device) const {
    return impl->GetMotionMappingForDevice(device);
}

GCAnalogFactory* InputSubsystem::GetGCAnalogs() {
    return impl->gcanalog.get();
}

const GCAnalogFactory* InputSubsystem::GetGCAnalogs() const {
    return impl->gcanalog.get();
}

GCButtonFactory* InputSubsystem::GetGCButtons() {
    return impl->gcbuttons.get();
}

const GCButtonFactory* InputSubsystem::GetGCButtons() const {
    return impl->gcbuttons.get();
}

UDPMotionFactory* InputSubsystem::GetUDPMotions() {
    return impl->udpmotion.get();
}

const UDPMotionFactory* InputSubsystem::GetUDPMotions() const {
    return impl->udpmotion.get();
}

UDPTouchFactory* InputSubsystem::GetUDPTouch() {
    return impl->udptouch.get();
}

const UDPTouchFactory* InputSubsystem::GetUDPTouch() const {
    return impl->udptouch.get();
}

void InputSubsystem::ReloadInputDevices() {
    if (!impl->udp) {
        return;
    }
    impl->udp->ReloadUDPClient();
}

std::vector<std::unique_ptr<Polling::DevicePoller>> InputSubsystem::GetPollers(
    Polling::DeviceType type) const {
#ifdef HAVE_SDL2
    return impl->sdl->GetPollers(type);
#else
    return {};
#endif
}

std::string GenerateKeyboardParam(int key_code) {
    Common::ParamPackage param{
        {"engine", "keyboard"},
        {"code", std::to_string(key_code)},
    };
    return param.Serialize();
}

std::string GenerateAnalogParamFromKeys(int key_up, int key_down, int key_left, int key_right,
                                        int key_modifier, float modifier_scale) {
    Common::ParamPackage circle_pad_param{
        {"engine", "analog_from_button"},
        {"up", GenerateKeyboardParam(key_up)},
        {"down", GenerateKeyboardParam(key_down)},
        {"left", GenerateKeyboardParam(key_left)},
        {"right", GenerateKeyboardParam(key_right)},
        {"modifier", GenerateKeyboardParam(key_modifier)},
        {"modifier_scale", std::to_string(modifier_scale)},
    };
    return circle_pad_param.Serialize();
}
} // namespace InputCommon
