#ifndef AES_USB_COMM_H
#define AES_USB_COMM_H

#include <string>
#include "vector"
#include "stdint.h"

#ifdef __OHOS__
#include <usb/usb_ddk_api.h>
#include <usb/usb_ddk_types.h>
#include "hid/hid_ddk_api.h"
#include "hid/hid_ddk_types.h"
#else

#endif

#define AES_BUFFER_MAX 60

typedef uint8_t AES_Buffer;

class AES_USB {
    class HIDDevice;
    HIDDevice* hidDevice;
    typedef enum {
        // 缓冲区过大或错误,缓冲区最大60字节
        BUFFER_MAX_ERROR = -2,
        // 设备不存在
        DEVICE_NOT_EXIST = -1,
        // UserPIN 错误
        DEVICE_USER_PIN_ERROR = -3,
    } HID_ERROR;
    char devicePIN[21];
public:
    typedef struct OH_USB_Driver{
        int32_t vendor_id;
        int32_t product_id;
        int64_t deviceId;
    }OH_USB_Driver;
    /**
     * @brief 初始化USB
     * @return int 0成功 -1失败
     */
    static int init();
    /**
     * @brief 释放USB
     */
    static void release();
    AES_USB();
    ~AES_USB();
    /**
     * @brief 打开设备
     * @param vendor_id 设备VID
     * @param product_id 设备PID
     * @param drivers USB列表(只有鸿蒙需要)
     * @return int 1成功 -1失败
     */
    int open(uint16_t vendor_id, uint16_t product_id,const std::vector<OH_USB_Driver>& drivers = std::vector<OH_USB_Driver>()) const;
    /**
     * @brief 设置用户PIN
     * @param pin 用户PIN
     * @return 设置用户PIN结果 0: 失败 1: 成功 -1: PIN长度错误
     */
    int setUserPIN(const std::string& pin);
    /**
     * @brief 写入指定的VID和PID
     * @param vendor_id VID
     * @param product_id PID
     * @return int 0成功 -1失败
     */
    int set_vid_pid(uint16_t vendor_id, uint16_t product_id) const;
    /**
     * @brief 读取VID和PID值
     * @param vendor_id VID
     * @param product_id PID
     * @return int 1: 成功 -1: 失败
     */
    int read_vid_pid(uint16_t* vendor_id, uint16_t* product_id) const;
    /**
     * @brief 写入数据
     * @param data 数据
     * @param length 数据长度
     * @param timeout 超时时间
     * @return int32_t 大于0成功 小于等于0失败 -> HID_ERROR
     */
    int32_t write(uint8_t* data, int length, int timeout = 1000 * 1000) const;
    /**
     * @brief 读取数据
     * @param data 数据
     * @param length 数据长度
     * @param timeout 超时
     * @return int32_t 大于0成功 小于等于0失败 -> HID_ERROR
     */
    int32_t read(uint8_t* data, int length, int timeout = 1000 * 1000) const;
    /**
     * @brief 关闭设备
     */
    void close() const;
    //---------------厂家函数------------------------------------------------------------------------------
    static int resetAll();
    typedef struct PID_Key {
        uint16_t vendor_id;
        uint16_t product_id;
        uint8_t PIN[20];
    } PID_Key;
    static int readID_Pin(PID_Key* key);
    static int writeID_Pin(const PID_Key* key);
    //---------------厂家函数------------------------------------------------------------------------------
};

#endif