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/typings/models/user'
9 private static instance: PeerTubeSocket
11 private userNotificationSockets: { [ userId: number ]: SocketIO.Socket[] } = {}
13 private constructor () {}
15 init (server: Server) {
16 const io = SocketIO(server)
18 io.of('/user-notifications')
19 .use(authenticateSocket)
20 .on('connection', socket => {
21 const userId = socket.handshake.query.user.id
23 logger.debug('User %d connected on the notification system.', userId)
25 if (!this.userNotificationSockets[userId]) this.userNotificationSockets[userId] = []
27 this.userNotificationSockets[userId].push(socket)
29 socket.on('disconnect', () => {
30 logger.debug('User %d disconnected from SocketIO notifications.', userId)
32 this.userNotificationSockets[userId] = this.userNotificationSockets[userId].filter(s => s !== socket)
37 sendNotification (userId: number, notification: UserNotificationModelForApi) {
38 const sockets = this.userNotificationSockets[userId]
42 const notificationMessage = notification.toFormattedJSON()
43 for (const socket of sockets) {
44 socket.emit('new-notification', notificationMessage)
48 static get Instance () {
49 return this.instance || (this.instance = new this())
53 // ---------------------------------------------------------------------------