/*-------------------------------------------------------------------*\
|  DarkProjectKeyboardController.cpp                                  |
|                                                                     |
|  Driver for DarkProjectKeyboard USB Controller                      |
|                                                                     |
|  Chris M (DrNo)          8 Apr 2022                                 |
|                                                                     |
\*-------------------------------------------------------------------*/

#include "LogManager.h"
#include "DarkProjectKeyboardController.h"

static uint8_t packet_map[88] =
{
/*00        ESC  F1   F2   F3   F4   F5   F6   F7   F8   F9  */
             5,  11,  17,  23,  29,  35,  41,  47,  53,  59,

/*10        F10  F11  F12  PRT  SLK  PBK   `    1    2    3  */
            65,  71,  77,  83,  89,  95,   0,   6,  12,  18,

/*20         4    5    6    7    8    9    0    -    =   BSP */
            24,  30,  36,  42,  48,  54,  60,  66,  72,  78,

/*30        INS  HME  PUP  TAB   Q    W    E    R    T    Y  */
            84,  90,  96,   1,   7,  13,  19,  25,  31,  37,

/*40         U    I    O    P    [    ]    \   DEL  END  PDN */
            43,  49,  55,  61,  67,  73,  79,  85,  91,  97,

/*50        CAP   A    S    D    F    G    H    J    K    L  */
             2,   8,  14,  20,  26,  32,  38,  44,  50,  56,

/*60         ;    '   ENT  LSH   Z    X    C    V    B    N  */
            62,  68,  80,   3,  15,  21,  27,  33,  39,  45,

/*70         M    ,    .    /   RSH  UP  LCTL LWIN LALT SPC  */
            51,  57,  63,  69,  81,  93,   4,  10,  16,  34,

/*80       RALT RFNC MENU RCTL  LFT  DWN  RGT                */
            52,  58,  64,  76,  88,  94, 100

/* Missing Indexes 9, 22, 28, 40, 46, 70, 74, 75, 82, 86, 87, 92, 98, 99, 101 */
};

DarkProjectKeyboardController::DarkProjectKeyboardController(hid_device* dev_handle, const char* path)
{
    dev                 = dev_handle;
    location            = path;
}

DarkProjectKeyboardController::~DarkProjectKeyboardController()
{
    hid_close(dev);
}

std::string DarkProjectKeyboardController::GetDeviceName()
{
    const int szTemp    = HID_MAX_STR;
    wchar_t tmpName[szTemp];

    hid_get_manufacturer_string(dev, tmpName, szTemp);
    std::wstring wName  = std::wstring(tmpName);
    std::string name    = std::string(wName.begin(), wName.end());

    return name;
}

std::string DarkProjectKeyboardController::GetSerial()
{
    const int szTemp    = HID_MAX_STR;
    wchar_t   tmpName[szTemp];

    hid_get_serial_number_string(dev, tmpName, szTemp);
    std::wstring wName  = std::wstring(tmpName);
    std::string serial  = std::string(wName.begin(), wName.end());

    return serial;
}

std::string DarkProjectKeyboardController::GetLocation()
{
    return("HID: " + location);
}

void DarkProjectKeyboardController::SetLedsDirect(std::vector<RGBColor> colors)
{
    uint8_t RGbuffer[DARKPROJECTKEYBOARD_PACKET_SIZE]           = { 0x08, 0x07, 0x00, 0x00, 0x00 };
    uint8_t BAbuffer[DARKPROJECTKEYBOARD_PACKET_SIZE]           = { 0x08, 0x07, 0x00, 0x01, 0x00 };

    /*-----------------------------------------------------------------*\
    | Set up Direct packet                                              |
    |   packet_map is the index of the Key from full_matrix_map and     |
    |   the value is the position in the direct packet buffer           |
    \*-----------------------------------------------------------------*/
    for(size_t i = 0; i < colors.size(); i++)
    {
        RGBColor key                                            = colors[i];
        uint16_t offset                                         = packet_map[i];

        RGbuffer[DARKPROJECTKEYBOARD_RED_BLUE_BYTE + offset]    = RGBGetRValue(key);
        RGbuffer[DARKPROJECTKEYBOARD_GREEN_BYTE + offset]       = RGBGetGValue(key);
        BAbuffer[DARKPROJECTKEYBOARD_RED_BLUE_BYTE + offset]    = RGBGetBValue(key);
    }

    hid_write(dev, RGbuffer, DARKPROJECTKEYBOARD_PACKET_SIZE);
    hid_write(dev, BAbuffer, DARKPROJECTKEYBOARD_PACKET_SIZE);
}