/*
 * Copyright (c) 2024-2025, stelar7 <dudedbz@gmail.com>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <LibGC/HeapVector.h>
#include <LibGC/Ptr.h>
#include <LibGC/RootVector.h>
#include <LibGC/Weak.h>
#include <LibWeb/DOM/EventTarget.h>
#include <LibWeb/IndexedDB/Internal/ObjectStore.h>
#include <LibWeb/StorageAPI/StorageKey.h>

namespace Web::IndexedDB {

// https://www.w3.org/TR/IndexedDB/#database-construct
class Database : public Bindings::PlatformObject {
    WEB_NON_IDL_PLATFORM_OBJECT(Database, Bindings::PlatformObject);
    GC_DECLARE_ALLOCATOR(Database);

public:
    void set_version(u64 version) { m_version = version; }
    u64 version() const { return m_version; }
    String name() const { return m_name; }

    void set_upgrade_transaction(GC::Ptr<IDBTransaction> transaction) { m_upgrade_transaction = transaction; }
    [[nodiscard]] GC::Ptr<IDBTransaction> upgrade_transaction() { return m_upgrade_transaction; }

    void associate(GC::Ref<IDBDatabase> connection);
    void dissociate(IDBDatabase& connection);
    using AssociatedConnections = GC::HeapVector<GC::Ref<IDBDatabase>>;
    GC::Ref<AssociatedConnections> associated_connections_as_heap_vector();
    GC::Ref<AssociatedConnections> associated_connections_as_heap_vector_except(IDBDatabase& connection);
    GC::RootVector<GC::Ref<IDBDatabase>> associated_connections_as_root_vector();

    ReadonlySpan<GC::Ref<ObjectStore>> object_stores() { return m_object_stores; }
    GC::Ptr<ObjectStore> object_store_with_name(String const& name) const;
    void add_object_store(GC::Ref<ObjectStore> object_store) { m_object_stores.append(object_store); }
    void remove_object_store(GC::Ref<ObjectStore> object_store)
    {
        m_object_stores.remove_first_matching([&](auto& entry) { return entry == object_store; });
    }

    [[nodiscard]] static Vector<GC::Weak<Database>> for_key(StorageAPI::StorageKey const&);
    [[nodiscard]] static Optional<Database&> for_key_and_name(StorageAPI::StorageKey const&, String const&);
    [[nodiscard]] static ErrorOr<GC::Ref<Database>> create_for_key_and_name(JS::Realm&, StorageAPI::StorageKey const&, String const&);
    [[nodiscard]] static ErrorOr<void> delete_for_key_and_name(StorageAPI::StorageKey const&, String const&);

    static void for_each_database(AK::Function<void(Database&)> const& visitor);

    [[nodiscard]] static GC::Ref<Database> create(JS::Realm&, String const&);
    virtual ~Database();

    void wait_for_connections_to_close(ReadonlySpan<GC::Ref<IDBDatabase>> connections, GC::Ref<GC::Function<void()>> after_all);
    void check_pending_connection_wait();

protected:
    explicit Database(IDBDatabase& database);

    explicit Database(JS::Realm& realm, String name)
        : PlatformObject(realm)
        , m_name(move(name))
    {
    }

    virtual void visit_edges(Visitor&) override;

private:
    struct PendingConnectionWait {
        Vector<GC::Weak<IDBDatabase>> connections;
        GC::Ref<GC::Function<void()>> callback;
    };

    Vector<GC::Weak<IDBDatabase>> m_associated_connections;
    Optional<PendingConnectionWait> m_pending_connection_wait;

    // A database has a name which identifies it within a specific storage key.
    String m_name;

    // A database has a version. When a database is first created, its version is 0 (zero).
    u64 m_version { 0 };

    // A database has at most one associated upgrade transaction, which is either null or an upgrade transaction, and is initially null.
    GC::Ptr<IDBTransaction> m_upgrade_transaction;

    // A database has zero or more object stores which hold the data stored in the database.
    Vector<GC::Ref<ObjectStore>> m_object_stores;
};

}