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

#pragma once

#include <chrono>
#include <vector>
#include <hidapi.h>
#include "DRGBDevices.h"
#include "RGBController.h"

#define DRGB_V4_ONE_PACKAGE_SIZE    316
#define DRGB_V4_PACKAGE_SIZE        340
#define DRGB_V3_PACKAGE_SIZE        21
#define DRGB_V2_PACKAGE_SIZE        20

class DRGBController
{
public:
    DRGBController(hid_device* dev_handle, const char* path,  unsigned short pid, std::string dev_name);
    ~DRGBController();

    std::string     GetFirmwareString();
    std::string     GetLocationString();
    std::string     GetNameString();
    std::string     GetSerialString();

    unsigned short  GetDevicePID();
    unsigned char   GetNumChannels();
    unsigned short  GetLEDsPerChannel();
    unsigned short  GetVersion();

    std::string     GetChannelName(unsigned char channel);

    void            SetChannelLEDs(unsigned char channel, RGBColor * colors, unsigned int num_colors);
    void            SendPacket(unsigned char* colors, unsigned int buf_packets, unsigned int LEDtotal);
    void            SendPacketFS(unsigned char* colors, unsigned int buf_packets, unsigned int Array);

    void            KeepaliveThread();

private:
    hid_device*             dev;
    std::string             location;
    std::string             name;
    std::thread*            keepalive_thread;
    std::atomic<bool>       keepalive_thread_run;
    std::chrono::time_point<std::chrono::steady_clock> last_commit_time;
    unsigned char           version[4] = {0, 0, 0,0};
    unsigned int            device_index;
};