/*
 * Copyright (c) 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 "executor.h"

#include <sstream>

#include "framework_executor_callback.h"
#include "iam_check.h"
#include "iam_common_defines.h"
#include "iam_logger.h"
#include "iam_mem.h"
#include "iam_para2str.h"
#include "iam_ptr.h"

#define LOG_LABEL UserIam::Common::LABEL_USER_AUTH_EXECUTOR

namespace OHOS {
namespace UserIam {
namespace UserAuth {
Executor::Executor(std::shared_ptr<ExecutorMgrWrapper> executorMgrWrapper,
    std::shared_ptr<IAuthExecutorHdi> executorHdi, uint16_t hdiId)
    : executorMgrWrapper_(executorMgrWrapper),
      executorHdi_(executorHdi),
      hdiId_(hdiId)
{
    auto hdi = executorHdi_;
    IF_FALSE_LOGE_AND_RETURN(hdi != nullptr);
    ExecutorInfo executorInfo = {};
    IF_FALSE_LOGE_AND_RETURN(hdi->GetExecutorInfo(executorInfo) == ResultCode::SUCCESS);
    authType_ = executorInfo.authType;
    std::ostringstream ss;
    uint32_t combineExecutorId =
        Common::CombineUint16ToUint32(hdiId_, static_cast<uint16_t>(executorInfo.executorSensorHint));
    const uint32_t uint32HexWidth = 8;
    ss << "Executor(Id:0x" << std::setfill('0') << std::setw(uint32HexWidth) << std::hex << combineExecutorId << ")";
    description_ = ss.str();
}

void Executor::OnHdiConnect()
{
    IAM_LOGI("%{public}s start", GetDescription());
    // register resource pool depends on hdi connect, after hid connect re-register resource pool
    OnFrameworkReady();
}

void Executor::OnHdiDisconnect()
{
    IAM_LOGI("%{public}s start", GetDescription());
    executorHdi_ = nullptr;
    RespondCallbackOnDisconnect();
}

void Executor::OnFrameworkReady()
{
    IAM_LOGI("%{public}s start", GetDescription());
    ExecutorInfo executorInfo = {};
    auto hdi = executorHdi_;
    if (hdi == nullptr) {
        IAM_LOGE("executorHdi_ is disconnected, skip framework ready process");
        return;
    }
    ResultCode ret = hdi->GetExecutorInfo(executorInfo);
    if (ret != ResultCode::SUCCESS) {
        IAM_LOGE("Get executor info failed");
        return;
    }
    RegisterExecutorCallback(executorInfo);
}

void Executor::RegisterExecutorCallback(ExecutorInfo &executorInfo)
{
    IAM_LOGI("%{public}s start", GetDescription());
    uint32_t combineExecutorId =
        Common::CombineUint16ToUint32(hdiId_, static_cast<uint16_t>(executorInfo.executorSensorHint));
    executorInfo.executorSensorHint = combineExecutorId;
    if (executorCallback_ == nullptr) {
        std::lock_guard<std::mutex> lock(mutex_);
        if (executorCallback_ == nullptr) {
            auto localExecutorCallback = Common::MakeShared<FrameworkExecutorCallback>(weak_from_this());
            IF_FALSE_LOGE_AND_RETURN(localExecutorCallback != nullptr);
            executorCallback_ = localExecutorCallback;
        }
    }
    IF_FALSE_LOGE_AND_RETURN(executorMgrWrapper_ != nullptr);
    executorMgrWrapper_->Register(executorInfo, executorCallback_);
    IAM_LOGI("register executor callback ok, executor id %{public}s",
        GET_MASKED_STRING(executorInfo.executorSensorHint).c_str());
}

void Executor::AddCommand(std::shared_ptr<IAsyncCommand> command)
{
    IAM_LOGI("%{public}s start", GetDescription());
    IF_FALSE_LOGE_AND_RETURN(command != nullptr);
    std::lock_guard<std::mutex> lock(mutex_);
    command2Respond_.insert(command);
}

void Executor::RemoveCommand(std::shared_ptr<IAsyncCommand> command)
{
    IAM_LOGI("%{public}s start", GetDescription());
    std::lock_guard<std::mutex> lock(mutex_);
    command2Respond_.erase(command);
}

void Executor::RespondCallbackOnDisconnect()
{
    IAM_LOGI("%{public}s start", GetDescription());
    std::set<std::shared_ptr<IAsyncCommand>> command2NotifyOnHdiDisconnect;
    {
        // cmd->OnHdiDisconnect will invoke RemoveCommand thus modify command2Respond_, make a copy before call
        // OnHdiDisconnect
        std::lock_guard<std::mutex> lock(mutex_);
        command2NotifyOnHdiDisconnect =
            std::set<std::shared_ptr<IAsyncCommand>>(command2Respond_.begin(), command2Respond_.end());
    }

    for (const auto &cmd : command2NotifyOnHdiDisconnect) {
        if (cmd == nullptr) {
            IAM_LOGE("cmd is null");
            continue;
        }
        cmd->OnHdiDisconnect();
    }
    IAM_LOGI("success");
}

std::shared_ptr<IAuthExecutorHdi> Executor::GetExecutorHdi()
{
    return executorHdi_;
}

const char *Executor::GetDescription()
{
    return description_.c_str();
}

int32_t Executor::GetAuthType() const
{
    return authType_;
}
} // namespace UserAuth
} // namespace UserIam
} // namespace OHOS