import { Socket } from 'dgram' import { Server } from 'http' import * as SocketIO from 'socket.io' import { MVideo } from '@server/types/models' import { UserNotificationModelForApi } from '@server/types/models/user' import { LiveVideoEventPayload, LiveVideoEventType } from '@shared/models' import { logger } from '../helpers/logger' import { authenticateSocket } from '../middlewares' import { isIdValid } from '@server/helpers/custom-validators/misc' class PeerTubeSocket { private static instance: PeerTubeSocket private userNotificationSockets: { [ userId: number ]: SocketIO.Socket[] } = {} private liveVideosNamespace: SocketIO.Namespace private constructor () {} init (server: Server) { const io = SocketIO(server) io.of('/user-notifications') .use(authenticateSocket) .on('connection', socket => { const userId = socket.handshake.query.user.id logger.debug('User %d connected on the notification system.', userId) if (!this.userNotificationSockets[userId]) this.userNotificationSockets[userId] = [] this.userNotificationSockets[userId].push(socket) socket.on('disconnect', () => { logger.debug('User %d disconnected from SocketIO notifications.', userId) this.userNotificationSockets[userId] = this.userNotificationSockets[userId].filter(s => s !== socket) }) }) this.liveVideosNamespace = io.of('/live-videos') .on('connection', socket => { socket.on('subscribe', ({ videoId }) => { if (!isIdValid(videoId)) return socket.join(videoId) }) socket.on('unsubscribe', ({ videoId }) => { if (!isIdValid(videoId)) return socket.leave(videoId) }) }) } sendNotification (userId: number, notification: UserNotificationModelForApi) { const sockets = this.userNotificationSockets[userId] if (!sockets) return logger.debug('Sending user notification to user %d.', userId) const notificationMessage = notification.toFormattedJSON() for (const socket of sockets) { socket.emit('new-notification', notificationMessage) } } sendVideoLiveNewState (video: MVideo) { const data: LiveVideoEventPayload = { state: video.state } const type: LiveVideoEventType = 'state-change' logger.debug('Sending video live new state notification of %s.', video.url) this.liveVideosNamespace .in(video.id) .emit(type, data) } static get Instance () { return this.instance || (this.instance = new this()) } } // --------------------------------------------------------------------------- export { PeerTubeSocket }