// Copyright 2018 the V8 project authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #ifndef V8_ASYNC_HOOKS_WRAPPER_H_ #define V8_ASYNC_HOOKS_WRAPPER_H_ #include <stack> #include "include/v8.h" #include "src/objects.h" namespace v8 { typedef double async_id_t; struct AsyncContext { async_id_t execution_async_id; async_id_t trigger_async_id; }; class AsyncHooksWrap { public: explicit AsyncHooksWrap(Isolate* isolate) { enabled_ = false; isolate_ = isolate; } void Enable(); void Disable(); bool IsEnabled() const { return enabled_; } inline v8::Local<v8::Function> init_function() const; inline void set_init_function(v8::Local<v8::Function> value); inline v8::Local<v8::Function> before_function() const; inline void set_before_function(v8::Local<v8::Function> value); inline v8::Local<v8::Function> after_function() const; inline void set_after_function(v8::Local<v8::Function> value); inline v8::Local<v8::Function> promiseResolve_function() const; inline void set_promiseResolve_function(v8::Local<v8::Function> value); private: Isolate* isolate_; Persistent<v8::Function> init_function_; Persistent<v8::Function> before_function_; Persistent<v8::Function> after_function_; Persistent<v8::Function> promiseResolve_function_; bool enabled_; }; class AsyncHooks { public: explicit AsyncHooks(Isolate* isolate) { isolate_ = isolate; AsyncContext ctx; ctx.execution_async_id = 1; ctx.trigger_async_id = 0; asyncContexts.push(ctx); current_async_id = 1; Initialize(); } ~AsyncHooks() { Deinitialize(); } async_id_t GetExecutionAsyncId() const; async_id_t GetTriggerAsyncId() const; Local<Object> CreateHook(const v8::FunctionCallbackInfo<v8::Value>& args); Persistent<FunctionTemplate> async_hook_ctor; private: std::vector<AsyncHooksWrap*> async_wraps_; Isolate* isolate_; Persistent<ObjectTemplate> async_hooks_templ; Persistent<Private> async_id_smb; Persistent<Private> trigger_id_smb; void Initialize(); void Deinitialize(); static void ShellPromiseHook(PromiseHookType type, Local<Promise> promise, Local<Value> parent); static void PromiseHookDispatch(PromiseHookType type, Local<Promise> promise, Local<Value> parent, AsyncHooksWrap* wrap, AsyncHooks* hooks); std::stack<AsyncContext> asyncContexts; async_id_t current_async_id; }; } // namespace v8 #endif // V8_ASYNC_HOOKS_WRAPPER_H_