server/lib/peertube-socket.ts

   1 import * as SocketIO from 'socket.io'
   2 import { authenticateSocket } from '../middlewares'
   3 import { logger } from '../helpers/logger'
   4 import { Server } from 'http'
   5 import { UserNotificationModelForApi } from '@server/types/models/user'
   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         if (!this.userNotificationSockets[userId]) this.userNotificationSockets[userId] = []
  26
  27         this.userNotificationSockets[userId].push(socket)
  28
  29         socket.on('disconnect', () => {
  30           logger.debug('User %d disconnected from SocketIO notifications.', userId)
  31
  32           this.userNotificationSockets[userId] = this.userNotificationSockets[userId].filter(s => s !== socket)
  33         })
  34       })
  35   }
  36
  37   sendNotification (userId: number, notification: UserNotificationModelForApi) {
  38     const sockets = this.userNotificationSockets[userId]
  39
  40     if (!sockets) return
  41
  42     const notificationMessage = notification.toFormattedJSON()
  43     for (const socket of sockets) {
  44       socket.emit('new-notification', notificationMessage)
  45     }
  46   }
  47
  48   static get Instance () {
  49     return this.instance || (this.instance = new this())
  50   }
  51 }
  52
  53 // ---------------------------------------------------------------------------
  54
  55 export {
  56   PeerTubeSocket
  57 }