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         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: UserNotificationModel) {
  38     const sockets = this.userNotificationSockets[userId]
  39
  40     if (!sockets) return
  41
  42     for (const socket of sockets) {
  43       socket.emit('new-notification', notification.toFormattedJSON())
  44     }
  45   }
  46
  47   static get Instance () {
  48     return this.instance || (this.instance = new this())
  49   }
  50 }
  51
  52 // ---------------------------------------------------------------------------
  53
  54 export {
  55   PeerTubeSocket
  56 }