#pragma once
#include <wtf/HashSet.h>
#include <wtf/Lock.h>
#include <wtf/RefPtr.h>
#include <wtf/RetainPtr.h>
#include <wtf/RunLoop.h>
#include <wtf/SharedTask.h>
#include <wtf/ThreadSafeRefCounted.h>
#if USE(CF)
#include <CoreFoundation/CoreFoundation.h>
#endif
namespace JSC {
class JSLock;
class VM;
class JSRunLoopTimer : public ThreadSafeRefCounted<JSRunLoopTimer> {
public:
typedef void TimerNotificationType();
using TimerNotificationCallback = RefPtr<WTF::SharedTask<TimerNotificationType>>;
class Manager {
#if USE(CF)
static void timerDidFireCallback(CFRunLoopTimerRef, void*);
#else
void timerDidFireCallback();
#endif
void timerDidFire();
public:
using EpochTime = Seconds;
static Manager& shared();
void registerVM(VM&);
void unregisterVM(VM&);
void scheduleTimer(JSRunLoopTimer&, Seconds nextFireTime);
void cancelTimer(JSRunLoopTimer&);
Optional<Seconds> timeUntilFire(JSRunLoopTimer&);
#if USE(CF)
void didChangeRunLoop(VM&, CFRunLoopRef newRunLoop);
#endif
private:
Lock m_lock;
struct PerVMData {
PerVMData() = default;
#if USE(CF)
PerVMData(Manager&) { }
#else
PerVMData(Manager&);
#endif
PerVMData(PerVMData&&) = default;
PerVMData& operator=(PerVMData&&) = default;
~PerVMData();
#if USE(CF)
void setRunLoop(Manager*, CFRunLoopRef);
RetainPtr<CFRunLoopTimerRef> timer;
RetainPtr<CFRunLoopRef> runLoop;
CFRunLoopTimerContext context;
#else
RunLoop* runLoop;
std::unique_ptr<RunLoop::Timer<Manager>> timer;
#endif
Vector<std::pair<Ref<JSRunLoopTimer>, EpochTime>> timers;
};
HashMap<Ref<JSLock>, PerVMData> m_mapping;
};
JSRunLoopTimer(VM*);
JS_EXPORT_PRIVATE virtual ~JSRunLoopTimer();
virtual void doWork(VM&) = 0;
void setTimeUntilFire(Seconds intervalInSeconds);
void cancelTimer();
bool isScheduled() const { return m_isScheduled; }
JS_EXPORT_PRIVATE void addTimerSetNotification(TimerNotificationCallback);
JS_EXPORT_PRIVATE void removeTimerSetNotification(TimerNotificationCallback);
JS_EXPORT_PRIVATE Optional<Seconds> timeUntilFire();
protected:
static const Seconds s_decade;
Ref<JSLock> m_apiLock;
private:
friend class Manager;
void timerDidFire();
HashSet<TimerNotificationCallback> m_timerSetCallbacks;
Lock m_timerCallbacksLock;
Lock m_lock;
bool m_isScheduled { false };
};
}