Expose ScriptEnvironmentPreparer to Rust

Signed-off-by: Delan Azabani <[email protected]>

Author: delan

mozjs-sys/src/jsglue.cpp

@@ -173,6 +173,21 @@ class RustJobQueue : public JS::JobQueue {
   }
 };
 
+struct RustEnvironmentPreparer : public js::ScriptEnvironmentPreparer {
+  JSContext* cx;
+  explicit RustEnvironmentPreparer(JSContext* cx) : cx(cx) {}
+  void invoke(JS::HandleObject global, Closure& closure) override {
+    MOZ_ASSERT(JS_IsGlobalObject(global));
+    MOZ_ASSERT(!JS_IsExceptionPending(cx));
+
+    AutoRealm ar(cx, global);
+    AutoReportException are(cx);
+    if (!closure(cx)) {
+      return;
+    }
+  }
+}
+
 struct JSExternalStringCallbacksTraps {
   void (*latin1Finalize)(const void* privateData, JS::Latin1Char* chars);
   void (*utf16Finalize)(const void* privateData, char16_t* chars);
@@ -1106,6 +1121,10 @@ JS::JobQueue* CreateJobQueue(const JobQueueTraps* aTraps, const void* aQueue,
 
 void DeleteJobQueue(JS::JobQueue* queue) { delete queue; }
 
+void InitScriptEnvironmentPreparer(JSContext* cx) {
+  js::SetScriptEnvironmentPreparer(cx, new RustEnvironmentPreparer(cx));
+}
+
 JSExternalStringCallbacks* CreateJSExternalStringCallbacks(
     const JSExternalStringCallbacksTraps* aTraps, void* privateData) {
   return new RustJSExternalStringCallbacks(*aTraps, privateData);