From a160d2b500a105e26a9e6ac3f32d7a42437a1462 Mon Sep 17 00:00:00 2001
From: Darshan Sen <darshan.sen@postman.com>
Date: Sat, 31 Jul 2021 11:53:45 +0530
Subject: [PATCH] src: return MaybeLocal from a couple of functions

Functions affected:
* InitializeContext()
* InitializeContextForSnapshot()
* InitializePrimordials()

Signed-off-by: Darshan Sen <darshan.sen@postman.com>
---
 src/api/environment.cc | 33 +++++++++++++++++++--------------
 src/node.h             |  3 ++-
 src/node_internals.h   |  6 ++++--
 3 files changed, 25 insertions(+), 17 deletions(-)

diff --git a/src/api/environment.cc b/src/api/environment.cc
index bd9de560d08120..01c8d84ed495ab 100644
--- a/src/api/environment.cc
+++ b/src/api/environment.cc
@@ -501,7 +501,7 @@ MaybeLocal<Object> GetPerContextExports(Local<Context> context) {
 
   Local<Object> exports = Object::New(isolate);
   if (context->Global()->SetPrivate(context, key, exports).IsNothing() ||
-      !InitializePrimordials(context))
+      InitializePrimordials(context).IsEmpty())
     return MaybeLocal<Object>();
   return handle_scope.Escape(exports);
 }
@@ -514,7 +514,7 @@ Local<Context> NewContext(Isolate* isolate,
   auto context = Context::New(isolate, nullptr, object_template);
   if (context.IsEmpty()) return context;
 
-  if (!InitializeContext(context)) {
+  if (InitializeContext(context).IsEmpty()) {
     return Local<Context>();
   }
 
@@ -581,16 +581,21 @@ void InitializeContextRuntime(Local<Context> context) {
   }
 }
 
-bool InitializeContextForSnapshot(Local<Context> context) {
+MaybeLocal<Context> InitializeContextForSnapshot(Local<Context> context) {
   Isolate* isolate = context->GetIsolate();
   HandleScope handle_scope(isolate);
 
   context->SetEmbedderData(ContextEmbedderIndex::kAllowWasmCodeGeneration,
                            True(isolate));
-  return InitializePrimordials(context);
+
+  if (InitializePrimordials(context).IsEmpty()) {
+    return MaybeLocal<Context>();
+  }
+
+  return context;
 }
 
-bool InitializePrimordials(Local<Context> context) {
+MaybeLocal<Object> InitializePrimordials(Local<Context> context) {
   // Run per-context JS files.
   Isolate* isolate = context->GetIsolate();
   Context::Scope context_scope(context);
@@ -606,7 +611,7 @@ bool InitializePrimordials(Local<Context> context) {
   if (!primordials->SetPrototype(context, Null(isolate)).FromJust() ||
       !GetPerContextExports(context).ToLocal(&exports) ||
       !exports->Set(context, primordials_string, primordials).FromJust()) {
-    return false;
+    return MaybeLocal<Object>();
   }
 
   static const char* context_files[] = {"internal/per_context/primordials",
@@ -623,27 +628,27 @@ bool InitializePrimordials(Local<Context> context) {
             context, *module, &parameters, nullptr);
     Local<Function> fn;
     if (!maybe_fn.ToLocal(&fn)) {
-      return false;
+      return MaybeLocal<Object>();
     }
     MaybeLocal<Value> result =
         fn->Call(context, Undefined(isolate), arraysize(arguments), arguments);
     // Execution failed during context creation.
-    // TODO(joyeecheung): deprecate this signature and return a MaybeLocal.
     if (result.IsEmpty()) {
-      return false;
+      return MaybeLocal<Object>();
     }
   }
 
-  return true;
+  return primordials;
 }
 
-bool InitializeContext(Local<Context> context) {
-  if (!InitializeContextForSnapshot(context)) {
-    return false;
+MaybeLocal<Context> InitializeContext(Local<Context> context) {
+  if (InitializeContextForSnapshot(context).IsEmpty()) {
+    return MaybeLocal<Context>();
   }
 
   InitializeContextRuntime(context);
-  return true;
+
+  return context;
 }
 
 uv_loop_t* GetCurrentEventLoop(Isolate* isolate) {
diff --git a/src/node.h b/src/node.h
index 066c0eadfb6b05..75af41839ab1d3 100644
--- a/src/node.h
+++ b/src/node.h
@@ -367,7 +367,8 @@ NODE_EXTERN v8::Local<v8::Context> NewContext(
 
 // Runs Node.js-specific tweaks on an already constructed context
 // Return value indicates success of operation
-NODE_EXTERN bool InitializeContext(v8::Local<v8::Context> context);
+NODE_EXTERN v8::MaybeLocal<v8::Context> InitializeContext(
+    v8::Local<v8::Context> context);
 
 // If `platform` is passed, it will be used to register new Worker instances.
 // It can be `nullptr`, in which case creating new Workers inside of
diff --git a/src/node_internals.h b/src/node_internals.h
index 8f7929994f3243..9c0f34844b469b 100644
--- a/src/node_internals.h
+++ b/src/node_internals.h
@@ -92,8 +92,10 @@ void SignalExit(int signal, siginfo_t* info, void* ucontext);
 std::string GetProcessTitle(const char* default_title);
 std::string GetHumanReadableProcessName();
 
-void InitializeContextRuntime(v8::Local<v8::Context>);
-bool InitializePrimordials(v8::Local<v8::Context> context);
+// TODO(RaisinTen): return a v8::MaybeLocal<v8::Context>.
+void InitializeContextRuntime(v8::Local<v8::Context> context);
+v8::MaybeLocal<v8::Object> InitializePrimordials(
+    v8::Local<v8::Context> context);
 
 class NodeArrayBufferAllocator : public ArrayBufferAllocator {
  public: