BloomPatched/src/DebugToolDrivers/USB/UsbDevice.hpp

#pragma once

#include <cstdint>
#include <memory>
#include <vector>
#include <optional>
#include <libusb-1.0/libusb.h>

#include "src/DebugToolDrivers/DebugTool.hpp"

namespace Bloom::Usb
{
    class UsbDevice
    {
    public:
        UsbDevice(std::uint16_t vendorId, std::uint16_t productId): vendorId(vendorId), productId(productId) {};
        ~UsbDevice() = default;

        void init();

        [[nodiscard]] libusb_device* getLibUsbDevice() const {
            return this->libUsbDevice;
        }

        void setLibUsbDevice(libusb_device* libUsbDevice) {
            this->libUsbDevice = libUsbDevice;
        }

        [[nodiscard]] std::uint16_t getVendorId() const {
            return this->vendorId;
        }

        [[nodiscard]] std::uint16_t getProductId() const {
            return this->productId;
        }

        /**
         * Selects a specific configuration on the device, using the configuration index.
         *
         * @param configIndex
         */
        virtual void setConfiguration(int configIndex);

    protected:
        libusb_context* libUsbContext = nullptr;
        libusb_device* libUsbDevice = nullptr;
        libusb_device_handle* libUsbDeviceHandle = nullptr;
        std::uint16_t vendorId;
        std::uint16_t productId;

        std::vector<libusb_device*> findMatchingDevices(
            std::optional<std::uint16_t> vendorId = std::nullopt, std::optional<std::uint16_t> productId = std::nullopt
        );

        void close();
    };
}