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

#include <errors.h>
#include <ipc_skeleton.h>

#include "transient_task_log.h"
#include "background_task_subscriber_proxy.h"
#include "ibackground_task_subscriber_ipc_interface_code.h"

namespace OHOS {
namespace BackgroundTaskMgr {
namespace {
constexpr int32_t APP_FIRST_UID = 10000;
}

BackgroundTaskSubscriberStub::BackgroundTaskSubscriberStub() {}
BackgroundTaskSubscriberStub::~BackgroundTaskSubscriberStub() {}

ErrCode BackgroundTaskSubscriberStub::OnRemoteRequest(uint32_t code,
    MessageParcel& data, MessageParcel& reply, MessageOption& option)
{
    std::u16string descriptor = BackgroundTaskSubscriberStub::GetDescriptor();
    std::u16string remoteDescriptor = data.ReadInterfaceToken();
    if (descriptor != remoteDescriptor) {
        BGTASK_LOGE("BackgroundTaskSubscriberStub Local descriptor not match remote.");
        return ERR_TRANSACTION_FAILED;
    }

    auto uid = IPCSkeleton::GetCallingUid();
    if (uid >= APP_FIRST_UID) {
        BGTASK_LOGE("BackgroundTaskSubscriberStub OnRemoteRequest failed, illegal calling uid %d.", uid);
        return ERR_INVALID_DATA;
    }

    return OnRemoteRequestInner(code, data, reply, option);
}

ErrCode BackgroundTaskSubscriberStub::OnRemoteRequestInner(uint32_t code,
    MessageParcel& data, MessageParcel& reply, MessageOption& option)
{
    switch (code) {
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_CONNECTED): {
            return HandleOnConnected();
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_DISCONNECTED): {
            return HandleOnDisconnected();
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_TRANSIENT_TASK_START):
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_TRANSIENT_TASK_END):
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_TRANSIENT_TASK_ERR):
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_APP_TRANSIENT_TASK_START):
            [[fallthrough]];
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_APP_TRANSIENT_TASK_END): {
            return HandleTransientTask(code, data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_CONTINUOUS_TASK_START): {
            return HandleOnContinuousTaskStart(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_CONTINUOUS_TASK_UPDATE): {
            return HandleOnContinuousTaskUpdate(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_CONTINUOUS_TASK_STOP): {
            return HandleOnContinuousTaskCancel(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_APP_CONTINUOUS_TASK_STOP): {
            return HandleOnAppContinuousTaskStop(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_APP_EFFICIENCY_RESOURCES_APPLY): {
            return HandleOnAppEfficiencyResourcesApply(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_APP_EFFICIENCY_RESOURCES_RESET): {
            return HandleOnAppEfficiencyResourcesReset(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_PROC_EFFICIENCY_RESOURCES_APPLY): {
            return HandleOnProcEfficiencyResourcesApply(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_PROC_EFFICIENCY_RESOURCES_RESET): {
            return HandleOnProcEfficiencyResourcesReset(data);
        }
        default:
            return IPCObjectStub::OnRemoteRequest(code, data, reply, option);
    }
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleTransientTask(uint32_t code, MessageParcel &data)
{
    switch (code) {
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_TRANSIENT_TASK_START): {
            return HandleOnTransientTaskStart(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_TRANSIENT_TASK_END): {
            return HandleOnTransientTaskEnd(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_TRANSIENT_TASK_ERR): {
            return HandleOnTransientTaskErr(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_APP_TRANSIENT_TASK_START): {
            return HandleOnAppTransientTaskStart(data);
        }
        case static_cast<uint32_t>(IBackgroundTaskSubscriberInterfaceCode::ON_APP_TRANSIENT_TASK_END): {
            return HandleOnAppTransientTaskEnd(data);
        }
    }
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnConnected()
{
    OnConnected();
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnDisconnected()
{
    OnDisconnected();
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnTransientTaskStart(MessageParcel& data)
{
    auto info = TransientTaskAppInfo::Unmarshalling(data);
    if (info == nullptr) {
        BGTASK_LOGE("HandleOnTransientTaskStart Read parcel failed.");
        return ERR_INVALID_DATA;
    }
    OnTransientTaskStart(info);
    return ERR_NONE;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnTransientTaskEnd(MessageParcel& data)
{
    auto info = TransientTaskAppInfo::Unmarshalling(data);
    if (info == nullptr) {
        BGTASK_LOGE("HandleOnTransientTaskEnd Read parcel failed.");
        return ERR_INVALID_DATA;
    }
    OnTransientTaskEnd(info);
    return ERR_NONE;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnTransientTaskErr(MessageParcel& data)
{
    auto info = TransientTaskAppInfo::Unmarshalling(data);
    if (info == nullptr) {
        BGTASK_LOGE("HandleOnTransientTaskErr Read parcel failed.");
        return ERR_INVALID_DATA;
    }
    OnTransientTaskErr(info);
    return ERR_NONE;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnAppTransientTaskStart(MessageParcel& data)
{
    auto info = TransientTaskAppInfo::Unmarshalling(data);
    if (info == nullptr) {
        BGTASK_LOGE("HandleOnAppTransientTaskStart Read parcel failed.");
        return ERR_INVALID_DATA;
    }
    OnAppTransientTaskStart(info);
    return ERR_NONE;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnAppTransientTaskEnd(MessageParcel& data)
{
    auto info = TransientTaskAppInfo::Unmarshalling(data);
    if (info == nullptr) {
        BGTASK_LOGE("HandleOnAppTransientTaskEnd Read parcel failed.");
        return ERR_INVALID_DATA;
    }
    OnAppTransientTaskEnd(info);
    return ERR_NONE;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnContinuousTaskStart(MessageParcel &data)
{
    std::shared_ptr<ContinuousTaskCallbackInfo> continuousTaskCallbackInfo
        = std::shared_ptr<ContinuousTaskCallbackInfo>(data.ReadParcelable<ContinuousTaskCallbackInfo>());
    if (!continuousTaskCallbackInfo) {
        BGTASK_LOGE("HandleOnContinuousTaskStart ContinuousTaskCallbackInfo ReadParcelable failed");
        return ERR_BGTASK_PARCELABLE_FAILED;
    }

    OnContinuousTaskStart(continuousTaskCallbackInfo);
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnContinuousTaskUpdate(MessageParcel &data)
{
    std::shared_ptr<ContinuousTaskCallbackInfo> continuousTaskCallbackInfo
        = std::shared_ptr<ContinuousTaskCallbackInfo>(data.ReadParcelable<ContinuousTaskCallbackInfo>());
    if (!continuousTaskCallbackInfo) {
        BGTASK_LOGE("HandleOnContinuousTaskUpdate ContinuousTaskCallbackInfo ReadParcelable failed");
        return ERR_BGTASK_PARCELABLE_FAILED;
    }

    OnContinuousTaskUpdate(continuousTaskCallbackInfo);
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnContinuousTaskCancel(MessageParcel &data)
{
    std::shared_ptr<ContinuousTaskCallbackInfo> continuousTaskCallbackInfo
        = std::shared_ptr<ContinuousTaskCallbackInfo>(data.ReadParcelable<ContinuousTaskCallbackInfo>());
    if (!continuousTaskCallbackInfo) {
        BGTASK_LOGE("HandleOnContinuousTaskCancel ContinuousTaskCallbackInfo ReadParcelable failed");
        return ERR_BGTASK_PARCELABLE_FAILED;
    }

    OnContinuousTaskStop(continuousTaskCallbackInfo);
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnAppContinuousTaskStop(MessageParcel &data)
{
    int32_t uid;
    if (!data.ReadInt32(uid)) {
        BGTASK_LOGE("HandleOnAppContinuousTaskStop read uid failed");
        return ERR_BGTASK_PARCELABLE_FAILED;
    }

    OnAppContinuousTaskStop(uid);
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnAppEfficiencyResourcesApply(MessageParcel &data)
{
    std::shared_ptr<ResourceCallbackInfo> resourceCallbackInfo
        = std::shared_ptr<ResourceCallbackInfo>(data.ReadParcelable<ResourceCallbackInfo>());
    if (!resourceCallbackInfo) {
        BGTASK_LOGE("HandleOnAppEfficiencyResourcesApply ReadParcelable failed");
        return ERR_BGTASK_PARCELABLE_FAILED;
    }
    OnAppEfficiencyResourcesApply(resourceCallbackInfo);
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnAppEfficiencyResourcesReset(MessageParcel &data)
{
    std::shared_ptr<ResourceCallbackInfo> resourceCallbackInfo
        = std::shared_ptr<ResourceCallbackInfo>(data.ReadParcelable<ResourceCallbackInfo>());
    if (!resourceCallbackInfo) {
        BGTASK_LOGE("HandleOnAppEfficiencyResourcesReset ReadParcelable failed");
        return ERR_BGTASK_PARCELABLE_FAILED;
    }
    OnAppEfficiencyResourcesReset(resourceCallbackInfo);
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnProcEfficiencyResourcesApply(MessageParcel &data)
{
    std::shared_ptr<ResourceCallbackInfo> resourceCallbackInfo
        = std::shared_ptr<ResourceCallbackInfo>(data.ReadParcelable<ResourceCallbackInfo>());
    if (!resourceCallbackInfo) {
        BGTASK_LOGE("HandleOnProcEfficiencyResourcesApply ContinuousTaskCallbackInfo ReadParcelable failed");
        return ERR_BGTASK_PARCELABLE_FAILED;
    }
    OnProcEfficiencyResourcesApply(resourceCallbackInfo);
    return ERR_OK;
}

ErrCode BackgroundTaskSubscriberStub::HandleOnProcEfficiencyResourcesReset(MessageParcel &data)
{
    std::shared_ptr<ResourceCallbackInfo> resourceCallbackInfo
        = std::shared_ptr<ResourceCallbackInfo>(data.ReadParcelable<ResourceCallbackInfo>());
    if (!resourceCallbackInfo) {
        BGTASK_LOGE("HandleOnProcEfficiencyResourcesReset ReadParcelable failed");
        return ERR_BGTASK_PARCELABLE_FAILED;
    }
    OnProcEfficiencyResourcesReset(resourceCallbackInfo);
    return ERR_OK;
}
}  // namespace BackgroundTaskMgr
}  // namespace OHOS