path: root/src/loop-play.js

// This isn't actually the code for the `play` command! That's in `play.js`.

// NOTE TO FUTURE SELF AND READERS:
// Please be careful to discern the target of methods such as waitForDownload.
// DownloadController and PlayController are messes and have lots of functions
// of the same name but completely different purposes. (Also keep an eye out
// for similarly/identically-named events between the two classes.)

'use strict'

const { spawn } = require('child_process')
const FIFO = require('fifo-js')
const EventEmitter = require('events')
const fs = require('fs')
const util = require('util')
const killProcess = require('./kill-process')
const { HistoryController, generalPicker } = require('./pickers')

const writeFile = util.promisify(fs.writeFile)

const {
  getDownloaderFor, byName: downloadersByName, makeConverter
} = require('./downloaders')

const {
  getItemPathString, safeUnlink, parentSymbol, sourceSymbol
} = require('./playlist-utils')

const { processTemplateString } = require('./general-util')

function getTimeStrings({curHour, curMin, curSec, lenHour, lenMin, lenSec}) {
  // Multiplication casts to numbers; addition prioritizes strings.
  // Thanks, JavaScript!
  const curSecTotal = (3600 * curHour) + (60 * curMin) + (1 * curSec)
  const lenSecTotal = (3600 * lenHour) + (60 * lenMin) + (1 * lenSec)
  const percentVal = (100 / lenSecTotal) * curSecTotal
  const percentDone = (
    (Math.trunc(percentVal * 100) / 100).toFixed(2) + '%'
  )

  const leftSecTotal = lenSecTotal - curSecTotal
  let leftHour = Math.floor(leftSecTotal / 3600)
  let leftMin = Math.floor((leftSecTotal - leftHour * 3600) / 60)
  let leftSec = Math.floor(leftSecTotal - leftHour * 3600 - leftMin * 60)

  const pad = val => val.toString().padStart(2, '0')
  curMin = pad(curMin)
  curSec = pad(curSec)
  lenMin = pad(lenMin)
  lenSec = pad(lenSec)
  leftMin = pad(leftMin)
  leftSec = pad(leftSec)

  // We don't want to display hour counters if the total length is less
  // than an hour.
  let timeDone, timeLeft, duration
  if (parseInt(lenHour) > 0) {
    timeDone = `${curHour}:${curMin}:${curSec}`
    timeLeft = `${leftHour}:${leftMin}:${leftSec}`
    duration = `${lenHour}:${lenMin}:${lenSec}`
  } else {
    timeDone = `${curMin}:${curSec}`
    timeLeft = `${leftMin}:${leftSec}`
    duration = `${lenMin}:${lenSec}`
  }

  return {percentDone, timeDone, timeLeft, duration}
}

class Player extends EventEmitter {
  constructor() {
    super()

    this.disablePlaybackStatus = false
  }

  set process(newProcess) {
    this._process = newProcess
    this._process.on('exit', code => {
      if (code !== 0 && !this._killed) {
        this.emit('crashed', code)
      }

      this._killed = false
    })
  }

  get process() {
    return this._process
  }

  playFile(file) {}
  seekAhead(secs) {}
  seekBack(secs) {}
  volUp(amount) {}
  volDown(amount) {}
  togglePause() {}

  async kill() {
    if (this.process) {
      this._killed = true
      await killProcess(this.process)
    }
  }

  printStatusLine(data) {
    // Quick sanity check - we don't want to print the status line if it's
    // disabled! Hopefully printStatusLine won't be called in that case, but
    // if it is, we should be careful.
    if (!this.disablePlaybackStatus) {
      this.emit('printStatusLine', data)
    }
  }
}

class MPVPlayer extends Player {
  getMPVOptions(file) {
    return ['--no-audio-display', file]
  }

  playFile(file) {
    // The more powerful MPV player. MPV is virtually impossible for a human
    // being to install; if you're having trouble with it, try the SoX player.

    this.process = spawn('mpv', this.getMPVOptions(file))

    this.process.stderr.on('data', data => {
      if (this.disablePlaybackStatus) {
        return
      }

      const match = data.toString().match(
        /(..):(..):(..) \/ (..):(..):(..) \(([0-9]+)%\)/
      )

      if (match) {
        const [
          curHour, curMin, curSec, // ##:##:##
          lenHour, lenMin, lenSec, // ##:##:##
          percent // ###%
        ] = match.slice(1)

        this.printStatusLine(getTimeStrings({curHour, curMin, curSec, lenHour, lenMin, lenSec}))
      }
    })

    return new Promise(resolve => {
      this.process.once('close', resolve)
    })
  }
}

