/*
 * Copyright (c) 2021-2023 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 INPUT_HUB_H
#define INPUT_HUB_H

#include <atomic>
#include <mutex>
#include <map>
#include <unordered_map>

#include <libevdev/libevdev.h>
#include <linux/input.h>
#include <sys/epoll.h>
#include <sys/inotify.h>

#include "constants_dinput.h"

namespace OHOS {
namespace DistributedHardware {
namespace DistributedInput {
struct AffectDhIds {
    std::vector<std::string> sharingDhIds;
    std::vector<std::string> noSharingDhIds;
};

class InputHub {
public:
    InputHub();
    ~InputHub();
    size_t StartCollectInputEvents(RawEvent *buffer, size_t bufferSize);
    size_t StartCollectInputHandler(InputDeviceEvent *buffer, size_t bufferSize);
    void StopCollectInputEvents();
    void StopCollectInputHandler();
    size_t DeviceIsExists(InputDeviceEvent *buffer, size_t bufferSize);
    std::vector<InputDevice> GetAllInputDevices();
    // return efftive dhids
    AffectDhIds SetSupportInputType(bool enabled, const uint32_t &inputTypes);
    // return efftive dhids
    AffectDhIds SetSharingDevices(bool enabled, std::vector<std::string> dhIds);
    void GetDevicesInfoByType(const uint32_t inputTypes, std::map<int32_t, std::string> &datas);
    void GetDevicesInfoByDhId(std::vector<std::string> dhidsVec, std::map<int32_t, std::string> &datas);
    void GetShareMousePathByDhId(std::vector<std::string> dhIds, std::string &path, std::string &dhId);
    void GetShareKeyboardPathsByDhIds(std::vector<std::string> dhIds, std::vector<std::string> &shareDhidsPaths,
        std::vector<std::string> &shareDhIds);
    bool IsAllDevicesStoped();
    void ScanInputDevices(const std::string &dirName);

private:
    struct Device  {
        Device* next;
        int fd; // may be -1 if device is closed
        const int32_t id;
        const std::string path;
        InputDevice identifier;
        uint32_t classes;
        uint8_t keyBitmask[(KEY_MAX + 1) / 8];
        uint8_t absBitmask[(ABS_MAX + 1) / 8];
        uint8_t relBitmask[(REL_MAX + 1) / 8];

        Device(int fd, int32_t id, const std::string &path, const InputDevice &identifier);
        ~Device();
        void Close();
        bool enabled; // initially true
        bool isShare;
        int32_t Enable();
        int32_t Disable();
        bool HasValidFd() const;
        const bool isVirtual; // set if fd < 0 is passed to constructor
    };

    struct AbsInfo {
        uint32_t absX;
        uint32_t absY;
        int32_t absXIndex;
        int32_t absYIndex;
    };

    int32_t Initialize();
    int32_t Release();

    size_t GetEvents(RawEvent *buffer, size_t bufferSize);
    size_t ReadInputEvent(int32_t readSize, Device &device);
    void GetDeviceHandler();
    int32_t RefreshEpollItem(bool isSleep);

    int32_t OpenInputDeviceLocked(const std::string &devicePath);
    int32_t QueryInputDeviceInfo(int fd, InputDevice &identifier);
    void QueryEventInfo(int fd, InputDevice &identifier);
    struct libevdev* GetLibEvDev(int fd);
    void GetEventTypes(struct libevdev *dev, InputDevice &identifier);
    int32_t GetEventKeys(struct libevdev *dev, InputDevice &identifier);
    int32_t GetABSInfo(struct libevdev *dev, InputDevice &identifier);
    int32_t GetRELTypes(struct libevdev *dev, InputDevice &identifier);
    void GetProperties(struct libevdev *dev, InputDevice &identifier);
    int32_t MakeDevice(int fd, std::unique_ptr<Device> device);
    void GenerateDescriptor(InputDevice &identifier) const;
    std::string StringPrintf(const char *format, ...) const;

    int32_t RegisterFdForEpoll(int fd);
    int32_t RegisterDeviceForEpollLocked(const Device &device);
    void AddDeviceLocked(std::unique_ptr<Device> device);
    void CloseDeviceLocked(Device &device);
    void CloseDeviceForAllLocked(Device &device);
    int32_t UnregisterDeviceFromEpollLocked(const Device &device) const;
    int32_t UnregisterFdFromEpoll(int fd) const;
    int32_t ReadNotifyLocked();
    void CloseDeviceByPathLocked(const std::string &devicePath);
    void CloseAllDevicesLocked();
    void JudgeDeviceOpenOrClose(const inotify_event &event);
    Device* GetDeviceByPathLocked(const std::string &devicePath);
    Device* GetDeviceByFdLocked(int fd);
    Device* GetSupportDeviceByFd(int fd);
    bool IsDeviceRegistered(const std::string &devicePath);

    bool ContainsNonZeroByte(const uint8_t *array, uint32_t startIndex, uint32_t endIndex);
    int64_t ProcessEventTimestamp(const input_event &event);
    bool IsTouchPad(const InputDevice &inputDevice);

    /*
     * this macro is used to tell if "bit" is set in "array"
     * it selects a byte from the array, and does a boolean AND
     * operation with a byte that only has the relevant bit set.
     * eg. to check for the 12th bit, we do (array[1] & 1<<4)
     */
    bool TestBit(uint32_t bit, const uint8_t *array);
    /* this macro computes the number of bytes needed to represent a bit array of the specified size */
    uint32_t SizeofBitArray(uint32_t bit);
    void RecordEventLog(const RawEvent *event);
    void RecordDeviceLog(const int32_t deviceId, const std::string &devicePath, const InputDevice &identifier);
    void HandleTouchScreenEvent(struct input_event readBuffer[], const size_t count, std::vector<bool> &needFilted);
    int32_t QueryLocalTouchScreenInfo(int fd);
    bool CheckTouchPointRegion(struct input_event readBuffer[], const AbsInfo &absInfo);
    size_t CollectEvent(RawEvent *buffer, size_t &capacity, Device *device, struct input_event readBuffer[],
        const size_t count);
    /*
     * isEnable: true for sharing dhid, false for no sharing dhid
     */
    void SaveAffectDhId(bool isEnable, const std::string &dhId, AffectDhIds &affDhIds);

    int epollFd_;
    int iNotifyFd_;
    int inputWd_;

    std::vector<std::unique_ptr<Device>> openingDevices_;
    std::vector<std::unique_ptr<Device>> closingDevices_;
    std::unordered_map<int32_t, std::unique_ptr<Device>> devices_;
    std::mutex devicesMutex_;

    std::atomic<bool> needToScanDevices_;
    std::string deviceId_;
    std::string touchDescriptor;
    std::atomic<int32_t> nextDeviceId_;

    // The array of pending epoll events and the index of the next event to be handled.
    struct epoll_event mPendingEventItems[EPOLL_MAX_EVENTS];
    std::atomic<int32_t> pendingEventCount_;
    std::atomic<int32_t> pendingEventIndex_;
    std::atomic<bool> pendingINotify_;
    std::mutex operationMutex_;

    std::atomic<bool> deviceChanged_;
    std::atomic<uint32_t> inputTypes_;
    std::atomic<bool> isStartCollectEvent_;
    std::atomic<bool> isStartCollectHandler_;
    std::unordered_map<std::string, bool> sharedDHIds_;
};
} // namespace DistributedInput
} // namespace DistributedHardware
} // namespace OHOS

#endif // INPUT_HUB_H