/*
 * 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 "daudio_manager_callback.h"

#include <string>
#include <hdf_base.h>

#include "daudio_constants.h"
#include "daudio_errorcode.h"
#include "daudio_hdf_operate.h"
#include "daudio_hdi_handler.h"
#include "daudio_hitrace.h"
#include "daudio_log.h"
#include "daudio_util.h"

#undef DH_LOG_TAG
#define DH_LOG_TAG "DAudioHdiHandler"

namespace OHOS {
namespace DistributedHardware {
IMPLEMENT_SINGLE_INSTANCE(DAudioHdiHandler);

DAudioHdiHandler::DAudioHdiHandler()
{
    DHLOGD("Distributed audio hdi handler construct.");
}

DAudioHdiHandler::~DAudioHdiHandler()
{
    DHLOGD("Distributed audio hdi handler deconstructed.");
}

int32_t DAudioHdiHandler::InitHdiHandler()
{
    DHLOGI("Init hdi handler.");
    if (audioSrvHdf_ != nullptr) {
        return DH_SUCCESS;
    }

    DAUDIO_SYNC_TRACE(DAUDIO_LOAD_HDF_DRIVER);
    DHLOGD("Load hdf driver start.");
    int32_t ret = DaudioHdfOperate::GetInstance().LoadDaudioHDFImpl();
    if (ret != DH_SUCCESS) {
        DHLOGE("Load hdf driver failed, ret: %d", ret);
        return ret;
    }
    DHLOGD("Load hdf driver end.");

    audioSrvHdf_ = IDAudioManager::Get(HDF_AUDIO_SERVICE_NAME.c_str(), false);
    if (audioSrvHdf_ == nullptr) {
        DHLOGE("Can not get hdf audio manager.");
        return ERR_DH_AUDIO_HDI_PROXY_NOT_INIT;
    }

    DHLOGI("Init hdi handler success.");
    return DH_SUCCESS;
}

int32_t DAudioHdiHandler::UninitHdiHandler()
{
    DHLOGI("Unload hdf driver start.");
    if (audioSrvHdf_ == nullptr) {
        DHLOGI("Audio hdi proxy not init. Do not need to uninit.");
        return DH_SUCCESS;
    }
    int32_t ret = DaudioHdfOperate::GetInstance().UnLoadDaudioHDFImpl();
    if (ret != DH_SUCCESS) {
        DHLOGE("Unload hdf driver failed, ret: %d", ret);
        return ret;
    }
    DHLOGI("Uninit hdi handler success.");
    return DH_SUCCESS;
}

int32_t DAudioHdiHandler::RegisterAudioDevice(const std::string &devId, const int32_t dhId,
    const std::string &capability, const std::shared_ptr<IDAudioHdiCallback> &callbackObjParam)
{
    DHLOGI("Register audio device, adpname: %s, dhId: %d", GetAnonyString(devId).c_str(), dhId);
    if (audioSrvHdf_ == nullptr) {
        DHLOGE("Audio hdi proxy not init.");
        return ERR_DH_AUDIO_HDI_PROXY_NOT_INIT;
    }
    std::string searchKey;
    switch (GetDevTypeByDHId(dhId)) {
        case AUDIO_DEVICE_TYPE_SPEAKER:
            searchKey = devId + "Speaker";
            break;
        case AUDIO_DEVICE_TYPE_MIC:
            searchKey = devId + "Mic";
            break;
        case AUDIO_DEVICE_TYPE_UNKNOWN:
        default:
            DHLOGE("Unknown audio device.");
            return ERR_DH_AUDIO_HDI_UNKNOWN_DEVTYPE;
    }
    {
        std::lock_guard<std::mutex> devLck(devMapMtx_);
        auto call = mapAudioMgrCallback_.find(searchKey);
        if (call == mapAudioMgrCallback_.end()) {
            const sptr<DAudioManagerCallback> callbackptr(new DAudioManagerCallback(callbackObjParam));
            mapAudioMgrCallback_.emplace(searchKey, callbackptr);
        }
        auto dhIds = mapAudioMgrDhIds_.find(devId);
        if (dhIds != mapAudioMgrDhIds_.end()) {
            dhIds->second.insert(dhId);
        } else {
            std::set<int32_t> newDhIds;
            newDhIds.insert(dhId);
            mapAudioMgrDhIds_.emplace(devId, newDhIds);
        }
    }

    auto iter = mapAudioMgrCallback_.find(searchKey);
    int32_t res = audioSrvHdf_->RegisterAudioDevice(devId, dhId, capability, iter->second);
    if (res != HDF_SUCCESS) {
        DHLOGE("Call hdf proxy register failed, res: %d", res);
        return ERR_DH_AUDIO_HDI_CALL_FAILED;
    }
    return DH_SUCCESS;
}

int32_t DAudioHdiHandler::UnRegisterAudioDevice(const std::string &devId, const int32_t dhId)
{
    DHLOGI("Unregister audio device, adpname: %s, dhId: %d", GetAnonyString(devId).c_str(), dhId);
    if (audioSrvHdf_ == nullptr) {
        DHLOGE("Audio hdi proxy not init");
        return ERR_DH_AUDIO_HDI_PROXY_NOT_INIT;
    }

    int32_t res = audioSrvHdf_->UnRegisterAudioDevice(devId, dhId);
    if (res != HDF_SUCCESS) {
        DHLOGE("Call hdf proxy unregister failed, res: %d", res);
        return ERR_DH_AUDIO_HDI_CALL_FAILED;
    }

    {
        std::lock_guard<std::mutex> devLck(devMapMtx_);
        auto iter = mapAudioMgrDhIds_.find(devId);
        if (iter == mapAudioMgrDhIds_.end()) {
            DHLOGE("Can not find register devId. devId: %s", GetAnonyString(devId).c_str());
            return ERR_DH_AUDIO_HDI_CALLBACK_NOT_EXIST;
        }

        iter->second.erase(dhId);
        if (iter->second.empty()) {
            mapAudioMgrDhIds_.erase(devId);
        }
    }
    return DH_SUCCESS;
}

int32_t DAudioHdiHandler::NotifyEvent(const std::string &devId, const int32_t dhId,
    const AudioEvent &audioEvent)
{
    DHLOGD("Notify event adpname: %s, dhId: %d, event type: %d, event content: %s.",
        GetAnonyString(devId).c_str(), dhId, audioEvent.type, audioEvent.content.c_str());
    if (audioSrvHdf_ == nullptr) {
        DHLOGE("Audio hdi proxy not init");
        return ERR_DH_AUDIO_HDI_PROXY_NOT_INIT;
    }
    OHOS::HDI::DistributedAudio::Audioext::V1_0::DAudioEvent newEvent = {AUDIO_EVENT_UNKNOWN, audioEvent.content};
    switch (audioEvent.type) {
        case AudioEventType::NOTIFY_OPEN_SPEAKER_RESULT:
            newEvent.type = AUDIO_EVENT_OPEN_SPK_RESULT;
            break;
        case AudioEventType::NOTIFY_CLOSE_SPEAKER_RESULT:
            newEvent.type = AUDIO_EVENT_CLOSE_SPK_RESULT;
            break;
        case AudioEventType::NOTIFY_OPEN_MIC_RESULT:
            newEvent.type = AUDIO_EVENT_OPEN_MIC_RESULT;
            break;
        case AudioEventType::NOTIFY_CLOSE_MIC_RESULT:
            newEvent.type = AUDIO_EVENT_CLOSE_MIC_RESULT;
            break;
        case AudioEventType::VOLUME_CHANGE:
            newEvent.type = AUDIO_EVENT_VOLUME_CHANGE;
            break;
        case AudioEventType::SPEAKER_CLOSED:
            newEvent.type = AUDIO_EVENT_SPK_CLOSED;
            break;
        case AudioEventType::MIC_CLOSED:
            newEvent.type = AUDIO_EVENT_MIC_CLOSED;
            break;
        case AudioEventType::AUDIO_FOCUS_CHANGE:
            newEvent.type = AUDIO_EVENT_FOCUS_CHANGE;
            break;
        case AudioEventType::AUDIO_RENDER_STATE_CHANGE:
            newEvent.type = AUDIO_EVENT_RENDER_STATE_CHANGE;
            break;
        default:
            DHLOGE("Unsupport audio event.");
            break;
    }

    int32_t res = audioSrvHdf_->NotifyEvent(devId, dhId, newEvent);
    if (res != HDF_SUCCESS) {
        DHLOGE("Call hdf proxy NotifyEvent failed, res: %d", res);
        return ERR_DH_AUDIO_HDI_CALL_FAILED;
    }
    return DH_SUCCESS;
}
} // namespace DistributedHardware
} // namespace OHOS