/*
 * Copyright (c) 2021-2022, Idan Horowitz <idan.horowitz@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/SinglyLinkedList.h>
#include <LibJS/Heap/GCPtr.h>
#include <LibJS/Runtime/FunctionObject.h>
#include <LibJS/Runtime/GlobalObject.h>
#include <LibJS/Runtime/JobCallback.h>
#include <LibJS/Runtime/Object.h>
#include <LibJS/Runtime/Value.h>
#include <LibJS/Runtime/WeakContainer.h>

namespace JS {

class FinalizationRegistry final
    : public Object
    , public WeakContainer {
    JS_OBJECT(FinalizationRegistry, Object);
    JS_DECLARE_ALLOCATOR(FinalizationRegistry);

public:
    virtual ~FinalizationRegistry() override = default;

    void add_finalization_record(Cell& target, Value held_value, Cell* unregister_token);
    bool remove_by_token(Cell& unregister_token);
    ThrowCompletionOr<void> cleanup(GCPtr<JobCallback> = {});

    virtual void remove_dead_cells(Badge<Heap>) override;

    Realm& realm() { return *m_realm; }
    Realm const& realm() const { return *m_realm; }

    JobCallback& cleanup_callback() { return *m_cleanup_callback; }
    JobCallback const& cleanup_callback() const { return *m_cleanup_callback; }

private:
    FinalizationRegistry(Realm&, NonnullGCPtr<JobCallback>, Object& prototype);

    virtual void visit_edges(Visitor& visitor) override;

    NonnullGCPtr<Realm> m_realm;
    NonnullGCPtr<JobCallback> m_cleanup_callback;

    struct FinalizationRecord {
        GCPtr<Cell> target;
        Value held_value;
        GCPtr<Cell> unregister_token;
    };
    SinglyLinkedList<FinalizationRecord> m_records;
};

}