DZ-Musicbot/utils/queueManager.js

111 lines
2.8 KiB
JavaScript
Raw Normal View History

2024-08-17 11:11:10 -04:00
const { createAudioPlayer, createAudioResource, AudioPlayerStatus, joinVoiceChannel } = require('@discordjs/voice');
const fs = require('fs');
const queueMap = new Map();
const playerMap = new Map();
2024-08-17 11:23:30 -04:00
const currentTrackMap = new Map();
2024-08-17 14:58:17 -04:00
const repeatMap = new Map();
2024-08-17 11:11:10 -04:00
2024-08-17 14:58:17 -04:00
function addToQueue(guildId, filePath, title) {
2024-08-17 11:11:10 -04:00
if (!queueMap.has(guildId)) {
queueMap.set(guildId, []);
}
2024-08-17 14:58:17 -04:00
queueMap.get(guildId).push({ filePath, title });
2024-08-17 11:11:10 -04:00
}
function getQueue(guildId) {
return queueMap.get(guildId) || [];
}
function getCurrentTrack(guildId) {
return currentTrackMap.get(guildId) || null;
}
2024-08-17 14:58:17 -04:00
function toggleRepeat(guildId) {
const currentRepeat = repeatMap.get(guildId) || false;
repeatMap.set(guildId, !currentRepeat);
return !currentRepeat;
}
2024-08-17 11:11:10 -04:00
function playNextInQueue(guildId, voiceChannel) {
const queue = getQueue(guildId);
2024-08-17 14:58:17 -04:00
const currentTrack = getCurrentTrack(guildId);
2024-08-17 14:58:17 -04:00
if (currentTrack) {
return false;
}
2024-08-17 11:11:10 -04:00
2024-08-17 14:58:17 -04:00
if (queue.length === 0) return false;
2024-08-17 11:11:10 -04:00
const connection = joinVoiceChannel({
channelId: voiceChannel.id,
guildId: guildId,
adapterCreator: voiceChannel.guild.voiceAdapterCreator,
});
2024-08-17 14:58:17 -04:00
const audioPlayer = playerMap.get(guildId) || createAudioPlayer();
playerMap.set(guildId, audioPlayer);
2024-08-17 11:11:10 -04:00
2024-08-17 14:58:17 -04:00
const trackToPlay = queue.shift();
currentTrackMap.set(guildId, trackToPlay);
2024-08-17 11:11:10 -04:00
2024-08-17 14:58:17 -04:00
if (!fs.existsSync(trackToPlay.filePath)) {
console.error('Audio file not found:', trackToPlay.filePath);
2024-08-17 11:11:10 -04:00
return false;
}
2024-08-17 14:58:17 -04:00
const resource = createAudioResource(trackToPlay.filePath);
2024-08-17 11:11:10 -04:00
audioPlayer.play(resource);
connection.subscribe(audioPlayer);
audioPlayer.on(AudioPlayerStatus.Idle, () => {
currentTrackMap.delete(guildId);
2024-08-17 14:58:17 -04:00
2024-08-17 11:11:10 -04:00
if (queue.length > 0) {
playNextInQueue(guildId, voiceChannel);
} else {
connection.destroy();
}
2024-08-17 14:58:17 -04:00
fs.unlink(trackToPlay.filePath, (err) => {
if (err) console.error('Error deleting file:', trackToPlay.filePath, err);
});
2024-08-17 11:11:10 -04:00
});
audioPlayer.on('error', (err) => {
console.error('AudioPlayer error:', err);
currentTrackMap.delete(guildId);
2024-08-17 11:11:10 -04:00
connection.destroy();
});
return true;
2024-08-17 11:11:10 -04:00
}
2024-08-17 14:58:17 -04:00
function skipTrack(guildId, voiceChannel) {
const player = playerMap.get(guildId);
2024-08-17 11:11:10 -04:00
const queue = getQueue(guildId);
2024-08-17 14:58:17 -04:00
const repeat = repeatMap.get(guildId);
2024-08-17 14:58:17 -04:00
if (!player) {
console.error('No player found for this guild.');
return;
}
2024-08-17 14:58:17 -04:00
currentTrackMap.delete(guildId);
if (repeat) {
repeatMap.set(guildId, false);
}
player.stop();
if (queue.length > 0) {
playNextInQueue(guildId, voiceChannel);
} else {
2024-08-17 14:58:17 -04:00
player.stop();
2024-08-17 11:11:10 -04:00
}
}
2024-08-17 14:58:17 -04:00
module.exports = { addToQueue, getQueue, getCurrentTrack, playNextInQueue, skipTrack, toggleRepeat };