carmi/src/unwrapable-proxy.js

'use strict'

const {proxyHandler} = require('./proxyHandler');
const proxyToObjMap = new WeakMap();
const unwrappedDeep = new WeakMap();

function wrap(val) {
  if (proxyToObjMap.has(val)) {
    return val;
  }
  const res = new Proxy(val, proxyHandler);
  proxyToObjMap.set(res, val);
  return res;
}

function unwrap(proxy) {
  let res = proxy;
  while (proxyToObjMap.has(proxy)) {
    res = proxyToObjMap.get(proxy);
    proxy = res;
  }
  if (Array.isArray(res)) {
    if (!unwrappedDeep.has(res)) {
      unwrappedDeep.set(res, true);
      res.forEach((val, key) => {
        res[key] = unwrap(val);
      });
    }
  } else if (typeof res === 'object' && res !== null) {
    if (!unwrappedDeep.has(res)) {
      unwrappedDeep.set(res, true);
      Object.getOwnPropertyNames(res).forEach(key => {
        res[key] = unwrap(res[key]);
      });
    }
  }
  return res;
}

module.exports = {
  unwrap,
  wrap
};