class ControllableMPVPlayer extends MPVPlayer {
  getMPVOptions(file) {
    return ['--input-file=' + this.fifo.path, ...super.getMPVOptions(file)]
  }

  playFile(file) {
    this.fifo = new FIFO()

    return super.playFile(file)
  }

  sendCommand(command) {
    if (this.fifo) {
      this.fifo.write(command)
    }
  }

  seekAhead(secs) {
    this.sendCommand(`seek +${parseFloat(secs)}`)
  }

  seekBack(secs) {
    this.sendCommand(`seek -${parseFloat(secs)}`)
  }

  volUp(amount) {
    this.sendCommand(`add volume +${parseFloat(amount)}`)
  }

  volDown(amount) {
    this.sendCommand(`add volume -${parseFloat(amount)}`)
  }

  togglePause() {
    this.sendCommand('cycle pause')
  }

  kill() {
    if (this.fifo) {
      this.fifo.close()
      delete this.fifo
    }

    return super.kill()
  }
}

class SoXPlayer extends Player {
  playFile(file) {
    // SoX's play command is useful for systems that don't have MPV. SoX is
    // much easier to install (and probably more commonly installed, as well).
    // You don't get keyboard controls such as seeking or volume adjusting
    // with SoX, though.

    this.process = spawn('play', [file])

    this.process.stdout.on('data', data => {
      process.stdout.write(data.toString())
    })

    // Most output from SoX is given to stderr, for some reason!
    this.process.stderr.on('data', data => {
      // The status line starts with "In:".
      if (data.toString().trim().startsWith('In:')) {
        if (this.disablePlaybackStatus) {
          return
        }

        const timeRegex = '([0-9]*):([0-9]*):([0-9]*)\.([0-9]*)'
        const match = data.toString().trim().match(new RegExp(
          `^In:([0-9.]+%)\\s*${timeRegex}\\s*\\[${timeRegex}\\]`
        ))

        if (match) {
          const percentStr = match[1]

          // SoX takes a loooooot of math in order to actually figure out the
          // duration, since it outputs the current time and the remaining time
          // (but not the duration).

          const [
            curHour, curMin, curSec, curSecFrac, // ##:##:##.##
            remHour, remMin, remSec, remSecFrac // ##:##:##.##
          ] = match.slice(2).map(n => parseInt(n))

          const duration = Math.round(
            (curHour + remHour) * 3600 +
            (curMin + remMin) * 60 +
            (curSec + remSec) * 1 +
            (curSecFrac + remSecFrac) / 100
          )

          const lenHour = Math.floor(duration / 3600)
          const lenMin = Math.floor((duration - lenHour * 3600) / 60)
          const lenSec = Math.floor(duration - lenHour * 3600 - lenMin * 60)

          this.printStatusLine(getTimeStrings({curHour, curMin, curSec, lenHour, lenMin, lenSec}))
        }
      }
    })

    return new Promise(resolve => {
      this.process.on('close', () => resolve())
    })
  }
}

class DownloadController extends EventEmitter {
  constructor(playlist, converterProgram) {
    super()

    Object.assign(this, {playlist, converterProgram})
  }

  async init() {
    this.converter = await makeConverter(this.converterProgram)
  }

  waitForDownload() {
    // Returns a promise that resolves when a download is
    // completed.  Note that this isn't necessarily the download
    // that was initiated immediately before a call to
    // waitForDownload (if any), since that download may have
    // been canceled (see cancel).  You can also listen for the
    // 'downloaded' event instead.

    return new Promise((resolve, reject) => {
      const onDownloaded = file => { clear(); resolve(file) }
      const onErrored = err => { clear(); reject(err) }

      const clear = () => {
        this.removeListener('downloaded', onDownloaded)
        this.removeListener('errored', onErrored)
      }

      this.once('downloaded', onDownloaded)
      this.once('errored', onErrored)
    })
  }

