| import { debugLog, waitForEvent } from '../utils.js' | |
| import config from '../../config.js' | |
| import constants from '../../constants.js' | |
| import sources from '../sources.js' | |
| import Filters from '../filters.js' | |
| import inputHandler from './inputHandler.js' | |
| import voiceUtils from '../voice/utils.js' | |
| import discordVoice from '@performanc/voice' | |
| globalThis.nodelinkPlayersCount = 0 | |
| globalThis.nodelinkPlayingPlayersCount = 0 | |
| class VoiceConnection { | |
| constructor(guildId, client) { | |
| nodelinkPlayersCount++ | |
| this.client = { | |
| userId: client.userId, | |
| ws: client.ws | |
| } | |
| this.cache = { | |
| url: null, | |
| protocol: null, | |
| track: null | |
| } | |
| this.config = { | |
| guildId, | |
| track: null, | |
| volume: 100, | |
| paused: false, | |
| filters: {}, | |
| voice: { | |
| token: null, | |
| endpoint: null, | |
| sessionId: null | |
| } | |
| } | |
| this._setupVoice() | |
| } | |
| _setupVoice() { | |
| this.connection = discordVoice.joinVoiceChannel({ guildId: this.config.guildId, userId: this.client.userId, encryption: config.audio.encryption }) | |
| this.connection.on('speakStart', (userId, ssrc) => inputHandler.handleStartSpeaking(ssrc, userId, this.config.guildId)) | |
| this.connection.on('stateChange', async (oldState, newState) => { | |
| switch (newState.status) { | |
| case 'disconnected': { | |
| debugLog('websocketClosed', 2, { track: this.config.track?.info, exception: constants.VoiceWSCloseCodes[newState.closeCode] }) | |
| this.connection.destroy() | |
| this.connection = null | |
| this._stopTrack() | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'WebSocketClosedEvent', | |
| guildId: this.config.guildId, | |
| code: newState.code, | |
| reason: constants.VoiceWSCloseCodes[newState.code], | |
| byRemote: true | |
| })) | |
| break | |
| } | |
| } | |
| }) | |
| this.connection.on('playerStateChange', (_oldState, newState) => { | |
| if (newState.status === 'idle' && [ 'stopped', 'finished' ].includes(newState.reason)) { | |
| nodelinkPlayingPlayersCount-- | |
| debugLog('trackEnd', 2, { track: this.config.track.info, reason: newState.reason }) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackEndEvent', | |
| guildId: this.config.guildId, | |
| track: this.config.track, | |
| reason: newState.reason | |
| })) | |
| this._stopTrack() | |
| } | |
| if (newState.status === 'playing' && newState.reason === 'requested') { | |
| nodelinkPlayingPlayersCount++ | |
| debugLog('trackStart', 2, { track: this.config.track.info }) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackStartEvent', | |
| guildId: this.config.guildId, | |
| track: this.config.track | |
| })) | |
| } | |
| }) | |
| this.connection.on('error', (error) => { | |
| debugLog('trackException', 2, { track: this.config.track?.info, exception: error.message }) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackExceptionEvent', | |
| guildId: this.config.guildId, | |
| track: this.config.track, | |
| exception: { | |
| message: error.message, | |
| severity: 'fault', | |
| cause: `${error.name}: ${error.message}` | |
| } | |
| })) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackEndEvent', | |
| guildId: this.config.guildId, | |
| track: this.config.track, | |
| reason: 'loadFailed' | |
| })) | |
| this._stopTrack() | |
| }) | |
| } | |
| _stopTrack() { | |
| this.cache = { | |
| url: null, | |
| protocol: null, | |
| track: null | |
| } | |
| this.config = { | |
| ...this.config, | |
| track: null, | |
| paused: false | |
| } | |
| } | |
| _getRealTime() { | |
| return this.connection.statistics.packetsExpected * 20 | |
| } | |
| updateVoice(buffer) { | |
| this.config.voice = buffer | |
| if (!this.connection) this._setupVoice() | |
| this.connection.voiceStateUpdate({ | |
| session_id: buffer.sessionId | |
| }) | |
| this.connection.voiceServerUpdate({ | |
| token: buffer.token, | |
| endpoint: buffer.endpoint | |
| }) | |
| if (!this.connection.ws) this.connection.connect() | |
| } | |
| destroy() { | |
| if (this.connection) { | |
| this.connection.destroy() | |
| this.connection = null | |
| } | |
| this._stopTrack() | |
| } | |
| async getResource(decodedTrack, urlInfo) { | |
| const streamInfo = await sources.getTrackStream(decodedTrack, urlInfo.url, urlInfo.protocol, urlInfo.additionalData) | |
| if (streamInfo.exception) return streamInfo | |
| return { stream: voiceUtils.createAudioResource(streamInfo.stream, urlInfo.format) } | |
| } | |
| async play(track, decodedTrack, noReplace) { | |
| if (noReplace && this.config.track) return this.config | |
| const urlInfo = await sources.getTrackURL(decodedTrack) | |
| if (urlInfo.exception) { | |
| this._stopTrack() | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackExceptionEvent', | |
| guildId: this.config.guildId, | |
| track: { | |
| encoded: track, | |
| info: decodedTrack | |
| }, | |
| exception: urlInfo.exception | |
| })) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackEndEvent', | |
| guildId: this.config.guildId, | |
| track: { | |
| encoded: track, | |
| info: decodedTrack, | |
| userData: this.config.track?.userData | |
| }, | |
| reason: 'loadFailed' | |
| })) | |
| return this.config | |
| } | |
| if (this.config.track?.encoded) { | |
| debugLog('trackEnd', 2, { track: this.config.track.info, reason: 'replaced' }) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackEndEvent', | |
| guildId: this.config.guildId, | |
| track: this.config.track, | |
| reason: 'replaced' | |
| })) | |
| debugLog('trackStart', 2, { track: decodedTrack }) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackStartEvent', | |
| guildId: this.config.guildId, | |
| track: { | |
| encoded: track, | |
| info: decodedTrack, | |
| userData: this.config.track?.userData | |
| } | |
| })) | |
| } | |
| let resource = null | |
| if (Object.keys(this.config.filters).length > 0) { | |
| const filter = new Filters() | |
| this.config.filters = filter.configure(this.config.filters) | |
| resource = await filter.getResource(decodedTrack, urlInfo.protocol, urlInfo.url, null, null, this.cache.ffmpeg, urlInfo.additionalData) | |
| } else { | |
| resource = await this.getResource(decodedTrack, urlInfo) | |
| } | |
| if (resource.exception) { | |
| this._stopTrack() | |
| debugLog('trackException', 2, { track: decodedTrack, exception: resource.exception.message }) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackExceptionEvent', | |
| guildId: this.config.guildId, | |
| track: { | |
| encoded: track, | |
| info: decodedTrack, | |
| userData: this.config.track?.userData | |
| }, | |
| exception: resource.exception | |
| })) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackEndEvent', | |
| guildId: this.config.guildId, | |
| track: { | |
| encoded: track, | |
| info: decodedTrack, | |
| userData: this.config.track?.userData | |
| }, | |
| reason: 'loadFailed' | |
| })) | |
| return this.config | |
| } | |
| this.cache.url = urlInfo.url | |
| this.cache.protocol = urlInfo.protocol | |
| this.config.track = { encoded: track, info: decodedTrack } | |
| this.config.paused = false | |
| if (this.config.volume !== 100) | |
| resource.stream.setVolume(this.config.volume / 100) | |
| if (!this.connection) | |
| return this.config | |
| if (!this.connection.udpInfo?.secretKey) | |
| await waitForEvent(this.connection, 'stateChange', (_oldState, newState) => newState.status === 'connected', config.options.threshold || undefined) | |
| const oldResource = this.connection.audioStream | |
| this.connection.play(resource.stream) | |
| if (oldResource) | |
| resource.stream.once('readable', () => oldResource.destroy()) | |
| return this.config | |
| } | |
| stop() { | |
| if (!this.config.track) return this.config | |
| if (this.connection.audioStream) this.connection.stop() | |
| else this._stopTrack() | |
| } | |
| volume(volume) { | |
| if (this.connection.audioStream) | |
| this.connection.audioStream.setVolume(volume / 100) | |
| this.config.volume = volume | |
| return this.config | |
| } | |
| pause(pause) { | |
| if (this.connection.audioStream) { | |
| if (pause) this.connection.pause() | |
| else this.connection.unpause() | |
| } | |
| this.config.paused = pause | |
| return this.config | |
| } | |
| async filters(filters) { | |
| if (!this.config.track?.encoded || !config.filters.enabled) return this.config | |
| const filter = new Filters() | |
| this.config.filters = filter.configure(filters, this.config.track.info) | |
| if (!this.config.track) return this.config | |
| const resource = await filter.getResource(this.config.track.info, this.cache.protocol, this.cache.url, this._getRealTime(), filters.endTime, null, null) | |
| if (resource.exception) { | |
| this._stopTrack() | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackExceptionEvent', | |
| guildId: this.config.guildId, | |
| track: this.config.track, | |
| exception: resource.exception | |
| })) | |
| this.client.ws.send(JSON.stringify({ | |
| op: 'event', | |
| type: 'TrackEndEvent', | |
| guildId: this.config.guildId, | |
| track: this.config.track, | |
| reason: 'loadFailed' | |
| })) | |
| return this.config | |
| } | |
| resource.stream.setVolume(filters.volume || (this.config.volume / 100)) | |
| this.config.volume = (filters.volume * 100) || this.config.volume | |
| if (!this.connection) | |
| return this.config | |
| if (!this.connection.udpInfo?.secretKey) | |
| await waitForEvent(this.connection, 'stateChange', (_oldState, newState) => newState.status === 'connected', config.options.threshold || undefined) | |
| this.connection.play(resource.stream) | |
| return this.config | |
| } | |
| } | |
| export default VoiceConnection | |