From 856149860eac8077e31c3be35157d7f3b1b2c1ff Mon Sep 17 00:00:00 2001 From: hydrabolt Date: Fri, 14 Aug 2015 16:02:40 +0100 Subject: [PATCH] Added more functionality --- hydrabot/.gitignore | 1 + hydrabot/authority.js | 37 +++++++++++++++++++++++++++ hydrabot/commands.js | 14 ++++++++++ hydrabot/hydrabot.js | 59 ++++++++++++++++++++++++++++++++++++++++++- 4 files changed, 110 insertions(+), 1 deletion(-) create mode 100644 hydrabot/authority.js create mode 100644 hydrabot/commands.js diff --git a/hydrabot/.gitignore b/hydrabot/.gitignore index d344ba6b0..070bc70aa 100644 --- a/hydrabot/.gitignore +++ b/hydrabot/.gitignore @@ -1 +1,2 @@ config.json +authority.json diff --git a/hydrabot/authority.js b/hydrabot/authority.js new file mode 100644 index 000000000..1f7c1c4af --- /dev/null +++ b/hydrabot/authority.js @@ -0,0 +1,37 @@ +var fs = require( "fs" ); + +var authCache = {}; + +exports.init = function() { + try { + var fd = fs.openSync( "./authority.json", "wx" ); + exports.writeCache(); + } catch ( e ) { + if ( e.errno !== -4075 ){ + throw e; + }else{ + authCache = JSON.parse(fs.readFileSync("./authority.json", "utf8")); + } + } +} + +exports.getLevel = function(user){ + + if(authCache[user.id]) + return authCache[user.id]; + else + return 0; + +} + +exports.setLevel = function(user, level){ + authCache[user.id] = level; + exports.writeCache(); +} + +exports.writeCache = function() { + fs.writeFile( './authority.json', JSON.stringify(authCache), function( err ) { + if ( err ) + console.log("Error saving Authority Caches - " + err.code); + } ); +} diff --git a/hydrabot/commands.js b/hydrabot/commands.js new file mode 100644 index 000000000..a26f25aae --- /dev/null +++ b/hydrabot/commands.js @@ -0,0 +1,14 @@ +var Authority = require("./authority.js"); + +Commands = []; + +Commands["info"] = { + oplevel : 0, + fn : function(bot, params, message){ + + bot.reply(message, "Info!"); + + } +} + +exports.Commands = Commands; diff --git a/hydrabot/hydrabot.js b/hydrabot/hydrabot.js index 2695d3b42..594140c7f 100644 --- a/hydrabot/hydrabot.js +++ b/hydrabot/hydrabot.js @@ -7,9 +7,21 @@ var Discord = require( "../" ); // structure : { "email" : "discordEmail", "password" : "discordPassword" } var BotConfig = require( "./config.json" ); +// Load the commands file +var Commands = require( "./commands.js" ).Commands; + +// Load the Authority handler +var Authority = require( "./authority.js" ); + +// Initialise it +Authority.init(); + // Create a new Discord Client var hydrabot = new Discord.Client(); +// An array of single character prefixes the bot will respond to +var commandPrefixes = [ "$", "£", "`" ]; + // Log the client in using the auth details in config.json hydrabot.login( BotConfig.email, BotConfig.password ); @@ -29,6 +41,51 @@ hydrabot.on( "disconnected", function( obj ) { hydrabot.on( "message", function( message ) { - console.log( message ); + // if the message doesn't begin with a valid command prefix exit + if ( commandPrefixes.indexOf( message.content.charAt( 0 ) ) == -1 ) + return; + + var command = "", + params = []; //set the message details + + // remove the prefix from the start of the message + message.content = message.content.substr( 1 ); + + // split the message by slashes. This will yield something + // like: ["command", "a", "b", "c"]. + var chunks = message.content.split( "/" ); + + for ( key in chunks ) { //loop through the chunks and trim them + chunks[ key ] = chunks[ key ].trim(); + } + + command = chunks[ 0 ]; //the first param will be the command + params = chunks.slice( 1 ); + + // it's less messy if we outsource to another function + handleMessage( command, params, message ); } ); + +function handleMessage( command, params, message ) { + var channel = message.channel; // set the channel variable to message.channel + var sender = message.author; // set the sender variable to the author of the message + var isPM = ( message.channel instanceof Discord.PMChannel ); // set isPM to true if the channel is a Private Message Channel. + + if ( Commands[ command ] ) { + + console.log(Authority.getLevel( message.author )); + if ( Authority.getLevel( message.author ) >= Commands[ command ].oplevel ) { + //user has authority to do this + Commands[ command ].fn( hydrabot, params, message ); + + } else { + //user doesn't have authority + hydrabolt.reply( message, "you don't have authority to do this!" ); + } + + } else { + hydrabot.reply( message, "that command was not found!" ); + } + +}