123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899 |
- 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 = new SocketIO.Server(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, { state: video.state })
- this.liveVideosNamespace
- .in(video.id)
- .emit(type, data)
- }
- sendVideoViewsUpdate (video: MVideo) {
- const data: LiveVideoEventPayload = { views: video.views }
- const type: LiveVideoEventType = 'views-change'
- logger.debug('Sending video live views update notification of %s.', video.url, { views: video.views })
- this.liveVideosNamespace
- .in(video.id)
- .emit(type, data)
- }
- static get Instance () {
- return this.instance || (this.instance = new this())
- }
- }
- // ---------------------------------------------------------------------------
- export {
- PeerTubeSocket
- }
|