stack-chan

// Imports the Google Cloud client library

import fs from 'fs'
import { writeFile } from 'fs/promises'
import fetch from 'node-fetch'
import path from 'path'
import { hideBin } from 'yargs/helpers'
import yargs from 'yargs/yargs'

/**
 * get Speech data from VoiceVox server
 * @param {*} speechText
 * @param {*} options
 * @returns Promise<Blob>
 */
async function getSpeech(speechText, options) {
  // windows resolve localhost returens IPv6 address(::1)
  // but voicevox only listening IPv4 address(127.0.0.1)
  //const host = options?.host ?? 'localhost'
  const host = options?.host ?? '127.0.0.1'
  const port = options?.port ?? 50021
  const speakerId = options?.speakerId ?? 1
  const speed = options?.speed ?? 1
  const pitch = options?.pitch ?? 0
  const intonation = options?.intonation ?? 1
  const sampleRate = options?.sampleRate ?? 11025
  const speakerName = await getSpeakerName(speakerId, options)
  console.log(`${speakerName}  Text:${speechText}`)
  if (speechText?.length === 0) {
    throw new Error('speechText required')
  }
  console.log(`http://${host}:${port}/audio_query?text=${speechText}&speaker=${speakerId}`)
  return fetch(encodeURI(`http://${host}:${port}/audio_query?text=${speechText}&speaker=${speakerId}`), {
    method: 'POST',
    cache: 'no-cache',
    headers: {
      'Content-Type': 'application/x-www-form-urlencoded',
    },
  })
    .then((response) => response.json())
    .then((json) => {
      // change voice parameter
      json.speedScale = speed
      json.pitchScale = pitch
      json.intonationScale = intonation
      json.outputSamplingRate = sampleRate
      return fetch(encodeURI(`http://${host}:${port}/synthesis?speaker=${speakerId}`), {
        method: 'POST',
        cache: 'no-cache',
        headers: {
          'Content-Type': 'application/json',
        },
        body: JSON.stringify(json),
      })
    })
    .then((res) => {
      return res.arrayBuffer()
    })
}

async function getSpeakerName(speakerId, options) {
  const host = options?.host ?? '127.0.0.1'
  const port = options?.port ?? 50021
  // get speaker name
  return fetch(encodeURI(`http://${host}:${port}/speakers`), {
    method: 'GET',
    cache: 'no-cache',
    headers: {
      'Content-Type': 'application/x-www-form-urlencoded',
    },
  })
    .then((response) => response.json())
    .then((speakers) => {
      let result = ''
      for (const speaker of speakers) {
        for (const style of speaker.styles) {
          if (typeof speakerId === 'number') {
            if (speakerId === style.id) {
              result = `${speaker.name}(${style.name}) SpeakerId:${style.id}`
            }
          } else {
            result += `${speaker.name}(${style.name}) SpeakerId:${style.id}\n`
          }
        }
      }
      return result ? result : 'Speaker Id Not Found'
    })
}

async function generateSpeech(options) {
  // import speechs list
  const { speeches } = await import(`file://${options.input}`)
  // The text to synthesize
  for (const [key, text] of Object.entries(speeches)) {
    const speech = await getSpeech(text, options)
    console.log(`write ${key}.wav`)
    writeFile(`${options.output}/${key}.wav`, Buffer.from(speech))
    console.log('ended')
  }
}

async function main() {
  const options = []
  const argv = yargs(hideBin(process.argv)).argv
  const cwd = process.cwd() // save CurrentDiredtory (firmware Directory)
  process.chdir(process.env.INIT_CWD) // change CurrntDirectory to npm run was executed direcory

  options.host = argv.host
  options.port = argv.port
  options.speakerId = argv.speaker
  options.sampleRate = argv.sample
  options.speed = argv.speed
  options.pitch = argv.pitch
  options.intnation = argv.intnation

  if (argv.list) {
    // --list specified only show charactor lists
    console.log(await getSpeakerName(argv.list, options))
    process.exit(0)
  }

  const input = argv.input
  if (input) {
    console.log(`cwd   :${process.cwd()}`)
    console.log(`input :${input}`)
    options.input = path.resolve(input)
    if (input !== options.input) {
      console.log(`      (${options.input})`)
    }
  } else {
    if (fs.existsSync('./speeches.js')) {
      console.log(`cwd   :${process.cwd()}`)
      console.log('input :./speeches.js')
      options.input = path.resolve('./speeches.js')
    } else {
      process.chdir(cwd) // change CurrentDirectory to saved(firmware Directory)
      console.log(`cwd   :${process.cwd()}`)
      console.log('input :./stackchan/assets/sounds/speeches_ja.js')
      options.input = path.resolve('./stackchan/assets/sounds/speeches_ja.js')
    }
    console.log(`      (${options.input})`)
  }

  const output = argv.output
  if (output) {
    options.output = path.resolve(output)
    console.log(`output:${output}`)
    if (output !== options.output) {
      console.log(`      (${options.output})`)
    }
  } else {
    if (fs.existsSync(`${path.dirname(options.input)}/assets`)) {
      console.log('output:./assets')
      options.output = `${path.dirname(options.input)}/assets`
      console.log(`      (${options.output})`)
    } else {
      options.output = path.dirname(options.input)
      console.log(`output:${options.output}`)
    }
  }
  await generateSpeech(options)
}

main()