//
// Created by maxmati on 4/18/15.
//

// DPI change support added by SouceCalve on 27/11/25.


#include "Mouse.h"
#include <cstring>
#include <iostream>

using std::cout; using std::endl;

void Mouse::init() {
    int ret = libusb_init(&context);
    if(ret < 0) {
        cout<<"Init Error "<<ret<<endl;
        return;
    }

    libusb_set_debug(context, LIBUSB_LOG_LEVEL_INFO);

    discoverDevices();

}

void Mouse::discoverDevices() {
    libusb_device** devs;
    ssize_t cnt = libusb_get_device_list(context, &devs);
    if(cnt < 0) {
        cout<<"Get Device Error"<<endl;
        return;
    }

    for(int i = 0; i < cnt; ++i) {
        libusb_device_descriptor desc;
        libusb_get_device_descriptor(devs[i], &desc);
        if (isCompatibleDevice(desc)) {
            switch (int status = libusb_open(devs[i], &currentDevice)) {
                case 0:
                    break;
                case LIBUSB_ERROR_NO_MEM:
                    cout << "LIBUSB_ERROR_NO_MEM" << endl;
                    continue;
                case LIBUSB_ERROR_ACCESS:
                    cout << "LIBUSB_ERROR_ACCESS" << endl;
                    continue;
                case LIBUSB_ERROR_NO_DEVICE:
                    cout << "LIBUSB_ERROR_NO_DEVICE" << endl;
                    continue;
                default:
                    cout << "Status: " << status << endl;
                    continue;
            }

            if(libusb_kernel_driver_active(currentDevice, 2) == 1)
                if(libusb_detach_kernel_driver(currentDevice, 2) != 0) {
                    libusb_close(currentDevice);
                    continue;
                }

            devices.insert(std::pair<int, libusb_device_handle*>(libusb_get_device_address(devs[i]), currentDevice));

        }

    }
    libusb_free_device_list(devs, 1);

    if(devices.size() == 0){
        cout<<"No suitable device found."<<endl;
        return;
    }

    currentDevice = devices.begin()->second;
}

bool Mouse::isCompatibleDevice(libusb_device_descriptor &desc) {
    if(desc.idVendor != A4TECH_VID)
        return false;

    for(size_t i = 0; i < COMPATIBLE_PIDS_SIZE; ++i)
        if(desc.idProduct == COMPATIBLE_PIDS[i])
            return true;

    return false;
}

Mouse::~Mouse() {
    for (auto& dev : devices)
        libusb_close(dev.second);
    if(context != nullptr)
        libusb_exit(context);
}

int Mouse::setBackLightLevel(uint8_t level) {
    uint8_t data[72] = {A4TECH_MAGIC, BACKLIGHT_OPCODE,0x00,0x00,
                        BACKLIGHT_WRITE,0x00,0x00,0x00, level, 0x00, };

    if(level < 0 || level > 3){
        return -1;
    }

    if(writeToMouse(data, sizeof(data)) < 0)
        return -2;

    return 0;
}

int Mouse::writeToMouse(uint8_t data[], size_t size) {
    int res = libusb_control_transfer(currentDevice, 0x21, 9,
                                      0x0307, 2, data, size, 10000);
    switch (res){
        case LIBUSB_ERROR_TIMEOUT:
            cout << "LIBUSB_ERROR_TIMEOUT" << endl;
            return -1;
        case LIBUSB_ERROR_PIPE:
            cout << "LIBUSB_ERROR_PIPE" << endl;
            return -1;
        case LIBUSB_ERROR_NO_DEVICE:
            cout << "LIBUSB_ERROR_NO_DEVICE" << endl;
            return -1;
        case LIBUSB_ERROR_BUSY:
            cout << "LIBUSB_ERROR_BUSY" << endl;
            return -1;
        case LIBUSB_ERROR_INVALID_PARAM:
            cout << "LIBUSB_ERROR_INVALID_PARAM" << endl;
        default:
            return 0;
    }
}

int Mouse::readFromMouse(uint8_t *request, size_t requestSize, uint8_t *response, size_t responseSize) {
    if(writeToMouse(request, requestSize) < 0){
        return -1;
    }

    int res =  libusb_control_transfer(currentDevice,0xa1,1,
                                       0x0307,2,response,responseSize,10000);
    if(res < 0){
        cout<<"Unnable to receive data"<<endl;
        return -2;
    }
    return 0;

}

uint8_t Mouse::getBackLightLevel() {
    uint8_t request[72] = {A4TECH_MAGIC, BACKLIGHT_OPCODE,0x00,0x00,
                        BACKLIGHT_READ,0x00,};
    uint8_t response[72];

    readFromMouse(request, sizeof(request), response, sizeof(response));

    return response[8];
}

uint8_t Mouse::setDPI(uint8_t dpi){
    if(dpi < 0 || dpi > 40) dpi=32;
    uint8_t data[72] = {A4TECH_MAGIC, DPI_SET_OPCODE,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x1a,0x1a,0x88,0x02,0x00, };
    memcpy(data + 10, dpi_binary+dpi*3, 3); 
    return writeToMouse(data, sizeof(data));
}

void Mouse::listDevices() {
    std::cout<<"Available devices:"<<endl;
    for (auto&devHand : devices){
        libusb_device* device = libusb_get_device(devHand.second);

        libusb_device_descriptor desc;
        libusb_get_device_descriptor(device, &desc);

        std::string name;

        switch (desc.idProduct){
            case BLOODY_V5_PID:
                name = "Bloody V5";
                break;
            case BLOODY_V7_PID:
                name = "Bloody V7";
                break;
            case BLOODY_V8_PID:
                name = "Bloody V8";
                break;
            case BLOODY_R7_PID:
                name = "Bloody R7";
                break;
            case BLOODY_R8_1_PID:
                name = "Bloody R8-1";
                break;
            case BLOODY_R3_PID:
                name = "Bloody R3";
                break;
            case BLOODY_AL9_PID:
                name = "Bloody AL9";
                break;
            case BLOODY_R70_PID:
                name = "Bloody R70";
                break;
            case BLOODY_A7_PID:
                name = "Bloody A7";
                break;
            case BLOODY_A9_PID:
                name = "Bloody A9";
                break;
            case BLOODY_RT5_PID:
                name = "Bloody RT5";
                break;
            case BLOODY_V8M_PID:
                name = "Bloody V8M";
		        break;
            case BLOODY_R8_PID:
                name = "Bloody R8";
                break;
            case BLOODY_J95S_PID:
                name = "Bloody J95S";
                break;
            case BLOODY_RT7_PID:
                name = "Bloody RT7";
                break;
            default:
                name = "Unknown";
        }

        std::cout<< devHand.first<<":"<<name<<endl;
    }

}

bool Mouse::selectDevice(int address) {
    if(devices.count(address) == 0)
        return false;

    currentDevice = devices.at(address);
    return true;
}