Unhardcode security manager methods

TurboWarp · Feb 25, 2023 · 773e1c1 · 773e1c1
1 parent 9e69ecd
commit 773e1c1
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/src/scaffolding/scaffolding.js b/src/scaffolding/scaffolding.js
@@ -518,19 +518,14 @@ class Scaffolding extends EventTarget {
     return variable;
   }
 
-  setExtensionSecurityManager ({getSandboxMode, canLoadExtensionFromProject}) {
+  setExtensionSecurityManager (newManager) {
     const securityManager = this.vm.extensionManager.securityManager;
     if (!securityManager) {
       console.warn('setExtensionSecurityManager not supported: there is no security manager');
       return;
     }
-
-    if (typeof getSandboxMode !== 'undefined') {
-      securityManager.getSandboxMode = wrapAsFunctionIfNotFunction(getSandboxMode);
-    }
-
-    if (typeof canLoadExtensionFromProject !== 'undefined') {
-      securityManager.canLoadExtensionFromProject = wrapAsFunctionIfNotFunction(canLoadExtensionFromProject);
+    for (const [methodName, fn] of Object.entries(newManager)) {
+      securityManager[methodName] = wrapAsFunctionIfNotFunction(fn);
     }
   }