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

#include "message_parcel.h"
#include "string_ex.h"

#include "ithermal_action_callback.h"
#include "ithermal_temp_callback.h"
#include "ithermal_level_callback.h"
#include "thermal_common.h"
#include "thermal_srv_ipc_interface_code.h"
#include "thermal_srv_sensor_info.h"
#include "xcollie/xcollie.h"

namespace OHOS {
namespace PowerMgr {
namespace {
constexpr int PARAM_MAX_NUM = 10;
}
int ThermalSrvStub::OnRemoteRequest(uint32_t code, MessageParcel& data, MessageParcel& reply, MessageOption& option)
{
    THERMAL_HILOGD(COMP_SVC,
        "ThermalSrvStub::OnRemoteRequest, cmd = %{public}d, flags = %{public}d",
        code, option.GetFlags());
    std::u16string descriptor = ThermalSrvStub::GetDescriptor();
    std::u16string remoteDescriptor = data.ReadInterfaceToken();
    if (descriptor != remoteDescriptor) {
        THERMAL_HILOGE(COMP_SVC, "ThermalSrvStub::OnRemoteRequest failed, descriptor is not matched!");
        return E_GET_THERMAL_SERVICE_FAILED;
    }
    const int DFX_DELAY_S = 60;
    int id = HiviewDFX::XCollie::GetInstance().SetTimer("ThermalSrvStub", DFX_DELAY_S, nullptr, nullptr,
        HiviewDFX::XCOLLIE_FLAG_LOG);
    int32_t ret = CheckRequestCode(code, data, reply, option);
    HiviewDFX::XCollie::GetInstance().CancelTimer(id);
    return ret;
}

int32_t ThermalSrvStub::CheckRequestCode(const uint32_t code, MessageParcel& data, MessageParcel& reply,
    MessageOption& option)
{
    switch (code) {
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::REG_THERMAL_TEMP_CALLBACK): {
            return SubscribeThermalTempCallbackStub(data);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::UNREG_THERMAL_TEMP_CALLBACK): {
            return UnSubscribeThermalTempCallbackStub(data);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::REG_THERMAL_LEVEL_CALLBACK): {
            return SubscribeThermalLevelCallbackStub(data);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::UNREG_THERMAL_LEVEL_CALLBACK): {
            return UnSubscribeThermalLevelCallbackStub(data);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::REG_THERMAL_ACTION_CALLBACK): {
            return SubscribeThermalActionCallbackStub(data);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::UNREG_THERMAL_ACTION_CALLBACK): {
            return UnSubscribeThermalActionCallbackStub(data);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::GET_SENSOR_INFO): {
            return GetThermalSrvSensorInfoStub(data, reply);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::GET_TEMP_LEVEL): {
            return GetThermalevelStub(reply);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::GET_THERMAL_INFO): {
            return GetThermalInfoStub(reply);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::SET_SCENE): {
            return SetSceneStub(data);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::UPDATE_THERMAL_STATE): {
            return UpdateThermalStateStub(data);
        }
        case static_cast<int>(PowerMgr::ThermalMgrInterfaceCode::SHELL_DUMP): {
            return ShellDumpStub(data, reply);
        }
        default: {
            return IPCObjectStub::OnRemoteRequest(code, data, reply, option);
        }
    }
}

int32_t ThermalSrvStub::SubscribeThermalTempCallbackStub(MessageParcel& data)
{
    THERMAL_HILOGD(COMP_SVC, "Enter");
    sptr<IRemoteObject> obj = data.ReadRemoteObject();
    THERMAL_RETURN_IF_WITH_RET((obj == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    sptr<IThermalTempCallback> callback = iface_cast<IThermalTempCallback>(obj);
    THERMAL_RETURN_IF_WITH_RET((callback == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    std::vector<std::string> typeList;
    if (!data.ReadStringVector(&typeList)) {
        THERMAL_HILOGI(COMP_SVC, "failed to read type list");
        return ERR_INVALID_VALUE;
    }
    SubscribeThermalTempCallback(typeList, callback);
    return ERR_OK;
}

int32_t ThermalSrvStub::UnSubscribeThermalTempCallbackStub(MessageParcel& data)
{
    THERMAL_HILOGD(COMP_SVC, "Enter");
    sptr<IRemoteObject> obj = data.ReadRemoteObject();
    THERMAL_RETURN_IF_WITH_RET((obj == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    sptr<IThermalTempCallback> callback = iface_cast<IThermalTempCallback>(obj);
    THERMAL_RETURN_IF_WITH_RET((callback == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    UnSubscribeThermalTempCallback(callback);
    return ERR_OK;
}

int32_t ThermalSrvStub::SubscribeThermalLevelCallbackStub(MessageParcel& data)
{
    THERMAL_HILOGD(COMP_SVC, "SubscribeThermalLevelCallbackStub Enter");
    sptr<IRemoteObject> obj = data.ReadRemoteObject();
    THERMAL_RETURN_IF_WITH_RET((obj == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    sptr<IThermalLevelCallback> callback = iface_cast<IThermalLevelCallback>(obj);
    THERMAL_RETURN_IF_WITH_RET((callback == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    SubscribeThermalLevelCallback(callback);
    return ERR_OK;
}

int32_t ThermalSrvStub::UnSubscribeThermalLevelCallbackStub(MessageParcel& data)
{
    THERMAL_HILOGD(COMP_SVC, "Enter");
    sptr<IRemoteObject> obj = data.ReadRemoteObject();
    THERMAL_RETURN_IF_WITH_RET((obj == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    sptr<IThermalLevelCallback> callback = iface_cast<IThermalLevelCallback>(obj);
    THERMAL_RETURN_IF_WITH_RET((callback == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    UnSubscribeThermalLevelCallback(callback);
    return ERR_OK;
}

int32_t ThermalSrvStub::SubscribeThermalActionCallbackStub(MessageParcel& data)
{
    THERMAL_HILOGD(COMP_SVC, "Enter");
    sptr<IRemoteObject> obj = data.ReadRemoteObject();
    THERMAL_RETURN_IF_WITH_RET((obj == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    sptr<IThermalActionCallback> callback = iface_cast<IThermalActionCallback>(obj);
    THERMAL_RETURN_IF_WITH_RET((callback == nullptr), E_READ_PARCEL_ERROR_THERMAL);

    std::vector<std::string> actionList;
    if (!data.ReadStringVector(&actionList)) {
        THERMAL_HILOGI(COMP_SVC, "failed to read action list");
        return ERR_INVALID_VALUE;
    }

    std::string desc;
    THERMAL_READ_PARCEL_WITH_RET(data, String, desc, E_READ_PARCEL_ERROR_THERMAL);

    SubscribeThermalActionCallback(actionList, desc, callback);
    return ERR_OK;
}

int32_t ThermalSrvStub::UnSubscribeThermalActionCallbackStub(MessageParcel& data)
{
    THERMAL_HILOGD(COMP_SVC, "Enter");
    sptr<IRemoteObject> obj = data.ReadRemoteObject();
    THERMAL_RETURN_IF_WITH_RET((obj == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    sptr<IThermalActionCallback> callback = iface_cast<IThermalActionCallback>(obj);
    THERMAL_RETURN_IF_WITH_RET((callback == nullptr), E_READ_PARCEL_ERROR_THERMAL);
    UnSubscribeThermalActionCallback(callback);
    return ERR_OK;
}

int32_t ThermalSrvStub::GetThermalSrvSensorInfoStub(MessageParcel& data, MessageParcel& reply)
{
    ThermalSrvSensorInfo sensorInfo;
    uint32_t type = 0;

    THERMAL_READ_PARCEL_WITH_RET(data, Uint32, type, E_READ_PARCEL_ERROR_THERMAL);
    THERMAL_HILOGD(COMP_SVC, "type is %{public}d", type);
    bool ret = GetThermalSrvSensorInfo(static_cast<SensorType>(type), sensorInfo);
    if (ret) {
        if (!reply.WriteParcelable(&sensorInfo)) {
            THERMAL_HILOGE(COMP_SVC, "write failed");
            return -1;
        }
    }
    return ERR_OK;
}

int32_t ThermalSrvStub::GetThermalevelStub(MessageParcel& reply)
{
    THERMAL_HILOGD(COMP_SVC, "Enter");
    ThermalLevel level;
    GetThermalLevel(level);
    THERMAL_WRITE_PARCEL_WITH_RET(reply, Uint32, static_cast<uint32_t>(level), ERR_OK);
    return ERR_OK;
}

int32_t ThermalSrvStub::GetThermalInfoStub(MessageParcel& reply)
{
    THERMAL_HILOGD(COMP_SVC, "Enter");
    bool ret = false;
    ret = GetThermalInfo();
    if (!reply.WriteBool(ret)) {
        THERMAL_HILOGE(COMP_FWK, "WriteBool fail");
        return E_READ_PARCEL_ERROR_THERMAL;
    }
    return ERR_OK;
}

int32_t ThermalSrvStub::SetSceneStub(MessageParcel& data)
{
    THERMAL_HILOGD(COMP_SVC, "Enter");
    std::string scene;

    THERMAL_READ_PARCEL_WITH_RET(data, String, scene, E_READ_PARCEL_ERROR_THERMAL);
    SetScene(scene);
    return ERR_OK;
}

int32_t ThermalSrvStub::UpdateThermalStateStub(MessageParcel& data)
{
    THERMAL_HILOGD(COMP_SVC, "Enter");
    std::string tag;
    std::string val;
    bool isImmed = false;
    THERMAL_READ_PARCEL_WITH_RET(data, String, tag, E_READ_PARCEL_ERROR_THERMAL);
    THERMAL_READ_PARCEL_WITH_RET(data, String, val, E_READ_PARCEL_ERROR_THERMAL);
    THERMAL_READ_PARCEL_WITH_RET(data, Bool, isImmed, E_READ_PARCEL_ERROR_THERMAL);
    UpdateThermalState(tag, val, isImmed);
    return ERR_OK;
}

int32_t ThermalSrvStub::ShellDumpStub(MessageParcel& data, MessageParcel& reply)
{
    uint32_t argc;
    std::vector<std::string> args;

    if (!data.ReadUint32(argc)) {
        THERMAL_HILOGE(COMP_SVC, "Readback fail!");
        return E_READ_PARCEL_ERROR_THERMAL;
    }

    if (argc >= PARAM_MAX_NUM) {
        THERMAL_HILOGE(COMP_SVC, "params exceed limit");
        return E_EXCEED_PARAM_LIMIT;
    }

    for (uint32_t i = 0; i < argc; i++) {
        std::string arg = data.ReadString();
        if (arg.empty()) {
            THERMAL_HILOGE(COMP_SVC, "read value fail:%{public}d", i);
            return E_READ_PARCEL_ERROR_THERMAL;
        }
        args.push_back(arg);
    }

    std::string ret = ShellDump(args, argc);
    if (!reply.WriteString(ret)) {
        THERMAL_HILOGE(COMP_SVC, "PowerMgrStub:: Dump Writeback Fail!");
        return E_READ_PARCEL_ERROR_THERMAL;
    }
    return ERR_OK;
}
} // namespace PowerMgr
} // namespace OHOS