Compare commits
No commits in common. "master" and "watson" have entirely different histories.
|
|
@ -1,6 +1,4 @@
|
||||||
node_modules
|
node_modules
|
||||||
voice_tmp
|
voice_tmp
|
||||||
.env
|
.env
|
||||||
*.db
|
*.db
|
||||||
.DS_Store
|
|
||||||
gkey.json
|
|
||||||
|
|
@ -1,31 +0,0 @@
|
||||||
const Voice = require("@discordjs/voice");
|
|
||||||
|
|
||||||
module.exports = class AudioQueue {
|
|
||||||
constructor(connection, api) {
|
|
||||||
this.connection = connection;
|
|
||||||
this.api = api;
|
|
||||||
this.queue = [];
|
|
||||||
this.current = undefined;
|
|
||||||
this.api.player.on(Voice.AudioPlayerStatus.Idle, this.handleStop.bind(this));
|
|
||||||
}
|
|
||||||
playNext() {
|
|
||||||
if (this.queue.length == 0) {
|
|
||||||
this.current = undefined;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
this.current = this.api.play(this.queue[0]);
|
|
||||||
}
|
|
||||||
handleStop(current) {
|
|
||||||
this.queue.shift();
|
|
||||||
this.playNext();
|
|
||||||
}
|
|
||||||
add(element) {
|
|
||||||
this.queue.push(element);
|
|
||||||
if (this.queue.length == 1) this.playNext();
|
|
||||||
}
|
|
||||||
flush() {
|
|
||||||
this.current.setVolume(0);
|
|
||||||
this.queue=[];
|
|
||||||
this.playNext();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,5 +0,0 @@
|
||||||
FROM node:18-alpine
|
|
||||||
COPY . .
|
|
||||||
RUN apk add ffmpeg
|
|
||||||
RUN npm install
|
|
||||||
ENTRYPOINT ["node", "index.js"]
|
|
||||||
62
adapter.js
62
adapter.js
|
|
@ -1,62 +0,0 @@
|
||||||
const { Snowflake, Client, Guild, VoiceBasedChannel, Events, Status, GatewayDispatchEvents } = require("discord.js");
|
|
||||||
|
|
||||||
const adapters = new Map();
|
|
||||||
const trackedClients = new Set();
|
|
||||||
const trackedShards = new Map();
|
|
||||||
|
|
||||||
|
|
||||||
function trackClient(client) {
|
|
||||||
if (trackedClients.has(client)) return;
|
|
||||||
trackedClients.add(client);
|
|
||||||
client.ws.on(GatewayDispatchEvents.VoiceServerUpdate, (payload) => {
|
|
||||||
adapters.get(payload.guild_id)?.onVoiceServerUpdate(payload);
|
|
||||||
});
|
|
||||||
client.ws.on(GatewayDispatchEvents.VoiceStateUpdate, (payload) => {
|
|
||||||
if (payload.guild_id && payload.session_id && payload.user_id === client.user?.id) {
|
|
||||||
adapters.get(payload.guild_id)?.onVoiceStateUpdate(payload);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
client.on(Events.ShardDisconnect, (_, shardId) => {
|
|
||||||
const guilds = trackedShards.get(shardId);
|
|
||||||
if (guilds) {
|
|
||||||
for (const guildID of guilds.values()) {
|
|
||||||
adapters.get(guildID)?.destroy();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
trackedShards.delete(shardId);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
function trackGuild(guild) {
|
|
||||||
let guilds = trackedShards.get(guild.shardId);
|
|
||||||
if (!guilds) {
|
|
||||||
guilds = new Set();
|
|
||||||
trackedShards.set(guild.shardId, guilds);
|
|
||||||
}
|
|
||||||
guilds.add(guild.id);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
module.exports = function (channel) {
|
|
||||||
return (methods) => {
|
|
||||||
adapters.set(channel.guild.id, methods);
|
|
||||||
trackClient(channel.client);
|
|
||||||
trackGuild(channel.guild);
|
|
||||||
return {
|
|
||||||
sendPayload(data) {
|
|
||||||
console.log(channel.guild.shard.status);
|
|
||||||
// if (channel.guild.shard.status === Status.READY) {
|
|
||||||
console.log("Sending shard data");
|
|
||||||
channel.guild.shard.send(data);
|
|
||||||
return true;
|
|
||||||
// }
|
|
||||||
console.log("Unable to send channel payload");
|
|
||||||
return false;
|
|
||||||
},
|
|
||||||
destroy() {
|
|
||||||
console.log("Destroying adapter");
|
|
||||||
return adapters.delete(channel.guild.id);
|
|
||||||
},
|
|
||||||
};
|
|
||||||
};
|
|
||||||
}
|
|
||||||
13
example.env
13
example.env
|
|
@ -1,13 +0,0 @@
|
||||||
TOKEN=DISCORD_BOT_TOKEN_HERE
|
|
||||||
GUILD=GUILD_ID_HERE
|
|
||||||
CHANNEL=VOICE_CHANNEL_ID_HERE
|
|
||||||
STRING_SET=en
|
|
||||||
VOICE_TMP_PATH=./voice_tmp/
|
|
||||||
DB_FILE=DATABASE_PATH_HERE
|
|
||||||
PREFIX=+
|
|
||||||
ANNOUNCEMENT_ENGINE=espeak
|
|
||||||
ANNOUNCEMENT_VOICE=en
|
|
||||||
watsonURL=WATSON_URL_HERE
|
|
||||||
watsonAPIKey=WATSON_API_KEY_HERE
|
|
||||||
TTS_CHANNEL=CANTTALK_TEXT_CHANNEL_ID_HERE
|
|
||||||
GOOGLE_APPLICATION_CREDENTIALS=GOOGLE_CLOUD_KEY_HERE
|
|
||||||
119
index.js
119
index.js
|
|
@ -1,87 +1,43 @@
|
||||||
const Discord = require('discord.js');
|
const Discord = require('discord.js');
|
||||||
const Voice = require("@discordjs/voice");
|
|
||||||
const adapterCreator = require("./adapter");
|
|
||||||
require('dotenv').config();
|
require('dotenv').config();
|
||||||
const fetch = require('node-fetch');
|
const fetch = require('node-fetch');
|
||||||
const fs = require('fs');
|
const fs = require('fs');
|
||||||
const sha1 = require('sha1');
|
const sha1 = require('sha1');
|
||||||
const sqlite3 = require('sqlite3');
|
const sqlite = require('sqlite3');
|
||||||
const { open } = require('sqlite')
|
|
||||||
|
|
||||||
let joinedVoiceChannels = [];
|
let joinedVoiceChannels = [];
|
||||||
let joinedVoiceChannelConnections = new Map();
|
|
||||||
|
|
||||||
let modules = [];
|
let modules = [];
|
||||||
|
|
||||||
let commandHandlers = new Map();
|
let commandHandlers = new Map();
|
||||||
const player = Voice.createAudioPlayer();
|
|
||||||
|
|
||||||
const rest = new Discord.REST({ version: '10' }).setToken(process.env["TOKEN"]);
|
const bot = new Discord.Client();
|
||||||
const bot = new Discord.Client({
|
|
||||||
intents: [
|
|
||||||
Discord.GatewayIntentBits.GuildMembers,
|
|
||||||
Discord.GatewayIntentBits.GuildMessageReactions,
|
|
||||||
Discord.GatewayIntentBits.GuildMessages,
|
|
||||||
Discord.GatewayIntentBits.GuildPresences,
|
|
||||||
Discord.GatewayIntentBits.GuildVoiceStates,
|
|
||||||
Discord.GatewayIntentBits.Guilds,
|
|
||||||
Discord.GatewayIntentBits.MessageContent
|
|
||||||
]
|
|
||||||
});
|
|
||||||
|
|
||||||
async function initDB() {
|
const db = new sqlite.Database(process.env.DB_FILE);
|
||||||
console.log(__dirname);
|
|
||||||
api.db = await open({
|
|
||||||
filename: process.env["DB_FILE"],
|
|
||||||
driver: sqlite3.Database
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
const api = {
|
const api = {
|
||||||
player: player,
|
db: db,
|
||||||
db: undefined,
|
|
||||||
queue: undefined,
|
|
||||||
strings: require('./strings/' + process.env.STRING_SET + '.json'),
|
|
||||||
ttsEngines: (() => {
|
ttsEngines: (() => {
|
||||||
let engines = {};
|
let engines={};
|
||||||
console.log(`Registering TTS engines...`);
|
console.log(`Registering TTS engines...`);
|
||||||
const engineDirectories = fs.readdirSync('./tts');
|
const engineDirectories = fs.readdirSync('./tts');
|
||||||
engineDirectories.forEach((dir) => {
|
engineDirectories.forEach((dir) => {
|
||||||
if (dir.startsWith('.')) return;
|
if(dir.startsWith('.')) return;
|
||||||
eng = require(`./tts/${dir}/index.js`);
|
eng=require(`./tts/${dir}/index.js`);
|
||||||
engines[dir] = new eng;
|
engines[dir]=new eng;
|
||||||
console.log(`Loading ./tts/${dir}/index.js`)
|
console.log(`Loading ./tts/${dir}/index.js`)
|
||||||
})
|
})
|
||||||
return engines;
|
return engines;
|
||||||
})(),
|
})(),
|
||||||
announcementVoice: process.env.ANNOUNCEMENT_VOICE,
|
|
||||||
announcementEngine: undefined,
|
|
||||||
|
|
||||||
play: (file) => {
|
|
||||||
return player.play(Voice.createAudioResource(file));
|
|
||||||
},
|
|
||||||
respond: (message, text, voiceText) => {
|
|
||||||
let toSend = message.member.displayName + ", " + (voiceText ? voiceText : text);
|
|
||||||
if (message.member.voice.channel) {
|
|
||||||
api.queue.add(__dirname + "/sysmsg.wav");
|
|
||||||
api.speak(message.member.voice.channel, toSend);
|
|
||||||
} else {
|
|
||||||
message.reply(text);
|
|
||||||
}
|
|
||||||
},
|
|
||||||
|
|
||||||
getActiveVoiceChannel: () => joinedVoiceChannels[0],
|
|
||||||
|
|
||||||
isInVoiceChannel: (channel) => {
|
isInVoiceChannel: (channel) => {
|
||||||
return joinedVoiceChannels.includes(channel);
|
return joinedVoiceChannels.includes(channel);
|
||||||
},
|
},
|
||||||
|
|
||||||
getConnectionForVoiceChannel: (channel) => {
|
getConnectionForVoiceChannel: (channel) => {
|
||||||
return joinedVoiceChannelConnections.get(channel);
|
return bot.voice.connections.find((conn) => conn.channel === channel);
|
||||||
},
|
},
|
||||||
|
|
||||||
generateVoice: async (string, engine, voice, params) => {
|
generateVoice: async (string, engine, voice, params) => {
|
||||||
const hash = sha1(voice + string);
|
const hash = sha1(voice+string);
|
||||||
const filepath = process.env.VOICE_TMP_PATH + hash + '.' + engine.fileExtension;
|
const filepath = process.env.VOICE_TMP_PATH + hash + '.' + engine.fileExtension;
|
||||||
if (!fs.existsSync(filepath)) {
|
if (!fs.existsSync(filepath)) {
|
||||||
await engine.getSpeechFile(string, filepath, voice, params);
|
await engine.getSpeechFile(string, filepath, voice, params);
|
||||||
|
|
@ -91,30 +47,22 @@ const api = {
|
||||||
|
|
||||||
joinChannel: async (channel) => {
|
joinChannel: async (channel) => {
|
||||||
if (!api.isInVoiceChannel(channel)) {
|
if (!api.isInVoiceChannel(channel)) {
|
||||||
|
const res = await channel.join();
|
||||||
const res = Voice.joinVoiceChannel({
|
|
||||||
channelId: channel.id,
|
|
||||||
guildId: channel.guild.id,
|
|
||||||
adapterCreator: adapterCreator(channel)
|
|
||||||
});
|
|
||||||
res.subscribe(player);
|
|
||||||
joinedVoiceChannels.push(channel);
|
joinedVoiceChannels.push(channel);
|
||||||
joinedVoiceChannelConnections.set(channel, res);
|
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|
||||||
leaveChannel: async (channel) => {
|
leaveChannel: async (channel) => {
|
||||||
if (joinedVoiceChannels.includes(channel)) {
|
if (joinedVoiceChannels.includes(channel)) {
|
||||||
let con = joinedVoiceChannelConnections.get(channel);
|
|
||||||
joinedVoiceChannels = joinedVoiceChannels.filter((chan) => chan !== channel);
|
joinedVoiceChannels = joinedVoiceChannels.filter((chan) => chan !== channel);
|
||||||
con.disconnect();
|
await channel.leave();
|
||||||
joinedVoiceChannelConnections.delete(channel);
|
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|
||||||
speak: async (channel, message, engine = api.announcementEngine, voice = api.announcementVoice, params = {}) => {
|
speak: async (channel, message, engine=api.ttsEngines.gtranslate, voice='en-us', params={}) => {
|
||||||
|
const conn = api.getConnectionForVoiceChannel(channel);
|
||||||
const filepath = await api.generateVoice(message, engine, voice, params);
|
const filepath = await api.generateVoice(message, engine, voice, params);
|
||||||
api.queue.add(filepath);
|
if (conn) conn.play(filepath);
|
||||||
},
|
},
|
||||||
|
|
||||||
registerCommand: async (commandString, commandFunc) => {
|
registerCommand: async (commandString, commandFunc) => {
|
||||||
|
|
@ -123,36 +71,27 @@ const api = {
|
||||||
}
|
}
|
||||||
|
|
||||||
function registerModules() {
|
function registerModules() {
|
||||||
console.log(`Registering modules...`);
|
console.log(`Registering modules...`);
|
||||||
const moduleDirectories = fs.readdirSync('./modules');
|
const moduleDirectories = fs.readdirSync('./modules');
|
||||||
moduleDirectories.forEach((dir) => {
|
moduleDirectories.forEach((dir) => {
|
||||||
if (dir.startsWith('.')) return;
|
if(dir.startsWith('.')) return;
|
||||||
modules.push(require(`./modules/${dir}`));
|
modules.push(require(`./modules/${dir}/index.js`));
|
||||||
console.log(`Loading ./modules/${dir}/index.js`)
|
console.log(`Loading ./modules/${dir}/index.js`)
|
||||||
})
|
})
|
||||||
modules.forEach((mod) => mod(bot, api));
|
modules.forEach((mod) => mod(bot, api));
|
||||||
}
|
}
|
||||||
|
|
||||||
function handleMessage(message) {
|
function handleMessage(message) {
|
||||||
console.log(`I got message`);
|
|
||||||
if (message.content.startsWith(process.env.PREFIX)) {
|
if (message.content.startsWith(process.env.PREFIX)) {
|
||||||
const args = message.content.split(" ");
|
const args = message.contents.split(" ");
|
||||||
const command = args[0].substr(1, args[0].length);
|
const command = args[0].substr(1, args[0].length);
|
||||||
const execution = commandHandlers.get(command);
|
const execution = commandHandlers.get(command);
|
||||||
if (command) {
|
if (command) {
|
||||||
if (execution) execution(args, message);
|
command(args, message);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
registerModules();
|
||||||
api.announcementEngine = api.ttsEngines[process.env.ANNOUNCEMENT_ENGINE];
|
|
||||||
|
|
||||||
async function start() {
|
|
||||||
await initDB();
|
|
||||||
registerModules();
|
|
||||||
}
|
|
||||||
bot.login(process.env.TOKEN);
|
bot.login(process.env.TOKEN);
|
||||||
bot.on('messageCreate', handleMessage);
|
bot.on('message', handleMessage);
|
||||||
|
|
||||||
start();
|
|
||||||
Binary file not shown.
|
|
@ -1,49 +0,0 @@
|
||||||
const printf=require('printf');
|
|
||||||
const fs = require('fs');
|
|
||||||
const path = require('path');
|
|
||||||
|
|
||||||
module.exports = async (bot, api) => {
|
|
||||||
bot.on('messageCreate', async (message) => {
|
|
||||||
if (!message.content.startsWith(process.env.PREFIX)) {
|
|
||||||
if (message.channel.id == process.env.TTS_CHANNEL) {
|
|
||||||
let chan=message.member.voice.channel;
|
|
||||||
let userRow = await api.db.get('select * from TTSPreferences where user_id=?', message.author.id);
|
|
||||||
if (!userRow) {
|
|
||||||
await api.db.run('insert into TTSPreferences (user_id,engine,voice) values (?,?,?)', [message.author.id, api.announcementEngine.shortName, api.announcementVoice]);
|
|
||||||
userRow = await api.db.get('select * from TTSPreferences where user_id=?', message.author.id);
|
|
||||||
}
|
|
||||||
if (api.ttsEngines[userRow.engine]) {
|
|
||||||
api.speak(chan,message.content, api.ttsEngines[userRow.engine], userRow.voice)
|
|
||||||
} else {
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
api.registerCommand('myvoice', async (args, message) => {
|
|
||||||
let userEngine, userVoice;
|
|
||||||
if (args.length > 3) {
|
|
||||||
return api.respond(message, printf(api.strings.TOO_MANY_ARGUMENTS));
|
|
||||||
}
|
|
||||||
if (api.ttsEngines[args[1]]) {
|
|
||||||
userEngine = args[1];
|
|
||||||
if (api.ttsEngines[userEngine].validateVoice(args[2].toLowerCase())) {
|
|
||||||
userVoice = args[2].toLowerCase();
|
|
||||||
api.respond(message, printf(api.strings.USER_VOICE_CHANGED, userVoice, api.ttsEngines[userEngine].longName));
|
|
||||||
} else {
|
|
||||||
userVoice = api.ttsEngines[userEngine].getDefaultVoice();
|
|
||||||
api.respond(message, printf(api.strings.INVALID_VOICE, userVoice, api.ttsEngines[userEngine].longName));
|
|
||||||
}
|
|
||||||
await api.db.run('update TTSPreferences set engine=?, voice=? where user_id=?', userEngine, userVoice, message.author.id);
|
|
||||||
} else {
|
|
||||||
api.respond(message, printf(api.strings.INVALID_ENGINE, args[1]));
|
|
||||||
}
|
|
||||||
});
|
|
||||||
api.registerCommand('random', async (args, message) => {
|
|
||||||
const files = fs.readdirSync(process.env["VOICE_TMP_PATH"]);
|
|
||||||
const rnd = files[Math.floor(Math.random()*files.length)];
|
|
||||||
console.log(rnd);
|
|
||||||
api.queue.add(__dirname + "/../../sysmsg.wav");
|
|
||||||
api.queue.add(process.env["VOICE_TMP_PATH"] + "/" + rnd);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
@ -1,34 +1,25 @@
|
||||||
const printf = require('printf');
|
module.exports = function(bot, api) {
|
||||||
const AudioQueue = require('../../AudioQueue.js')
|
bot.on('voiceStateUpdate', async (oldState, newState) => {
|
||||||
|
if (newState.member.user.bot) return;
|
||||||
module.exports = function (bot, api) {
|
if (oldState.channel && newState.channel) return;
|
||||||
bot.on('voiceStateUpdate', async (oldState, newState) => {
|
const channel = oldState.channel || newState.channel;
|
||||||
if (newState.member.user.bot) return;
|
if (!channel) return;
|
||||||
if (oldState.channel && newState.channel) return;
|
if (channel.members.array().length < 2) {
|
||||||
const channel = oldState.channel || newState.channel;
|
return await api.leaveChannel(channel);
|
||||||
if (!channel) return;
|
}
|
||||||
if (channel.members.size < 2) {
|
await api.joinChannel(channel);
|
||||||
api.AudioQueue.flush();
|
let joined = false;
|
||||||
await api.leaveChannel(channel);
|
if (!oldState.channel) {
|
||||||
|
joined = true;
|
||||||
}
|
}
|
||||||
await api.joinChannel(channel);
|
|
||||||
let joined = false;
|
let username = newState.member.displayName;
|
||||||
if (!oldState.channel) {
|
let str = "";
|
||||||
joined = true;
|
if (!joined) {
|
||||||
let conn = api.getConnectionForVoiceChannel(channel);
|
str = username + " left the channel";
|
||||||
if (!api.queue) api.queue = new AudioQueue(conn, api);
|
} else {
|
||||||
}
|
str = username + " joined the channel";
|
||||||
|
}
|
||||||
let username = newState.member.displayName;
|
api.speak(channel, str);
|
||||||
let str = "";
|
})
|
||||||
if (!joined) {
|
}
|
||||||
str = printf(api.strings.USER_LEFT, username);
|
|
||||||
} else {
|
|
||||||
str = printf(api.strings.USER_JOINED, username);
|
|
||||||
}
|
|
||||||
const filepath = await api.generateVoice(str, api.announcementEngine, api.announcementVoice);
|
|
||||||
api.queue.add(__dirname + "/sysmsg.wav");
|
|
||||||
api.queue.add(filepath);
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -1,23 +0,0 @@
|
||||||
let ChatGPTAPI = null;
|
|
||||||
|
|
||||||
module.exports = function (bot, api) {
|
|
||||||
import("chatgpt").then((mod) => {
|
|
||||||
ChatGPTAPI = mod.ChatGPTAPI;
|
|
||||||
});
|
|
||||||
api.registerCommand('chat', async (args, message) => {
|
|
||||||
const response = await getChatGPTResponse(message.content.slice(6).trim());
|
|
||||||
api.respond(message, response);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
async function getChatGPTResponse(prompt) {
|
|
||||||
const api = new ChatGPTAPI({
|
|
||||||
apiKey: process.env.OPENAI_API_KEY,
|
|
||||||
completionParams: {
|
|
||||||
model: 'gpt-4o'
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
const res = await api.sendMessage(prompt);
|
|
||||||
return res.text;
|
|
||||||
}
|
|
||||||
|
|
@ -1,29 +0,0 @@
|
||||||
const gtranslate = require('node-google-translate-skidz');
|
|
||||||
|
|
||||||
module.exports = function (bot, api) {
|
|
||||||
api.registerCommand('mangle', async (args, message) => {
|
|
||||||
let str = message.content.slice(8).trim();
|
|
||||||
let langs = process.env.MANGLE_LANGS.split(',');
|
|
||||||
let lang;
|
|
||||||
let i = 0;
|
|
||||||
for (let lang of langs) {
|
|
||||||
if (i >= langs.length - 1) break;
|
|
||||||
let translationResult = await translate(str, lang, langs[i + 1]);
|
|
||||||
str = translationResult.translation;
|
|
||||||
i++;
|
|
||||||
}
|
|
||||||
api.respond(message, str);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
async function translate(text, fromLang, toLang) {
|
|
||||||
return new Promise((resolve, reject) => {
|
|
||||||
gtranslate({
|
|
||||||
text: text,
|
|
||||||
source: fromLang,
|
|
||||||
target: toLang
|
|
||||||
}, (res) => {
|
|
||||||
resolve(res);
|
|
||||||
})
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
@ -1,17 +0,0 @@
|
||||||
const fetch = require('node-fetch');
|
|
||||||
|
|
||||||
module.exports = function (bot, api) {
|
|
||||||
api.registerCommand('randomquote', async (args, message) => {
|
|
||||||
const data = await fetch(process.env["QDB_URL"], {
|
|
||||||
headers: {
|
|
||||||
Authorization: 'Basic ' + Buffer.from(`${process.env["QDB_USER"]}:${process.env["QDB_PASS"]}`).toString('base64')
|
|
||||||
}
|
|
||||||
});
|
|
||||||
const quotes = await data.json();
|
|
||||||
const quote = quotes[Math.floor(Math.random()*quotes.length)];
|
|
||||||
let chan=message.member.voice.channel;
|
|
||||||
// api.queue.add(__dirname + "/sysmsg.wav");
|
|
||||||
// api.speak(chan, `${quote.author}, on ${quote.medium}: ${quote.quote}`);
|
|
||||||
api.respond(message, `Here's your quote: ${quote.author}, on ${author.medium}: ${quote.quote}`);
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
@ -1,6 +0,0 @@
|
||||||
module.exports = function (bot, api) {
|
|
||||||
api.registerCommand("summon", async (args, message) => {
|
|
||||||
await api.joinChannel(message.member.voice.channel);
|
|
||||||
api.respond(message, `Hi!`);
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
@ -1,23 +0,0 @@
|
||||||
const printf=require('printf');
|
|
||||||
|
|
||||||
module.exports = function (bot, api) {
|
|
||||||
api.registerCommand('announcevoice', (args, message) => {
|
|
||||||
let channel = api.getActiveVoiceChannel();
|
|
||||||
if (args.length > 3) {
|
|
||||||
return api.respond(message, printf(api.strings.TOO_MANY_ARGUMENTS));
|
|
||||||
}
|
|
||||||
if (api.ttsEngines[args[1]]) {
|
|
||||||
api.announcementEngine = api.ttsEngines[args[1]];
|
|
||||||
if (api.announcementEngine.validateVoice(args[2])) {
|
|
||||||
api.announcementVoice = args[2];
|
|
||||||
api.respond(message, printf(api.strings.SYSTEM_VOICE_CHANGED, api.announcementVoice, api.announcementEngine.longName));
|
|
||||||
} else {
|
|
||||||
api.announcementVoice = api.announcementEngine.getDefaultVoice();
|
|
||||||
api.respond(message, printf(api.strings.INVALID_VOICE, api.announcementVoice, api.announcementEngine.longName));
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
api.respond(message, printf(api.strings.INVALID_ENGINE, args[1]));
|
|
||||||
}
|
|
||||||
});
|
|
||||||
api.registerCommand('flush',()=>api.queue.flush());
|
|
||||||
}
|
|
||||||
|
|
@ -1,14 +1,10 @@
|
||||||
const AudioQueue=require('../../AudioQueue.js')
|
module.exports = function(bot, api) {
|
||||||
|
bot.on('ready', async () => {
|
||||||
module.exports = function (bot, api) {
|
console.log("Bot initialized and listening");
|
||||||
bot.on('ready', async () => {
|
const guild = await bot.guilds.fetch(process.env.GUILD);
|
||||||
console.log("Bot initialized and listening");
|
const channel = await bot.channels.fetch(process.env.CHANNEL);
|
||||||
const guild = await bot.guilds.fetch(process.env.GUILD);
|
await api.joinChannel(channel);
|
||||||
const channel = await bot.channels.fetch(process.env.CHANNEL);
|
|
||||||
await api.joinChannel(channel);
|
api.speak(channel, `Hi! I'm alive. It is now ${new Date().toLocaleTimeString()} on ${new Date().toLocaleDateString()}`,api.ttsEngines.espeak, "en");
|
||||||
let conn=api.getConnectionForVoiceChannel(channel);
|
})
|
||||||
if (!api.queue) api.queue=new AudioQueue(conn, api);
|
|
||||||
api.queue.add(__dirname + "/../../sysstart.wav");
|
|
||||||
api.speak(channel, api.strings.WELCOME);
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
@ -1,47 +0,0 @@
|
||||||
const printf = require('printf')
|
|
||||||
const isStringInt = require('is-string-int');
|
|
||||||
const levenshtein = require('fast-levenshtein')
|
|
||||||
|
|
||||||
module.exports = function (bot, api) {
|
|
||||||
bot.currentWBW = "";
|
|
||||||
api.registerCommand('wbw', async (args, message) => {
|
|
||||||
if (args.length == 1) {
|
|
||||||
return api.respond(message, bot.currentWBW ? printf(api.strings.CURRENT_STORY, bot.currentWBW) : printf(api.strings.NO_STORY));
|
|
||||||
}
|
|
||||||
if (args.length > 2) {
|
|
||||||
return api.respond(message, printf(api.strings.TOO_MANY_ARGUMENTS));
|
|
||||||
} else {
|
|
||||||
if (isStringInt(args[1])) {
|
|
||||||
let story = await api.db.get('select * from WBWStories where story_id=?', parseInt(args[1]))
|
|
||||||
if (!story) {
|
|
||||||
return api.respond(message, api.strings.WBW_INVALID_ID)
|
|
||||||
} else {
|
|
||||||
return api.respond(message, story.story_text)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
let lastUser = await api.db.get('select value from BotState where key="last_wbw"');
|
|
||||||
if (message.author.id == lastUser.value && bot.currentWBW != "") {
|
|
||||||
let lastWord = (bot.currentWBW.indexOf(" ") == bot.currentWBW.lastIndexOf(" ")) ? bot.currentWBW : bot.currentWBW.slice(bot.currentWBW.slice(0,-1).lastIndexOf(' ') + 1);
|
|
||||||
console.log(args[1], lastWord, levenshtein.get(args[1], lastWord))
|
|
||||||
if (levenshtein.get(args[1], lastWord) <= 3) {
|
|
||||||
bot.currentWBW = bot.currentWBW.replace(new RegExp(lastWord + "([^" + lastWord + "]*)$"), args[1] + "$1 ");
|
|
||||||
api.respond(message, printf(api.strings.WBW_REPLACED, lastWord, args[1]))
|
|
||||||
} else {
|
|
||||||
return api.respond(message, printf(api.strings.WBW_TOO_DIFFERENT))
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
bot.currentWBW += args[1] + ' ';
|
|
||||||
api.respond(message, printf(api.strings.WBW_NEW_WORD))
|
|
||||||
let toSay = bot.currentWBW.indexOf(".") == -1 ? bot.currentWBW : bot.currentWBW.slice(bot.currentWBW.lastIndexOf('.') + 2);
|
|
||||||
api.speak(message.member.voice.channel, toSay)
|
|
||||||
await api.db.run('update BotState set value=? where key="last_wbw"', message.author.id);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
api.registerCommand('newwbw', async (args, message) => {
|
|
||||||
await api.db.run('insert into WBWStories (story_text) values(?)', bot.currentWBW);
|
|
||||||
bot.currentWBW = '';
|
|
||||||
api.respond(message, printf(api.strings.WBW_RESET))
|
|
||||||
})
|
|
||||||
}
|
|
||||||
File diff suppressed because it is too large
Load Diff
55
package.json
55
package.json
|
|
@ -1,34 +1,21 @@
|
||||||
{
|
{
|
||||||
"name": "tardis-bot",
|
"name": "tardis-bot",
|
||||||
"version": "1.0.0",
|
"version": "1.0.0",
|
||||||
"description": "",
|
"description": "",
|
||||||
"main": "index.js",
|
"main": "index.js",
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"test": "echo \"Error: no test specified\" && exit 1",
|
"test": "echo \"Error: no test specified\" && exit 1"
|
||||||
"start": "node index.js"
|
},
|
||||||
},
|
"keywords": [],
|
||||||
"keywords": [],
|
"author": "",
|
||||||
"author": "",
|
"license": "ISC",
|
||||||
"license": "ISC",
|
"dependencies": {
|
||||||
"dependencies": {
|
"discord.js": "^12.5.3",
|
||||||
"@discordjs/voice": "^0.15.0",
|
"dotenv": "^8.2.0",
|
||||||
"@google-cloud/text-to-speech": "^3.1.3",
|
"google-tts-api": "^2.0.2",
|
||||||
"chatgpt": "^5.1.2",
|
"node-fetch": "^2.6.1",
|
||||||
"discord.js": "^14.8.0",
|
"opusscript": "^0.0.8",
|
||||||
"dotenv": "^8.2.0",
|
"sha1": "^1.1.1",
|
||||||
"fast-levenshtein": "^3.0.0",
|
"sqlite3": "^5.0.2"
|
||||||
"google-tts-api": "^2.0.2",
|
}
|
||||||
"is-string-int": "^1.0.1",
|
}
|
||||||
"libsodium-wrappers": "^0.7.11",
|
|
||||||
"microsoft-cognitiveservices-speech-sdk": "^1.16.0",
|
|
||||||
"node-fetch": "^2.6.1",
|
|
||||||
"node-google-translate-skidz": "^1.1.2",
|
|
||||||
"opusscript": "^0.0.8",
|
|
||||||
"printf": "^0.6.1",
|
|
||||||
"sam-js": "^0.1.2",
|
|
||||||
"sha1": "^1.1.1",
|
|
||||||
"sqlite": "^4.0.21",
|
|
||||||
"sqlite3": "^5.0.2",
|
|
||||||
"wavefile": "^11.0.0"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -1,17 +0,0 @@
|
||||||
{
|
|
||||||
"WELCOME": "Beep boop. I'm a bot. Hi.",
|
|
||||||
"USER_JOINED": "%s joined the channel.",
|
|
||||||
"USER_LEFT": "%s left the channel.",
|
|
||||||
"SYSTEM_VOICE_CHANGED": "My new voice is %s from %s",
|
|
||||||
"USER_VOICE_CHANGED": "Your new voice is %s from %s",
|
|
||||||
"INVALID_ENGINE": "%s is not a valid engine name.",
|
|
||||||
"INVALID_VOICE": "invalid voice name. Using default voice %s for %s instead.",
|
|
||||||
"TOO_MANY_ARGUMENTS": "too many arguments for command.",
|
|
||||||
"CURRENT_STORY": "Here's the current story: %s",
|
|
||||||
"NO_STORY": "No story in progress at the moment.",
|
|
||||||
"WBW_REPLACED": "Replaced %s with %s",
|
|
||||||
"WBW_TOO_DIFFERENT": "This word is too different from the last word.",
|
|
||||||
"WBW_NEW_WORD": "Added to story.",
|
|
||||||
"WBW_RESET": "The story has been reset.",
|
|
||||||
"WBW_INVALID_ID": "No story with that ID."
|
|
||||||
}
|
|
||||||
|
|
@ -1,10 +0,0 @@
|
||||||
{
|
|
||||||
"WELCOME": "Hola hola, soy un bot.",
|
|
||||||
"USER_JOINED": "%s se ha unido al canal.",
|
|
||||||
"USER_LEFT": "%s ha salido del canal.",
|
|
||||||
"SYSTEM_VOICE_CHANGED": "Mi nueva voz es %s de %s",
|
|
||||||
"USER_VOICE_CHANGED": "Tu nueva voz es %s de %s",
|
|
||||||
"INVALID_ENGINE": "%s no es un nombre de motor válido.",
|
|
||||||
"INVALID_VOICE": "Nombre de voz no válido. Usando voz por defecto %s para %s.",
|
|
||||||
"TOO_MANY_ARGUMENTS": "Demasiados argumentos para el comando."
|
|
||||||
}
|
|
||||||
BIN
sysmsg.wav
BIN
sysmsg.wav
Binary file not shown.
BIN
sysstart.wav
BIN
sysstart.wav
Binary file not shown.
9
t.js
9
t.js
|
|
@ -1,9 +0,0 @@
|
||||||
const t=require('node-google-translate-skidz')
|
|
||||||
|
|
||||||
t({
|
|
||||||
text: 'this is a test of the meow',
|
|
||||||
'source': 'en',
|
|
||||||
target: 'ta'
|
|
||||||
}, (res) => {
|
|
||||||
console.log(res);
|
|
||||||
});
|
|
||||||
Binary file not shown.
|
|
@ -1,22 +1,16 @@
|
||||||
const fs = require('fs');
|
const fs=require('fs');
|
||||||
|
|
||||||
module.exports = class {
|
module.exports=class {
|
||||||
constructor(shortName, longName, fileExtension, supportedParameters = []) {
|
constructor(longName, fileExtension, supportedParameters=[]) {
|
||||||
this.shortName = shortName;
|
this.longName=longName;
|
||||||
this.longName = longName;
|
this.fileExtension=fileExtension;
|
||||||
this.fileExtension = fileExtension;
|
|
||||||
}
|
}
|
||||||
getInternalVoiceName(str) {
|
async getSpeech(text, voice, params) {}
|
||||||
return this.voices ? this.voices[str] : str;
|
|
||||||
}
|
|
||||||
getDefaultVoice() { }
|
|
||||||
validateVoice(voice) { return this.voices ? this.voices[voice] : true; }
|
|
||||||
async getSpeech(text, voice = this.getDefaultVoice(), params) { }
|
|
||||||
async getSpeechFile(text, filepath, voice, params) {
|
async getSpeechFile(text, filepath, voice, params) {
|
||||||
const data = await this.getSpeech(text, voice, params);
|
const data = await this.getSpeech(text, voice, params);
|
||||||
const contents = await data.arrayBuffer();
|
const contents = await data.arrayBuffer();
|
||||||
const buf = Buffer.from(contents);
|
const buf = Buffer.from(contents);
|
||||||
fs.writeFileSync(filepath, buf);
|
fs.writeFileSync(filepath, buf);
|
||||||
return filepath;
|
return filepath;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -1,57 +0,0 @@
|
||||||
const BaseEngine = require('../BaseEngine');
|
|
||||||
const sdk = require("microsoft-cognitiveservices-speech-sdk");
|
|
||||||
const fetch = require('node-fetch');
|
|
||||||
|
|
||||||
module.exports = class AzureTTS extends BaseEngine {
|
|
||||||
constructor() {
|
|
||||||
super("azure", "Microsoft Azure TTS", "wav");
|
|
||||||
this.voices = {};
|
|
||||||
this.populateVoiceList();
|
|
||||||
}
|
|
||||||
|
|
||||||
getDefaultVoice() {
|
|
||||||
return "Aria";
|
|
||||||
}
|
|
||||||
|
|
||||||
getSpeechFile(text, filepath, voice = this.getDefaultVoice(), params = {}) {
|
|
||||||
return new Promise((resolve, reject) => {
|
|
||||||
const speechConfig = sdk.SpeechConfig.fromSubscription(process.env.AZURE_API_KEY, process.env.AZURE_REGION);
|
|
||||||
speechConfig.speechSynthesisOutputFormat = sdk.SpeechSynthesisOutputFormat.Riff24Khz16BitMonoPcm;
|
|
||||||
speechConfig.speechSynthesisVoiceName = this.voices[voice];
|
|
||||||
const audioConfig = sdk.AudioConfig.fromAudioFileOutput(filepath);
|
|
||||||
const synthesizer = new sdk.SpeechSynthesizer(speechConfig, audioConfig);
|
|
||||||
synthesizer.speakTextAsync(text,
|
|
||||||
result => {
|
|
||||||
synthesizer.close();
|
|
||||||
if (result) {
|
|
||||||
// return result as stream
|
|
||||||
resolve(filepath);
|
|
||||||
}
|
|
||||||
},
|
|
||||||
error => {
|
|
||||||
console.log(error);
|
|
||||||
synthesizer.close();
|
|
||||||
reject(error);
|
|
||||||
});
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
async populateVoiceList() {
|
|
||||||
const opts = {
|
|
||||||
headers: {
|
|
||||||
'Ocp-Apim-Subscription-Key': process.env.AZURE_API_KEY
|
|
||||||
}
|
|
||||||
}
|
|
||||||
const res = await fetch(process.env.AZURE_LIST_ENDPOINT, opts);
|
|
||||||
const json = await res.json();
|
|
||||||
json.forEach((voice) => {
|
|
||||||
if (this.voices[voice.DisplayName.toLowerCase()]) {
|
|
||||||
if (voice.Name.includes('Neural')) {
|
|
||||||
this.voices[voice.DisplayName.toLowerCase()] = voice.ShortName;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
this.voices[voice.DisplayName.toLowerCase()] = voice.ShortName;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,47 +0,0 @@
|
||||||
const BaseEngine = require('../BaseEngine');
|
|
||||||
const fetch = require('node-fetch');
|
|
||||||
const querystring = require('querystring');
|
|
||||||
|
|
||||||
module.exports = class extends BaseEngine {
|
|
||||||
constructor() {
|
|
||||||
super('eleven',"Eleven Labs TTS", "mp3");
|
|
||||||
this.voices = {};
|
|
||||||
this.populateVoiceList();
|
|
||||||
}
|
|
||||||
async populateVoiceList() {
|
|
||||||
const url = "https://api.elevenlabs.io/v1/voices";
|
|
||||||
const authorization = process.env.XI_API_KEY;
|
|
||||||
const opts = {
|
|
||||||
method: "get",
|
|
||||||
headers: {
|
|
||||||
'xi-api-key': authorization
|
|
||||||
},
|
|
||||||
}
|
|
||||||
const res = await fetch(url, opts);
|
|
||||||
const voices = await res.json();
|
|
||||||
voices.voices.forEach((i) => {
|
|
||||||
let voiceName = i.name.toLowerCase();
|
|
||||||
this.voices[voiceName] = i.voice_id;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
getDefaultVoice() {
|
|
||||||
return 'Guillem';
|
|
||||||
}
|
|
||||||
async getSpeech(text, voice = this.getSpeechVoice(), params = {}) {
|
|
||||||
const url = "https://api.elevenlabs.io/v1/text-to-speech/" + this.getInternalVoiceName(voice);
|
|
||||||
const authorization = process.env.XI_API_KEY;
|
|
||||||
const opts = {
|
|
||||||
method: "post",
|
|
||||||
headers: {
|
|
||||||
'Content-Type': 'application/json',
|
|
||||||
'voice_id': this.getInternalVoiceName(voice),
|
|
||||||
'xi-api-key': authorization
|
|
||||||
},
|
|
||||||
body: JSON.stringify({
|
|
||||||
model_id: 'eleven_multilingual_v2',
|
|
||||||
text: text
|
|
||||||
})
|
|
||||||
};
|
|
||||||
return fetch(url, opts);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
@ -1,15 +1,12 @@
|
||||||
const BaseEngine = require('../BaseEngine')
|
const BaseEngine=require('../BaseEngine')
|
||||||
const { spawn } = require('child_process')
|
const {spawn} = require('child_process')
|
||||||
|
|
||||||
module.exports = class extends BaseEngine {
|
module.exports=class extends BaseEngine {
|
||||||
constructor() {
|
constructor() {
|
||||||
super('espeak','ESpeak', 'wav')
|
super('ESpeak','wav')
|
||||||
}
|
}
|
||||||
getDefaultVoice() {
|
async getSpeechFile(text, filepath, voice='en', params={}) {
|
||||||
return 'en';
|
let proc=await spawn('espeak', ['-v', voice, '-w',filepath, '--stdin']);
|
||||||
}
|
|
||||||
async getSpeechFile(text, filepath, voice = this.getDefaultVoice(), params = {}) {
|
|
||||||
let proc = await spawn('espeak', ['-v', voice, '-w', filepath, '--stdin']);
|
|
||||||
proc.stdin.end(text);
|
proc.stdin.end(text);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -1,36 +0,0 @@
|
||||||
const BaseEngine = require('../BaseEngine')
|
|
||||||
const sdk = require('@google-cloud/text-to-speech');
|
|
||||||
const fs = require('fs');
|
|
||||||
const util = require('util');
|
|
||||||
|
|
||||||
module.exports = class GoogleCloudTTS extends BaseEngine {
|
|
||||||
constructor() {
|
|
||||||
super('google', 'Google Cloud TTS', 'wav');
|
|
||||||
this.client = new sdk.TextToSpeechClient();
|
|
||||||
this.client.initialize();
|
|
||||||
this.voices = {};
|
|
||||||
this.populateVoiceList();
|
|
||||||
}
|
|
||||||
async populateVoiceList() {
|
|
||||||
const [result] = await this.client.listVoices({});
|
|
||||||
const voiceList = result.voices;
|
|
||||||
voiceList.forEach((voice) => {
|
|
||||||
|
|
||||||
this.voices[voice.name.toLowerCase()] = { name: voice.name, lang: voice.languageCodes[0] };
|
|
||||||
});
|
|
||||||
}
|
|
||||||
getDefaultVoice() {
|
|
||||||
return 'en-US-Wavenet-A';
|
|
||||||
}
|
|
||||||
async getSpeechFile(text, filepath, voice = this.getDefaultVoice(), params = {}) {
|
|
||||||
const request = {
|
|
||||||
input: { text: text },
|
|
||||||
voice: { name: this.voices[voice].name, languageCode: this.voices[voice].lang },
|
|
||||||
audioConfig: { audioEncoding: 'LINEAR16' },
|
|
||||||
};
|
|
||||||
let [response] = await this.client.synthesizeSpeech(request);
|
|
||||||
const writeFile = util.promisify(fs.writeFile);
|
|
||||||
await writeFile(filepath, response.audioContent, 'binary');
|
|
||||||
return filepath;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,16 +1,13 @@
|
||||||
const BaseEngine = require('../BaseEngine');
|
const BaseEngine=require('../BaseEngine');
|
||||||
const fetch = require('node-fetch');
|
const fetch = require('node-fetch');
|
||||||
const tts = require('google-tts-api');
|
const tts = require('google-tts-api');
|
||||||
|
|
||||||
module.exports = class extends BaseEngine {
|
module.exports= class extends BaseEngine {
|
||||||
constructor() {
|
constructor() {
|
||||||
super('gtranslate', "Google Translate TTS", "mp3");
|
super("Google Translate TTS","mp3");
|
||||||
}
|
}
|
||||||
getDefaultVoice() {
|
async getSpeech(text, voice='en-us', params={}) {
|
||||||
return 'en-us';
|
const url = tts.getAudioUrl(text, {lang: voice});
|
||||||
}
|
|
||||||
async getSpeech(text, voice = this.getDefaultVoice(), params = {}) {
|
|
||||||
const url = tts.getAudioUrl(text, { lang: voice });
|
|
||||||
return fetch(url);
|
return fetch(url);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
@ -1,37 +0,0 @@
|
||||||
const BaseEngine = require('../BaseEngine');
|
|
||||||
const fetch = require('node-fetch');
|
|
||||||
const tts = require('google-tts-api');
|
|
||||||
|
|
||||||
module.exports = class extends BaseEngine {
|
|
||||||
constructor() {
|
|
||||||
super('openai', "OpenAI TTS", "mp3");
|
|
||||||
this.voices = {
|
|
||||||
alloy: "alloy",
|
|
||||||
echo: "echo",
|
|
||||||
fable: "fable",
|
|
||||||
onyx: "onyx",
|
|
||||||
nova: "nova",
|
|
||||||
shimmer: "shimmer"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
getDefaultVoice() {
|
|
||||||
return 'Alloy';
|
|
||||||
}
|
|
||||||
async getSpeech(text, voice = this.getDefaultVoice(), params = {}) {
|
|
||||||
const url = `https://api.openai.com/v1/audio/speech`;
|
|
||||||
const opts = {
|
|
||||||
method: "post",
|
|
||||||
headers: {
|
|
||||||
'Content-Type': 'application/json',
|
|
||||||
Authorization: `Bearer ${process.env["OPENAI_API_KEY"]}`
|
|
||||||
},
|
|
||||||
body: JSON.stringify({
|
|
||||||
model: "tts-1-hd",
|
|
||||||
input: text,
|
|
||||||
voice: voice,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
console.log(opts);
|
|
||||||
return fetch(url, opts);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
@ -1,25 +0,0 @@
|
||||||
const BaseEngine = require('../BaseEngine')
|
|
||||||
const Sam = require('sam-js');
|
|
||||||
const wavefile = require('wavefile');
|
|
||||||
const fs = require('fs');
|
|
||||||
|
|
||||||
module.exports = class extends BaseEngine {
|
|
||||||
constructor() {
|
|
||||||
super('sam', 'Software Automatic Mouth', 'wav')
|
|
||||||
}
|
|
||||||
getDefaultVoice() {
|
|
||||||
return 'sam';
|
|
||||||
}
|
|
||||||
async getSpeechFile(text, filepath, voice = this.getDefaultVoice(), params = {}) {
|
|
||||||
let sam = new Sam();
|
|
||||||
let phonetic = false;
|
|
||||||
if (text[0] == "$") {
|
|
||||||
text = text.slice(1);
|
|
||||||
phonetic = true;
|
|
||||||
}
|
|
||||||
const buf = sam.buf8(text, phonetic);
|
|
||||||
const file = new wavefile.WaveFile();
|
|
||||||
file.fromScratch(1, 22050, 8, buf);
|
|
||||||
fs.writeFileSync(filepath, file.toBuffer());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,47 +0,0 @@
|
||||||
const BaseEngine = require('../BaseEngine');
|
|
||||||
const fetch = require('node-fetch');
|
|
||||||
const querystring = require('querystring');
|
|
||||||
const fs = require("fs");
|
|
||||||
|
|
||||||
module.exports = class extends BaseEngine {
|
|
||||||
constructor() {
|
|
||||||
super('unreal', "Unreal Speech TTS", "mp3");
|
|
||||||
this.voices = {
|
|
||||||
scarlett: 'Scarlett',
|
|
||||||
liv: 'Liv',
|
|
||||||
dan: 'Dan',
|
|
||||||
will: 'Will',
|
|
||||||
amy: 'Amy'
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
getDefaultVoice() {
|
|
||||||
return 'Liv';
|
|
||||||
}
|
|
||||||
|
|
||||||
async getSpeechFile(text, filepath, voice = this.getDefaultVoice(), params = {}) {
|
|
||||||
const url = "https://api.v6.unrealspeech.com/speech";
|
|
||||||
const authorization = process.env.UNREAL_API_KEY;
|
|
||||||
const opts = {
|
|
||||||
method: "post",
|
|
||||||
headers: {
|
|
||||||
'Content-Type': 'application/json',
|
|
||||||
'Authorization': `Bearer ${authorization}`
|
|
||||||
},
|
|
||||||
body: JSON.stringify({
|
|
||||||
Bitrate: "320k",
|
|
||||||
Temperature: 0.1,
|
|
||||||
VoiceId: this.getInternalVoiceName(voice),
|
|
||||||
Text: text,
|
|
||||||
AudioFormat: "mp3"
|
|
||||||
})
|
|
||||||
};
|
|
||||||
const res = await fetch(url, opts);
|
|
||||||
const json = await res.json();
|
|
||||||
const data = await fetch(json.OutputUri);
|
|
||||||
const contents = await data.arrayBuffer();
|
|
||||||
const buf = Buffer.from(contents);
|
|
||||||
fs.writeFileSync(filepath, buf);
|
|
||||||
return filepath;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
@ -1,41 +1,17 @@
|
||||||
const BaseEngine = require('../BaseEngine');
|
const BaseEngine=require('../BaseEngine');
|
||||||
const fetch = require('node-fetch');
|
const fetch = require('node-fetch');
|
||||||
const querystring = require('querystring');
|
const querystring = require('querystring');
|
||||||
|
|
||||||
module.exports = class extends BaseEngine {
|
module.exports= class extends BaseEngine {
|
||||||
constructor() {
|
constructor() {
|
||||||
super('watson',"IBM Watson TTS", "ogg");
|
super("IBM Watson TTS","ogg");
|
||||||
this.voices = {};
|
|
||||||
this.populateVoiceList();
|
|
||||||
}
|
}
|
||||||
async populateVoiceList() {
|
async getSpeech(text, voice='en-us', params={}) {
|
||||||
const url = process.env.watsonURL + "/v1/voices";
|
const url = process.env.watsonURL+"/v1/synthesize";
|
||||||
const authorization = this.IBMAuthString();
|
let buff=new Buffer('apikey:'+process.env.watsonAPIKey);
|
||||||
const opts = {
|
let b64auth=buff.toString('base64');
|
||||||
method: "get",
|
const authorization='Basic '+b64auth;
|
||||||
headers: {
|
const opts={
|
||||||
'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",
|
method: "post",
|
||||||
headers: {
|
headers: {
|
||||||
'Content-Type': 'application/json',
|
'Content-Type': 'application/json',
|
||||||
|
|
@ -45,6 +21,6 @@ module.exports = class extends BaseEngine {
|
||||||
text: text
|
text: text
|
||||||
})
|
})
|
||||||
};
|
};
|
||||||
return fetch(url, opts);
|
return fetch(url,opts);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
Loading…
Reference in New Issue