@range3/holo/src/cli.js

Hololive Tools for CLI

'use strict'

const commander = require('commander')
const chalk = require('chalk')
const { DateTime } = require('luxon')
const terminalImage = require('./terminal-image')
const supportsSixel = require('supports-sixel')
const got = require('got')
const ProgramSchedule = require('./program-schedule')

const now = DateTime.local()
const groupIds = {
  all: ProgramSchedule.GROUP_ALL,
  live: ProgramSchedule.GROUP_HOLOLIVE,
  stars: ProgramSchedule.GROUP_HOLOSTARS,
  innk: ProgramSchedule.GROUP_INNK,
  id: ProgramSchedule.GROUP_HOLOID,
  en: ProgramSchedule.GROUP_HOLOEN,
}

commander
  .name('holo')
  .option('-z, --zone <value>', 'TimeZone', process.env.TZ || 'Asia/Tokyo')
  .option('-g, --group <type...>', `Group Names (${Object.keys(groupIds).join(' | ')})`, ['live'])
  .option('-n, --num <number>', 'Print n programs', Number, 100)
  .option('-b, --before <number>', 'Print programs that started later than N minutes ago', Number, 10)
  .option('-a, --after <number>', 'Print programs whose broadcast start is earlier than N minutes later', Number, 180)
  .option('--no-current', 'Disable printing current programs on air')
  .option('--no-thumbnail', 'Disable printing thumbnails')
  .option('-H, --height <number>', 'Height of thumbnails', Number, 10)
  .parse(process.argv)

if (commander.group.some(g => !groupIds[g])) {
  commander.help()
}
const groups = commander.group.includes('all')
  ? [groupIds.all]
  : Array.from(new Set(commander.group.map(g => groupIds[g])))

;(async () => {
  try {
    const sixelIsSupported = await supportsSixel()

    const schedules = groups.map(group => new ProgramSchedule({
      group,
      zone: commander.zone,
    }))
    await Promise.all(schedules.map(schedule => schedule.fetch()))

    const programs = await Promise.all(
      schedules
        .map(schedule => schedule.toArray())
        .flat()
        .filter(program =>
          (program.isOnAir && commander.current) ||
          (program.dateTime <= now.plus({ minutes: commander.after }) &&
          program.dateTime >= now.minus({ minutes: commander.before })))
        .sort((a, b) => a.dateTime - b.dateTime)
        .slice(0, commander.num)
        .map(async program => {
          let thumbnailBuffer = null
          try {
            thumbnailBuffer = commander.thumbnail ? await got(program.thumbnail).buffer() : null
          } catch (e) {
            if (e instanceof got.HTTPError) {
              thumbnailBuffer = null
            } else {
              throw e
            }
          }
          return {
            thumbnailBuffer,
            ...program,
          }
        }))
    for await (const program of programs) {
      const onAir = program.isOnAir ? ` ${chalk.white.bgRed('ON AIR')}` : ''
      if (commander.thumbnail) {
        try {
          process.stdout.write(
            await terminalImage.buffer(
              program.thumbnailBuffer, { height: commander.height, sixel: sixelIsSupported }))
        } catch (err) {
          console.error(err)
        }
      }
      process.stdout.write(
          `${program.dateTime.toLocaleString(DateTime.DATETIME_SHORT)}${onAir} ${program.name} ${program.link}`,
      )
      process.stdout.write('\n')
    }
  } catch (error) {
    console.error(error)
  }
})()