/*---------------------------------------------------------*\
| DRGBController.cpp                                        |
|                                                           |
|   Driver for DRGBmods                                     |
|                                                           |
|   Zhi Yan                                     25 Jun 2024 |
|                                                           |
|   This file is part of the OpenRGB project                |
|   SPDX-License-Identifier: GPL-2.0-or-later               |
\*---------------------------------------------------------*/

#include <cstring>
#include "DRGBController.h"
#include "StringUtils.h"

using namespace std::chrono_literals;

DRGBController::DRGBController(hid_device* dev_handle, const char* path, unsigned short pid, std::string dev_name)
{
    dev         = dev_handle;
    location    = path;
    name        = dev_name;

    /*-----------------------------------------------------*\
    | Loop through all known devices to look for a PID      |
    | match                                                 |
    \*-----------------------------------------------------*/
    for(unsigned int i = 0; i < DRGB_NUM_DEVICES; i++)
    {
        if(drgb_device_list[i]->pid == pid)
        {
            /*---------------------------------------------*\
            | Set device index                              |
            \*---------------------------------------------*/
            device_index = i;
        }
    }

    /*-----------------------------------------------------*\
    | Exit hardware effects.  Start a thread to continuously|
    | send a keepalive packet every 500ms                   |
    \*-----------------------------------------------------*/
    keepalive_thread_run = 1;
    keepalive_thread     = new std::thread(&DRGBController::KeepaliveThread, this);
}

DRGBController::~DRGBController()
{
    keepalive_thread_run = 0;
    keepalive_thread->join();
    delete keepalive_thread;
    hid_close(dev);
}

std::string DRGBController::GetFirmwareString()
{
    return "v"+std::to_string(version[0]) + "." + std::to_string(version[1]) + "." + std::to_string(version[2]) + "." + std::to_string(version[3]);
}

std::string DRGBController::GetLocationString()
{
    return("HID: " + location);
}

std::string DRGBController::GetNameString()
{
    return(name);
}

std::string DRGBController::GetSerialString()
{
    wchar_t serial_string[128];
    int ret = hid_get_serial_number_string(dev, serial_string, 128);

    if(ret != 0)
    {
        return("");
    }

    return(StringUtils::wstring_to_string(serial_string));
}

unsigned short DRGBController::GetDevicePID()
{
    return(drgb_device_list[device_index]->pid);
}

unsigned char DRGBController::GetNumChannels()
{
    return(drgb_device_list[device_index]->channels);
}

unsigned short DRGBController::GetLEDsPerChannel()
{
    return(drgb_device_list[device_index]->leds_per_channel);
}

unsigned short DRGBController::GetVersion()
{
    return(drgb_device_list[device_index]->version);
}

std::string DRGBController::GetChannelName(unsigned char channel)
{
    std::string channel_name;

    if(drgb_device_list[device_index]->channels == 6)
    {
        if(channel == 0)
        {
            channel_name = "Strimer ATX" + std::to_string(channel + 1);
        }
        else if(channel < 3)
        {
            channel_name = "Channel C" + std::to_string(channel);
        }
        else if(channel == 3)
        {
            channel_name = "Strimer GPU" + std::to_string(channel - 2);
        }
        else if(channel < 6)
        {
            channel_name = "Channel D" + std::to_string(channel - 3);
        }
    }
    else if(drgb_device_list[device_index]->channels == 10 || drgb_device_list[device_index]->channels == 12)
    {
        channel_name = "Channel " + std::to_string(channel + 1);
    }
    else if(drgb_device_list[device_index]->channels == 14)
    {
        if(channel < 4)
        {
            channel_name = "LCD " + std::to_string(channel + 1);
        }
        else if(channel < 6)
        {
            channel_name = "LED " + std::to_string(channel + 1);
        }
        else if(channel < 16)
        {
            channel_name = "ARGB " + std::to_string(channel - 5);
        }
    }
    else if(channel < 8)
    {
        channel_name = "Channel A" + std::to_string(channel + 1);
    }
    else if(channel < 16)
    {
        channel_name = "Channel B" + std::to_string(channel - 7);
    }
    else if(drgb_device_list[device_index]->channels == 30)
    {
        if(channel < 24)
        {
            channel_name = "Channel C" + std::to_string(channel - 15);
        }
        else if(channel < 30)
        {
            channel_name = "Channel D" + std::to_string(channel - 23);
        }
    }
    else if(channel < 22)
    {
        channel_name = "Channel C" + std::to_string(channel - 15);
    }
    else if(channel < 28)
    {
        channel_name = "Channel D" + std::to_string(channel - 21);
    }
    else if(channel < 36)
    {
        channel_name = "Channel E" + std::to_string(channel - 27);
    }

    return(channel_name);
}

