File size: 2,479 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 |
import { debugLog } from '../utils.js'
import config from '../../config.js'
import voiceUtils from '../voice/utils.js'
import discordVoice from '@performanc/voice'
import prism from 'prism-media'
const Connections = {}
function setupConnection(ws, req, parsedClientName) {
const userId = req.headers['user-id']
const guildId = req.headers['guild-id']
ws.on('close', (code, reason) => {
debugLog('disconnectCD', 3, { ...parsedClientName, code, reason, guildId })
delete Connections[userId]
})
ws.on('error', (err) => {
debugLog('disconnectCD', 3, { ...parsedClientName, error: `Error: ${err.message}`, guildId })
delete Connections[userId]
})
Connections[userId] = {
ws,
guildId
}
}
function handleStartSpeaking(ssrc, userId, guildId) {
const opusStream = discordVoice.getSpeakStream(ssrc)
const stream = new voiceUtils.NodeLinkStream(opusStream, config.voiceReceive.type === 'pcm' ? [ new prism.opus.Decoder({ rate: 48000, channels: 2, frameSize: 960 }) ] : [])
let timeout = null
const startSpeakingResponse = JSON.stringify({
op: 'speak',
type: 'startSpeakingEvent',
data: {
userId,
guildId
}
})
Object.keys(Connections).forEach((botId) => {
if (Connections[botId].guildId !== guildId) return;
Connections[botId].ws.send(startSpeakingResponse)
})
let buffer = []
stream.on('data', (chunk) => {
if (timeout) {
clearTimeout(timeout)
timeout = null
}
if (Object.keys(Connections).length === 0) {
stream.destroy()
buffer = null
return;
}
buffer.push(chunk)
})
stream.on('end', () => {
let i = 0
const connectionsArray = Object.keys(Connections)
if (connectionsArray.length === 0) {
buffer = []
return;
}
timeout = setTimeout(() => {
const endSpeakingResponse = JSON.stringify({
op: 'speak',
type: 'endSpeakingEvent',
data: {
userId,
guildId,
data: Buffer.concat(buffer).toString('base64'),
type: config.voiceReceive.type
}
})
connectionsArray.forEach((botId) => {
if (Connections[botId].guildId !== guildId) return;
Connections[botId].ws.send(endSpeakingResponse)
i++
})
buffer = []
debugLog('sentDataCD', 3, { clientsAmount: i, guildId })
}, config.voiceReceive.timeout)
})
}
export default {
setupConnection,
handleStartSpeaking
} |