lbry-wunderbot/bot/modules/speech.js
2017-11-01 10:50:36 -07:00

245 lines
No EOL
8 KiB
JavaScript

let request = require('request');
let wget = require('wget');
let fs = require('fs');
let config = require('config');
let hasSpeechBotChannels = require('../helpers.js').hasSpeechBotChannels;
let inPrivate = require('../helpers.js').inPrivate;
let ChannelID = config.get('speechbot').mainchannel;
//debug output "true/false" outputs same error as slack message in console if set to true
//if set to false console will be left blank like normal
//some have more info on file details of error
let FullDebug = "true"
//outputs response from speech, very bulk reply
let ResponseDebug = "false"
exports.commands = [
"speech" // command that is in this file, every command needs it own export as shown below
]
exports.speech = {
usage: "<name>",
description: "gets top claim from spee.ch, coming soon post to spee.ch",
process: function(bot,msg,suffix){
if(!hasSpeechBotChannels(msg) && !inPrivate(msg)){
msg.channel.send('Please use <#' + ChannelID + '> or DMs to talk to speech bot.');
return;
}
var command = "!speech"
words = suffix.trim().split(' ').filter( function(n){return n !== "";} );
var imagename = words[0];
//check if image name is help, if it is then do help message
if (imagename == "help") {
doHelp(bot,msg,suffix)
return;
} else {
//check if imagename is defined if not do error
if (imagename === undefined) {
if (FullDebug === "true") {
var message = "`no name provided`"
console.log('no name provided');
msg.channel.send(message);
doHelp(bot,msg,suffix)
return
} else {
var message = "`no name provided`"
msg.channel.send(message);
doHelp(bot,msg,suffix)
return
}
}
//set second word to url
var filepath = words[1];
//check if a url is provided if none do help message
if (filepath === undefined) {
if (FullDebug === "true") {
var message = "`no url provided, fetching image from:`\n" +
"https://spee.ch/" + imagename
console.log('no url provided');
msg.channel.send(message);
return
} else {
var message = "`no url provided, fetching image from:`\n" +
"https://spee.ch/" + imagename
msg.channel.send(message);
return
}}
//prepare url for other uses
//we will just set filepath to url to be safe
var url = filepath;
//parse first 4 letters of url should be http
var linkvalid = url.slice(0, 4)
//check of url provided begins with http in not throw error and help message
if (linkvalid !== "http") {
if (FullDebug === "true") {
var message = '`error not a valid url, please start with http or https`'
console.log('invalid url provided: ' + filepath);
msg.channel.send(message);
return
} else {
var message = '`error not a valid url, please start with http or https`'
msg.channel.send(message);
return;
}}
//function to check if url is an image
var isUriImage = function(uri) {
//make sure we remove any nasty GET params
uri = uri.split('?')[0];
//moving on, split the uri into parts that had dots before them
var parts = uri.split('.');
//get the last part ( should be the extension )
var extension = parts[parts.length-1];
//define some image types to test against
var imageTypes = ['jpg','jpeg','tiff','png','gif','bmp'];
//check if the extension matches anything in the list. if it does set true if not set false
if(imageTypes.indexOf(extension) !== -1) {
return true;
} else {
return false
}
}
//check if url is an image if its not throw error and help message
if (isUriImage(url) === false) {
if (FullDebug === "true"){
var message = '`error not a valid image url, be sure the link includes a file type`'
console.log('invalid url provided: ' + url);
msg.channel.send(message);
return
} else {
var message = '`error not a valid image url, be sure the link includes a file type`'
msg.channel.send(message);
return
}
}
//set third word to nsfw, with it being an optional functionality
var eighteen = words[2];
//check is NSFW if yes or no sets proper value if none
if (eighteen == "" || eighteen == "none" || eighteen == undefined || eighteen == null || eighteen == "no"|| eighteen == "false" || eighteen == false || eighteen == "n") {
eighteen = "no";
} else {
eighteen = "yes"
}
//prepare url for wget
var source = url;
//parse the filename to use to save file
filepath = source.split('/').pop();
//set proper directory for downloading image
var outputFile = 'speech-uploads/' + filepath;
//set download directory to current working directory
var dir = process.cwd() ;
//set full path to directory for speech uploading
var fullpath = dir + '\\speech-uploads\\' + filepath;
//download url via wget
var download = wget.download(url, outputFile);
//check if url is reachable if not throw error
download.on('error', function(err) {
if (FullDebug === "true") {
console.log("error could not reach: " + url + " : " + err);
var message = '`error url could not be reached`'
msg.channel.send(message);
return
} else {
var message = '`error url could not be reached`'
msg.channel.send(message);
return
}
});
download.on('end', (output) => {
//if no errors and file ready -> do the request
output && doSteps(bot, imagename, url, eighteen);
});}
//send help message
function doHelp(bot,msg,suffix) {
msg.channel.send(
{
"embed": {
"title": "",
"description": '**!speech `<Name>`** : *displays top claim on speech* \n\n\n' +
"**COMING SOON POSTING TO SPEECH** \n\n" +
'**!speech `<Name> <URL> <NSFW>`** : *Uploads Image URL to Spee.ch* \n' +
'**NOTE : dont include spaces in name (NSFW is optional true/false, if left blank will defualt to false)** \n' +
'EXAMPLE : `!speech my-image-name https://url/to/image.png false`',
"color": 7976557,
"author": {
"name": "Speech Bot Help",
"icon_url": "https://i.imgur.com/yWf5USu.png"
}
}
}
);
}
//send post request to speech
function doSteps(bot, imagename, url, eighteen) {
request.post(
//url to send post request
'https://spee.ch/api/publish',
//json payload
{ json: { name: imagename ,file: fullpath,nsfw: eighteen } },
//get response from server
function (error, response, body) {
//output response if ResponseDebug set to true
if (ResponseDebug === "true") {
console.log(response);
console.log(error);
console.log(body.success);
console.log(body.message);
}
//check speech response for file path error, if found throw internal error!
if (body.message === "no files found in request") {
if (FullDebug === "true") {
console.log("no file found: " + fullpath);
var message = '`Failed to upload file internally!!`\n please contact <@244245498746241025> or another moderator if the issue persists';
msg.channel.send(message);
return
} else {
var message = '`Failed to upload file internally!!`\n please contact <@244245498746241025> or another moderator if the issue persists';
msg.channel.send(message);
return
}
}
//check speech response for filename error, if found throw internal error!
if (body.message === "no name field found in request") {
if (FullDebug === "true") {
console.log("no name field found: " + imagename);
var message = '`Failed to upload file internally!!`\n please contact <@244245498746241025> or another moderator if the issue persists';
msg.channel.send(message);
return
} else {
var message = '`Failed to upload file internally!!`\n please contact <@244245498746241025> or another moderator if the issue persists';
msg.channel.send(message);
return
}
}
//if no errors post this message
var message = 'uploading... \n "name":"' + imagename + '",\n "URL": "' + url + '",\n "nsfw":"' + eighteen + '"\n to spee.ch';
console.log('uploading... \n "name":"' + imagename + '",\n "file name": "' + filepath + '",\n "url":"' + url + '"\n "path":"' + fullpath + '"\n "nsfw": "' + eighteen + '"' );
msg.channel.send(message);
}
);
};
}
}