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

#include <vector>

#include "display_info.h"
#include "display_change_info.h"
#include "dm_common.h"
#include "marshalling_helper.h"
#include "screen_info.h"
#include "window_manager_hilog.h"

namespace OHOS::Rosen {
namespace {
constexpr HiviewDFX::HiLogLabel LABEL = {LOG_CORE, HILOG_DOMAIN_DISPLAY, "DisplayManagerAgentStub"};
}

int32_t DisplayManagerAgentStub::OnRemoteRequest(uint32_t code, MessageParcel& data,
    MessageParcel& reply, MessageOption& option)
{
    WLOGFD("code:%{public}u", code);
    if (data.ReadInterfaceToken() != GetDescriptor()) {
        WLOGFE("InterfaceToken check failed");
        return -1;
    }
    switch (code) {
        case TRANS_ID_NOTIFY_DISPLAY_POWER_EVENT: {
            return ProcNotifyDisplayPowerEvent(data);
        }
        case TRANS_ID_NOTIFY_DISPLAY_STATE_CHANGED: {
            return ProcNotifyDisplayStateChanged(data);
        }
        case TRANS_ID_ON_SCREEN_CONNECT: {
            return ProcScreenConnect(data);
        }
        case TRANS_ID_ON_SCREEN_DISCONNECT: {
            return ProcScreenDisconnect(data);
        }
        case TRANS_ID_ON_SCREEN_CHANGED: {
            return ProcScreenChanged(data);
        }
        case TRANS_ID_ON_SCREENGROUP_CHANGED: {
            return ProcScreenGroupChanged(data);
        }
        case TRANS_ID_ON_DISPLAY_CONNECT: {
            return ProcDisplayConnect(data);
        }
        case TRANS_ID_ON_DISPLAY_DISCONNECT: {
            return ProcDisplayDisconnect(data);
        }
        case TRANS_ID_ON_DISPLAY_CHANGED: {
            return ProcDisplayChanged(data);
        }
        case TRANS_ID_ON_SCREEN_SHOT: {
            return ProcScreenShot(data);
        }
        case TRANS_ID_ON_PRIVATE_WINDOW: {
            return ProcPrivateWindow(data);
        }
        case TRANS_ID_ON_PRIVATE_WINDOW_LIST: {
            return ProcPrivateWindowList(data);
        }
        case TRANS_ID_ON_FOLD_STATUS_CHANGED: {
            return ProcFoldStatusChanged(data);
        }
        case TRANS_ID_ON_DISPLAY_CHANGE_INFO_CHANGED: {
            return ProcDisplayChangeInfoChanged(data);
        }
        case TRANS_ID_ON_DISPLAY_MODE_CHANGED: {
            return ProcDisplayModechanged(data);
        }
        case TRANS_ID_ON_AVAILABLE_AREA_CHANGED: {
            return ProcAvailableAreaChanged(data);
        }
        case TRANS_ID_ON_FOLD_ANGLE_CHANGED: {
            return ProcFoldAngleChanged(data);
        }
        case TRANS_ID_ON_CAPTURE_STATUS_CHANGED: {
            return ProcCaptureStatusChanged(data);
        }
        default: {
            WLOGFW("unknown transaction code %{public}d", code);
            return IPCObjectStub::OnRemoteRequest(code, data, reply, option);
        }
    }
    return 0;
}

int32_t DisplayManagerAgentStub::ProcFoldAngleChanged(MessageParcel& data)
{
    std::vector<float> foldAngles;
    if (!data.ReadFloatVector(&foldAngles)) {
        WLOGFE("Read foldAngles failed");
        return -1;
    }
    NotifyFoldAngleChanged(foldAngles);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcCaptureStatusChanged(MessageParcel& data)
{
    bool isCapture = data.ReadBool();
    NotifyCaptureStatusChanged(isCapture);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcNotifyDisplayPowerEvent(MessageParcel& data)
{
    DisplayPowerEvent event = static_cast<DisplayPowerEvent>(data.ReadUint32());
    EventStatus status = static_cast<EventStatus>(data.ReadUint32());
    NotifyDisplayPowerEvent(event, status);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcNotifyDisplayStateChanged(MessageParcel& data)
{
    DisplayState state = static_cast<DisplayState>(data.ReadUint32());
    DisplayId id = static_cast<DisplayId>(data.ReadUint64());
    NotifyDisplayStateChanged(id, state);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcScreenConnect(MessageParcel& data)
{
    sptr<ScreenInfo> screenInfo = data.ReadParcelable<ScreenInfo>();
    OnScreenConnect(screenInfo);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcScreenDisconnect(MessageParcel& data)
{
    ScreenId screenId;
    if (!data.ReadUint64(screenId)) {
        WLOGFE("Read ScreenId failed");
        return -1;
    }
    OnScreenDisconnect(screenId);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcScreenChanged(MessageParcel& data)
{
    sptr<ScreenInfo> screenInfo = data.ReadParcelable<ScreenInfo>();
    uint32_t event;
    if (!data.ReadUint32(event)) {
        WLOGFE("Read ScreenChangeEvent failed");
        return -1;
    }
    OnScreenChange(screenInfo, static_cast<ScreenChangeEvent>(event));
    return 0;
}

int32_t DisplayManagerAgentStub::ProcScreenGroupChanged(MessageParcel& data)
{
    std::string trigger;
    if (!data.ReadString(trigger)) {
        WLOGFE("Read trigger failed");
        return -1;
    }
    std::vector<sptr<ScreenInfo>> screenInfos;
    if (!MarshallingHelper::UnmarshallingVectorParcelableObj<ScreenInfo>(data, screenInfos)) {
        WLOGFE("Read ScreenInfo failed");
        return -1;
    }
    uint32_t event;
    if (!data.ReadUint32(event)) {
        WLOGFE("Read ScreenChangeEvent failed");
        return -1;
    }
    OnScreenGroupChange(trigger, screenInfos, static_cast<ScreenGroupChangeEvent>(event));
    return 0;
}

int32_t DisplayManagerAgentStub::ProcDisplayConnect(MessageParcel& data)
{
    sptr<DisplayInfo> displayInfo = data.ReadParcelable<DisplayInfo>();
    OnDisplayCreate(displayInfo);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcDisplayDisconnect(MessageParcel& data)
{
    DisplayId displayId;
    if (!data.ReadUint64(displayId)) {
        WLOGFE("Read DisplayId failed");
        return -1;
    }
    OnDisplayDestroy(displayId);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcDisplayChanged(MessageParcel& data)
{
    sptr<DisplayInfo> displayInfo = data.ReadParcelable<DisplayInfo>();
    uint32_t event;
    if (!data.ReadUint32(event)) {
        WLOGFE("Read DisplayChangeEvent failed");
        return -1;
    }
    OnDisplayChange(displayInfo, static_cast<DisplayChangeEvent>(event));
    return 0;
}

int32_t DisplayManagerAgentStub::ProcScreenShot(MessageParcel& data)
{
    sptr<ScreenshotInfo> snapshotInfo = data.ReadParcelable<ScreenshotInfo>();
    OnScreenshot(snapshotInfo);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcPrivateWindow(MessageParcel& data)
{
    bool hasPrivate = data.ReadBool();
    NotifyPrivateWindowStateChanged(hasPrivate);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcPrivateWindowList(MessageParcel& data)
{
    DisplayId displayId = static_cast<DisplayId>(data.ReadUint64());
    std::vector<std::string> privacyWindowList;
    data.ReadStringVector(&privacyWindowList);
    NotifyPrivateStateWindowListChanged(displayId, privacyWindowList);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcFoldStatusChanged(MessageParcel& data)
{
    uint32_t foldStatus;
    if (!data.ReadUint32(foldStatus)) {
        WLOGFE("Read FoldStatus failed");
        return -1;
    }
    NotifyFoldStatusChanged(static_cast<FoldStatus>(foldStatus));
    return 0;
}

int32_t DisplayManagerAgentStub::ProcDisplayChangeInfoChanged(MessageParcel& data)
{
    sptr<DisplayChangeInfo> info;
    info = DisplayChangeInfo::Unmarshalling(data);
    if (!info) {
        WLOGFE("Read DisplayChangeInfo failed");
        return -1;
    }
    NotifyDisplayChangeInfoChanged(info);
    return 0;
}

int32_t DisplayManagerAgentStub::ProcDisplayModechanged(MessageParcel& data)
{
    uint32_t displayMode;
    if (!data.ReadUint32(displayMode)) {
        WLOGFE("Read FoldDisplayMode failed");
        return -1;
    }
    NotifyDisplayModeChanged(static_cast<FoldDisplayMode>(displayMode));
    return 0;
}

int32_t DisplayManagerAgentStub::ProcAvailableAreaChanged(MessageParcel& data)
{
    DMRect rect;
    rect.posX_ = data.ReadInt32();
    rect.posY_ = data.ReadInt32();
    rect.width_ = data.ReadUint32();
    rect.height_ = data.ReadUint32();
    NotifyAvailableAreaChanged(rect);
    return 0;
}
} // namespace OHOS::Rosen