/*
 * Copyright (c) 2021-2022 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 "key_command_manager.h"
#include "ability_manager_client.h"
#include "file_ex.h"
#include "mmi_log.h"
#include "ohos/aafwk/base/string_wrapper.h"
#include "timer_manager.h"

namespace OHOS {
namespace MMI {
    namespace {
        constexpr int32_t MAX_PREKEYS_NUM = 4;
        constexpr int32_t INVALID_VALUE = -1;
        constexpr OHOS::HiviewDFX::HiLogLabel LABEL = { LOG_CORE, MMI_LOG_DOMAIN, "KeyCommandManager" };
    }

KeyCommandManager::KeyCommandManager()
{
    const std::string configFile = GetConfigFilePath();
    ResolveConfig(configFile);
    Print();
}

std::string KeyCommandManager::GenerateKey(const ShortcutKey& key)
{
    std::set<int32_t> preKeys = key.preKeys;
    std::stringstream oss;
    for (const auto preKey: preKeys) {
        oss << preKey << ",";
    }
    oss << key.finalKey << ",";
    oss << key.triggerType;
    return std::string(oss.str());
}

std::string KeyCommandManager::GetConfigFilePath()
{
    std::string defaultConfig = "/product/multimodalinput/ability_launch_config.json";
    return FileExists(defaultConfig) ? defaultConfig : "/system/etc/multimodalinput/ability_launch_config.json";
}

void KeyCommandManager::ResolveConfig(const std::string configFile)
{
    if (!FileExists(configFile)) {
        MMI_LOGE("config file %{public}s not exist", configFile.c_str());
        return;
    }
    MMI_LOGD("config file path:%{public}s", configFile.c_str());
    std::ifstream reader(configFile);
    if (!reader.is_open()) {
        MMI_LOGE("config file open failed");
        return;
    }
    json configJson;
    reader >> configJson;
    reader.close();
    if (configJson.empty()) {
        MMI_LOGE("config file is empty");
        return;
    }
    json shortkeys = configJson["Shortkeys"];
    if (!shortkeys.is_array() || shortkeys.empty()) {
        MMI_LOGE("shortkeys in config file is empty");
        return;
    }
    for (size_t i = 0; i < shortkeys.size(); i++) {
        ShortcutKey shortcutKey;
        if (!ConvertToShortcutKey(shortkeys[i], shortcutKey)) {
            continue;
        }
        std::string key = GenerateKey(shortcutKey);
        auto res = shortcutKeys_.find(key);
        if (res == shortcutKeys_.end()) {
            shortcutKeys_.emplace(key, shortcutKey);
        }
    }
}

bool KeyCommandManager::ConvertToShortcutKey(const json &jsonData, ShortcutKey &shortcutKey)
{
    json preKey = jsonData["preKey"];
    if (!preKey.is_array() || preKey.size() > MAX_PREKEYS_NUM) {
        MMI_LOGE("preKey number must less and equal four");
        return false;
    }

    for (size_t i = 0; i < preKey.size(); i++) {
        if (!preKey[i].is_number() || preKey[i] < 0) {
            MMI_LOGE("preKey must be number and bigger and equal 0");
            return false;
        }
        auto ret = shortcutKey.preKeys.emplace(preKey[i]);
        if (!ret.second) {
            MMI_LOGE("preKey must be unique");
            return false;
        }
    }
    if (!jsonData["finalKey"].is_number()) {
        MMI_LOGE("finalKey must be number");
        return false;
    }
    shortcutKey.finalKey = jsonData["finalKey"];
    if ((!jsonData["trigger"].is_string()) ||
        (jsonData["trigger"].get<std::string>() != "key_up" && jsonData["trigger"].get<std::string>() != "key_down")) {
        MMI_LOGE("trigger must be one of [key_up, key_down]");
        return false;
    }
    if (jsonData["trigger"].get<std::string>() == "key_up") {
        shortcutKey.triggerType = KeyEvent::KEY_ACTION_UP;
    } else {
        shortcutKey.triggerType = KeyEvent::KEY_ACTION_DOWN;
    }
    if ((!jsonData["keyDownDuration"].is_number()) || (jsonData["keyDownDuration"] < 0)) {
        MMI_LOGE("keyDownDuration must be number and bigger and equal zero");
        return false;
    }
    shortcutKey.keyDownDuration = jsonData["keyDownDuration"];
    if (!PackageAbility(jsonData["ability"], shortcutKey.ability)) {
        MMI_LOGE("package ability failed!");
        return false;
    }
    return true;
}

bool KeyCommandManager::PackageAbility(const json &jsonAbility, Ability &ability)
{
    if (!jsonAbility.is_object()) {
        MMI_LOGE("ability must be object");
        return false;
    }
    if (!jsonAbility["entities"].is_array()) {
        MMI_LOGE("entities must be array!");
        return false;
    }
    if (!jsonAbility["params"].is_array()) {
        MMI_LOGE("params must be array!");
        return false;
    }
    ability.bundleName = jsonAbility["bundleName"];
    ability.abilityName = jsonAbility["abilityName"];
    ability.action = jsonAbility["action"];
    ability.type = jsonAbility["type"];
    ability.deviceId = jsonAbility["deviceId"];
    ability.uri = jsonAbility["uri"];
    for (size_t i = 0; i < jsonAbility["entities"].size(); i++) {
        ability.entities.push_back(jsonAbility["entities"][i]);
    }
    json params = jsonAbility["params"];
    for (size_t i = 0; i < params.size(); i++) {
        if (!params[i].is_object()) {
            MMI_LOGE("param must be object");
            return false;
        }
        ability.params.emplace(params[i]["key"], params[i]["value"]);
    }
    return true;
}

void KeyCommandManager::Print()
{
    uint32_t count = shortcutKeys_.size();
    MMI_LOGD("shortcutKey count:%{public}u", count);
    for (const auto &item : shortcutKeys_) {
        auto &shortcutKey = item.second;
        for (auto prekey: shortcutKey.preKeys) {
            MMI_LOGD("preKey:%{public}d", prekey);
        }
        MMI_LOGD("finalKey:%{public}d,keyDownDuration:%{public}d,triggerType:%{public}d,"
            " bundleName:%{public}s,abilityName:%{public}s", shortcutKey.finalKey,
            shortcutKey.keyDownDuration, shortcutKey.triggerType,
            shortcutKey.ability.bundleName.c_str(), shortcutKey.ability.abilityName.c_str());
    }
}

bool KeyCommandManager::HandlerEvent(const std::shared_ptr<KeyEvent> event)
{
    MMI_LOGD("enter");
    if (IsKeyMatch(lastMatchedKey_, event)) {
        MMI_LOGE("The same event is waiting timeout, skip");
        return true;
    }
    if (lastMatchedKey_.timerId >= 0) {
        MMI_LOGE("remove timer timeid:%{public}d", lastMatchedKey_.timerId);
        TimerMgr->RemoveTimer(lastMatchedKey_.timerId);
    }
    ResetLastMatchedKey();
    for (auto iter = shortcutKeys_.begin(); iter != shortcutKeys_.end(); ++iter) {
        ShortcutKey &shortcutKey = iter->second;
        if (!IsKeyMatch(shortcutKey, event)) {
            MMI_LOGD("not matched, next");
            continue;
        }
        for (const auto &prekey: shortcutKey.preKeys) {
            MMI_LOGD("eventkey matched, preKey:%{public}d", prekey);
        }
        MMI_LOGD("eventkey matched, finalKey:%{public}d,bundleName:%{public}s",
            shortcutKey.finalKey, shortcutKey.ability.bundleName.c_str());
        if (shortcutKey.triggerType == KeyEvent::KEY_ACTION_DOWN) {
            return HandleKeyDown(shortcutKey);
        } else if (shortcutKey.triggerType == KeyEvent::KEY_ACTION_UP) {
            return HandleKeyUp(event, shortcutKey);
        } else {
            return HandleKeyCancel(shortcutKey);
        }
    }
    MMI_LOGD("leave");
    return false;
}

bool KeyCommandManager::IsKeyMatch(const ShortcutKey &shortcutKey, const std::shared_ptr<KeyEvent> &key)
{
    MMI_LOGD("enter");
    if (key->GetKeyCode() != shortcutKey.finalKey || shortcutKey.triggerType != key->GetKeyAction()) {
        return false;
    }
    if ((shortcutKey.preKeys.size()) != (key->GetKeyItems().size() - 1)) {  // KeyItems contain finalkey, so decrease 1
        return false;
    }
    for (const auto &item : key->GetKeyItems()) {
        int32_t keyCode = item.GetKeyCode();
        if (keyCode == key->GetKeyCode()) {
            continue;
        }
        auto res = shortcutKey.preKeys.find(keyCode);
        if (res == shortcutKey.preKeys.end()) {
            return false;
        }
    }
    MMI_LOGD("matched...");
    return true;
}

bool KeyCommandManager::HandleKeyDown(ShortcutKey &shortcutKey)
{
    MMI_LOGD("enter");
    if (shortcutKey.keyDownDuration == 0) {
        MMI_LOGD("Start launch ability immediately");
        LaunchAbility(shortcutKey);
    } else {
        shortcutKey.timerId = TimerMgr->AddTimer(shortcutKey.keyDownDuration, 1, [this, shortcutKey] () {
            MMI_LOGD("Timer callback");
            LaunchAbility(shortcutKey);
        });
        if (shortcutKey.timerId < 0) {
            MMI_LOGE("Timer add failed");
            return false;
        }
        MMI_LOGD("add timer success, timeid:%{public}d", shortcutKey.timerId);
        lastMatchedKey_ = shortcutKey;
    }
    return true;
}

bool KeyCommandManager::HandleKeyUp(const std::shared_ptr<KeyEvent> &keyEvent, const ShortcutKey &shortcutKey)
{
    MMI_LOGD("enter");
    if (shortcutKey.keyDownDuration == 0) {
        MMI_LOGD("Start launch ability immediately");
        LaunchAbility(shortcutKey);
        return true;
    } else {
        const KeyEvent::KeyItem* keyItem = keyEvent->GetKeyItem();
        CHKPF(keyItem);
        auto upTime = keyEvent->GetActionTime();
        auto downTime = keyItem->GetDownTime();
        MMI_LOGD("UpTime:%{public}" PRId64 ",downTime:%{public}" PRId64 ",keyDownDuration:%{public}d",
            upTime, downTime, shortcutKey.keyDownDuration);
        if (upTime - downTime >= static_cast<int64_t>(shortcutKey.keyDownDuration) * 1000) {
            MMI_LOGD("Skip, upTime - downTime >= duration");
            return false;
        }
        MMI_LOGD("Start launch ability immediately");
        LaunchAbility(shortcutKey);
        return true;
    }
}

bool KeyCommandManager::HandleKeyCancel(ShortcutKey &shortcutKey)
{
    MMI_LOGD("enter");
    if (shortcutKey.timerId < 0) {
        MMI_LOGE("Skip, timerid < 0");
    }
    auto timerId = shortcutKey.timerId;
    shortcutKey.timerId = -1;
    TimerMgr->RemoveTimer(timerId);
    MMI_LOGD("Leave, timerId: %{public}d", timerId);
    return false;
}

void KeyCommandManager::LaunchAbility(ShortcutKey key)
{
    AAFwk::Want want;
    want.SetElementName(key.ability.deviceId, key.ability.bundleName, key.ability.abilityName);
    want.SetAction(key.ability.action);
    want.SetUri(key.ability.uri);
    want.SetType(key.ability.uri);
    for (const auto &entity : key.ability.entities) {
        want.AddEntity(entity);
    }
    AAFwk::WantParams wParams;
    for (const auto &item : key.ability.params) {
        wParams.SetParam(item.first, AAFwk::String::Box(item.second));
    }
    want.SetParams(wParams);
    MMI_LOGD("Start launch ability, bundleName:%{public}s", key.ability.bundleName.c_str());
    ErrCode err = AAFwk::AbilityManagerClient::GetInstance()->StartAbility(want);
    if (err != ERR_OK) {
        MMI_LOGE("LaunchAbility failed, bundleName:%{public}s,err:%{public}d", key.ability.bundleName.c_str(), err);
    }
    ResetLastMatchedKey();
    MMI_LOGD("End launch ability, bundleName:%{public}s", key.ability.bundleName.c_str());
}

void KeyCommandManager::ResetLastMatchedKey()
{
    lastMatchedKey_.preKeys.clear();
    lastMatchedKey_.finalKey = INVALID_VALUE;
    lastMatchedKey_.timerId = INVALID_VALUE;
}

std::shared_ptr<IKeyCommandManager> IKeyCommandManager::GetInstance()
{
    if (keyCommand_ == nullptr) {
        keyCommand_ = std::make_shared<KeyCommandManager>();
    }
    return keyCommand_;
}
} // namespace MMI
} // namespace OHOS