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

#include "ability_manager_client.h"
#include "memory_guard.h"
#include "window.h"
#include "window_manager_hilog.h"
#include "xcollie/watchdog.h"
#include "perform_reporter.h"

namespace OHOS {
namespace Rosen {
namespace {
constexpr HiviewDFX::HiLogLabel LABEL = {LOG_CORE, HILOG_DOMAIN_WINDOW, "InnerManager"};
}
WM_IMPLEMENT_SINGLE_INSTANCE(WindowInnerManager)

WindowInnerManager::WindowInnerManager() : eventHandler_(nullptr), eventLoop_(nullptr),
    state_(InnerWMRunningState::STATE_NOT_START)
{
}

WindowInnerManager::~WindowInnerManager()
{
    Stop();
}

bool WindowInnerManager::Init()
{
    // create handler for inner command at server
    eventLoop_ = AppExecFwk::EventRunner::Create(INNER_WM_THREAD_NAME);
    if (eventLoop_ == nullptr) {
        return false;
    }
    eventHandler_ = std::make_shared<AppExecFwk::EventHandler>(eventLoop_);
    if (eventHandler_ == nullptr) {
        return false;
    }
    int ret = HiviewDFX::Watchdog::GetInstance().AddThread(INNER_WM_THREAD_NAME, eventHandler_);
    if (ret != 0) {
        WLOGFE("Add watchdog thread failed");
    }
    eventHandler_->PostTask([]() { MemoryGuard cacheGuard; }, "wms:Init:cacheGuard",
        0, AppExecFwk::EventQueue::Priority::IMMEDIATE);
    moveDragController_ = new MoveDragController();
    if (!moveDragController_->Init()) {
        WLOGFE("Init window drag controller failed");
        return false;
    }

    WLOGI("init window inner manager service success.");
    return true;
}

void WindowInnerManager::Start(bool enableRecentholder)
{
    isRecentHolderEnable_ = enableRecentholder;
    if (state_ == InnerWMRunningState::STATE_RUNNING) {
        WLOGI("window inner manager service has already started.");
    }
    if (!Init()) {
        WLOGI("failed to init window inner manager service.");
        return;
    }
    state_ = InnerWMRunningState::STATE_RUNNING;
    eventLoop_->Run();

    pid_ = getpid();
    WLOGI("window inner manager service start success.");
}

void WindowInnerManager::Stop()
{
    WLOGI("stop window inner manager service.");
    if (eventLoop_ != nullptr) {
        eventLoop_->Stop();
        eventLoop_.reset();
    }
    if (eventHandler_ != nullptr) {
        eventHandler_.reset();
    }
    if (moveDragController_ != nullptr) {
        moveDragController_->Stop();
    }
    state_ = InnerWMRunningState::STATE_NOT_START;
}

void WindowInnerManager::CreateInnerWindow(std::string name, DisplayId displayId, Rect rect,
    WindowType type, WindowMode mode)
{
    bool recentHolderWindowFlag = isRecentHolderEnable_;
    auto task = [name, displayId, rect, mode, type, recentHolderWindowFlag]() {
        if (type == WindowType::WINDOW_TYPE_PLACEHOLDER && recentHolderWindowFlag) {
            PlaceHolderWindow::GetInstance().Create(name, displayId, rect, mode);
        }
    };
    PostTask(task, "CreateInnerWindow");
    return;
}

void WindowInnerManager::DestroyInnerWindow(DisplayId displayId, WindowType type)
{
    bool recentHolderWindowFlag = isRecentHolderEnable_;
    auto task = [type, recentHolderWindowFlag]() {
        if (type == WindowType::WINDOW_TYPE_PLACEHOLDER && recentHolderWindowFlag) {
            PlaceHolderWindow::GetInstance().Destroy();
        }
    };
    PostTask(task, "DestroyInnerWindow");
    return;
}

void WindowInnerManager::UpdateInnerWindow(DisplayId displayId, WindowType type, uint32_t width, uint32_t height)
{
    bool recentHolderWindowFlag = isRecentHolderEnable_;
    auto task = [type, width, height, recentHolderWindowFlag]() {
        if (type == WindowType::WINDOW_TYPE_PLACEHOLDER && recentHolderWindowFlag) {
            PlaceHolderWindow::GetInstance().Update(width, height);
        }
    };
    PostTask(task, "UpdateInnerWindow");
    return;
}

void WindowInnerManager::MinimizeAbility(const wptr<WindowNode> &node, bool isFromUser)
{
    // asynchronously calls the MinimizeAbility of AbilityManager
    auto weakNode = node.promote();
    if (weakNode == nullptr) {
        WLOGFE("minimize ability failed.");
        return;
    }
    wptr<IRemoteObject> weakToken(weakNode->abilityToken_);
    WLOGFD("minimize window %{public}u,  isfromuser: %{public}d", weakNode->GetWindowId(), isFromUser);
    auto task = [weakToken, isFromUser]() {
        auto token = weakToken.promote();
        if (token == nullptr) {
            WLOGE("minimize ability failed, because window token is nullptr.");
            return;
        }
        AAFwk::AbilityManagerClient::GetInstance()->MinimizeAbility(token, isFromUser);
    };
    PostTask(task, "MinimizeAbility");
}

void WindowInnerManager::TerminateAbility(const wptr<WindowNode> &node)
{
    // asynchronously calls the TerminateAbility of AbilityManager
    auto weakNode = node.promote();
    if (weakNode == nullptr) {
        WLOGFE("terminate ability failed.");
        return;
    }
    wptr<IRemoteObject> weakToken(weakNode->abilityToken_);
    WLOGFD("terminate window %{public}u", weakNode->GetWindowId());
    auto task = [weakToken]() {
        auto token = weakToken.promote();
        if (token == nullptr) {
            WLOGE("terminate ability failed, because window token is nullptr.");
            return;
        }
        AAFwk::Want resultWant;
        AAFwk::AbilityManagerClient::GetInstance()->TerminateAbility(token, -1, &resultWant);
    };
    PostTask(task, "TerminateAbility");
}

void WindowInnerManager::CloseAbility(const wptr<WindowNode> &node)
{
    // asynchronously calls the CloseAbility of AbilityManager
    auto weakNode = node.promote();
    if (weakNode == nullptr) {
        WLOGFE("close ability failed.");
        return;
    }
    wptr<IRemoteObject> weakToken(weakNode->abilityToken_);
    WLOGFD("close window %{public}u", weakNode->GetWindowId());
    auto task = [weakToken]() {
        auto token = weakToken.promote();
        if (token == nullptr) {
            WLOGE("close ability failed, because window token is nullptr.");
            return;
        }
        AAFwk::Want resultWant;
        AAFwk::AbilityManagerClient::GetInstance()->CloseAbility(token);
    };
    PostTask(task, "CloseAbility");
}

void WindowInnerManager::CompleteFirstFrameDrawing(const wptr<WindowNode> &node)
{
    // asynchronously calls the CloseAbility of AbilityManager
    auto weakNode = node.promote();
    if (weakNode == nullptr) {
        WLOGFE("CompleteFirstFrameDrawing failed.");
        return;
    }
    wptr<IRemoteObject> weakToken(weakNode->abilityToken_);
    WLOGFD("CompleteFirstFrameDrawing %{public}u", weakNode->GetWindowId());
    auto task = [weakToken]() {
        auto token = weakToken.promote();
        if (token == nullptr) {
            WLOGE("CompleteFirstFrameDrawing failed, because window token is nullptr.");
            return;
        }
        AAFwk::AbilityManagerClient::GetInstance()->CompleteFirstFrameDrawing(token);
    };
    PostTask(task, "CompleteFirstFrameDrawing");
}

void WindowInnerManager::UpdateMissionSnapShot(const wptr<WindowNode> &node, std::shared_ptr<Media::PixelMap> pixelMap)
{
    // asynchronously calls the UpdateMissionSnapShot of AbilityManager
    auto weakNode = node.promote();
    if (weakNode == nullptr) {
        WLOGFE("UpdateMissionSnapShot failed.");
        return;
    }
    wptr<IRemoteObject> weakToken(weakNode->abilityToken_);
    WLOGFD("Update id %{public}u", weakNode->GetWindowId());
    auto task = [weakToken, pixelMap]() {
        auto token = weakToken.promote();
        if (token == nullptr) {
            WLOGE("UpdateMissionSnapShot failed, because window token is nullptr.");
            return;
        }
        if (pixelMap == nullptr) {
            WLOGE("UpdateMissionSnapShot failed, because pixelMap is nullptr.");
            return;
        }
        AAFwk::AbilityManagerClient::GetInstance()->UpdateMissionSnapShot(token, pixelMap);
    };
    PostTask(task, "UpdateMissionSnapShot");
}

void WindowInnerManager::PostTask(InnerTask &&task, std::string name, EventPriority priority)
{
    if (eventHandler_ == nullptr) {
        WLOGFE("listener handler is nullptr");
        return;
    }
    bool ret = eventHandler_->PostTask(task, "wms:" + name, 0, priority); // 0 is task delay time
    if (!ret) {
        WLOGFE("post listener callback task failed.");
        return;
    }
    return;
}

pid_t WindowInnerManager::GetPid()
{
    return pid_;
}

void WindowInnerManager::SetInputEventConsumer()
{
    if (moveDragController_ == nullptr) {
        return;
    }
    moveDragController_->SetInputEventConsumer();
}

void WindowInnerManager::NotifyDisplayLimitRectChange(const std::map<DisplayId, Rect>& limitRectMap)
{
    if (moveDragController_ == nullptr) {
        return;
    }
    moveDragController_->HandleDisplayLimitRectChange(limitRectMap);
}

bool WindowInnerManager::NotifyServerReadyToMoveOrDrag(uint32_t windowId, sptr<WindowProperty>& windowProperty,
    sptr<MoveDragProperty>& moveDragProperty)
{
    if (moveDragController_->GetActiveWindowId() != INVALID_WINDOW_ID) {
        WLOGFW("Is already in dragging or moving state, invalid operation");
        return false;
    }
    moveDragController_->HandleReadyToMoveOrDrag(windowId, windowProperty, moveDragProperty);
    WLOGI("NotifyServerReadyToMoveOrDrag, windowId: %{public}u", windowId);
    return true;
}

void WindowInnerManager::NotifyWindowEndUpMovingOrDragging(uint32_t windowId)
{
    if (moveDragController_->GetActiveWindowId() != windowId) {
        return;
    }
    moveDragController_->HandleEndUpMovingOrDragging(windowId);
    WLOGI("NotifyWindowEndUpMovingOrDragging, windowId: %{public}u", windowId);
}

void WindowInnerManager::NotifyWindowRemovedOrDestroyed(uint32_t windowId)
{
    if (moveDragController_->GetActiveWindowId() != windowId) {
        return;
    }
    moveDragController_->HandleWindowRemovedOrDestroyed(windowId);
    WLOGI("NotifyWindowRemovedOrDestroyed, windowId: %{public}u", windowId);
}

void WindowInnerManager::ConsumePointerEvent(const std::shared_ptr<MMI::PointerEvent>& pointerEvent)
{
    uint32_t windowId = static_cast<uint32_t>(pointerEvent->GetAgentWindowId());
    if (moveDragController_->GetActiveWindowId() != windowId ||
        moveDragController_->GetActiveWindowId() == INVALID_WINDOW_ID) {
        WLOGFE("active winId or inputEvent winId is invalid, windowId: %{public}u, activeWinId: %{public}u",
            windowId, moveDragController_->GetActiveWindowId());
        return;
    }
    moveDragController_->ConsumePointerEvent(pointerEvent);
}

void WindowInnerManager::StartWindowInfoReportLoop()
{
    if (isReportTaskStart_ || eventHandler_ == nullptr) {
        return;
    }
    auto task = [this]() {
        WindowInfoReporter::GetInstance().ReportRecordedInfos();
        isReportTaskStart_ = false;
        StartWindowInfoReportLoop();
    };
    int64_t delayTime = 1000 * 60 * 60; // an hour.
    bool ret = eventHandler_->PostTask(task, "wms:WindowInfoReport", delayTime);
    if (!ret) {
        WLOGFE("post listener callback task failed. the task name is WindowInfoReport");
        return;
    }
    isReportTaskStart_ = true;
}

void WindowInnerManager::SetWindowRoot(const sptr<WindowRoot> &windowRoot)
{
    if (moveDragController_ != nullptr) {
        moveDragController_->SetWindowRoot(windowRoot);
    }
}
} // Rosen
} // OHOS