refactor(api/internal/monkeypatch.js): improve installed native perse…

…rvation
socketsupply · Oct 11, 2023 · 75252c4 · 75252c4
1 parent 2609596
commit 75252c4
Showing 1 changed file with 19 additions and 10 deletions.
diff --git a/api/internal/monkeypatch.js b/api/internal/monkeypatch.js
@@ -21,34 +21,43 @@ export function init () {
     }
 
     const actualName = name.split('.').slice(-1)[0]
-    let nativeImplementation = null
 
     if (typeof prefix === 'string') {
       name = `${prefix}.${name}`
     }
 
-    nativeImplementation = target[actualName] || null
     if (typeof target[actualName] === 'object' && target[actualName] !== null) {
       for (const key in implementation) {
+        const nativeImplementation = target[actualName][key] || null
         // let this fail, the environment implementation may not be writable
         try {
           target[actualName][key] = implementation[key]
         } catch {}
+
+        if (nativeImplementation !== null) {
+          const nativeName = ['_', 'native', ...name.split('.'), key].join('_')
+          Object.defineProperty(globalThis, nativeName, {
+            enumerable: false,
+            configurable: false,
+            value: nativeImplementation
+          })
+        }
       }
     } else {
+      const nativeImplementation = target[actualName] || null
       // let this fail, the environment implementation may not be writable
       try {
         target[actualName] = implementation
       } catch {}
-    }
 
-    if (nativeImplementation !== null) {
-      const nativeName = ['_', 'native', ...name.split('.')].join('_')
-      Object.defineProperty(globalThis, nativeName, {
-        enumerable: false,
-        configurable: false,
-        value: nativeImplementation
-      })
+      if (nativeImplementation !== null) {
+        const nativeName = ['_', 'native', ...name.split('.')].join('_')
+        Object.defineProperty(globalThis, nativeName, {
+          enumerable: false,
+          configurable: false,
+          value: nativeImplementation
+        })
+      }
     }
   }