File size: 5,093 Bytes
b58c6cb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 |
import http from 'node:http'
import { URL } from 'node:url'
import connectionHandler from './handler.js'
import inputHandler from './inputHandler.js'
import config from '../../config.js'
import { debugLog, parseClientName } from '../utils.js'
import { WebSocketServer } from '../ws.js'
if (typeof config.server.port !== 'number')
throw new Error('Port must be a number.')
if (typeof config.server.password !== 'string')
throw new Error('Password must be a string.')
if (typeof config.options.threshold !== 'boolean' && typeof config.options.threshold !== 'number')
throw new Error('Threshold must be a boolean or a number.')
if (typeof config.options.playerUpdateInterval !== 'boolean' && typeof config.options.playerUpdateInterval !== 'number')
throw new Error('Player update interval must be a boolean or a number.')
if (typeof config.options.statsInterval !== 'boolean' && typeof config.options.statsInterval !== 'number')
throw new Error('Stats interval must be a boolean or a number.')
if (typeof config.options.maxResultsLength !== 'number')
throw new Error('Max results length must be a number.')
if (typeof config.options.maxAlbumPlaylistLength !== 'number')
throw new Error('Max album playlist length must be a number.')
if (typeof config.options.maxCaptionsLength !== 'number')
throw new Error('Max captions length must be a number.')
if (typeof config.options.bypassAgeRestriction !== 'boolean')
throw new Error('Bypass age restriction must be a boolean.')
if (!['bandcamp', 'deezer', 'soundcloud', 'youtube', 'ytmusic'].includes(config.search.defaultSearchSource))
throw new Error('Default search source must be either "bandcamp", "deezer", "soundcloud", "youtube" or "ytmusic".')
if (config.search.fallbackSearchSource === 'soundcloud')
throw new Error('SoundCloud is not supported as a fallback source.')
if (config.search.sources.spotify.enabled && !config.search.sources.spotify.market)
throw new Error('Spotify is enabled but no market was provided.')
if (config.search.sources.deezer.enabled && config.search.sources.deezer.decryptionKey === 'DISABLED')
throw new Error('Deezer is enabled but no decryption key or API key was provided.')
if (config.search.sources.soundcloud.enabled && config.search.sources.soundcloud.clientId === 'DISABLED')
throw new Error('SoundCloud is enabled but no client ID was provided.')
if (![ 'high', 'medium', 'low', 'lowest' ].includes(config.audio.quality))
throw new Error('Audio quality must be either "high", "medium", "low" or "lowest".')
if (![ 'xsalsa20_poly1305', 'xsalsa20_poly1305_suffix', 'xsalsa20_poly1305_lite' ].includes(config.audio.encryption))
throw new Error('Encryption must be either "xsalsa20_poly1305", "xsalsa20_poly1305_suffix" or "xsalsa20_poly1305_lite".')
if (typeof config.voiceReceive.timeout !== 'number')
throw new Error('Voice receive timeout must be a number.')
if (![ 'opus', 'pcm' ].includes(config.voiceReceive.type))
throw new Error('Voice receive type must be either "opus" or "pcm".')
const server = http.createServer(connectionHandler.requestHandler)
const v4 = new WebSocketServer()
v4.on('/v4/websocket', connectionHandler.configureConnection)
v4.on('/connection/data', inputHandler.setupConnection)
server.on('upgrade', (req, socket, head) => {
const { pathname } = new URL(req.url, `http://${req.headers.host}`)
if (req.headers.authorization !== config.server.password) {
debugLog('disconnect', 3, { name: 'Unknown', version: '0.0.0', code: 401, reason: 'Invalid password' })
req.socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n')
return req.socket.destroy()
}
const parsedClientName = parseClientName(req.headers['client-name'])
if (!parsedClientName) {
debugLog('connect', 1, { name: req.headers['client-name'], error: 'Client-name doesn\'t conform to NAME/VERSION format.' })
req.socket.write('HTTP/1.1 400 Bad Request\r\n\r\n')
return req.socket.destroy()
}
req.clientInfo = parsedClientName
if (pathname === '/v4/websocket') {
debugLog('connect', 3, parsedClientName)
v4.handleUpgrade(req, socket, head, { 'isNodeLink': true }, (ws) => v4.emit('/v4/websocket', ws, req, parsedClientName))
}
if (pathname === '/connection/data') {
if (!req.headers['guild-id'] || !req.headers['user-id']) {
debugLog('connectCD', 1, { ...parsedClientName, error: `"${!req.headers['guild-id'] ? 'guild-id' : 'user-id'}" header not provided.` })
req.socket.write('HTTP/1.1 400 Bad Request\r\n\r\n')
return req.socket.destroy()
}
debugLog('connectCD', 3, { ...parsedClientName, guildId: req.headers['guild-id'] })
v4.handleUpgrade(req, socket, head, {}, (ws) => v4.emit('/connection/data', ws, req, parsedClientName))
}
})
v4.on('error', (err) => {
debugLog('error', 3, { error: err.message })
})
server.on('error', (err) => {
debugLog('http', 1, { error: err.message })
})
server.listen(config.server.port || 2333, () => {
console.log(`[\u001b[32mwebsocket\u001b[37m]: Listening on port \u001b[94m${config.server.port || 2333}\u001b[37m.`)
}) |