/*
 * Copyright (C) 2021 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#ifndef HDC_HOST_USB_H
#define HDC_HOST_USB_H
#include "host_common.h"

namespace Hdc {
class HdcHostUSB : public HdcUSBBase {
public:
    HdcHostUSB(const bool serverOrDaemonIn, void *ptrMainBase, void *ctxUSBin);
    virtual ~HdcHostUSB();
    int Initial();
    void InitLogging(void *ctxUSB);
    int SendUSBRaw(HSession hSession, uint8_t *data, const int length) override;
    HSession ConnectDetectDaemon(const HSession hSession, const HDaemonInfo pdi);
    void Stop();
    void RemoveIgnoreDevice(string &mountInfo);

private:
    enum UsbCheckStatus {
        HOST_USB_IGNORE = 1,
        HOST_USB_READY,
        HOST_USB_REGISTER,
    };
    static int LIBUSB_CALL HotplugHostUSBCallback(libusb_context *ctx, libusb_device *device,
                                                  libusb_hotplug_event event, void *userData);
    static void UsbWorkThread(void *arg);  // 3rd thread
    static void WatchUsbNodeChange(uv_timer_t *handle);
    static void KickoutZombie(HSession hSession);
    static void LIBUSB_CALL USBBulkCallback(struct libusb_transfer *transfer);
    int StartupUSBWork();
    int CheckActiveConfig(libusb_device *device, HUSB hUSB, libusb_device_descriptor& desc);
    int OpenDeviceMyNeed(HUSB hUSB);
    int CheckDescriptor(HUSB hUSB, libusb_device_descriptor& desc);
    bool IsDebuggableDev(const struct libusb_interface_descriptor *ifDescriptor);
    bool ReadyForWorkThread(HSession hSession) override;
    bool FindDeviceByID(HUSB hUSB, const char *usbMountPoint, libusb_context *ctxUSB);
    bool DetectMyNeed(libusb_device *device, string &sn);
    void RestoreHdcProtocol(HUSB hUsb, const uint8_t *buf, int bufSize);
    void UpdateUSBDaemonInfo(HUSB hUSB, HSession hSession, uint8_t connStatus);
    void BeginUsbRead(HSession hSession);
    void ReviewUsbNodeLater(string &nodeKey);
    void CancelUsbIo(HSession hSession) override;
    int UsbToHdcProtocol(uv_stream_t *stream, uint8_t *appendData, int dataSize) override;
    int SubmitUsbBio(HSession hSession, bool sendOrRecv, uint8_t *buf, int bufSize);

    libusb_context *ctxUSB;
    uv_timer_t devListWatcher;
    map<string, UsbCheckStatus> mapIgnoreDevice;

private:
    uv_thread_t threadUsbWork;
};
}  // namespace Hdc
#endif