/*---------------------------------------------------------*\
| AsusAuraMonitorController.cpp                             |
|                                                           |
|   Driver for ASUS Aura monitor                            |
|                                                           |
|   Mola19                                      08 Mar 2022 |
|                                                           |
|   This file is part of the OpenRGB project                |
|   SPDX-License-Identifier: GPL-2.0-only                   |
\*---------------------------------------------------------*/

#include <cstring>
#include "AsusAuraMonitorController.h"
#include "LogManager.h"
#include "StringUtils.h"

AuraMonitorController::AuraMonitorController(hid_device* dev_handle, const char* path, uint16_t pid)
{
    dev         = dev_handle;
    location    = path;
    device_pid  = pid;
}

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

std::string AuraMonitorController::GetDeviceLocation()
{
    return("HID: " + location);
}

std::string AuraMonitorController::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));
}

void AuraMonitorController::BeginUpdate()
{
    unsigned char usb_buf[8];

    if (device_pid == AURA_ROG_PG32UQ_PID || device_pid == AURA_ROG_STRIX_XG32VC_PID)
    {
        memset(usb_buf, 0x00, sizeof(usb_buf));

        usb_buf[0x00]   = 0x03;
        usb_buf[0x01]   = 0x02;
        usb_buf[0x02]   = 0xA1;
        usb_buf[0x03]   = 0x80;

        usb_buf[0x04]   = 0x20;
        hid_send_feature_report(dev, usb_buf, 8);

        usb_buf[0x04]   = 0x30;
        hid_send_feature_report(dev, usb_buf, 8);
    }
}

void AuraMonitorController::UpdateLed
    (
    int             led,
    unsigned char   red,
    unsigned char   green,
    unsigned char   blue
    )
{
    unsigned char usb_buf[8];

    memset(usb_buf, 0x00, sizeof(usb_buf));

    usb_buf[0x00]   = 0x03;
    usb_buf[0x01]   = 0x02;
    usb_buf[0x02]   = 0xA1;
    usb_buf[0x03]   = 0x80;

    unsigned char offset = (device_pid == AURA_ROG_STRIX_XG279Q_PID) ? 0 : 16;

    usb_buf[0x04]   = offset + led * 3;
    usb_buf[0x05]   = red;

    hid_send_feature_report(dev, usb_buf, 8);

    usb_buf[0x04]   = offset + led * 3 + 1;
    usb_buf[0x05]   = blue;

    hid_send_feature_report(dev, usb_buf, 8);

    usb_buf[0x04]   = offset + led * 3 + 2;
    usb_buf[0x05]   = green;

    hid_send_feature_report(dev, usb_buf, 8);
}

void AuraMonitorController::ApplyChanges()
{
    unsigned char usb_buf[8];

    memset(usb_buf, 0x00, sizeof(usb_buf));

    usb_buf[0x00]   = 0x03;
    usb_buf[0x01]   = 0x02;
    usb_buf[0x02]   = 0xA1;
    usb_buf[0x03]   = 0x80;
    usb_buf[0x04]   = 0xA0;
    usb_buf[0x05]   = 0x01;

    hid_send_feature_report(dev, usb_buf, 8);
}