Don't throw on unknown messages about webgpu-samples HOT 4 CLOSED

postMesasge({a: {b: 123}});

onmessage = (e) => {
  console.log(e.data.a.b);
}

const GlobalCtor = globalThis.constructor;

const GlobalPropertyDescriptors = Object.getOwnPropertyDescriptors(globalThis);

const isOurMessage = e => e.data.sig === 'webgpu_inspector';
const listenerToWrapperFn = new WeakMap();

function makeWrapper(listener) {
  const wrapper = function(event) {
    if (isOurMessage(event)) {
      // do something with our message
      console.log('got message from us', event.data);
    } else {
      return listener.call(this, event);
    }
  };
  listenerToWrapperFn.set(listener, wrapper);
  return wrapper;
}

Object.defineProperty(globalThis, 'onmessage', {
  get() {
    const listener = GlobalPropertyDescriptors.onmessage.get.call(this);
    return listenerToWrapperFn.get(listener) ?? listener;
  },
  set(listener) {
    GlobalPropertyDescriptors.onmessage.set.call(this, makeWrapper(listener));
  }
});

GlobalCtor.prototype.addEventListener = (function(origFn) {
  return function(name, listener, ...args) {
    listener = name === 'message' ? makeWrapper(listener) : listener;
    return origFn.call(this, name, listener, ...args);
  }

})(GlobalCtor.prototype.addEventListener);

GlobalCtor.prototype.removeEventListener = (function(origFn) {
  return function(name, listener, ...args) {
    listener = listenerToWrapperFn.get(listener) ?? listener;
    return origFn.call(this, name, listener, ...args);
  };
})(GlobalCtor.prototype.removeEventListener);