1 files changed, 96 insertions, 0 deletions
diff --git a/deps/v8/src/d8/async-hooks-wrapper.h b/deps/v8/src/d8/async-hooks-wrapper.h
new file mode 100644
index 0000000000..f339b6e316
--- /dev/null
+++ b/deps/v8/src/d8/async-hooks-wrapper.h
@@ -0,0 +1,96 @@
+// 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_D8_ASYNC_HOOKS_WRAPPER_H_
+#define V8_D8_ASYNC_HOOKS_WRAPPER_H_
+
+#include <stack>
+
+#include "include/v8.h"
+#include "src/objects/objects.h"
+
+namespace v8 {
+
+using async_id_t = double;
+
+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_D8_ASYNC_HOOKS_WRAPPER_H_