#ifndef JSDOMPromise_h
#define JSDOMPromise_h
#if ENABLE(PROMISES)
#include "JSDOMBinding.h"
#include <heap/StrongInlines.h>
#include <runtime/JSPromiseDeferred.h>
#include <wtf/Optional.h>
namespace WebCore {
class DeferredWrapper {
public:
DeferredWrapper(JSC::ExecState*, JSDOMGlobalObject*, JSC::JSPromiseDeferred*);
template<class ResolveResultType>
void resolve(const ResolveResultType&);
template<class RejectResultType>
void reject(const RejectResultType&);
JSDOMGlobalObject& globalObject() const;
private:
void callFunction(JSC::ExecState&, JSC::JSValue function, JSC::JSValue resolution);
void resolve(JSC::ExecState& state, JSC::JSValue resolution) { callFunction(state, m_deferred->resolve(), resolution); }
void reject(JSC::ExecState& state, JSC::JSValue resolution) { callFunction(state, m_deferred->reject(), resolution); }
JSC::Strong<JSDOMGlobalObject> m_globalObject;
JSC::Strong<JSC::JSPromiseDeferred> m_deferred;
};
void rejectPromiseWithExceptionIfAny(JSC::ExecState&, JSDOMGlobalObject&, JSC::JSPromiseDeferred&);
inline JSC::JSValue callPromiseFunction(JSC::ExecState& state, JSC::EncodedJSValue promiseFunction(JSC::ExecState*, JSC::JSPromiseDeferred*))
{
JSDOMGlobalObject& globalObject = *JSC::jsCast<JSDOMGlobalObject*>(state.lexicalGlobalObject());
JSC::JSPromiseDeferred& promiseDeferred = *JSC::JSPromiseDeferred::create(&state, &globalObject);
promiseFunction(&state, &promiseDeferred);
rejectPromiseWithExceptionIfAny(state, globalObject, promiseDeferred);
ASSERT(!state.hadException());
return promiseDeferred.promise();
}
template <typename Value, typename Error>
class DOMPromise {
public:
DOMPromise(DeferredWrapper&& wrapper) : m_wrapper(WTF::move(wrapper)) { }
DOMPromise(DOMPromise&& promise) : m_wrapper(WTF::move(promise.m_wrapper)) { }
DOMPromise(const DOMPromise&)= delete;
DOMPromise& operator=(DOMPromise const&) = delete;
void resolve(const Value& value) { m_wrapper.resolve<Value>(value); }
void reject(const Error& error) { m_wrapper.reject<Error>(error); }
private:
DeferredWrapper m_wrapper;
};
template<typename Value, typename Error>
class DOMPromiseWithCallback {
public:
DOMPromiseWithCallback(DeferredWrapper&& wrapper) : m_wrapper(WTF::move(wrapper)) { }
DOMPromiseWithCallback(std::function<void(const Value&)> resolve, std::function<void(const Error&)> reject)
: m_resolveCallback(WTF::move(resolve))
, m_rejectCallback(WTF::move(reject))
{
ASSERT(m_resolveCallback);
ASSERT(m_rejectCallback);
}
void resolve(const Value&);
void reject(const Error&);
private:
Optional<DOMPromise<Value, Error>> m_wrapper;
std::function<void(const Value&)> m_resolveCallback;
std::function<void(const Error&)> m_rejectCallback;
};
template<typename Value, typename Error>
class DOMPromiseIteratorWithCallback {
public:
DOMPromiseIteratorWithCallback(DeferredWrapper&& wrapper) : m_wrapper(WTF::move(wrapper)) { }
DOMPromiseIteratorWithCallback(std::function<void(const Value&)> resolve, std::function<void()> resolveEnd, std::function<void(const Error&)> reject)
: m_resolveCallback(WTF::move(resolve))
, m_resolveEndCallback(WTF::move(resolveEnd))
, m_rejectCallback(WTF::move(reject))
{
ASSERT(m_resolveCallback);
ASSERT(m_resolveEndCallback);
ASSERT(m_rejectCallback);
}
void resolve(const Value&);
void resolveEnd();
void reject(const Error&);
private:
Optional<DeferredWrapper> m_wrapper;
std::function<void(const Value&)> m_resolveCallback;
std::function<void()> m_resolveEndCallback;
std::function<void(const Error&)> m_rejectCallback;
};
template<class ResolveResultType>
inline void DeferredWrapper::resolve(const ResolveResultType& result)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
resolve(*exec, toJS(exec, m_globalObject.get(), result));
}
template<class RejectResultType>
inline void DeferredWrapper::reject(const RejectResultType& result)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
reject(*exec, toJS(exec, m_globalObject.get(), result));
}
template<>
inline void DeferredWrapper::reject(const std::nullptr_t&)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
reject(*exec, JSC::jsNull());
}
template<>
inline void DeferredWrapper::reject(const JSC::JSValue& value)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
reject(*exec, value);
}
template<>
inline void DeferredWrapper::reject<ExceptionCode>(const ExceptionCode& ec)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
reject(*exec, createDOMException(exec, ec));
}
template<>
inline void DeferredWrapper::resolve<String>(const String& result)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
resolve(*exec, jsString(exec, result));
}
template<>
inline void DeferredWrapper::resolve<bool>(const bool& result)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
resolve(*exec, JSC::jsBoolean(result));
}
template<>
inline void DeferredWrapper::resolve<JSC::JSValue>(const JSC::JSValue& value)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
resolve(*exec, value);
}
template<>
inline void DeferredWrapper::resolve<Vector<unsigned char>>(const Vector<unsigned char>& result)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
RefPtr<ArrayBuffer> buffer = ArrayBuffer::create(result.data(), result.size());
resolve(*exec, toJS(exec, m_globalObject.get(), buffer.get()));
}
template<>
inline void DeferredWrapper::resolve(const std::nullptr_t&)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
resolve(*exec, JSC::jsUndefined());
}
template<>
inline void DeferredWrapper::reject<String>(const String& result)
{
ASSERT(m_deferred);
ASSERT(m_globalObject);
JSC::ExecState* exec = m_globalObject->globalExec();
JSC::JSLockHolder locker(exec);
reject(*exec, jsString(exec, result));
}
template<typename Value, typename Error>
inline void DOMPromiseWithCallback<Value, Error>::resolve(const Value& value)
{
if (m_resolveCallback) {
m_resolveCallback(value);
return;
}
m_wrapper.value().resolve(value);
}
template<typename Value, typename Error>
inline void DOMPromiseWithCallback<Value, Error>::reject(const Error& error)
{
if (m_rejectCallback) {
m_rejectCallback(error);
return;
}
m_wrapper.value().reject(error);
}
template<typename Value, typename Error>
inline void DOMPromiseIteratorWithCallback<Value, Error>::resolve(const Value& value)
{
if (m_resolveCallback) {
m_resolveCallback(value);
return;
}
JSDOMGlobalObject& globalObject = m_wrapper.value().globalObject();
m_wrapper.value().resolve(toJSIterator(*globalObject.globalExec(), globalObject, value));
}
template<typename Value, typename Error>
inline void DOMPromiseIteratorWithCallback<Value, Error>::resolveEnd()
{
if (m_resolveEndCallback) {
m_resolveEndCallback();
return;
}
m_wrapper.value().resolve(toJSIteratorEnd(*m_wrapper.value().globalObject().globalExec()));
}
template<typename Value, typename Error>
inline void DOMPromiseIteratorWithCallback<Value, Error>::reject(const Error& error)
{
if (m_rejectCallback) {
m_rejectCallback(error);
return;
}
m_wrapper.value().reject(error);
}
}
#endif // ENABLE(PROMISES)
#endif // JSDOMPromise_h