kea-socket.io/src/utils.js

Socket.io binding for Kea store

import { getPluginContext } from 'kea';

export const isFunction = (obj) => typeof obj === 'function';

export const isSocketIo = (obj) =>
  !!obj &&
  ((isFunction(obj.on) && isFunction(obj.emit)) || (Array.isArray(obj) && obj.every((item) => isSocketIo(item))));

export const trimNamespace = (namespaced) => namespaced.split('/').pop();

export const getEmitters = (name) => {
  const { emitters = {} } = getPluginContext('kea-socket.io');
  if (name) {
    return emitters[name];
  }
  return emitters;
};

export const getCurrentName = (name, prefix = '') => {
  if (name.indexOf(prefix) === 0) {
    return name.replace(prefix, '');
  }
  return name;
};

export const wildcardMiddleware = (Emitter) => {
  const emit = Emitter.prototype.emit;

  function onevent(packet) {
    const args = packet.data || [];
    if (packet.id != null) {
      args.push(this.ack(packet.id));
    }
    emit.call(this, '*', packet);
    return emit.apply(this, args);
  }

  return (socket, next) => {
    if (socket.onevent !== onevent) {
      socket.onevent = onevent;
    }
    return next ? next() : null;
  };
};