/*
 * Copyright (c) 2024, Tim Flynn <trflynn89@ladybird.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <LibCore/Forward.h>
#include <LibGC/Function.h>
#include <LibGC/Ptr.h>
#include <LibJS/Heap/Cell.h>
#include <LibWeb/Export.h>

namespace Web::WebDriver {

class WEB_API HeapTimer : public JS::Cell {
    GC_CELL(HeapTimer, JS::Cell);
    GC_DECLARE_ALLOCATOR(HeapTimer);

public:
    explicit HeapTimer();
    virtual ~HeapTimer() override;

    void start(u64 timeout_ms, GC::Ref<GC::Function<void()>> on_timeout);
    void stop_and_fire_timeout_handler();
    void stop();

    bool is_timed_out() const { return m_timed_out; }

private:
    virtual void visit_edges(JS::Cell::Visitor& visitor) override;

    NonnullRefPtr<Core::Timer> m_timer;
    GC::Ptr<GC::Function<void()>> m_on_timeout;
    bool m_timed_out { false };
};

}
