DLAP/AudioBackend.js

/**************************************************************************
 * 
 *  DLAP Bot: A Discord bot that plays local audio tracks.
 *  (C) Copyright 2022
 *  Programmed by Andrew Lee 
 *  
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
 * 
 ***************************************************************************/

import {
  createAudioPlayer,
  createAudioResource,
  getVoiceConnection,
  joinVoiceChannel,
  VoiceConnectionStatus
} from '@discordjs/voice'
import { MessageEmbed } from 'discord.js'
import config from './config.json' assert {type: 'json'}
import { readdirSync, writeFile } from 'node:fs'

export const player = createAudioPlayer();
export let audio;
export let files = readdirSync('music');
let fileData;

export let playerState;
let isAudioStatePaused;

export async function voiceInit(bot) {
  bot.channels.fetch(config.voiceChannel).then(async channel => {
    const connection = joinVoiceChannel({
      channelId: channel.id,
      guildId: channel.guild.id,
      adapterCreator: channel.guild.voiceAdapterCreator
    });

    connection.on(VoiceConnectionStatus.Ready, () => {
      console.log('Ready to blast some beats!');
    });

    connection.on(VoiceConnectionStatus.Destroyed, () => {
      console.log('Destroying beats...');
    });

    player.on('idle', () => {
      console.log("Beat has finished playing, shuffling the beats...");
      searchAudio(bot);
    })

    await searchAudio(bot);
    return connection.subscribe(player);
  }).catch(e => { console.error(e) })
}

export async function searchAudio(bot){
  //TODO: Eventually this system will need a rework so it won't repeat the same files.

  audio = files[Math.floor(Math.random() * files.length)];
  return await playAudio(bot);
}

export async function inputAudio(bot, integer) {
  audio = files[integer];
  return await playAudio(bot);
}

export async function playAudio(bot) {
  let resource = createAudioResource('music/' + audio);

  player.play(resource);

  console.log('Now playing: ' + audio);

  playerState = "Playing"
  isAudioStatePaused = false

  audio = audio.split('.').slice(0, -1).join('.');

  if (config.txtFile === true) {
    fileData = "Now Playing: " + audio
    writeFile("./now-playing.txt", fileData, (err) => {
      if (err)
        console.log(err);
    });
  }

  const statusEmbed = new MessageEmbed()
      .addField('Now Playing', `${audio}`)
      .setColor('#0066ff')

  let statusChannel = bot.channels.cache.get(config.statusChannel);
  if (!statusChannel) return console.error('The status channel does not exist! Skipping.');
  return await statusChannel.send({embeds: [statusEmbed]});

}

export async function destroyAudio(interaction) {
  if (config.txtFile === true) {
    fileData = "Now Playing: Nothing";
    writeFile("now-playing.txt", fileData, (err) => {
      if (err)
        console.log(err);
    });
  }

  audio = "Not Playing"
  playerState = "Stopped"
  isAudioStatePaused = true

  const connection = getVoiceConnection(interaction.guild.id);
  if (VoiceConnectionStatus.Ready) {
    player.stop();
    return connection.destroy();
  }
}

export function audioState() {
  if (isAudioStatePaused === false) {
    isAudioStatePaused = true
    playerState = "Paused"
  } else if (isAudioStatePaused === true) {
    isAudioStatePaused = false
    playerState = "Playing"
  }
}

export async function stopBot(bot, interaction) {
  const statusEmbed = new MessageEmbed()
      .setAuthor({name: bot.user.username, iconURL: bot.user.avatarURL()})
      .setDescription(`That\'s all folks! Powering down ${bot.user.username}...`)
      .setColor('#0066ff')
  let statusChannel = bot.channels.cache.get(config.statusChannel);
  if (!statusChannel) return console.error('The status channel does not exist! Skipping.');
  await statusChannel.send({embeds: [statusEmbed]});

  console.log(`Powering off ${bot.user.username}...`);
  await destroyAudio(interaction);
  bot.destroy();
  return process.exit(0);
}