nirvana-v3/packages/api/sockets/index.ts

import {
  ConnectToLineRequest,
  RtcReceiveSignalResponse,
  RtcSendSignalRequest,
  ServerRequestChannels,
  ServerResponseChannels,
  SomeoneConnectedResponse,
  SomeoneDisconnectedResponse,
  SomeoneTunedResponse,
  SomeoneUntunedFromLineResponse,
  StartBroadcastingRequest,
  StopBroadcastingRequest,
  TuneToLineRequest,
  UntuneFromLineRequest,
  UserStartedBroadcastingResponse,
  UserStoppedBroadcastingResponse,
} from '@nirvana/core/sockets/channels';

import GetAllSocketClients from '@nirvana/core/sockets/getAllActiveSocketClients';
import { JwtClaims } from '../middleware/auth';
import { LineMemberState } from '@nirvana/core/models/line.model';
import { LineService } from '../services/line.service';
import ReceiveSignal from '@nirvana/core/sockets/receiveSignal';
import SendSignal from '@nirvana/core/sockets/sendSignal';
import { UserService } from '../services/user.service';
import { UserStatus } from '@nirvana/core/models/user.model';
import { client } from '../services/database.service';
import { loadConfig } from '../config';

const jwt = require('jsonwebtoken');

const config = loadConfig();

// NOTE: client socket connections should never have to deal with socketIds
const socketIdsToUserIds: {
  [socketId: string]: string;
} = {};
const userIdsToSocketIds: {
  [userId: string]: string;
} = {};

export default function InitializeWs(io: any) {
  console.log('initializing web sockets');

  return io
    .use(function (socket: any, next: any) {
      console.log('authenticating user...');

      try {
        const { token } = socket.handshake.query;
        console.log(token);

        // verify jwt token with our api secret
        var decoded: JwtClaims = jwt.verify(token, config.JWT_TOKEN_SECRET);

        socket.userInfo = decoded;

        next();
      } catch (error) {
        console.error(error);
        next(new Error('WS Authentication Error'));
      }
    })
    .on('connection', function (socket: any) {
      const userInfo: JwtClaims = socket.userInfo;

      socketIdsToUserIds[socket.id] = userInfo.userId.toString();
      userIdsToSocketIds[userInfo.userId.toString()] = socket.id;

      console.log(`a user connected | user Id: ${userInfo.userId} and name: ${userInfo.name}`);

      socket.on('test', () => {
        console.log('asdf');
      });

      // ?verification that user is in a particular line to be tuned into it or just generally in it?

      /** CONNECT | User wants to subscribe to live emissions of a line */
      socket.on(ServerRequestChannels.CONNECT_TO_LINE, (req: ConnectToLineRequest) => {
        // add this user to the room
        console.log(`${socket.id} user CONNECTED room for line ${Object.keys(socket.rooms)}`);

        const roomName = `connectedLine:${req.lineId}`;
        socket.join(roomName);

        const clientUserIdsInRoom = [...(io.sockets.adapter.rooms.get(roomName) ?? [])].map(
          (otherUserSocketId: string) => socketIdsToUserIds[otherUserSocketId],
        );

        io.in(roomName).emit(
          ServerResponseChannels.SOMEONE_CONNECTED_TO_LINE,
          new SomeoneConnectedResponse(req.lineId, userInfo.userId, clientUserIdsInRoom),
        );
      });

      /** TUNE | User tunes into the line either temporarily or toggled in  */
      socket.on(ServerRequestChannels.TUNE_INTO_LINE, async (req: TuneToLineRequest) => {
        console.log(`${socket.id} user TUNED into room for line ${req.lineId}`);

        const roomName = `tunedLine:${req.lineId}`;
        socket.join(roomName);

        const clientUserIdsInRoom = [...(io.sockets.adapter.rooms.get(roomName) ?? [])].map(
          (otherUserSocketId: string) => socketIdsToUserIds[otherUserSocketId],
        );

        // we want to notify everyone connected to the line even if they are not tuned in
        const connectedLineRoomName = `connectedLine:${req.lineId}`;

        io.in(connectedLineRoomName).emit(
          ServerResponseChannels.SOMEONE_TUNED_INTO_LINE,
          new SomeoneTunedResponse(req.lineId, userInfo.userId, clientUserIdsInRoom),
        );
      });

      /**
       * Notify all connected users when someone UNTUNES from a room
       */
      socket.on(ServerRequestChannels.UNTUNE_FROM_LINE, async (req: UntuneFromLineRequest) => {
        const roomName = `tunedLine:${req.lineId}`;
        socket.leave(roomName);

        console.log(`${userInfo.userId} left room: ${roomName}`);

        // we want to notify everyone connected to the line even if they are not tuned in
        const connectedLineRoomName = `connectedLine:${req.lineId}`;

        io.in(connectedLineRoomName).emit(
          ServerResponseChannels.SOMEONE_UNTUNED_FROM_LINE,
          new SomeoneUntunedFromLineResponse(req.lineId, userInfo.userId),
        );
      });

      // TODO: use same pattern as tuning and untuning and send updated fresh list of current broadcasters but using another namespace/room for broadcasters in a line
      /** BROADCAST UPDATE | tell all connected, not just tuned into, that there is an update to someone broadcasting */
      socket.on(ServerRequestChannels.BROADCAST_TO_LINE, (req: StartBroadcastingRequest) => {
        const roomName = `connectedLine:${req.lineId}`;

        io.in(roomName).emit(
          ServerResponseChannels.SOMEONE_STARTED_BROADCASTING,
          new UserStartedBroadcastingResponse(req.lineId, userInfo.userId),
        );
      });

      socket.on(ServerRequestChannels.STOP_BROADCAST_TO_LINE, (req: StopBroadcastingRequest) => {
        const roomName = `connectedLine:${req.lineId}`;

        io.in(roomName).emit(
          ServerResponseChannels.SOMEONE_STOPPED_BROADCASTING,
          new UserStoppedBroadcastingResponse(req.lineId, userInfo.userId),
        );
      });

      // tell the proper other user to create a local peer object for the one on one mesh connection
      socket.on(ServerRequestChannels.RTC_SEND_SIGNAL, (req: RtcSendSignalRequest) => {
        const userSocketId = userIdsToSocketIds[req.userToCall];

        io.to(userSocketId).emit(
          ServerResponseChannels.RTC_RECEIVING_SIGNAL,
          new RtcReceiveSignalResponse(userInfo.userId, req.simplePeerSignal),
        );
      });

      // tell all people in all my lines that I am disconnecting or untuning
      // tell the tuned in folks the new list of
      socket.on('disconnecting', (reason: any) => {
        console.log(`someone disconnecting: ${reason}`);

        console.log(socket.rooms);

        for (const roomName of socket.rooms) {
          if (roomName !== socket.id) {
            const lineId = roomName.split(':')[1];

            const roomToTell = roomName.includes('tunedLine')
              ? `connectedLine:${lineId}`
              : roomName;

            io.in(roomToTell).emit(
              ServerResponseChannels.SOMEONE_DISCONNECTED_FROM_LINE,
              new SomeoneDisconnectedResponse(lineId, userInfo.userId),
            );
          }
        }
      });

      // ==== DISCONNECT ====
      socket.on('disconnect', () => {
        delete socketIdsToUserIds[socket.id];
        delete userIdsToSocketIds[userInfo.userId];
      });
    });
}