OpenHarmony-v4.1-Release/s

/*
 * 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 "core/components/stack/stack_element.h"

#include "bridge/declarative_frontend/view_stack_processor.h"
#include "core/components/bubble/bubble_element.h"
#include "core/components/bubble/render_bubble.h"
#include "core/components/common/properties/animation_option.h"
#include "core/components/dialog/dialog_component.h"
#include "core/components/dialog/dialog_element.h"
#include "core/components/dialog_tween/render_dialog_tween.h"
#include "core/components/drop_filter/drop_filter_element.h"
#include "core/components/page/page_element.h"
#include "core/components/picker/picker_base_element.h"
#include "core/components/popup/popup_component.h"
#include "core/components/popup/popup_theme.h"
#include "core/components/select_popup/select_popup_element.h"
#include "core/components/text_overlay/text_overlay_element.h"
#include "core/components_v2/inspector/inspector_composed_component.h"
#include "core/components_v2/inspector/inspector_composed_element.h"
#include "core/pipeline/base/composed_element.h"

namespace OHOS::Ace {

void StackElement::PushInstant(const RefPtr<Component>& newComponent, bool disableTouchEvent)
{
    PopupComponentInfo pushComponentInfo = { -1, "-1", Operation::DIRECT_PUSH, newComponent };
    popupComponentInfos_.emplace_back(pushComponentInfo);
    disableTouchEvent_ = disableTouchEvent;
    PerformBuild();
}

void StackElement::PushComponent(const RefPtr<Component>& newComponent, bool disableTouchEvent)
{
    PopupComponentInfo pushComponentInfo = { -1, "-1", Operation::DIRECT_PUSH, newComponent };
    popupComponentInfos_.emplace_back(pushComponentInfo);
    disableTouchEvent_ = disableTouchEvent;
    MarkDirty();
}

void StackElement::PopComponent()
{
    PopupComponentInfo popComponentInfo = { -1, "-1", Operation::DIRECT_POP, nullptr };
    popupComponentInfos_.emplace_back(popComponentInfo);
    MarkDirty();
}

void StackElement::PushToastComponent(const RefPtr<Component>& newComponent, int32_t toastId)
{
    PopupComponentInfo pushComponentInfo = { toastId, "-1", Operation::TOAST_PUSH, newComponent };
    popupComponentInfos_.emplace_back(pushComponentInfo);
    MarkDirty();
}

void StackElement::PopToastComponent(int32_t toastPopId)
{
    PopupComponentInfo popComponentInfo = { toastPopId, "-1", Operation::TOAST_POP, nullptr };
    popupComponentInfos_.emplace_back(popComponentInfo);
    MarkDirty();
}

void StackElement::PushPanel(const RefPtr<Component>& newComponent, bool disableTouchEvent)
{
    PopupComponentInfo pushComponentInfo = { -1, "-1", Operation::PANEL_PUSH, newComponent };
    popupComponentInfos_.emplace_back(pushComponentInfo);
    disableTouchEvent_ = disableTouchEvent;
    MarkDirty();
}

bool StackElement::HasOverlayChild()
{
    bool hasMultiChild = children_.size() > 1;
    // avoid pop toast to protect page not empty.
    for (const auto& child: children_) {
        if (!std::none_of(toastStack_.begin(), toastStack_.end(),
            [child](const ToastInfo& toast) { return toast.child == child; })) {
            return false;
        }
    }
    return hasMultiChild;
}

void StackElement::PopPanel()
{
    PopDialog();
}

bool StackElement::PushDialog(const RefPtr<Component>& newComponent, bool disableTouchEvent)
{
    auto context = context_.Upgrade();
    if (context) {
        AccessibilityEvent stackEvent;
        stackEvent.eventType = "ejectdismiss";
        context->SendEventToAccessibility(stackEvent);
    }
    PopupComponentInfo pushComponentInfo = { -1, "-1", Operation::DIALOG_PUSH, newComponent };
    CreateInspectorComponent(pushComponentInfo);

    popupComponentInfos_.emplace_back(pushComponentInfo);
    disableTouchEvent_ = disableTouchEvent;
    MarkDirty();
    return true;
}

bool StackElement::PopDialog(int32_t id)
{
    LOGI("StackElement::PopDialog id is %{public}d", id);
    auto context = context_.Upgrade();
    if (context) {
        AccessibilityEvent stackEvent;
        stackEvent.eventType = "ejectdismiss";
        context->SendEventToAccessibility(stackEvent);
    }
    PopupComponentInfo popComponentInfo = { id, "-1", Operation::DIALOG_POP, nullptr };
    popupComponentInfos_.emplace_back(popComponentInfo);
    MarkDirty();
    return true;
}

void StackElement::PopTextOverlay()
{
    PopupComponentInfo popComponentInfo = { -1, "-1", Operation::TEXT_OVERLAY_POP, nullptr };
    popupComponentInfos_.emplace_back(popComponentInfo);
    MarkDirty();
}

void StackElement::PopPopup(const ComposeId& id)
{
    PopupComponentInfo popComponentInfo = { -1, id, Operation::POPUP_POP, nullptr };
    popupComponentInfos_.emplace_back(popComponentInfo);
    MarkDirty();
}

void StackElement::PopMenu()
{
    PopupComponentInfo popComponentInfo = { -1, "-1", Operation::MENU_POP, nullptr };
    popupComponentInfos_.emplace_back(popComponentInfo);
    MarkDirty();
}

void StackElement::PopVideo()
{
    PopupComponentInfo popComponentInfo = { -1, "-1", Operation::VIDEO_POP, nullptr };
    popupComponentInfos_.emplace_back(popComponentInfo);
    MarkDirty();
}

void StackElement::PopInstant()
{
    auto child = children_.end();
    if (child != children_.begin()) {
        child--;
        UpdateChild(*child, nullptr);
    }
    EnableTouchEventAndRequestFocus();
}

void StackElement::PerformBuild()
{
    // rebuild popup component
    for (auto& info : popupComponentInfos_) {
        PerformPopupChild(info);
    }

    ComponentGroupElement::PerformBuild();
    popupComponentInfos_.clear();
}

void StackElement::PerformPopupChild(PopupComponentInfo& popupComponentInfo)
{
    switch (popupComponentInfo.operation) {
        case Operation::TOAST_PUSH:
            PerformPushToast(popupComponentInfo);
            break;
        case Operation::DIRECT_PUSH:
        case Operation::DIALOG_PUSH:
        case Operation::PANEL_PUSH:
            PerformPushChild(popupComponentInfo);
            break;
        case Operation::TOAST_POP:
            PerformPopToastById(popupComponentInfo.popId);
            break;
        case Operation::DIALOG_POP:
            PerformPopDialog(popupComponentInfo.popId);
            break;
        case Operation::TEXT_OVERLAY_POP:
            PerformPopTextOverlay();
            break;
        case Operation::POPUP_POP:
            PerformPopPopup(popupComponentInfo.id);
            break;
        case Operation::MENU_POP:
            PerformPopMenu();
            break;
        case Operation::VIDEO_POP:
            PerformPopVideo();
            break;
        case Operation::DIRECT_POP:
            PerformDirectPop();
            break;
        default:
            break;
    }
}

void StackElement::PerformPushToast(PopupComponentInfo& popupComponentInfo)
{
    if (!popupComponentInfo.IsValid() || popupComponentInfo.operation != Operation::TOAST_PUSH) {
        return;
    }
    PerformPopToast();
    // store toast element
    RefPtr<Element> toastElement = UpdateChild(nullptr, popupComponentInfo.component);
    if (toastElement) {
        ToastInfo toastInfo = { popupComponentInfo.popId, toastElement };
        toastStack_.emplace_back(toastInfo);
    }
    popupComponentInfo.component = nullptr;
}

void StackElement::PerformPushChild(PopupComponentInfo& popupComponentInfo)
{
    if (!popupComponentInfo.IsValid()) {
        return;
    }
    // store toast element
    if (!UpdateChild(nullptr, popupComponentInfo.component)) {
        return;
    }
    for (auto child = (++children_.rbegin()); child != children_.rend(); ++child) {
        auto renderNode = (*child)->GetRenderNode();
        if (renderNode) {
            renderNode->SetDisableTouchEvent(disableTouchEvent_);
        }
    }
    auto renderNode = GetRenderNode();
    if (!renderNode) {
        return;
    }
    renderNode->MarkNeedLayout();
    if (isPageElement()) {
        if (!focusNodes_.empty() && focusNodes_.back()->IsFocusable()) {
            focusNodes_.back()->RequestFocus();
        }
    }
    popupComponentInfo.component = nullptr;
}

void StackElement::PerformPopToastById(int32_t toastId)
{
    if (toastStack_.empty()) {
        return;
    }
    for (auto iter = toastStack_.end() - 1; iter >= toastStack_.begin(); --iter) {
        if (iter->toastId == toastId) {
            UpdateChild(iter->child, nullptr);
            toastStack_.erase(iter);
            break;
        }
    }
    EnableTouchEventAndRequestFocus();
}

void StackElement::PerformPopToast()
{
    if (!toastStack_.empty()) {
        UpdateChild(toastStack_.back().child, nullptr);
        toastStack_.pop_back();
    }
}

void StackElement::PerformPopDialog(int32_t id)
{
    if (id >= 0) {
        PerformPopDialogById(id);
        return;
    }

    bool hasDialog = std::any_of(children_.begin(), children_.end(), [](const RefPtr<Element>& child) {
        return AceType::InstanceOf<V2::InspectorComposedElement>(child) || AceType::InstanceOf<DialogElement>(child) ||
               AceType::InstanceOf<PickerBaseElement>(child) || AceType::InstanceOf<DropFilterElement>(child);
    });
    if (!hasDialog) {
        EnableTouchEventAndRequestFocus();
        return;
    }
    for (auto iter = children_.rbegin(); iter != children_.rend(); ++iter) {
        if (AceType::InstanceOf<V2::InspectorComposedElement>(*iter) || AceType::InstanceOf<DialogElement>(*iter) ||
            AceType::InstanceOf<PickerBaseElement>(*iter) || AceType::InstanceOf<DropFilterElement>(*iter)) {
            UpdateChild(*iter, nullptr);
            break;
        }
    }
    EnableTouchEventAndRequestFocus();
}

void StackElement::PerformPopDialogById(int32_t id)
{
    bool hasDialog = std::any_of(children_.begin(), children_.end(), [](const RefPtr<Element>& child) {
        return AceType::InstanceOf<V2::InspectorComposedElement>(child) || AceType::InstanceOf<DialogElement>(child);
    });
    if (!hasDialog) {
        EnableTouchEventAndRequestFocus();
        return;
    }
    for (auto iter = children_.rbegin(); iter != children_.rend(); ++iter) {
        auto dialogElement = AceType::DynamicCast<DialogElement>(*iter);
        if (dialogElement && dialogElement->GetDialogId() == id) {
            UpdateChild(*iter, nullptr);
            break;
        }
        auto inspectorComposedElement = AceType::DynamicCast<V2::InspectorComposedElement>(*iter);
        if (inspectorComposedElement) {
            dialogElement = inspectorComposedElement->GetContentElement<DialogElement>(DialogElement::TypeId());
            if (dialogElement && dialogElement->GetDialogId() == id) {
                UpdateChild(inspectorComposedElement, nullptr);
                break;
            }
        }
    }
    EnableTouchEventAndRequestFocus();
}

void StackElement::PerformPopTextOverlay()
{
    for (auto iter = children_.rbegin(); iter != children_.rend(); ++iter) {
        if (AceType::InstanceOf<TextOverlayElement>(*iter)) {
            UpdateChild(*iter, nullptr);
            break;
        }
    }
    if (IsFocusable()) {
        RequestFocus();
    }
    EnableTouchEventAndRequestFocus();
}

void StackElement::PerformPopPopup(const ComposeId& id)
{
    for (auto iter = children_.rbegin(); iter != children_.rend(); ++iter) {
        auto child = DynamicCast<TweenElement>(*iter);
        if (child && child->GetId() == id) {
            auto themeManager = GetThemeManager();
            if (!themeManager || !themeManager->GetTheme<PopupTheme>()) {
                LOGE("themeManager or get theme is null!");
                return;
            }

            auto context = context_.Upgrade();
            if (context && !context->GetOnShow()) {
                UpdateChild(child, nullptr);
                break;
            }

            auto theme = themeManager->GetTheme<PopupTheme>();
            auto hideAlphaAnimation = AceType::MakeRefPtr<CurveAnimation<float>>(1.0f, 0.0f, Curves::FAST_OUT_SLOW_IN);
            TweenOption hideOption;
            hideOption.SetDuration(theme->GetHideTime());
            hideOption.SetOpacityAnimation(hideAlphaAnimation);

            auto animator = child->GetController();
            animator->ClearAllListeners();
            child->SetOption(hideOption);
            child->ApplyOptions();
            child->ApplyKeyframes();
            animator->AddStopListener(
                [weakStack = AceType::WeakClaim(this), weakChild = AceType::WeakClaim(AceType::RawPtr(child))] {
                    auto lastStack = weakStack.Upgrade();
                    auto child = weakChild.Upgrade();
                    if (lastStack && child) {
                        lastStack->UpdateChild(child, nullptr);
                    }
                });
            animator->Play();
            break;
        }
    }
    if (IsFocusable()) {
        RequestFocus();
    }
    EnableTouchEventAndRequestFocus();
}

void StackElement::PerformPopMenu()
{
    for (auto iter = children_.rbegin(); iter != children_.rend(); ++iter) {
        if (AceType::InstanceOf<SelectPopupElement>(*iter)) {
            UpdateChild(*iter, nullptr);
            break;
        }
    }
    if (IsFocusable()) {
        RequestFocus();
    }
    EnableTouchEventAndRequestFocus();
}

void StackElement::PerformPopVideo()
{
    for (auto iter = children_.rbegin(); iter != children_.rend(); ++iter) {
        auto element = DynamicCast<ComposedElement>(*iter);
        if (element && StringUtils::EndWith(element->GetName(), "fullscreen")) {
            UpdateChild(*iter, nullptr);
            break;
        }
    }
    EnableTouchEventAndRequestFocus();
}

void StackElement::PerformDirectPop()
{
    auto child = children_.end();
    while (child != children_.begin()) {
        child--;
        bool isNotToast = std::none_of(
            toastStack_.begin(), toastStack_.end(), [child](const ToastInfo& toast) { return toast.child == *child; });
        if (isNotToast) {
            UpdateChild(*child, nullptr);
            break;
        }
    }
    EnableTouchEventAndRequestFocus();
}

bool StackElement::RequestNextFocus(bool vertical, bool reverse, const Rect& rect)
{
    if (!isPageElement()) {
        return GoToNextFocus(reverse, rect);
    }
    return false;
}

void StackElement::OnFocus()
{
    if (!isPageElement()) {
        FocusGroup::OnFocus();
        return;
    }
    if (focusNodes_.empty()) {
        itLastFocusNode_ = focusNodes_.end();
        return;
    }
    // Only focus on the top focusable child.
    itLastFocusNode_ = focusNodes_.end();
    while (itLastFocusNode_ != focusNodes_.begin()) {
        --itLastFocusNode_;
        (*itLastFocusNode_)->SetParentFocusable(IsParentFocusable());
        if ((*itLastFocusNode_)->RequestFocusImmediately()) {
            FocusNode::OnFocus();
            break;
        }
    }

    if (!IsCurrentFocus()) {
        itLastFocusNode_ = focusNodes_.end();
    } else {
        // lower focusable node can not be focus.
        auto iter = itLastFocusNode_;
        while (iter != focusNodes_.begin()) {
            --iter;
            (*iter)->SetParentFocusable(false);
        }
    }
}

void StackElement::OnBlur()
{
    FocusGroup::OnBlur();
    if (!isPageElement()) {
        return;
    }

    auto iter = focusNodes_.end();
    while (iter != focusNodes_.begin()) {
        --iter;
        (*iter)->SetParentFocusable(IsParentFocusable());
    }
}

void StackElement::EnableTouchEventAndRequestFocus()
{
    for (auto& child : children_) {
        auto renderNode = child->GetRenderNode();
        if (renderNode) {
            renderNode->SetDisableTouchEvent(false);
        }
    }
    if (IsFocusable()) {
        RequestFocus();
    }
}

void StackElement::CreateInspectorComponent(PopupComponentInfo& componentInfo) const
{
    auto dialog = AceType::DynamicCast<DialogComponent>(componentInfo.component);
    if (!dialog) {
        return;
    }
    auto inspectorTag = dialog->GetInspectorTag();
    if (V2::InspectorComposedComponent::HasInspectorFinished(inspectorTag)) {
        auto composedComponent = AceType::MakeRefPtr<V2::InspectorComposedComponent>(
            V2::InspectorComposedComponent::GenerateId(), inspectorTag);
        composedComponent->SetChild(componentInfo.component);
        componentInfo.component = composedComponent;
    }
}

bool StackElement::PopPopupIfExist() const
{
    auto bubbleElement = GetBubble(GetLastChild());
    if (!bubbleElement) {
        return false;
    }
    auto renderBubble = DynamicCast<RenderBubble>(bubbleElement->GetRenderNode());
    if (!renderBubble) {
        return false;
    }
    renderBubble->PopBubble();
    return true;
}

RefPtr<BubbleElement> StackElement::GetBubble(const RefPtr<Element>& element) const
{
    if (!element) {
        return nullptr;
    }

    auto bubble = DynamicCast<BubbleElement>(element);
    if (bubble) {
        return bubble;
    }

    return GetBubble(element->GetFirstChild());
}

bool StackElement::PopDialogIfExist() const
{
    auto dialogTweenElement = GetDialog(GetLastChild());
    if (!dialogTweenElement) {
        return false;
    }
    auto renderDialogTween = DynamicCast<RenderDialogTween>(dialogTweenElement->GetRenderNode());
    if (!renderDialogTween) {
        return false;
    }
    renderDialogTween->PopDialog();
    return true;
}

RefPtr<DialogTweenElement> StackElement::GetDialog(const RefPtr<Element>& element) const
{
    if (!element) {
        return nullptr;
    }

    auto dialog = DynamicCast<DialogTweenElement>(element);
    if (dialog) {
        return dialog;
    }

    return GetDialog(element->GetFirstChild());
}

} // namespace OHOS::Ace