How to add extra logic to Reatom context

ctx.ts

const ctxBase = createCtx();
const listeners = new Set();
const { subscribe } = ctx;
export const ctx = Object.assign({}, ctxBase, {
  subscribe(...a) {
    const un = subscribe.apply(this, a);

    // logs
    if (a.length === 1) return un;

    listeners.add(un);
    return () => {
      listeners.delete(un);
      un();
    };
  },
});
export const reset = () => {
  listeners.forEach((un) => un());
};