Implementation of v8::Context::SetPromiseHooks().

Author: iamstolis

graal-nodejs/deps/v8/src/graal/graal_context.cc

@@ -40,6 +40,7 @@
  */
 
 #include "graal_context.h"
+#include "graal_function.h"
 #include "graal_object.h"
 
 #include "graal_context-inl.h"
@@ -114,3 +115,11 @@ v8::Local<v8::Object> GraalContext::GetExtrasBindingObject() {
     GraalObject* graal_extras = GraalObject::Allocate(Isolate(), java_extras);
     return reinterpret_cast<v8::Object*> (graal_extras);
 }
+
+void GraalContext::SetPromiseHooks(v8::Local<v8::Function> init_hook, v8::Local<v8::Function> before_hook, v8::Local<v8::Function> after_hook, v8::Local<v8::Function> resolve_hook) {
+    jobject java_init_hook = init_hook.IsEmpty() ? nullptr : reinterpret_cast<GraalFunction*> (*init_hook)->GetJavaObject();
+    jobject java_before_hook = before_hook.IsEmpty() ? nullptr : reinterpret_cast<GraalFunction*> (*before_hook)->GetJavaObject();
+    jobject java_after_hook = after_hook.IsEmpty() ? nullptr : reinterpret_cast<GraalFunction*> (*after_hook)->GetJavaObject();
+    jobject java_resolve_hook = resolve_hook.IsEmpty() ? nullptr : reinterpret_cast<GraalFunction*> (*resolve_hook)->GetJavaObject();
+    JNI_CALL_VOID(Isolate(), GraalAccessMethod::context_set_promise_hooks, GetJavaObject(), java_init_hook, java_before_hook, java_after_hook, java_resolve_hook);
+}

graal-nodejs/deps/v8/src/graal/graal_context.h

@@ -60,6 +60,7 @@ class GraalContext : public GraalHandleContent {
     void UseDefaultSecurityToken();
     v8::Local<v8::Value> GetSecurityToken();
     v8::Local<v8::Object> GetExtrasBindingObject();
+    void SetPromiseHooks(v8::Local<v8::Function> init_hook, v8::Local<v8::Function> before_hook, v8::Local<v8::Function> after_hook, v8::Local<v8::Function> resolve_hook);
 
     inline v8::Isolate* GetIsolate() {
         return reinterpret_cast<v8::Isolate*> (Isolate());

graal-nodejs/deps/v8/src/graal/graal_isolate.cc

@@ -803,6 +803,7 @@ GraalIsolate::GraalIsolate(JavaVM* jvm, JNIEnv* env, v8::Isolate::CreateParams c
     ACCESS_METHOD(GraalAccessMethod::context_set_security_token, "contextSetSecurityToken", "(Ljava/lang/Object;Ljava/lang/Object;)V")
     ACCESS_METHOD(GraalAccessMethod::context_get_security_token, "contextGetSecurityToken", "(Ljava/lang/Object;)Ljava/lang/Object;")
     ACCESS_METHOD(GraalAccessMethod::context_get_extras_binding_object, "contextGetExtrasBindingObject", "(Ljava/lang/Object;)Ljava/lang/Object;")
+    ACCESS_METHOD(GraalAccessMethod::context_set_promise_hooks, "contextSetPromiseHooks", "(Ljava/lang/Object;Ljava/lang/Object;Ljava/lang/Object;Ljava/lang/Object;Ljava/lang/Object;)V")
     ACCESS_METHOD(GraalAccessMethod::find_dynamic_object_fields, "findDynamicObjectFields", "(Ljava/lang/Object;)[Ljava/lang/Object;")
     ACCESS_METHOD(GraalAccessMethod::proxy_is_function, "proxyIsFunction", "(Ljava/lang/Object;)Z")
     ACCESS_METHOD(GraalAccessMethod::proxy_get_handler, "proxyGetHandler", "(Ljava/lang/Object;)Ljava/lang/Object;")

graal-nodejs/deps/v8/src/graal/graal_isolate.h

@@ -264,6 +264,7 @@ enum GraalAccessMethod {
     context_set_embedder_data,
     context_get_embedder_data,
     context_get_extras_binding_object,
+    context_set_promise_hooks,
     try_catch_exception,
     try_catch_has_terminated,
     message_get_script_resource_name,

graal-nodejs/deps/v8/src/graal/v8.cc

@@ -3639,6 +3639,10 @@ namespace v8 {
         TRACE
     }
 
+    void Context::SetPromiseHooks(Local<Function> init_hook, Local<Function> before_hook, Local<Function> after_hook, Local<Function> resolve_hook) {
+        reinterpret_cast<GraalContext*> (this)->SetPromiseHooks(init_hook, before_hook, after_hook, resolve_hook);
+    }
+
     void Object::CheckCast(v8::Value* obj) {}
     void Promise::CheckCast(v8::Value* obj) {}
     void Function::CheckCast(v8::Value* obj) {}

graal-nodejs/mx.graal-nodejs/com.oracle.truffle.trufflenode/src/com/oracle/truffle/trufflenode/GraalJSAccess.java

@@ -2792,6 +2792,41 @@ public Object contextGetEmbedderData(Object realm, int index) {
         return data.getEmbedderData(index);
     }
 
+    public void contextSetPromiseHooks(Object context, Object initHook, Object beforeHook, Object afterHook, Object resolveHook) {
+        PromiseHook hook = new PromiseHook() {
+            @Override
+            public void promiseChanged(int changeType, DynamicObject promise, DynamicObject parentPromise) {
+                Object hook;
+                switch (changeType) {
+                    case PromiseHook.TYPE_INIT:
+                        hook = initHook;
+                        break;
+                    case PromiseHook.TYPE_RESOLVE:
+                        hook = resolveHook;
+                        break;
+                    case PromiseHook.TYPE_BEFORE:
+                        hook = beforeHook;
+                        break;
+                    case PromiseHook.TYPE_AFTER:
+                        hook = afterHook;
+                        break;
+                    default:
+                        throw Errors.shouldNotReachHere();
+                }
+                if (hook != null) {
+                    Object[] arguments;
+                    if (changeType == PromiseHook.TYPE_INIT) {
+                        arguments = new Object[]{promise, parentPromise};
+                    } else {
+                        arguments = new Object[]{promise};
+                    }
+                    JSRuntime.call(hook, Undefined.instance, arguments);
+                }
+            }
+        };
+        ((JSRealm) context).getContext().setPromiseHook(hook);
+    }
+
     public void isolateRunMicrotasks() {
         pollWeakCallbackQueue(false);
         try {

graal-nodejs/mx.graal-nodejs/com.oracle.truffle.trufflenode/src/svmnodejs.jniconfig

@@ -39,6 +39,7 @@
       { "name": "contextNew" },
       { "name": "contextSetEmbedderData" },
       { "name": "contextSetPointerInEmbedderData" },
+      { "name": "contextSetPromiseHooks" },
       { "name": "contextSetSecurityToken" },
       { "name": "create" },
       { "name": "dataViewNew" },