void DRGBController::SetChannelLEDs(unsigned char /*channel*/, RGBColor* /*colors*/, unsigned int /*num_colors*/)
{

}

void DRGBController::SendPacket(unsigned char* colors, unsigned int buf_packets , unsigned int LEDtotal)
{
    unsigned char   usb_buf[1025];
    unsigned int    buf_idx = 0;
    memset(usb_buf, 0x00, sizeof(usb_buf));
    usb_buf[0x00]   = 0x00;

    /*-----------------------------------------------------*\
    | Update last commit time                               |
    \*-----------------------------------------------------*/
    last_commit_time = std::chrono::steady_clock::now();

    unsigned int    HigCount = LEDtotal / 256 >= 1 ? 1 : 0;
    unsigned int    LowCount = LEDtotal >= DRGB_V4_ONE_PACKAGE_SIZE ? 60 : (LEDtotal % 256) ;
    LEDtotal = LEDtotal <= DRGB_V4_ONE_PACKAGE_SIZE ? 0 : (LEDtotal-DRGB_V4_ONE_PACKAGE_SIZE);
    for(unsigned int i = 0; i < buf_packets; i++)
    {
        usb_buf[1]  = i + 100 ;
        usb_buf[2]  = buf_packets + 99 ;
        usb_buf[3]  = HigCount;
        usb_buf[4]  = LowCount;
        buf_idx     = i*1020;
        memcpy(usb_buf + 5, colors + buf_idx, 1020);
        hid_write(dev, usb_buf, 1025);
        if(LEDtotal)
        {
            HigCount    = LEDtotal / 256 >= 1 ? 1 : 0;
            LowCount    = LEDtotal >= DRGB_V4_PACKAGE_SIZE ? 84 : (LEDtotal % 256) ;
            LEDtotal    = LEDtotal <= DRGB_V4_PACKAGE_SIZE ? 0 : (LEDtotal-DRGB_V4_PACKAGE_SIZE);
        }
    }
}

void DRGBController::SendPacketFS(unsigned char* colors, unsigned int buf_packets , unsigned int Array)
{
    unsigned char usb_buf[65] = {0};
    unsigned int current_index = 0;

    /*-----------------------------------------------------*\
    | Update last commit time                               |
    \*-----------------------------------------------------*/
    last_commit_time = std::chrono::steady_clock::now();

    if(Array == 0x64 || Array == 0x47)
    {
        const unsigned int offset = (Array == 0x64) ? 100 : 92;

        for(unsigned int i = 0; i < buf_packets; i++)
        {
            const bool is_last_packet = (i == buf_packets - 1);
            usb_buf[1] = is_last_packet ? (Array + offset + i) : (Array + i);
            current_index = i * 63;
            memcpy(usb_buf + 2, colors + current_index, 63);

            hid_write(dev, usb_buf, 65);
        }
    }
    else
    {
        memcpy(usb_buf + 1, colors, 64);
        hid_write(dev, usb_buf, 65);
    }
}

void DRGBController::KeepaliveThread()
{
    unsigned char   sleep_buf[65];
    sleep_buf[0] = 0x65;
    while(keepalive_thread_run.load())
    {
        if((std::chrono::steady_clock::now() - last_commit_time) > std::chrono::milliseconds(500))
        {
            SendPacketFS(sleep_buf, 1, 0);
        }
        std::this_thread::sleep_for(300ms);
    }
}