  async download(downloader, downloaderArg, converterOptions) {
    // Downloads a file.  This doesn't return anything; use
    // waitForDownload to get the result of this.
    // (The reasoning is that it's possible for a download to
    // be canceled and replaced with a new download (see cancel)
    // which would void the result of the old download.)
    // The resulting file is a WAV.

    this.cleanupListeners()

    let canceled = false

    this._handleCanceled = () => {
      canceled = true
      this.cleanupListeners()
    }

    this.once('canceled', this._handleCanceled)

    let downloadFile

    // TODO: Be more specific; 'errored' * 2 could instead be 'downloadErrored' and
    // 'convertErrored'.

    try {
      downloadFile = await downloader(downloaderArg)
    } catch(err) {
      this.emit('errored', 'Download error: ' + err)
      return
    }

    // If this current download has been canceled, we should get rid of the
    // download file (and shouldn't emit a download success).
    if (canceled) {
      await safeUnlink(downloadFile, this.playlist)
      return
    }

    let convertFile

    try {
      convertFile = await this.converter(converterOptions)(downloadFile)
    } catch(err) {
      this.emit('errored', 'Convert error: ' + err)
      return
    } finally {
      // Whether the convertion succeeds or not (hence 'finally'), we should
      // delete the temporary download file.
      await safeUnlink(downloadFile, this.playlist)
    }

    // Again, if canceled, we should delete temporary files and stop.
    if (canceled) {
      await safeUnlink(convertFile, this.playlist)
      return
    }

    this.emit('downloaded', convertFile)
    this.cleanupListeners()
  }

  cleanupListeners() {
    if (this._handleCanceled) {
      this.removeListener('canceled', this._handleCanceled)
    }
  }

  cancel() {
    // Cancels the current download.  This doesn't cancel any
    // waitForDownload promises, though -- you'll need to start
    // a new download to resolve those.

    this.emit('canceled')
    this.cleanupListeners()
  }
}

class PlayController extends EventEmitter {
  constructor({
    player, playlist, historyController, downloadController,
    statusLineTemplate = '%longIndex% (%percentDone%) %timeDone% / %duration%',
    useConverterOptions = true,
    trackDisplayFile = null // File to output current track path to.
  }) {
    super()

    this.player = player
    this.playlist = playlist
    this.historyController = historyController
    this.downloadController = downloadController
    this.useConverterOptions = useConverterOptions
    this.trackDisplayFile = trackDisplayFile

    this.currentTrack = null
    this.nextTrack = null
    this.nextFile = undefined // TODO: Why isn't this null?
    this.stopped = false
    this.shouldMoveNext = true
    this.failedCount = 0
    this.playFailCount = 0

    this.player.on('crashed', () => {
      if (this.currentTrack) {
        console.log('\x1b[31mFailed to play track \x1b[1m' +
          getItemPathString(this.currentTrack) + '\x1b[0m'
        )
      } else {
        console.log('\x1b[31mFailed to play track.\x1b[0m')
      }
      this.playFailCount++

      if (this.playFailCount >= 5) {
        console.error(
          '\x1b[31mFailed to play 5 tracks. Halting, to prevent damage to ' +
          'the computer.\x1b[0m'
        )

        process.exit(1)
        throw new Error('Intentionally halted - failed to play tracks.')
      }
    })

    this.player.on('printStatusLine', playerData => {
      let fullStatusLine = ''

      // ESC[K - clears the line going from the cursor position onwards.
      // This is here to avoid artefacts from a previously printed status line.
      fullStatusLine += '\x1b[K'

      const track = this.currentTrack

      let longIndex = ''
      if (track) {
        if (track.overallTrackIndex || track.groupTrackIndex) {
          longIndex += '('

          addTrackNumberInnards: {
            if (track.overallTrackIndex) {
              const [ cur, len ] = track.overallTrackIndex
              longIndex += `${cur + 1} / ${len}`

              if (track.groupTrackIndex) {
                const [ curGroup, lenGroup ] = track.groupTrackIndex

                // If the overall and group track indexes are equal to each
                // other, there's a fair chance we're just playing a single
                // group; so, we only display one index (and we don't show
                // "[All]"/"[Group]" labels).
                if (curGroup === cur && lenGroup === len) {
                  break addTrackNumberInnards
                } else {
                  longIndex += ' [All]; '
                }
              }
            }

            if (track.groupTrackIndex) {
              const [ cur, len ] = track.groupTrackIndex
              longIndex += `${cur + 1} / ${len}`

              if (track.overallTrackIndex) {
                longIndex += ' [Group]'
              }
            }
          }

          longIndex += ')'
        }
      }

      const groupIndexArr = (track && track.groupTrackIndex) || ['', '']

      fullStatusLine += processTemplateString(statusLineTemplate, Object.assign({
        index: track ? (track.overallTrackIndex[0] + 1) : '',
        trackCount: track ? (track.overallTrackIndex[1]) : '',
        indexGroup: groupIndexArr[0],
        trackCountGroup: groupIndexArr[1],
        longIndex,
        trackName: track.name
      }, playerData))

      // Carriage return - moves the cursor back to the start of the line,
      // so that the next status line is printed on top of this one.
      fullStatusLine += '\r'

      process.stdout.write(fullStatusLine)
    })
  }

