server/lib/peertube-socket.ts

   1 import * as SocketIO from 'socket.io'
   2 import { authenticateSocket } from '../middlewares'
   3 import { UserNotificationModel } from '../models/account/user-notification'
   4 import { logger } from '../helpers/logger'
   5 import { Server } from 'http'
   6
   7 class PeerTubeSocket {
   8
   9   private static instance: PeerTubeSocket
  10
  11   private userNotificationSockets: { [ userId: number ]: SocketIO.Socket } = {}
  12
  13   private constructor () {}
  14
  15   init (server: Server) {
  16     const io = SocketIO(server)
  17
  18     io.of('/user-notifications')
  19       .use(authenticateSocket)
  20       .on('connection', socket => {
  21         const userId = socket.handshake.query.user.id
  22
  23         logger.debug('User %d connected on the notification system.', userId)
  24
  25         this.userNotificationSockets[userId] = socket
  26
  27         socket.on('disconnect', () => {
  28           logger.debug('User %d disconnected from SocketIO notifications.', userId)
  29
  30           delete this.userNotificationSockets[userId]
  31         })
  32       })
  33   }
  34
  35   sendNotification (userId: number, notification: UserNotificationModel) {
  36     const socket = this.userNotificationSockets[userId]
  37
  38     if (!socket) return
  39
  40     socket.emit('new-notification', notification.toFormattedJSON())
  41   }
  42
  43   static get Instance () {
  44     return this.instance || (this.instance = new this())
  45   }
  46 }
  47
  48 // ---------------------------------------------------------------------------
  49
  50 export {
  51   PeerTubeSocket
  52 }