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

#pragma once

#include <LibWeb/Layout/Box.h>
#include <LibWeb/Layout/FormattingContext.h>

namespace Web::Layout {

class FlexFormattingContext final : public FormattingContext {
public:
    FlexFormattingContext(LayoutState&, LayoutMode, Box const& flex_container, FormattingContext* parent);
    ~FlexFormattingContext();

    virtual bool inhibits_floating() const override { return true; }

    virtual void run(AvailableSpace const&) override;
    virtual CSSPixels automatic_content_width() const override;
    virtual CSSPixels automatic_content_height() const override;

    Box const& flex_container() const { return context_box(); }

    StaticPositionRect calculate_static_position_rect(Box const&) const;

private:
    [[nodiscard]] bool should_treat_main_size_as_auto(Box const&) const;
    [[nodiscard]] bool should_treat_cross_size_as_auto(Box const&) const;

    [[nodiscard]] bool should_treat_main_max_size_as_none(Box const&) const;
    [[nodiscard]] bool should_treat_cross_max_size_as_none(Box const&) const;

    [[nodiscard]] CSSPixels adjust_main_size_through_aspect_ratio_for_cross_size_min_max_constraints(Box const&, CSSPixels main_size, CSS::Size const& min_cross_size, CSS::Size const& max_cross_size) const;
    [[nodiscard]] CSSPixels adjust_cross_size_through_aspect_ratio_for_main_size_min_max_constraints(Box const&, CSSPixels cross_size, CSS::Size const& min_main_size, CSS::Size const& max_main_size) const;

    [[nodiscard]] CSSPixels calculate_main_size_from_cross_size_and_aspect_ratio(CSSPixels cross_size, CSSPixelFraction aspect_ratio) const;
    [[nodiscard]] CSSPixels calculate_cross_size_from_main_size_and_aspect_ratio(CSSPixels main_size, CSSPixelFraction aspect_ratio) const;

    void dump_items() const;

    struct DirectionAgnosticMargins {
        CSSPixels main_before { 0 };
        CSSPixels main_after { 0 };
        CSSPixels cross_before { 0 };
        CSSPixels cross_after { 0 };

        bool main_before_is_auto { false };
        bool main_after_is_auto { false };
        bool cross_before_is_auto { false };
        bool cross_after_is_auto { false };
    };

    struct FlexItem {
        GC::Ref<Box> box;
        LayoutState::UsedValues& used_values;
        Optional<CSS::FlexBasis> used_flex_basis {};
        bool used_flex_basis_is_definite { false };
        bool main_size_was_resolved_from_aspect_ratio { false };
        CSSPixels flex_base_size { 0 };
        CSSPixels hypothetical_main_size { 0 };
        CSSPixels hypothetical_cross_size { 0 };
        CSSPixels hypothetical_cross_size_with_margins() { return hypothetical_cross_size + margins.cross_before + margins.cross_after + borders.cross_after + borders.cross_before + padding.cross_after + padding.cross_before; }
        CSSPixels target_main_size { 0 };
        bool frozen { false };
        Optional<double> flex_factor {};
        double scaled_flex_shrink_factor { 0 };
        double desired_flex_fraction { 0 };

        CSSPixels outer_hypothetical_main_size() const
        {
            return hypothetical_main_size + margins.main_before + margins.main_after + borders.main_before + borders.main_after + padding.main_before + padding.main_after;
        }

        CSSPixels outer_target_main_size() const
        {
            return target_main_size + margins.main_before + margins.main_after + borders.main_before + borders.main_after + padding.main_before + padding.main_after;
        }

        CSSPixels outer_flex_base_size() const
        {
            return flex_base_size + margins.main_before + margins.main_after + borders.main_before + borders.main_after + padding.main_before + padding.main_after;
        }

        // The used main size of this flex item. Empty until determined.
        Optional<CSSPixels> main_size {};

        // The used cross size of this flex item. Empty until determined.
        Optional<CSSPixels> cross_size {};

