qiao-tts/index.js

nodejs tts tool

'use strict';

var fs = require('fs');
var qiaoEncode = require('qiao-encode');
var qiaoAjax = require('qiao-ajax');
var qiao_log_js = require('qiao.log.js');
var qiaoFile = require('qiao-file');

// fs
const logger$1 = qiao_log_js.Logger('qiao-tts');

/**
 * moyinTTS
 * https://openapi.moyin.com/index
 * @param {*} options
 * @returns
 */
const moyinTTS = (options) => {
  const methodName = 'moyinTTS';

  // const
  const appKey = options.appKey;
  const appSecret = options.appSecret;
  const txt = options.txt;
  const audioPath = options.audioPath;
  logger$1.info(methodName, 'appKey', appKey);
  logger$1.info(methodName, 'appSecret', appSecret);
  logger$1.info(methodName, 'txt', txt);
  logger$1.info(methodName, 'audioPath', audioPath);

  // check
  if (!appKey) {
    logger$1.info(methodName, 'need appKey');
    return;
  }
  if (!appSecret) {
    logger$1.info(methodName, 'need appSecret');
    return;
  }
  if (!txt) {
    logger$1.info(methodName, 'need txt');
    return;
  }
  if (!audioPath) {
    logger$1.info(methodName, 'need audioPath');
    return;
  }

  // config
  const timestamp = Math.floor(Date.now() / 1000);
  const signature = qiaoEncode.md5(`${appKey}+${appSecret}+${timestamp}`, 'hex');
  const config = {
    text: txt,
    appkey: appKey,
    signature: signature,
    timestamp: timestamp,
    speaker: options.speaker || 'cissy_meet',
    audio_type: options.audio_type || 'mp3',
    // 发音人合成的语速，支持小数点后两位
    // 默认值：1.0
    // 可选值：0.5-2.0
    speed: options.speed || 1.0,
  };
  logger$1.info(methodName, 'config', config);

  // req
  return new Promise((resolve, reject) => {
    qiaoAjax
      .post('https://open.mobvoi.com/api/tts/v1', {
        data: config,
        responseType: 'stream',
      })
      .then((response) => {
        // not 200
        if (response.status !== 200) return reject(`response.status not 200, is ${response.status}`);

        // request error
        if (response.headers['content-type'] === 'application/json') {
          let jsonData = '';
          response.data.on('data', (chunk) => {
            jsonData += chunk.toString();
          });
          response.data.on('end', () => {
            try {
              const data = JSON.parse(jsonData);
              return reject(`request error, code is ${data.errorMessage.code}, msg is ${data.errorMessage.desc}`);
            } catch (error) {
              return reject(error);
            }
          });
        }

        // request success
        const writer = fs.createWriteStream(audioPath);
        response.data.pipe(writer);
        writer.on('finish', resolve);
        writer.on('error', reject);
      })
      .catch((err) => {
        return reject(err);
      });
  });
};

// qiao
const logger = qiao_log_js.Logger('qiao-tts');

/**
 * huoshanTTS
 * https://www.volcengine.com/product/tts
 * @param {*} options
 * @returns
 */
const huoshanTTS = async (options) => {
  const methodName = 'huoshanTTS';

  // const
  const appid = options.appid;
  const token = options.token;
  const txt = options.txt;
  const speaker = options.speaker;
  const audioPath = options.audioPath;
  logger.info(methodName, 'token', token);
  logger.info(methodName, 'txt', txt);
  logger.info(methodName, 'audioPath', audioPath);

  // check
  if (!appid) {
    logger.info(methodName, 'need appid');
    return;
  }
  if (!token) {
    logger.info(methodName, 'need token');
    return;
  }
  if (!txt) {
    logger.info(methodName, 'need txt');
    return;
  }
  if (!speaker) {
    logger.info(methodName, 'need speaker');
    return;
  }
  if (!audioPath) {
    logger.info(methodName, 'need audioPath');
    return;
  }

  // const
  const url = 'https://openspeech.bytedance.com/api/v1/tts';
  const uuidname = qiaoEncode.uuid();

  // config
  const config = {
    headers: {
      Authorization: `Bearer;${token}`,
    },
    data: {
      app: {
        appid: appid,
        token: token,
        cluster: 'volcano_tts',
      },
      user: {
        uid: 'xiaolouai',
      },
      audio: {
        voice_type: speaker,
        encoding: 'mp3',
        speed_ratio: 1,
      },
      request: {
        reqid: uuidname,
        text: txt,
        operation: 'query',
      },
    },
  };
  logger.info(methodName, 'config', config);

  // req
  const res = await qiaoAjax.post(url, config);
  if (res.status !== 200) {
    logger.error(methodName, 'status not 200', res.status);
    return;
  }
  if (res.data.code !== 3000) {
    logger.error(methodName, 'code not 3000', res.data);
    return;
  }

  // write
  const audioBuffer = Buffer.from(res.data.data, 'base64');
  const writeFileRes = await qiaoFile.writeFile(audioPath, audioBuffer);
  if (!writeFileRes) {
    logger.error(methodName, 'write file error');
    return;
  }

  // r
  return true;
};

exports.huoshanTTS = huoshanTTS;
exports.moyinTTS = moyinTTS;