avocado-old/packages/server/socket.js

const SocketServer = require('socket.io');

import {compose} from '@avocado/core';
import {EventEmitter} from '@avocado/mixins';
import {SocketIoParser, allPackets, idFromPacket} from '@avocado/packet';

const decorateServer = compose(
  EventEmitter,
);

export class Server extends decorateServer(class {}) {

  constructor(httpServer) {
    super();
    this.httpServer = httpServer;
    this.onConnect = this.onConnect.bind(this);
  }

  close(fn) {
    this.io.off('connect', this.onConnect);
    this.io.close(fn);
  }

  onConnect(socket) {
    this.emit('connect', new ServerSocket(socket));
  }

  open() {
    this.io = new SocketServer(this.httpServer, {
      parser: SocketIoParser,
      path: '/avocado',
      perMessageDeflate: false,
      serveClient: false,
    });
    this.io.on('connect', this.onConnect);
  }

}

const decorateSocket = compose(
  EventEmitter,
);

class ServerSocket extends decorateSocket(class {}) {

  constructor(socket) {
    super();
    this.socket = socket;
    for (const Packet of allPackets()) {
      const id = idFromPacket(Packet);
      this.socket.on(id, (packet) => {
        this.emit('packet', packet);
      });
    }
    this.socket.on('disconnect', (...args) => {
      this.emit('disconnect', ...args);
    });
  }

  disconnect() {
    this.socket.disconnect(true);
  }

  send(packet) {
    const id = idFromPacket(packet.constructor);
    this.socket.emit(id, packet.data);
  }

}