/*
 * Copyright (c) 2018-2022, Andreas Kling <andreas@ladybird.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <AK/Debug.h>
#include <LibWeb/HTML/NavigableContainer.h>
#include <LibWeb/Layout/NavigableContainerViewport.h>
#include <LibWeb/Layout/Viewport.h>
#include <LibWeb/Painting/BorderRadiusCornerClipper.h>
#include <LibWeb/Painting/DisplayListRecorder.h>
#include <LibWeb/Painting/NavigableContainerViewportPaintable.h>
#include <LibWeb/Painting/ViewportPaintable.h>

namespace Web::Painting {

GC_DEFINE_ALLOCATOR(NavigableContainerViewportPaintable);

GC::Ref<NavigableContainerViewportPaintable> NavigableContainerViewportPaintable::create(Layout::NavigableContainerViewport const& layout_box)
{
    return layout_box.heap().allocate<NavigableContainerViewportPaintable>(layout_box);
}

NavigableContainerViewportPaintable::NavigableContainerViewportPaintable(Layout::NavigableContainerViewport const& layout_box)
    : PaintableBox(layout_box)
{
}

Layout::NavigableContainerViewport const& NavigableContainerViewportPaintable::layout_box() const
{
    return static_cast<Layout::NavigableContainerViewport const&>(layout_node());
}

void NavigableContainerViewportPaintable::paint(PaintContext& context, PaintPhase phase) const
{
    if (!is_visible())
        return;

    PaintableBox::paint(context, phase);

    if (phase == PaintPhase::Foreground) {
        auto absolute_rect = this->absolute_rect();
        auto clip_rect = context.rounded_device_rect(absolute_rect);
        ScopedCornerRadiusClip corner_clip { context, clip_rect, normalized_border_radii_data(ShrinkRadiiForBorders::Yes) };

        auto const* hosted_document = layout_box().dom_node().content_document_without_origin_check();
        if (!hosted_document)
            return;
        auto const* hosted_paint_tree = hosted_document->paintable();
        if (!hosted_paint_tree)
            return;

        context.display_list_recorder().save();

        context.display_list_recorder().add_clip_rect(clip_rect.to_type<int>());

        DOM::Document::PaintConfig paint_config;
        paint_config.paint_overlay = context.should_paint_overlay();
        paint_config.should_show_line_box_borders = context.should_show_line_box_borders();
        paint_config.has_focus = context.has_focus();
        auto display_list = const_cast<DOM::Document*>(hosted_document)->record_display_list(paint_config);
        auto scroll_state_snapshot = hosted_document->paintable()->scroll_state().snapshot();
        context.display_list_recorder().paint_nested_display_list(display_list, move(scroll_state_snapshot), context.enclosing_device_rect(absolute_rect).to_type<int>());

        context.display_list_recorder().restore();

        if constexpr (HIGHLIGHT_FOCUSED_FRAME_DEBUG) {
            if (layout_box().dom_node().content_navigable()->is_focused()) {
                context.display_list_recorder().draw_rect(clip_rect.to_type<int>(), Color::Cyan);
            }
        }
    }
}

}