import { Server as WebSocketServer } from 'ws'
import { CONFIG, TRACKER_RATE_LIMITS } from '../initializers/constants'
import { VideoFileModel } from '../models/video/video-file'
+import { parse } from 'url'
const TrackerServer = bitTorrentTracker.Server
function createWebsocketTrackerServer (app: express.Application) {
const server = http.createServer(app)
- const wss = new WebSocketServer({ server: server, path: '/tracker/socket' })
+ const wss = new WebSocketServer({ noServer: true })
+
wss.on('connection', function (ws, req) {
- const ip = proxyAddr(req, CONFIG.TRUST_PROXY)
- ws['ip'] = ip
+ ws['ip'] = proxyAddr(req, CONFIG.TRUST_PROXY)
trackerServer.onWebSocketConnection(ws)
})
+ server.on('upgrade', (request, socket, head) => {
+ const pathname = parse(request.url).pathname
+
+ if (pathname === '/tracker/socket') {
+ wss.handleUpgrade(request, socket, head, ws => wss.emit('connection', ws, request))
+ }
+
+ // Don't destroy socket, we have Socket.IO too
+ })
+
return server
}