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

#pragma once

#include <AK/Function.h>
#include <LibGC/Ptr.h>
#include <LibGC/WeakContainer.h>
#include <LibWeb/Bindings/PlatformObject.h>
#include <LibWeb/Forward.h>

namespace Web::DOM {

// NOTE: HTMLCollection is in the DOM namespace because it's part of the DOM specification.

// This class implements a live, filtered view of a DOM subtree.
// When constructing an HTMLCollection, you provide a root node + a filter.
// The filter is a simple Function object that answers the question
// "is this Element part of the collection?"

class HTMLCollection
    : public Bindings::PlatformObject
    , public GC::WeakContainer {
    WEB_PLATFORM_OBJECT(HTMLCollection, Bindings::PlatformObject);
    GC_DECLARE_ALLOCATOR(HTMLCollection);

public:
    enum class Scope {
        Children,
        Descendants,
    };
    [[nodiscard]] static GC::Ref<HTMLCollection> create(ParentNode& root, Scope, ESCAPING Function<bool(Element const&)> filter, ESCAPING Function<bool(Element const&, Element const&)> sort = nullptr);

    virtual ~HTMLCollection() override;

    size_t length() const;
    Element* item(size_t index) const;
    Element* named_item(FlyString const& key) const;

    GC::RootVector<GC::Ref<Element>> collect_matching_elements() const;

    virtual Optional<JS::Value> item_value(size_t index) const override;
    virtual JS::Value named_item_value(FlyString const& name) const override;
    virtual Vector<FlyString> supported_property_names() const override;
    virtual bool is_supported_property_name(FlyString const&) const override;

protected:
    HTMLCollection(ParentNode& root, Scope, ESCAPING Function<bool(Element const&)> filter, ESCAPING Function<bool(Element const&, Element const&)> sort = nullptr);

    virtual void initialize(JS::Realm&) override;

    GC::Ref<ParentNode> root() { return *m_root; }
    GC::Ref<ParentNode const> root() const { return *m_root; }

private:
    virtual void visit_edges(Cell::Visitor&) override;
    virtual void remove_dead_cells(Badge<GC::Heap>) override;
    virtual GC::Cell const& owner_cell(Badge<GC::Heap>) const override;

    void update_cache_if_needed() const;
    void update_name_to_element_mappings_if_needed() const;

    mutable u64 m_cached_dom_tree_version { 0 };
    mutable Vector<GC::RawPtr<Element>> m_cached_elements;
    mutable OwnPtr<OrderedHashMap<FlyString, GC::RawPtr<Element>>> m_cached_name_to_element_mappings;

    GC::Ref<ParentNode> m_root;
    Function<bool(Element const&)> m_filter;
    Function<bool(Element const&, Element const&)> m_sort;

    Scope m_scope { Scope::Descendants };
};

}
