/*
 * Copyright (c) 2020-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.
 */

#include "input_event_hub.h"
#include "gfx_utils/graphic_log.h"

namespace OHOS {
namespace {
const uint32_t TOUCH_DEV_ID = 1;
const uint32_t MOUSE_DEV_ID = 2;
const uint32_t UNKNOW_DEV_ID = 32;
}
IInputInterface* InputEventHub::inputInterface_ = nullptr;
InputEventCb InputEventHub::callback_ = { 0 };
InputEventHub::ReadCallback InputEventHub::readCallback_ = nullptr;

InputEventHub* InputEventHub::GetInstance()
{
    static InputEventHub instance;
    return &instance;
}

InputEventHub::InputEventHub()
{
    for (uint8_t i = 0; i < MAX_INPUT_DEVICE_NUM; i++) {
        mountDevIndex_[i] = UNKNOW_DEV_ID;
    }
    openDev_ = 0;

    data_.deviceId = UNKNOW_DEV_ID;
    data_.state = 0;
    data_.timestamp = 0;
    data_.type = InputDevType::INDEV_TYPE_UNKNOWN;
    data_.x = 0;
    data_.y = 0;
}

void InputEventHub::SetUp()
{
    int32_t ret = GetInputInterface(&inputInterface_);
    if (ret != INPUT_SUCCESS) {
        GRAPHIC_LOGE("get input driver interface failed!");
        return;
    }
    uint8_t num = ScanInputDevice();
    if (num == 0) {
        GRAPHIC_LOGE("There is no device!");
        return;
    }
    for (uint8_t i = 0; i < num; i++) {
        if (inputInterface_ == nullptr || inputInterface_->iInputManager == nullptr) {
            GRAPHIC_LOGE("input interface or input manager is nullptr, open device failed!");
            return;
        }
        ret = inputInterface_->iInputManager->OpenInputDevice(mountDevIndex_[i]);
        if (ret == INPUT_SUCCESS && inputInterface_->iInputReporter != nullptr) {
            callback_.EventPkgCallback = EventCallback;
            ret = inputInterface_->iInputReporter->RegisterReportCallback(mountDevIndex_[i], &callback_);
            if (ret != INPUT_SUCCESS) {
                GRAPHIC_LOGE("device dose not exist, can't register callback to it!");
                return;
            }
            openDev_ = openDev_ | (1 << i);
        }
    }
}

void InputEventHub::TearDown()
{
    int32_t ret = 0;
    for (uint8_t i = 0; i < MAX_INPUT_DEVICE_NUM; i++) {
        if ((openDev_ & (1 << i)) == 0) {
            continue;
        }
        if (inputInterface_ == nullptr) {
            GRAPHIC_LOGE("input interface point is nullptr!");
            return;
        }
        if (inputInterface_->iInputReporter == nullptr || inputInterface_->iInputManager == nullptr) {
            GRAPHIC_LOGE("input interface or input manager is nullptr, open device failed!");
            return;
        }
        ret = inputInterface_->iInputReporter->UnregisterReportCallback(mountDevIndex_[i]);
        if (ret != INPUT_SUCCESS) {
            GRAPHIC_LOGE("Unregister callback failed!");
        }
        ret  = inputInterface_->iInputManager->CloseInputDevice(mountDevIndex_[i]);
        if (ret != INPUT_SUCCESS) {
            GRAPHIC_LOGE("Unmount device failed!");
        }
        openDev_ = openDev_ & ~(1 << i);
    }

    if (inputInterface_ != nullptr) {
        if (inputInterface_->iInputManager != nullptr) {
            free(inputInterface_->iInputManager);
        }
        if (inputInterface_->iInputReporter != nullptr) {
            free(inputInterface_->iInputReporter);
        }
        if (inputInterface_->iInputController != nullptr) {
            free(inputInterface_->iInputController);
        }
        free(inputInterface_);
        inputInterface_ = nullptr;
    }
}

void InputEventHub::EventCallback(const InputEventPackage **pkgs, uint32_t count, uint32_t devIndex)
{
    if (pkgs == nullptr || readCallback_ == nullptr || count == 0) {
        return;
    }

    RawEvent& data = InputEventHub::GetInstance()->data_;
    for (uint32_t i = 0; i < count; i++) {
        if (pkgs[i]->type == EV_REL) {
            data.type = InputDevType::INDEV_TYPE_MOUSE;
            if (pkgs[i]->code == REL_X)
                data.x += pkgs[i]->value;
            else if (pkgs[i]->code == REL_Y)
                data.y += pkgs[i]->value;
        } else if (pkgs[i]->type == EV_ABS) {
            data.type = InputDevType::INDEV_TYPE_TOUCH;
            if (pkgs[i]->code == ABS_MT_POSITION_X)
                data.x = pkgs[i]->value;
            else if (pkgs[i]->code == ABS_MT_POSITION_Y)
                data.y = pkgs[i]->value;
        } else if (pkgs[i]->type == EV_KEY) {
            if (pkgs[i]->code == BTN_MOUSE || pkgs[i]->code == BTN_TOUCH) {
                if (pkgs[i]->value == 0)
                    data.state = 0;
                else if (pkgs[i]->value == 1)
                    data.state = 1;
            }
        } else if (pkgs[i]->type == EV_SYN) {
            if (pkgs[i]->code == SYN_REPORT) {
                break;
            }
        }
    }

    readCallback_(&data);
}

InputDevType InputEventHub::GetDeviceType(uint32_t devIndex)
{
    switch (devIndex) {
        case TOUCH_DEV_ID: {
            return InputDevType::INDEV_TYPE_TOUCH;
        }
        case MOUSE_DEV_ID: {
            return InputDevType::INDEV_TYPE_MOUSE;
        }
        default: {
            return InputDevType::INDEV_TYPE_UNKNOWN;
        }
    }
}

uint8_t InputEventHub::ScanInputDevice()
{
    for (uint8_t i = 0; i < MAX_INPUT_DEVICE_NUM; i++) {
        mountDevIndex_[i] = UNKNOW_DEV_ID;
    }
    /* later will be change get device mode */
    uint32_t deviceNum = 0;
    mountDevIndex_[0] = TOUCH_DEV_ID;
    mountDevIndex_[1] = MOUSE_DEV_ID;
    deviceNum = 2; // 2:Number of current devices
    return deviceNum;
}
} // namespace OHOS