chomens-bot-js/plugins/music.js

/* eslint-disable require-jsdoc */
/* eslint-disable max-len */
const { Midi } = require('@tonejs/midi')
const { convertMidi } = require('../util/midi_converter')

const soundNames = {
  harp: 'minecraft:block.note_block.harp',
  basedrum: 'minecraft:block.note_block.basedrum',
  snare: 'minecraft:block.note_block.snare',
  hat: 'minecraft:block.note_block.hat',
  bass: 'minecraft:block.note_block.bass',
  flute: 'minecraft:block.note_block.flute',
  bell: 'minecraft:block.note_block.bell',
  guitar: 'minecraft:block.note_block.guitar',
  chime: 'minecraft:block.note_block.chime',
  xylophone: 'minecraft:block.note_block.xylophone',
  iron_xylophone: 'minecraft:block.note_block.iron_xylophone',
  cow_bell: 'minecraft:block.note_block.cow_bell',
  didgeridoo: 'minecraft:block.note_block.didgeridoo',
  bit: 'minecraft:block.note_block.bit',
  banjo: 'minecraft:block.note_block.banjo',
  pling: 'minecraft:block.note_block.pling'
}

function inject (bot) {
  bot.music = function () {}
  bot.music.song = null
  bot.music.loop = 0
  bot.music.queue = []
  let time = 0
  let startTime = 0
  let noteIndex = 0
  bot.music.skip = function () {
    if (bot.music.loop === 2) {
      bot.music.queue.push(bot.music.queue.shift())
      bot.music.play(bot.music.queue[0])
    } else {
      bot.music.queue.shift()
    }
    resetTime()
  }

  const interval = setInterval(async () => {
    try {
      if (!bot.music.queue.length) return
      bot.music.song = bot.music.queue[0]
      time = Date.now() - startTime
      while (bot.music.song.notes[noteIndex]?.time <= time) {
        const note = bot.music.song.notes[noteIndex]
        const floatingPitch = 2 ** ((note.pitch - 12) / 12.0)
        bot.core.run(`minecraft:execute as @a[tag=!nomusic] at @s run playsound ${soundNames[note.instrument]} record @s ~ ~ ~ ${note.volume} ${floatingPitch}`)
        noteIndex++
        bot.core.run('minecraft:title @a[tag=!nomusic] actionbar ' + JSON.stringify(toComponent()))
        if (noteIndex >= bot.music.song.notes.length) {
          if (bot.music.loop === 1) {
            resetTime()
            return
          }
          if (bot.music.loop === 2) {
            resetTime()
            bot.music.queue.push(bot.music.queue.shift())
            bot.music.play(bot.music.queue[0])
            return
          }
          bot.music.queue.shift()
          if (!bot.music.queue[0]) {
            bot.tellraw('@a', { text: 'Finished playing every song in the queue' })
            bot.music.stop()
            return
          }
          if (bot.music.queue[0].notes.length > 0) {
            if (bot.music.queue.length !== 1) resetTime()
            bot.music.play(bot.music.queue[0])
            return
          }
        }
      }
    } catch (e) {}
  }, 50)

  bot.once('end', () => {
    clearInterval(interval)
  })

  bot.music.load = function (buffer, fallbackName = '[unknown]') {
    // TODO: NBS Support
    const midi = new Midi(buffer)
    const song = convertMidi(midi)
    if (song.name === '') song.name = fallbackName
    return song
  }

  bot.music.play = function () {
    if (bot.music.queue.length === 1) resetTime()
  }

  bot.music.stop = function () {
    bot.music.song = null
    bot.music.loop = 0
    bot.music.queue = []
    resetTime()
  }

  function resetTime () {
    time = 0
    startTime = Date.now()
    noteIndex = 0
  }

  function formatTime (time) {
    const seconds = Math.floor(time / 1000)

    return `${Math.floor(seconds / 60)}:${(seconds % 60).toString().padStart(2, '0')}`
  }

  function toComponent () {
    const component = [
      { text: '[', color: 'dark_gray' },
      { text: 'ChomeNS Bot', color: 'yellow' },
      { text: '] ', color: 'dark_gray' },
      { text: 'Now Playing', color: 'gold' },
      { text: ' | ', color: 'dark_gray' },
      { text: bot.music.song.name, color: 'green' },
      { text: ' | ', color: 'dark_gray' },
      { text: formatTime(time), color: 'gray' },
      { text: ' / ', color: 'dark_gray' },
      { text: formatTime(bot.music.song.length), color: 'gray' },
      { text: ' | ', color: 'dark_gray' },
      { text: noteIndex, color: 'gray' },
      { text: ' / ', color: 'dark_gray' },
      { text: bot.music.song.notes.length, color: 'gray' }
    ]

    if (bot.music.loop === 1) {
      component.push({ text: ' | ', color: 'dark_gray' })
      component.push({ text: 'Looping Current', color: 'green' })
    }
    if (bot.music.loop === 2) {
      component.push({ text: ' | ', color: 'dark_gray' })
      component.push({ text: 'Looping All', color: 'green' })
    }

    return component
  }
}

module.exports = { inject }