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

#pragma once

#include <AK/Noncopyable.h>
#include <LibWeb/Layout/BlockContainer.h>
#include <LibWeb/Layout/LayoutState.h>
#include <LibWeb/Layout/TextNode.h>

namespace Web::Layout {

// This class iterates over all the inline-level objects within an inline formatting context.
// By repeatedly calling next() with the remaining available width on the current line,
// it returns an "Item" representing the next piece of inline-level content to be placed on the line.
class InlineLevelIterator {
    AK_MAKE_NONCOPYABLE(InlineLevelIterator);
    AK_MAKE_NONMOVABLE(InlineLevelIterator);

public:
    struct Item {
        enum class Type {
            Text,
            Element,
            ForcedBreak,
            AbsolutelyPositionedElement,
            FloatingElement,
        };
        Type type {};
        GC::Ptr<Layout::Node const> node {};
        RefPtr<Gfx::GlyphRun> glyph_run {};
        size_t offset_in_node { 0 };
        size_t length_in_node { 0 };
        CSSPixels width { 0.0f };
        CSSPixels padding_start { 0.0f };
        CSSPixels padding_end { 0.0f };
        CSSPixels border_start { 0.0f };
        CSSPixels border_end { 0.0f };
        CSSPixels margin_start { 0.0f };
        CSSPixels margin_end { 0.0f };
        bool is_collapsible_whitespace { false };
        bool can_break_before { false };

        CSSPixels border_box_width() const
        {
            return border_start + padding_start + width + padding_end + border_end;
        }
    };

    InlineLevelIterator(Layout::InlineFormattingContext&, LayoutState&, Layout::BlockContainer const& containing_block, LayoutState::UsedValues const& containing_block_used_values, LayoutMode);

    Optional<Item> next();
    CSSPixels next_non_whitespace_sequence_width();

private:
    void generate_all_items();
    Optional<Item> generate_next_item();
    Gfx::GlyphRun::TextType resolve_text_direction_from_context();
    void skip_to_next();
    void compute_next();

    void enter_text_node(Layout::TextNode const&);

    void enter_node_with_box_model_metrics(Layout::NodeWithStyleAndBoxModelMetrics const&);
    void exit_node_with_box_model_metrics();

    void add_extra_box_model_metrics_to_item(Item&, bool add_leading_metrics, bool add_trailing_metrics);

    Layout::Node const* next_inline_node_in_pre_order(Layout::Node const& current, Layout::Node const* stay_within);

    Layout::InlineFormattingContext& m_inline_formatting_context;
    Layout::LayoutState& m_layout_state;
    GC::Ref<BlockContainer const> m_containing_block;
    LayoutState::UsedValues const& m_containing_block_used_values;
    GC::Ptr<Layout::Node const> m_current_node;
    GC::Ptr<Layout::Node const> m_next_node;
    LayoutMode const m_layout_mode;

    struct TextNodeContext {
        Vector<TextNode::Chunk> chunks;
        size_t next_chunk_index { 0 };
        bool should_collapse_whitespace {};
        bool should_wrap_lines {};
        bool should_respect_linebreaks {};
        Optional<Gfx::GlyphRun::TextType> last_known_direction {};
    };

    Optional<TextNodeContext> m_text_node_context;

    struct ExtraBoxMetrics {
        CSSPixels margin { 0 };
        CSSPixels border { 0 };
        CSSPixels padding { 0 };
    };

    Optional<ExtraBoxMetrics> m_extra_leading_metrics;
    Optional<ExtraBoxMetrics> m_extra_trailing_metrics;

    Vector<GC::Ref<NodeWithStyleAndBoxModelMetrics const>> m_box_model_node_stack;

    // Pre-generated items for O(1) iteration and lookahead.
    Vector<Item> m_items;
    size_t m_next_item_index { 0 };

    // Accumulated width tracking for tab calculations during pre-generation.
    CSSPixels m_accumulated_width_for_tabs { 0 };

    bool m_previous_chunk_can_break_after { false };
};

}