OpenHarmony-v5.1.0-Release/s

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

#include "window_manager_hilog.h"

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

void ScreenSessionManagerClientStub::InitScreenChangeMap()
{
    if (HandleScreenChangeMap_.size() != 0) {
        WLOGFI("screen change map has init!");
        return;
    }
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SCREEN_CONNECTION_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
            return HandleOnScreenConnectionChanged(data, reply);
        };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_PROPERTY_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnPropertyChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_POWER_STATUS_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnPowerStatusChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SCREEN_EXTEND_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnScreenExtendChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SENSOR_ROTATION_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnSensorRotationChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_HOVER_STATUS_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnHoverStatusChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SCREEN_ORIENTATION_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnScreenOrientationChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SCREEN_ROTATION_LOCKED_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnScreenRotationLockedChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_DISPLAY_STATE_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnDisplayStateChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SCREEN_SHOT] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnScreenshot(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_IMMERSIVE_STATE_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnImmersiveStateChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_SET_DISPLAY_NODE_SCREEN_ID] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnSetDisplayNodeScreenId(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_GET_SURFACENODEID_FROM_MISSIONID] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnGetSurfaceNodeIdsFromMissionIdsChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_SET_FOLD_DISPLAY_MODE] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnUpdateFoldDisplayMode(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SWITCH_USER_CMD] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleSwitchUserCallback(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_SET_VIRTUAL_PIXEL_RATIO_SYSTEM] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleSetVirtualPixelRatioSystem(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_FOLDSTATUS_CHANGED_REPORT_UE] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnFoldStatusChangedReportUE(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SCREEN_CAPTURE_NOTIFY] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleScreenCaptureNotify(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SUPER_FOLD_STATUS_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnSuperFoldStatusChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_SECONDARY_REFLEXION_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnSecondaryReflexionChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_CAMERA_BACKSELFIE_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnCameraBackSelfieChanged(data, reply);
    };
    HandleScreenChangeMap_[ScreenSessionManagerClientMessage::TRANS_ID_ON_EXTEND_SCREEN_CONNECT_STATUS_CHANGED] =
        [this](MessageParcel& data, MessageParcel& reply) {
        return HandleOnExtendScreenConnectStatusChanged(data, reply);
    };
}

ScreenSessionManagerClientStub::ScreenSessionManagerClientStub()
{
    InitScreenChangeMap();
}

int ScreenSessionManagerClientStub::OnRemoteRequest(uint32_t code, MessageParcel& data, MessageParcel& reply,
    MessageOption& option)
{
    int handleRet = ERR_INVALID_STATE;
    if (data.ReadInterfaceToken() != GetDescriptor()) {
        WLOGFE("Failed to check interface token!");
        return handleRet;
    }
    ScreenSessionManagerClientMessage msgId = static_cast<ScreenSessionManagerClientMessage>(code);
    auto handleCall = HandleScreenChangeMap_.find(msgId);
    if (handleCall != HandleScreenChangeMap_.end() && handleCall->second != nullptr) {
        auto handleFunc = handleCall->second;
        handleRet = handleFunc(data, reply);
    } else {
        WLOGFE("Failed to find function handler!");
        handleRet = IPCObjectStub::OnRemoteRequest(code, data, reply, option);
    }
    return handleRet;
}