        CSSPixels main_offset { 0 };
        CSSPixels cross_offset { 0 };
        DirectionAgnosticMargins margins {};
        DirectionAgnosticMargins borders {};
        DirectionAgnosticMargins padding {};
        bool is_min_violation { false };
        bool is_max_violation { false };

        CSSPixels add_main_margin_box_sizes(CSSPixels content_size) const
        {
            return content_size + margins.main_before + margins.main_after + borders.main_before + borders.main_after + padding.main_before + padding.main_after;
        }

        CSSPixels add_cross_margin_box_sizes(CSSPixels content_size) const
        {
            return content_size + margins.cross_before + margins.cross_after + borders.cross_before + borders.cross_after + padding.cross_before + padding.cross_after;
        }
    };

    struct FlexLine {
        Vector<FlexItem&> items;
        CSSPixels cross_size { 0 };
        bool has_baseline_aligned_items { false };
        Optional<CSSPixels> remaining_free_space;
        double chosen_flex_fraction { 0 };

        double sum_of_flex_factor_of_unfrozen_items() const;
        double sum_of_scaled_flex_shrink_factor_of_unfrozen_items() const;
    };

    CSSPixels main_gap() const;
    CSSPixels cross_gap() const;
    [[nodiscard]] bool has_definite_main_size(LayoutState::UsedValues const&) const;
    [[nodiscard]] bool has_definite_cross_size(LayoutState::UsedValues const&) const;
    [[nodiscard]] bool has_definite_main_size(FlexItem const& item) const { return has_definite_main_size(item.used_values); }
    [[nodiscard]] bool has_definite_cross_size(FlexItem const& item) const { return has_definite_cross_size(item.used_values); }

    [[nodiscard]] CSSPixels inner_main_size(LayoutState::UsedValues const&) const;
    [[nodiscard]] CSSPixels inner_cross_size(LayoutState::UsedValues const&) const;
    [[nodiscard]] CSSPixels inner_main_size(FlexItem const& item) const { return inner_main_size(item.used_values); }
    [[nodiscard]] CSSPixels inner_cross_size(FlexItem const& item) const { return inner_cross_size(item.used_values); }
    bool has_main_min_size(Box const&) const;
    bool has_cross_min_size(Box const&) const;
    CSSPixels specified_main_max_size(FlexItem const&) const;
    CSSPixels specified_cross_max_size(FlexItem const&) const;
    bool is_cross_auto(Box const&) const;
    CSSPixels specified_main_min_size(FlexItem const&) const;
    CSSPixels specified_cross_min_size(FlexItem const&) const;
    CSSPixels calculate_inner_flex_container_cross_min_size() const;
    CSSPixels calculate_inner_flex_container_cross_max_size() const;
    bool has_main_max_size(Box const&) const;
    bool has_cross_max_size(Box const&) const;
    CSSPixels automatic_minimum_size(FlexItem const&) const;
    CSSPixels content_based_minimum_size(FlexItem const&) const;
    Optional<CSSPixels> specified_size_suggestion(FlexItem const&) const;
    Optional<CSSPixels> transferred_size_suggestion(FlexItem const&) const;
    CSSPixels content_size_suggestion(FlexItem const&) const;
    CSS::Size const& computed_main_size(Box const&) const;
    CSS::Size const& computed_main_min_size(Box const&) const;
    CSS::Size const& computed_main_max_size(Box const&) const;
    CSS::Size const& computed_cross_size(Box const&) const;
    CSS::Size const& computed_cross_min_size(Box const&) const;
    CSS::Size const& computed_cross_max_size(Box const&) const;

    CSSPixels get_pixel_width(FlexItem const&, CSS::Size const&) const;
    CSSPixels get_pixel_height(FlexItem const&, CSS::Size const&) const;

    bool flex_item_is_stretched(FlexItem const&) const;

    void set_main_size(Box const&, CSSPixels size);
    void set_cross_size(Box const&, CSSPixels size);
    void set_main_size(FlexItem&, CSSPixels size);
    void set_cross_size(FlexItem&, CSSPixels size);
    void set_offset(FlexItem&, CSSPixels main_offset, CSSPixels cross_offset);

