#pragma once

#include "libusbwrap/interface/IUsbDeviceFactory.hpp"

namespace libusbwrap {
class UsbDeviceFactory : public IUsbDeviceFactory {
   public:
    UsbDeviceFactory();
    ~UsbDeviceFactory();

    // delete copy ctor and assignment
    UsbDeviceFactory(const UsbDeviceFactory&)      = delete;
    UsbDeviceFactory& operator=(UsbDeviceFactory&) = delete;

    bool init();
    /**
     * @brief Gets all devices that are detected by libusb. Will allocate a shared_ptr for each Device
     * 
     * @return std::vector<std::shared_ptr<IUsbDevice>> Vector of all detected USB devices
     */
    std::vector<std::shared_ptr<IUsbDevice>> findAllDevices() override;
    /**
     * @brief Gets all devices with certain vid/pid combination. 
     *        If only one device of certain type is connected, vector is usually only one element
     * 
     * @param vid VendorId of the devices to find
     * @param pid ProductId of the devices to find
     * @return std::vector<std::shared_ptr<IUsbDevice>> Vector of detected USB devices based on vid/pid
     */
    std::vector<std::shared_ptr<IUsbDevice>> findDevices(uint16_t vid, uint16_t pid) override;

   private:
    // methods
    int refreshDeviceList();
    std::vector<std::shared_ptr<IUsbDevice>> buildMaskedDeviceVector(uint16_t vidMask,
                                                                     uint16_t pidMask, uint16_t vid,
                                                                     uint16_t pid);
    // members
    libusb_context* mLibusbCtx{nullptr};
    libusb_device** mLibusbDeviceList;
};
}  // namespace libusbwrap