int ScreenSessionManagerClientStub::HandleSwitchUserCallback(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleSwitchUserCallback");
    std::vector<int32_t> oldScbPids;
    data.ReadInt32Vector(&oldScbPids);
    int32_t currentScbPid = data.ReadInt32();
    SwitchUserCallback(oldScbPids, currentScbPid);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnScreenConnectionChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnScreenConnectionChanged");
    auto rsId = static_cast<ScreenId>(data.ReadUint64());
    auto name = data.ReadString();
    bool isExtend = data.ReadBool();
    auto innerName = data.ReadString();
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto screenEvent = static_cast<ScreenEvent>(data.ReadUint8());
    SessionOption option = {
        .rsId_ = rsId,
        .name_ = name,
        .isExtend_ = isExtend,
        .innerName_ = innerName,
        .screenId_ = screenId,
    };
    OnScreenConnectionChanged(option, screenEvent);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnPropertyChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnPropertyChanged");
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    ScreenProperty property;
    if (!RSMarshallingHelper::Unmarshalling(data, property)) {
        WLOGFE("Read property failed");
        return ERR_INVALID_DATA;
    }
    auto reason = static_cast<ScreenPropertyChangeReason>(data.ReadUint32());
    OnPropertyChanged(screenId, property, reason);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnPowerStatusChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnPowerStatusChanged");
    auto event = static_cast<DisplayPowerEvent>(data.ReadUint32());
    auto status = static_cast<EventStatus>(data.ReadUint32());
    auto reason = static_cast<PowerStateChangeReason>(data.ReadUint32());
    OnPowerStatusChanged(event, status, reason);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnSensorRotationChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnSensorRotationChanged");
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto sensorRotation = data.ReadFloat();
    OnSensorRotationChanged(screenId, sensorRotation);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnScreenExtendChanged(MessageParcel& data, MessageParcel& reply)
{
    auto mainScreenId = static_cast<ScreenId>(data.ReadUint64());
    auto extendScreenId = static_cast<ScreenId>(data.ReadUint64());
    WLOGI("mainScreenId=%{public}" PRIu64" extendScreenId=%{public}" PRIu64, mainScreenId, extendScreenId);
    OnScreenExtendChanged(mainScreenId, extendScreenId);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnScreenOrientationChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnScreenOrientationChanged");
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto screenOrientation = data.ReadFloat();
    OnScreenOrientationChanged(screenId, screenOrientation);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnScreenRotationLockedChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnScreenRotationLockedChanged");
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto isLocked = data.ReadBool();
    OnScreenRotationLockedChanged(screenId, isLocked);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnDisplayStateChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnDisplayStateChanged");
    auto defaultDisplayId = static_cast<DisplayId>(data.ReadUint64());
    auto displayInfo = data.ReadStrongParcelable<DisplayInfo>();
    auto mapSize = data.ReadUint32();
    std::map<DisplayId, sptr<DisplayInfo>> displayInfoMap;
    while (mapSize--) {
        auto displayId = data.ReadUint64();
        auto displayInfoValue = data.ReadStrongParcelable<DisplayInfo>();
        displayInfoMap.emplace(displayId, displayInfoValue);
    }
    auto type = static_cast<DisplayStateChangeType>(data.ReadUint32());
    OnDisplayStateChanged(defaultDisplayId, displayInfo, displayInfoMap, type);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnUpdateFoldDisplayMode(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnUpdateFoldDisplayMode");
    auto foldDisplayMode = static_cast<FoldDisplayMode>(data.ReadUint32());
    OnUpdateFoldDisplayMode(foldDisplayMode);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnGetSurfaceNodeIdsFromMissionIdsChanged(MessageParcel& data,
    MessageParcel& reply)
{
    std::vector<uint64_t> missionIds;
    data.ReadUInt64Vector(&missionIds);
    std::vector<uint64_t> surfaceNodeIds;
    data.ReadUInt64Vector(&surfaceNodeIds);
    bool isBlackList = data.ReadBool();
    OnGetSurfaceNodeIdsFromMissionIdsChanged(missionIds, surfaceNodeIds, isBlackList);
    if (!reply.WriteUInt64Vector(surfaceNodeIds)) {
        WLOGFE("Write surfaceNodeIds failed");
        return ERR_TRANSACTION_FAILED;
    }
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnScreenshot(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnScreenshot");
    auto displayId = static_cast<DisplayId>(data.ReadUint64());
    OnScreenshot(displayId);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnImmersiveStateChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnImmersiveStateChanged");
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    bool immersive = false;
    OnImmersiveStateChanged(screenId, immersive);
    if (!reply.WriteBool(immersive)) {
        WLOGFE("Write immersive failed");
        return ERR_TRANSACTION_FAILED;
    }
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnSetDisplayNodeScreenId(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnSetDisplayNodeScreenId");
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto displayNodeScreenId = static_cast<ScreenId>(data.ReadUint64());
    SetDisplayNodeScreenId(screenId, displayNodeScreenId);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleSetVirtualPixelRatioSystem(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleSetVirtualPixelRatioSystem");
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    float virtualPixelRatio = data.ReadFloat();
    SetVirtualPixelRatioSystem(screenId, virtualPixelRatio);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnFoldStatusChangedReportUE(MessageParcel& data, MessageParcel& reply)
{
    std::vector<std::string> screenFoldInfo;
    data.ReadStringVector(&screenFoldInfo);
    OnFoldStatusChangedReportUE(screenFoldInfo);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnHoverStatusChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnHoverStatusChanged");
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto hoverStatus = data.ReadInt32();
    auto needRotate = data.ReadBool();
    OnHoverStatusChanged(screenId, hoverStatus, needRotate);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleScreenCaptureNotify(MessageParcel& data, MessageParcel& reply)
{
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto uid = data.ReadInt32();
    auto clientName = data.ReadString();
    WLOGI("notify scb capture screenId=%{public}" PRIu64", uid=%{public}d.", screenId, uid);
    ScreenCaptureNotify(screenId, uid, clientName);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnCameraBackSelfieChanged(MessageParcel& data, MessageParcel& reply)
{
    WLOGD("HandleOnCameraBackSelfieChanged");
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    bool isCameraBackSelfie = data.ReadBool();
    OnCameraBackSelfieChanged(screenId, isCameraBackSelfie);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnSuperFoldStatusChanged(MessageParcel& data, MessageParcel& reply)
{
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto superFoldStatus = static_cast<SuperFoldStatus>(data.ReadUint32());
    WLOGI("super fold status screenId=%{public}" PRIu64", superFoldStatus=%{public}d.",
        screenId, static_cast<uint32_t>(superFoldStatus));
    OnSuperFoldStatusChanged(screenId, superFoldStatus);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnSecondaryReflexionChanged(MessageParcel& data, MessageParcel& reply)
{
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto isSecondaryReflexion = static_cast<bool>(data.ReadUint32());
    WLOGI("secondary reflexion screenId=%{public}" PRIu64", isSecondaryReflexion=%{public}d.",
        screenId, static_cast<uint32_t>(isSecondaryReflexion));
    OnSecondaryReflexionChanged(screenId, isSecondaryReflexion);
    return ERR_NONE;
}

int ScreenSessionManagerClientStub::HandleOnExtendScreenConnectStatusChanged(MessageParcel& data, MessageParcel& reply)
{
    auto screenId = static_cast<ScreenId>(data.ReadUint64());
    auto extendScreenConnectStatus = static_cast<ExtendScreenConnectStatus>(data.ReadUint32());
    WLOGI("extendScreenConnectStatus screenId=%{public}" PRIu64", extendScreenConnectStatus=%{public}d.",
        screenId, static_cast<uint32_t>(extendScreenConnectStatus));
    OnExtendScreenConnectStatusChanged(screenId, extendScreenConnectStatus);
    return ERR_NONE;
}
} // namespace OHOS::Rosen