-import { Socket } from 'dgram'
import { Server } from 'http'
import * as SocketIO from 'socket.io'
import { MVideo } from '@server/types/models'
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 constructor () {}
init (server: Server) {
- const io = SocketIO(server)
+ const io = new SocketIO.Server(server)
io.of('/user-notifications')
.use(authenticateSocket)
this.liveVideosNamespace = io.of('/live-videos')
.on('connection', socket => {
- socket.on('subscribe', ({ videoId }) => socket.join(videoId))
- socket.on('unsubscribe', ({ videoId }) => socket.leave(videoId))
+ socket.on('subscribe', ({ videoId }) => {
+ if (!isIdValid(videoId)) return
+
+ socket.join(videoId)
+ })
+
+ socket.on('unsubscribe', ({ videoId }) => {
+ if (!isIdValid(videoId)) return
+
+ socket.leave(videoId)
+ })
})
}
const data: LiveVideoEventPayload = { state: video.state }
const type: LiveVideoEventType = 'state-change'
- logger.debug('Sending video live new state notification of %s.', video.url)
+ 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)