#pragma once
#include "JSDOMBinding.h"
#include "JSDOMGlobalObject.h"
#include "ScriptExecutionContext.h"
#include <JavaScriptCore/JSObject.h>
#include <JavaScriptCore/Strong.h>
#include <JavaScriptCore/StrongInlines.h>
#include <wtf/Threading.h>
namespace WebCore {
class JSCallbackData {
WTF_MAKE_FAST_ALLOCATED;
public:
enum class CallbackType { Function, Object, FunctionOrObject };
JSDOMGlobalObject* globalObject() { return m_globalObject.get(); }
protected:
explicit JSCallbackData(JSDOMGlobalObject* globalObject)
: m_globalObject(globalObject)
{
}
~JSCallbackData()
{
#if !PLATFORM(IOS_FAMILY)
ASSERT(m_thread.ptr() == &Thread::current());
#endif
}
WEBCORE_EXPORT static JSC::JSValue invokeCallback(JSDOMGlobalObject&, JSC::JSObject* callback, JSC::JSValue thisValue, JSC::MarkedArgumentBuffer&, CallbackType, JSC::PropertyName functionName, NakedPtr<JSC::Exception>& returnedException);
private:
JSC::Weak<JSDOMGlobalObject> m_globalObject;
#if ASSERT_ENABLED
Ref<Thread> m_thread { Thread::current() };
#endif
};
class JSCallbackDataStrong : public JSCallbackData {
public:
JSCallbackDataStrong(JSC::JSObject* callback, JSDOMGlobalObject* globalObject, void* = nullptr)
: JSCallbackData(globalObject)
, m_callback(globalObject->vm(), callback)
{
}
JSC::JSObject* callback() { return m_callback.get(); }
JSC::JSValue invokeCallback(JSC::JSValue thisValue, JSC::MarkedArgumentBuffer& args, CallbackType callbackType, JSC::PropertyName functionName, NakedPtr<JSC::Exception>& returnedException)
{
auto* globalObject = this->globalObject();
if (!globalObject)
return { };
return JSCallbackData::invokeCallback(*globalObject, callback(), thisValue, args, callbackType, functionName, returnedException);
}
private:
JSC::Strong<JSC::JSObject> m_callback;
};
class JSCallbackDataWeak : public JSCallbackData {
public:
JSCallbackDataWeak(JSC::JSObject* callback, JSDOMGlobalObject* globalObject, void* owner)
: JSCallbackData(globalObject)
, m_callback(callback, &m_weakOwner, owner)
{
}
JSC::JSObject* callback() { return m_callback.get(); }
JSC::JSValue invokeCallback(JSC::JSValue thisValue, JSC::MarkedArgumentBuffer& args, CallbackType callbackType, JSC::PropertyName functionName, NakedPtr<JSC::Exception>& returnedException)
{
auto* globalObject = this->globalObject();
if (!globalObject)
return { };
return JSCallbackData::invokeCallback(*globalObject, callback(), thisValue, args, callbackType, functionName, returnedException);
}
void visitJSFunction(JSC::SlotVisitor&);
private:
class WeakOwner : public JSC::WeakHandleOwner {
bool isReachableFromOpaqueRoots(JSC::Handle<JSC::Unknown>, void* context, JSC::SlotVisitor&, const char**) override;
};
WeakOwner m_weakOwner;
JSC::Weak<JSC::JSObject> m_callback;
};
class DeleteCallbackDataTask : public ScriptExecutionContext::Task {
public:
template <typename CallbackDataType>
explicit DeleteCallbackDataTask(CallbackDataType* data)
: ScriptExecutionContext::Task(ScriptExecutionContext::Task::CleanupTask, [data = std::unique_ptr<CallbackDataType>(data)] (ScriptExecutionContext&) {
})
{
}
};
}