    void set_main_axis_first_margin(FlexItem&, CSSPixels margin);
    void set_main_axis_second_margin(FlexItem&, CSSPixels margin);

    void set_has_definite_main_size(FlexItem&);
    void set_has_definite_cross_size(FlexItem&);

    void copy_dimensions_from_flex_items_to_boxes();

    void generate_anonymous_flex_items();

    void determine_available_space_for_items(AvailableSpace const&);

    void determine_flex_base_size(FlexItem&);

    void collect_flex_items_into_flex_lines();

    void resolve_flexible_lengths();
    void resolve_flexible_lengths_for_line(FlexLine&);

    void resolve_cross_axis_auto_margins();

    void determine_hypothetical_cross_size_of_item(FlexItem&);

    void calculate_cross_size_of_each_flex_line();

    void handle_align_content_stretch();

    CSS::AlignItems alignment_for_item(Box const&) const;

    void determine_used_cross_size_of_each_flex_item();

    void distribute_any_remaining_free_space();

    void align_all_flex_items_along_the_cross_axis();

    void resolve_baseline_aligned_items();

    void align_all_flex_lines();

    bool is_row_layout() const { return m_flex_direction == CSS::FlexDirection::Row || m_flex_direction == CSS::FlexDirection::RowReverse; }
    bool is_single_line() const { return flex_container().computed_values().flex_wrap() == CSS::FlexWrap::Nowrap; }
    bool inline_axis_is_horizontal(Box const&) const;
    bool main_axis_is_horizontal() const;
    bool cross_axis_is_horizontal() const { return !main_axis_is_horizontal(); }
    bool main_axis_is_parallel_to_inline_axis(Box const&) const;
    bool cross_axis_is_reverse() const;
    bool is_direction_reverse() const;
    void populate_specified_margins(FlexItem&, CSS::FlexDirection) const;

    void determine_intrinsic_size_of_flex_container();
    [[nodiscard]] CSSPixels calculate_intrinsic_main_size_of_flex_container();
    [[nodiscard]] CSSPixels calculate_intrinsic_cross_size_of_flex_container();

    [[nodiscard]] CSSPixels calculate_cross_min_content_contribution(FlexItem const&, bool resolve_percentage_min_max_sizes) const;
    [[nodiscard]] CSSPixels calculate_cross_max_content_contribution(FlexItem const&, bool resolve_percentage_min_max_sizes) const;
    [[nodiscard]] CSSPixels calculate_main_min_content_contribution(FlexItem const&) const;
    [[nodiscard]] CSSPixels calculate_main_max_content_contribution(FlexItem const&) const;

    [[nodiscard]] CSSPixels calculate_min_content_main_size(FlexItem const&) const;
    [[nodiscard]] CSSPixels calculate_max_content_main_size(FlexItem const&) const;
    [[nodiscard]] CSSPixels calculate_min_content_cross_size(FlexItem const&) const;
    [[nodiscard]] CSSPixels calculate_max_content_cross_size(FlexItem const&) const;

    [[nodiscard]] CSSPixels calculate_fit_content_main_size(FlexItem const&) const;
    [[nodiscard]] CSSPixels calculate_fit_content_cross_size(FlexItem const&) const;

    [[nodiscard]] CSSPixels calculate_width_to_use_when_determining_intrinsic_height_of_item(FlexItem const&) const;

    virtual void parent_context_did_dimension_child_root_box() override;

    CSS::FlexBasis used_flex_basis_for_item(FlexItem const&) const;

    LayoutState::UsedValues& m_flex_container_state;

    Vector<FlexLine> m_flex_lines;
    Vector<FlexItem> m_flex_items;
    CSS::FlexDirection m_flex_direction {};

    struct AxisAgnosticAvailableSpace {
        AvailableSize main;
        AvailableSize cross;
        AvailableSpace space;
    };
    Optional<AxisAgnosticAvailableSpace> m_available_space_for_items;
    Optional<AvailableSpace> m_available_space;
};

}
