]>
Commit | Line | Data |
---|---|---|
cef534ed C |
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 | this.userNotificationSockets[userId] = socket | |
26 | ||
27 | socket.on('disconnect', () => { | |
28 | logger.debug('User %d disconnected from SocketIO notifications.', userId) | |
29 | ||
30 | delete this.userNotificationSockets[userId] | |
31 | }) | |
32 | }) | |
33 | } | |
34 | ||
35 | sendNotification (userId: number, notification: UserNotificationModel) { | |
36 | const socket = this.userNotificationSockets[userId] | |
37 | ||
38 | if (!socket) return | |
39 | ||
40 | socket.emit('new-notification', notification.toFormattedJSON()) | |
41 | } | |
42 | ||
43 | static get Instance () { | |
44 | return this.instance || (this.instance = new this()) | |
45 | } | |
46 | } | |
47 | ||
48 | // --------------------------------------------------------------------------- | |
49 | ||
50 | export { | |
51 | PeerTubeSocket | |
52 | } |