Class Socket

keyserver/src/socket/socket.js:150–759

};

class Socket {
  ws: WebSocket;
  httpRequest: $Request;
  viewer: ?Viewer;
  redis: ?RedisSubscriber;
  redisPromiseResolver: SequentialPromiseResolver<ServerServerSocketMessage>;

  stateCheckConditions: StateCheckConditions = {
    activityRecentlyOccurred: true,
    stateCheckOngoing: false,
  };
  stateCheckTimeoutID: ?TimeoutID;

  constructor(ws: WebSocket, httpRequest: $Request) {
    this.ws = ws;
    this.httpRequest = httpRequest;
    ws.on('message', this.onMessage);
    ws.on('close', this.onClose);
    this.resetTimeout();
    this.redisPromiseResolver = new SequentialPromiseResolver(this.sendMessage);
  }

  onMessage = async (
    messageString: string | Buffer | ArrayBuffer | Array<Buffer>,
  ): Promise<void> => {
    invariant(typeof messageString === 'string', 'message should be string');
    let responseTo = null;
    try {
      this.resetTimeout();
      const messageObject = JSON.parse(messageString);
      const clientSocketMessageWithClientIDs = checkInputValidator(
        clientSocketMessageInputValidator,
        messageObject,
        'socket message',
      );
      responseTo = clientSocketMessageWithClientIDs.id;
      if (
        clientSocketMessageWithClientIDs.type ===
        clientSocketMessageTypes.INITIAL
      ) {
        if (this.viewer) {
          // This indicates that the user sent multiple INITIAL messages.
          throw new ServerError('socket_already_initialized');
        }
        this.viewer = await fetchViewerForSocket(
          this.httpRequest,
          clientSocketMessageWithClientIDs,
        );
      }
      const { viewer } = this;
      if (!viewer) {
        // This indicates a non-INITIAL message was sent by the client before
        // the INITIAL message.
        throw new ServerError('socket_uninitialized');
      }
      if (viewer.sessionChanged) {
        // This indicates that the cookie was invalid, and we've assigned a new
        // anonymous one.