  async loopPlay() {
    // Null would imply there's NO up-next track, but really we just haven't
    // set it yet.
    this.nextTrack = undefined

    // Download the very first track.
    this.startNextDownload(this.historyController.getNextTrack())

    await this.waitForDownload()

    while (this.nextTrack && !this.stopped) {
      this.currentTrack = this.nextTrack

      const next = this.nextFile
      this.nextFile = undefined

      // Pre-download the track that's up next.
      this.startNextDownload(this.historyController.timeline[
        this.historyController.timelineIndex + 1
      ])

      if (next) {
        if (this.trackDisplayFile) {
          await writeFile(this.trackDisplayFile,
            getItemPathString(this.currentTrack[sourceSymbol])
          )
        }

        await this.playFile(next)

        // Now that we're done playing the file, we should delete it.. unless
        // it's the file that's coming up! (This would only happen in the case
        // that all temporary files are stored in the same folder, together;
        // indeed an unusual case, but technically possible.)
        if (next !== this.nextFile) {
          await safeUnlink(next, this.playlist)
        }
      }

      if (!this.shouldMoveNext) {
        this.shouldMoveNext = true
      } else {
        this.historyController.timelineIndex++
        this.historyController.fillTimeline()
      }

      await this.waitForDownload()
    }
  }

  waitForDownload() {
    return new Promise(resolve => {
      if (this.isDownloading) {
        this.once('downloaded', () => resolve())
      } else {
        resolve()
      }
    })
  }

  startNextDownload(picked) {
    this.isDownloading = true

    this.nextTrack = picked

    if (!picked) {
      return null
    }

    let downloader

    if (picked.downloader) {
      downloader = downloadersByName[picked.downloader]()

      if (!downloader) {
        console.error(
          `Invalid downloader for track ${picked.name}:`, downloader
        )

        return false
      }
    } else {
      downloader = getDownloaderFor(picked.downloaderArg)
    }

    if (picked.converterOptions && !Array.isArray(picked.converterOptions)) {
      throw new Error("The converterOptions track property must be an array")
    }

    this.downloadController.download(
      downloader, picked.downloaderArg,
      this.useConverterOptions ? picked.converterOptions : undefined
    )

    this.downloadController.waitForDownload()
      .then(file => {
        this.isDownloading = false
        this.nextFile = file
        this.failedCount = 0
        this.emit('downloaded')
      })
      .catch(err => {
        // TODO: Test this!!
        console.warn(
          "\x1b[31mFailed to download (or convert) track \x1b[1m" +
          getItemPathString(this.nextTrack) + "\x1b[0m"
        )
        console.warn(err)

        this.failedCount++

        if (this.failedCount >= 5) {
          console.error(
            "\x1b[31mFailed to download 5 tracks in a row. Halting, to " +
            "prevent damage to the computer.\x1b[0m"
          )

          process.exit(1)
          throw new Error('Intentionally halted - failed to download tracks.')
        }

        // A little bit blecht, but.. this works.
        // "When a track fails, remove it from the timeline, and start
        // downloading whatever track fills its place."
        // Only problem is if a track before the current timeline index fails,
        // maybe? (Since then the timelineIndex value might be messed up?)
        const tl = this.historyController.timeline
        const index = tl.indexOf(picked)
        tl.splice(index, 1)
        this.historyController.fillTimeline()
        this.startNextDownload(tl[index])
      })

    return picked
  }

  playFile(file) {
    return this.player.playFile(file)
  }

