@starbase/realtime/client.js

Starbase Realtime Communications

'use strict';

function Realtime(sock, auth=null) {

  let onError = null;
  let onConnected = null;
  let onDisconnected = null;
  let onWhoami = null;
  let onWhois = null;
  let onUsers = null;
  let onJoin = null;
  let onLeave = null;
  let onMessage = null;
  let onRooms = null;
  let onRoom = null;
  let onChat = null;
  let onAuth = null;
  let onAll = null;

  let callbacks = {};
  let ids = 0;

  sock.onState(s=>{
    if (s === 'connected') {
      if (onConnected && typeof onConnected === 'function') {
        onConnected({"type":"connected"});
      }
      if (onAll && typeof onAll === 'function') {
        onAll({"type":"connected"});
      }
    }

    if (s === 'disconnected') {
      if (onDisconnected && typeof onDisconnected === 'function') {
        onDisconnected({"type":"disconnected"});
      }
      if (onAll && typeof onAll === 'function') {
        onAll({"type":"disconnected"});
      }
    }

  });

  sock.onError(err=>{
    if (onError && typeof onError === 'function') {
      onError({"error":err});
    }

    if (onAll && typeof onAll === 'function') {
      onAll({"error":err});
    }

  });

  sock.onMessage(async m=>{
    if (m.msgID && callbacks[m.msgID]) {
      if (m.error) {
        callbacks[m.msgID].reject(m);
      } else {
        callbacks[m.msgID].resolve(m);
      }
    }

    if (m.type === 'whoami' && onWhoami && typeof onWhoami === 'function') {
      onWhoami(m);
    }

    if (m.type === 'whois' && onWhois && typeof onWhois === 'function') {
      onWhois(m);
    }

    if (m.type === 'users' && onUsers && typeof onUsers === 'function') {
      onUsers(m);
    }

    if (m.type === 'rooms' && onRooms && typeof onRooms === 'function') {
      onRooms(m);
    }

    if (m.type === 'room' && onRoom && typeof onRoom === 'function') {
      onRoom(m);
    }

    if (m.type === 'message' && onMessage && typeof onMessage === 'function') {
      onMessage(m);
    }

    if (m.type === 'chat' && onChat && typeof onChat === 'function') {
      onChat(m);
    }

    if (m.type === 'join' && onJoin && typeof onJoin === 'function') {
      onJoin(m);
    }
    if (m.type === 'leave' && onLeave && typeof onLeave === 'function') {
      onLeave(m);
    }

    if (m.type === 'auth' && onAuth && typeof onAuth === 'function') {
      onAuth(m);
    }

    if (onAll && typeof onAll === 'function') {
      onAll(m);
    }

  });

  return {
    "onWhoami":(cb) => {
      onWhoami = cb;
    },
    "onWhois":(cb) => {
      onWhois= cb;
    },
    "onUsers":(cb) => {
      onUsers = cb;
    },
    "onJoin":(cb) => {
      onJoin = cb;
    },
    "onLeave":(cb) => {
      onLeave = cb;
    },
    "onRoom":(cb) => {
      onRoom = cb;
    },
    "onRooms":(cb) => {
      onRooms = cb;
    },
    "onChat":(cb) => {
      onChat = cb;
    },
    "onMessage":(cb) => {
      onMessage = cb;
    },
    "onAuth":(cb) => {
      onAuth = cb;
    },
    "onAll":(cb) => {
      onAll = cb;
    },
    "onConnected":(cb) => {
      onConnected = cb;
    },
    "onDisconnected":(cb) => {
      onDisconnected = cb;
    },
    "onError":(cb) => {
      onError = cb;
    },
    "connect":() => {
      sock.connect();
    },
    "close":() => {
      sock.close();
    },
    "whoami":()=> {
      let msgID = ids++;
      sock.send({"msgID":msgID, "type":"whoami"});
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "whois":(username=null, id=null)=> {
      let msgID = ids++;
      if (id) {
        id = id.toString();
      }
      sock.send({"msgID":msgID, "type":"whois", "username":username, "id":id});
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "join":(room) => {
      let msgID = ids++;
      sock.send({"msgID":msgID, "type":"join","room":room});
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "leave":(room) => {
      let msgID = ids++;
      sock.send({"msgID":msgID, "type":"leave","room":room});
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "chat":(room, message) => {
      let msgID = ids++;
      sock.send({"msgID":msgID, "type":"chat","room":room.toString(), "chat":message});
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "message":(to, message) => {
      let msgID = ids++;
      sock.send({"msgID":msgID, "type":"message","to":to.toString(), "message":message});
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "users":() => {
      let msgID = ids++;
      sock.send({"msgID":msgID, "type":"users"});
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "rooms":() => {
      let msgID = ids++;
      sock.send({"msgID":msgID, "type":"rooms"});
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "room":(room) => {
      let msgID = ids++;
      sock.send({"msgID":msgID, "type":"room", "room":room});
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "auth":async () => {
      let msgID = ids++;
      if (auth) {
        sock.send({"msgID":msgID, "type":"auth", "token":await auth.getToken()});
      }
      return new Promise((resolve,reject)=>{
        callbacks[msgID] = {"resolve":resolve, "reject":reject};
      });
    },
    "sock":sock
  }

}