50 lines
1.3 KiB
JavaScript
50 lines
1.3 KiB
JavaScript
const BaseEngine = require('../BaseEngine');
|
|
const fetch = require('node-fetch');
|
|
const querystring = require('querystring');
|
|
|
|
module.exports = class extends BaseEngine {
|
|
constructor() {
|
|
super('watson',"IBM Watson TTS", "ogg");
|
|
this.voices = {};
|
|
this.populateVoiceList();
|
|
}
|
|
async populateVoiceList() {
|
|
const url = process.env.watsonURL + "/v1/voices";
|
|
const authorization = this.IBMAuthString();
|
|
const opts = {
|
|
method: "get",
|
|
headers: {
|
|
'Authorization': authorization
|
|
},
|
|
}
|
|
const res = await fetch(url, opts);
|
|
const voices = await res.json();
|
|
voices.voices.forEach((i) => {
|
|
let voiceName = i.description.substring(0, i.description.indexOf(':')).toLowerCase();
|
|
this.voices[voiceName] = i.name;
|
|
});
|
|
}
|
|
getDefaultVoice() {
|
|
return 'Michael';
|
|
}
|
|
IBMAuthString() {
|
|
let buff = new Buffer('apikey:' + process.env.watsonAPIKey);
|
|
let b64auth = buff.toString('base64');
|
|
return 'Basic ' + b64auth;
|
|
}
|
|
async getSpeech(text, voice = this.getSpeechVoice(), params = {}) {
|
|
const url = process.env.watsonURL + "/v1/synthesize?voice=" + this.getInternalVoiceName(voice);
|
|
const authorization = this.IBMAuthString();
|
|
const opts = {
|
|
method: "post",
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
'Authorization': authorization
|
|
},
|
|
body: JSON.stringify({
|
|
text: text
|
|
})
|
|
};
|
|
return fetch(url, opts);
|
|
}
|
|
}; |