  async skip() {
    // TODO: It would be nice if this returned the next track, but that
    // probably isn't possible with the current play/loop-setup.

    if (this.nextTrack !== this.historyController.getNextTrack(false)) {
      this.downloadController.cancel()
      this.startNextDownload(this.historyController.getNextTrack())
      this.shouldMoveNext = false
    }

    await this.player.kill()
    this.currentTrack = null
  }

  async skipBack() {
    // Usually the downloader moves forwards in time (so, the NEXT track will
    // be pre-downloaded). Here, we want to move back, so we need to override
    // the downloader ourselves.

    if (this.nextTrack !== this.historyController.getBackTrack(false)) {
      this.downloadController.cancel()
      this.startNextDownload(this.historyController.getBackTrack())
    }

    this.shouldMoveNext = false
    await this.player.kill()
    this.currentTrack = null
  }

  async skipUpNext() {
    if (this.nextFile) {
      await safeUnlink(this.nextFile, this.playlist)
    }

    const tl = this.historyController.timeline
    tl.splice(this.historyController.timelineIndex + 1, 1)
    this.historyController.fillTimeline()

    this.downloadController.cancel()
    return this.startNextDownload(tl[this.historyController.timelineIndex + 1])
  }

  async stop() {
    // TODO: How to bork download-controller files?? Wait for it to emit a
    // 'cleaned up' event? This whole program being split-up is a Baaaaad idea.
    this.downloadController.cancel()
    await this.player.kill()
    this.currentTrack = null
    this.stopped = true
  }

  logTrackInfo(upNextTrackCount = 3, previousTrackCount = undefined) {
    if (typeof previousTrackCount === 'undefined') {
      previousTrackCount = upNextTrackCount
    }

    const getColorMessage = t => {
      if (!t) return '\x1b[2m(No track)\x1b[0m'

      const path = getItemPathString(t[parentSymbol])

      return (
        `\x1b[1m${t.name} \x1b[0m@ ${path} \x1b[2m(${t.downloaderArg})\x1b[0m`
      )
    }

    const getCleanMessage = t => {
      if (!t) return '(No track)'

      const path = getItemPathString(t[parentSymbol])

      return `${t.name} @ ${path}`
    }

    const hc = this.historyController
    const tl = hc.timeline
    const tlI = hc.timelineIndex

    for (let i = Math.max(0, tlI - (previousTrackCount - 1)); i < tlI; i++) {
      console.log(`\x1b[2m(Prev) ${getCleanMessage(tl[i])}\x1b[0m`)
    }

    console.log(`\x1b[1m(Curr) \x1b[1m${getColorMessage(tl[tlI])}\x1b[0m`)

    for (let i = tlI + 1; i < Math.min(tlI + upNextTrackCount, tl.length); i++) {
      console.log(`(Next) ${getCleanMessage(tl[i])}`)
    }
  }
}

module.exports = async function startLoopPlay(
  playlist, {
    pickerOptions, playerCommand, converterCommand,
    useConverterOptions = true,
    disablePlaybackStatus = false,
    startTrack = null,
    trackDisplayFile = null
  }
) {
  // Looping play function. Takes a playlist and an object containing general
  // options (picker options, player command, and disable-playback-status).
  // Stops when the history controller returns null.

  let player
  if (playerCommand === 'sox' || playerCommand === 'play') {
    player = new SoXPlayer()
  } else if (playerCommand === 'mpv') {
    player = new ControllableMPVPlayer()
  } else if (
    playerCommand === 'mpv-nocontrolls' ||
    playerCommand === 'mpv-windows' ||
    playerCommand === 'mpv-nofifo'
  ) {
    player = new MPVPlayer()
  } else {
    if (playerCommand) {
      console.warn('Invalid player command given?', playerCommand)
    } else {
      console.warn('No player command given?')
    }

    return Promise.resolve()
  }

  Object.assign(player, {disablePlaybackStatus})

  const downloadController = new DownloadController(
    playlist, converterCommand
  )
  await downloadController.init()

  const historyController = new HistoryController(
    playlist, generalPicker, pickerOptions
  )

  if (startTrack) {
    historyController.timeline.push(startTrack)
  }

  const playController = new PlayController({
    player, playlist, historyController, downloadController,
    trackDisplayFile
  })

  Object.assign(playController, {useConverterOptions})

  const promise = playController.loopPlay()

  return {
    promise,
    playController,
    downloadController,
    player
  }
}