From 1ac2156ae46f9b94ac39cd8b241d0f1ac16f5cce Mon Sep 17 00:00:00 2001
From: Christopher Hiller <boneskull@boneskull.com>
Date: Thu, 24 Oct 2024 12:22:18 -0700
Subject: [PATCH] feat(compartment-mapper): add Compartment option to
 captureFromMap

This just allows `captureFromMap()` to receive a `Compartment` option which is passed thru to `link()`.
---
 packages/compartment-mapper/src/capture-lite.js | 4 ++++
 packages/compartment-mapper/src/types.js        | 1 +
 2 files changed, 5 insertions(+)

diff --git a/packages/compartment-mapper/src/capture-lite.js b/packages/compartment-mapper/src/capture-lite.js
index 6d489dfccb..6ef57bca0b 100644
--- a/packages/compartment-mapper/src/capture-lite.js
+++ b/packages/compartment-mapper/src/capture-lite.js
@@ -56,6 +56,8 @@ import { unpackReadPowers } from './powers.js';
 
 const { freeze, assign, create, fromEntries, entries, keys } = Object;
 
+const defaultCompartment = Compartment;
+
 /**
  * We attempt to produce compartment maps that are consistent regardless of
  * whether the packages were originally laid out on disk for development or
@@ -266,6 +268,7 @@ export const captureFromMap = async (powers, compartmentMap, options = {}) => {
     sourceMapHook = undefined,
     parserForLanguage: parserForLanguageOption = {},
     languageForExtension: languageForExtensionOption = {},
+    Compartment = defaultCompartment,
   } = options;
 
   const parserForLanguage = freeze(
@@ -309,6 +312,7 @@ export const captureFromMap = async (powers, compartmentMap, options = {}) => {
     parserForLanguage,
     languageForExtension,
     archiveOnly: true,
+    Compartment,
   });
   await compartment.load(entryModuleSpecifier);
   if (policy) {
diff --git a/packages/compartment-mapper/src/types.js b/packages/compartment-mapper/src/types.js
index f022c9f032..bdd3e4fc78 100644
--- a/packages/compartment-mapper/src/types.js
+++ b/packages/compartment-mapper/src/types.js
@@ -883,6 +883,7 @@ export {};
  * @property {Record<string, ParserImplementation>} [parserForLanguage]
  * @property {LanguageForExtension} [languageForExtension]
  * @property {ExitModuleImportNowHook} [importNowHook]
+ * @property {typeof Compartment} [Compartment]
  */
 
 /**