#include "Detector.h"
#include "LogManager.h"
#include "RGBController.h"
#include "RGBController_EVGAKeyboard.h"
#include "RGBController_EVGAMouse.h"
#include <hidapi/hidapi.h>

/*-----------------------------------------------------*\
| EVGA USB vendor ID                                    |
\*-----------------------------------------------------*/
#define EVGA_USB_VID                            0x3842

/*-----------------------------------------------------*\
| Keyboard product IDs                                  |
\*-----------------------------------------------------*/
#define Z15_ISO_PID                             0x260E
#define Z15_ANSI_PID                            0x2608
#define Z20_ANSI_PID                            0x260A

/*-----------------------------------------------------*\
| Mouse product IDs                                     |
\*-----------------------------------------------------*/
#define X20_WIRED_PID              0x2420
#define X20_WIRELESS_ADAPTER_PID   0x2402

void DetectEVGAKeyboardControllers(hid_device_info* info, const std::string& name)
{
    hid_device* dev = hid_open_path(info->path);

    if(dev)
    {
        EVGAKeyboardController*     controller      = new EVGAKeyboardController(dev, info->path, info->product_id);
        RGBController_EVGAKeyboard* rgb_controller  = new RGBController_EVGAKeyboard(controller);
        rgb_controller->name                        = name;

        ResourceManager::get()->RegisterRGBController(rgb_controller);
    }
}

void DetectEVGAMouse(hid_device_info* info, const std::string &, int connection_type)
{
    hid_device* dev = hid_open_path(info->path);
    if (dev)
    {
        EVGAMouseController*     controller     = new EVGAMouseController(dev, info->path, connection_type);
        RGBController_EVGAMouse* rgb_controller = new RGBController_EVGAMouse(controller);
        /*-------------------------*\
        | Constructor sets the name |
        \*-------------------------*/
        ResourceManager::get()->RegisterRGBController(rgb_controller);
    }
}

void DetectWiredEVGAMouse(hid_device_info* info, const std::string &name)
{
    DetectEVGAMouse(info, name, EVGA_PERIPHERAL_CONNECTION_TYPE_WIRED);
}

void DetectWirelessEVGAMouse(hid_device_info* info, const std::string &name)
{
    DetectEVGAMouse(info, name, EVGA_PERIPHERAL_CONNECTION_TYPE_WIRELESS);
}


REGISTER_HID_DETECTOR_IPU("EVGA Z15 Keyboard",     DetectEVGAKeyboardControllers,   EVGA_USB_VID,   Z15_ISO_PID,                1,  0x08,   0x4B);
REGISTER_HID_DETECTOR_IPU("EVGA Z15 Keyboard",     DetectEVGAKeyboardControllers,   EVGA_USB_VID,   Z15_ANSI_PID,               1,  0x08,   0x4B);
REGISTER_HID_DETECTOR_IPU("EVGA Z20 Keyboard",     DetectEVGAKeyboardControllers,   EVGA_USB_VID,   Z20_ANSI_PID,               1,  0x08,   0x4B);

REGISTER_HID_DETECTOR_IPU("EVGA X20 Gaming Mouse",  DetectWiredEVGAMouse,           EVGA_USB_VID,   X20_WIRED_PID,              2,  0xFFFF, 0);
REGISTER_HID_DETECTOR_IPU("EVGA X20 USB Receiver",  DetectWirelessEVGAMouse,        EVGA_USB_VID,   X20_WIRELESS_ADAPTER_PID,   2,  0xFFFF, 0);