summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorKevin Smith <git@kismith.co.uk>2011-02-27 22:45:32 (GMT)
committerKevin Smith <git@kismith.co.uk>2011-03-06 20:25:01 (GMT)
commit27d21b371f24272466a2d6a5bf2e2b717ee2d9fc (patch)
tree5f53281711d4f467933e4b3315241e4eee58a64c
parentd9c9df3b4ae5432552417fc4db74d62ab34f066d (diff)
downloadswift-27d21b371f24272466a2d6a5bf2e2b717ee2d9fc.zip
swift-27d21b371f24272466a2d6a5bf2e2b717ee2d9fc.tar.bz2
A start on Swiftob, a Swiften-based chatbot.
-rw-r--r--.project8
-rw-r--r--Swiften/MUC/MUC.cpp8
-rw-r--r--Swiften/MUC/MUC.h4
-rw-r--r--Swiftob/Commands.cpp163
-rw-r--r--Swiftob/Commands.h71
-rw-r--r--Swiftob/LuaCommands.cpp377
-rw-r--r--Swiftob/LuaCommands.h93
-rw-r--r--Swiftob/MUCs.cpp116
-rw-r--r--Swiftob/MUCs.h47
-rw-r--r--Swiftob/SConscript26
-rw-r--r--Swiftob/Storage.cpp47
-rw-r--r--Swiftob/Storage.h28
-rw-r--r--Swiftob/Swiftob.cpp117
-rw-r--r--Swiftob/Swiftob.h49
-rw-r--r--Swiftob/Users.cpp45
-rw-r--r--Swiftob/Users.h49
-rw-r--r--Swiftob/linit.c1
-rw-r--r--Swiftob/main.cpp37
-rw-r--r--Swiftob/scripts/agenda.lua94
-rw-r--r--Swiftob/scripts/echo.lua5
-rw-r--r--Swiftob/scripts/eval.lua14
-rw-r--r--Swiftob/scripts/version.lua32
22 files changed, 1424 insertions, 7 deletions
diff --git a/.project b/.project
index 8a42eaa..fcdfcdc 100644
--- a/.project
+++ b/.project
@@ -18,7 +18,7 @@
</dictionary>
<dictionary>
<key>org.eclipse.cdt.make.core.autoBuildTarget</key>
- <value>Sluift</value>
+ <value></value>
</dictionary>
<dictionary>
<key>org.eclipse.cdt.make.core.buildArguments</key>
@@ -29,10 +29,6 @@
<value>python</value>
</dictionary>
<dictionary>
- <key>org.eclipse.cdt.make.core.buildLocation</key>
- <value></value>
- </dictionary>
- <dictionary>
<key>org.eclipse.cdt.make.core.cleanBuildTarget</key>
<value>-c</value>
</dictionary>
@@ -54,7 +50,7 @@
</dictionary>
<dictionary>
<key>org.eclipse.cdt.make.core.fullBuildTarget</key>
- <value>Sluift</value>
+ <value></value>
</dictionary>
<dictionary>
<key>org.eclipse.cdt.make.core.stopOnError</key>
diff --git a/Swiften/MUC/MUC.cpp b/Swiften/MUC/MUC.cpp
index b8c23cd..68a5a86 100644
--- a/Swiften/MUC/MUC.cpp
+++ b/Swiften/MUC/MUC.cpp
@@ -208,6 +208,14 @@ void MUC::handleCreationConfigResponse(MUCOwnerPayload::ref /*unused*/, ErrorPay
}
}
+bool MUC::hasOccupant(const std::string& nick) {
+ return occupants.find(nick) != occupants.end();
+}
+
+MUCOccupant MUC::getOccupant(const std::string& nick) {
+ return occupants.find(nick)->second;
+}
+
//FIXME: Recognise Topic changes
//TODO: Invites(direct/mediated)
diff --git a/Swiften/MUC/MUC.h b/Swiften/MUC/MUC.h
index ef76a6a..278ef95 100644
--- a/Swiften/MUC/MUC.h
+++ b/Swiften/MUC/MUC.h
@@ -51,7 +51,9 @@ namespace Swift {
void handleIncomingMessage(Message::ref message);
/** Expose public so it can be called when e.g. user goes offline */
void handleUserLeft(LeavingType);
-
+ /** Get occupant information*/
+ MUCOccupant getOccupant(const std::string& nick);
+ bool hasOccupant(const std::string& nick);
public:
boost::signal<void (const std::string& /*nick*/)> onJoinComplete;
boost::signal<void (ErrorPayload::ref)> onJoinFailed;
diff --git a/Swiftob/Commands.cpp b/Swiftob/Commands.cpp
new file mode 100644
index 0000000..e39d23e
--- /dev/null
+++ b/Swiftob/Commands.cpp
@@ -0,0 +1,163 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include "Swiftob/Commands.h"
+
+#include <iostream>
+#include <boost/bind.hpp>
+
+#include <Swiften/Client/Client.h>
+
+typedef std::pair<std::string, Commands::Command*> NamedCommand;
+
+Commands::Commands(Users* users, Swift::Client* client, Storage* storage, MUCs* mucs) {
+ users_ = users;
+ client_ = client;
+ mucs_ = mucs;
+ storage_ = storage;
+ resetCommands();
+}
+
+void Commands::resetCommands() {
+ foreach (NamedCommand command, commands_) {
+ delete command.second;
+ }
+ commands_.clear();
+ registerCommand("quit", Owner, "Quit the bot", boost::bind(&Commands::handleQuitCommand, this, _1, _2, _3));
+ registerCommand("help", Anyone, "Get help", boost::bind(&Commands::handleHelpCommand, this, _1, _2, _3));
+ registerCommand("join", Owner, "Join a MUC", boost::bind(&Commands::handleJoinCommand, this, _1, _2, _3));
+ registerCommand("part", Owner, "Leave a MUC", boost::bind(&Commands::handlePartCommand, this, _1, _2, _3));
+ registerCommand("rehash", Owner, "Reload scripts", boost::bind(&Commands::handleRehashCommand, this, _1, _2, _3));
+ onReset();
+}
+
+void Commands::registerCommand(const std::string& name, RoleList roles, const std::string& description, boost::function<void(const std::string& /*command*/, const std::string& /*params*/, Swift::Message::ref)> callback) {
+ Command* command = new Command(roles, description);
+ commands_[name] = command;
+ command->onReceived.connect(callback);
+}
+
+bool Commands::hasCommand(const std::string& name) {
+ return commands_.find(name) != commands_.end();
+}
+
+bool Commands::runCommand(const std::string& name, const std::string& params, Swift::Message::ref message) {
+ Users::User::Role userRole = users_->getRoleForSender(message);
+ Command* command = commands_[name];
+ if (roleIn(userRole, command->getAllowedBy())) {
+ command->onReceived(name, params, message);
+ return true;
+ } else {
+ replyTo(message, "You may not run this command", true);
+ }
+ return false;
+}
+
+bool Commands::roleIn(const Users::User::Role userRole, RoleList roleList) {
+ switch (roleList) {
+ case Owner : return userRole == Users::User::Owner;
+ case Anyone : return true;
+ }
+ std::cerr << "Unrecognised role list" << std::endl;
+ return false;
+}
+
+void Commands::handleQuitCommand(const std::string& /*command*/, const std::string& /*params*/, Swift::Message::ref message) {
+ replyTo(message, "Shutting down");
+ std::cout << "Quitting at the behest of " << message->getFrom().toString() << std::endl;
+ exit(0);
+}
+
+void Commands::setRehashError(const std::string& error) {
+ if (!rehashError_.empty()) {
+ rehashError_ += "; ";
+ }
+ rehashError_ += error;
+}
+
+void Commands::handleRehashCommand(const std::string& /*command*/, const std::string& /*params*/, Swift::Message::ref message) {
+ rehashError_ = "";
+ replyTo(message, "Rehashing now.");
+ std::cout << "Rehashing at the behest of " << message->getFrom().toString() << std::endl;
+ resetCommands();
+ if (rehashError_.empty()) {
+ replyTo(message, "Rehash complete");
+ } else {
+ replyTo(message, "I have suffered a tremendous failure: " + rehashError_);
+ }
+}
+
+void Commands::handleJoinCommand(const std::string& /*command*/, const std::string& params, Swift::Message::ref message) {
+ Swift::JID room(params);
+ if (!room.isValid() || !room.getResource().empty() || room.getNode().empty()) {
+ replyTo(message, "Can't join " + room.toString() + ", not a valid room JID.");
+ return;
+ }
+ if (mucs_->contains(room)) {
+ replyTo(message, "I'm already (trying to be?) in " + room.toString() + ".");
+ return;
+ }
+ replyTo(message, "Trying to join " + room.toString() + ".");
+ mucs_->join(room, boost::bind(&Commands::handleJoinCommandSuccess, this, room, message), boost::bind(&Commands::handleJoinCommandFailure, this, room, _1, message));
+}
+
+void Commands::handlePartCommand(const std::string& /*command*/, const std::string& params, Swift::Message::ref message) {
+ Swift::JID room(params);
+ if (!room.isValid() || !room.getResource().empty() || room.getNode().empty()) {
+ replyTo(message, "Can't leave " + room.toString() + ", not a valid room JID.");
+ return;
+ }
+ if (mucs_->contains(room)) {
+ replyTo(message, "I'm not in " + room.toString() + ".");
+ return;
+ }
+ replyTo(message, "Leaving " + room.toString() + ".");
+ mucs_->part(room);
+}
+
+void Commands::handleJoinCommandSuccess(const Swift::JID& room, Swift::Message::ref message) {
+ replyTo(message, "Joined " + room.toString());
+}
+
+void Commands::handleJoinCommandFailure(const Swift::JID& room, const std::string& error, Swift::Message::ref message) {
+ replyTo(message, "Join to " + room.toString() + "failed. " + error);
+}
+
+void Commands::handleHelpCommand(const std::string& /*command*/, const std::string& /*params*/, Swift::Message::ref message) {
+ Users::User::Role userRole = users_->getRoleForSender(message);
+ std::string result("Available commands:");
+ std::string bang = message->getType() == Swift::Message::Groupchat ? "\n!" : "\n";
+ foreach (NamedCommand pair, commands_) {
+ if (roleIn(userRole, pair.second->getAllowedBy())) {
+ result += bang + pair.first + " - " + pair.second->getDescription();
+ }
+ }
+ replyTo(message, result, true);
+}
+
+/**
+ * \param outOfMUC Reply to the sender directly, don't spam MUCs with the reply
+ */
+void Commands::replyTo(Swift::Message::ref source, std::string replyBody, bool outOfMUC) {
+ Swift::Message::ref reply(new Swift::Message());
+ Swift::Message::Type type = source->getType();
+ reply->setType(type);
+ reply->setBody(type == Swift::Message::Groupchat ? source->getFrom().getResource() + ": " + replyBody : replyBody);
+ Swift::JID to = source->getFrom();
+ if (type == Swift::Message::Groupchat) {
+ if (outOfMUC) {
+ reply->setType(Swift::Message::Chat);
+ } else {
+ to = to.toBare();
+ }
+ }
+ reply->setTo(to);
+ if (client_->isAvailable()) {
+ client_->sendMessage(reply);
+ } else {
+ std::cout << "Dropping '" + reply->getBody() + "' -> " + reply->getTo().toString() + " on the floor due to missing connection." << std::endl;
+ }
+}
diff --git a/Swiftob/Commands.h b/Swiftob/Commands.h
new file mode 100644
index 0000000..e11078d
--- /dev/null
+++ b/Swiftob/Commands.h
@@ -0,0 +1,71 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#pragma once
+
+#include <map>
+#include <string>
+
+#include <Swiften/Base/boost_bsignals.h>
+#include <Swiften/Elements/Message.h>
+
+#include "Swiftob/Users.h"
+#include "Swiftob/MUCs.h"
+
+namespace Swift {
+ class Client;
+}
+
+class Storage;
+class Commands {
+ public:
+ enum RoleList {Anyone, Owner};
+ public:
+ class Command {
+ public:
+ Command(RoleList allowedBy, const std::string& description) : allowedBy_(allowedBy), description_(description) {
+ }
+
+ virtual ~Command() {};
+
+ boost::signal<void (const std::string&, const std::string&, Swift::Message::ref)> onReceived;
+ RoleList getAllowedBy() {return allowedBy_;}
+ std::string getDescription() {return description_;}
+ private:
+ RoleList allowedBy_;
+ std::string description_;
+ };
+
+ public:
+ Commands(Users* users, Swift::Client* client, Storage* storage, MUCs* mucs);
+ bool hasCommand(const std::string&);
+ bool runCommand(const std::string& command, const std::string& params, Swift::Message::ref message);
+ void replyTo(Swift::Message::ref source, std::string replyBody, bool outOfMUC = false);
+ void registerCommand(const std::string& name, RoleList roles, const std::string& description, boost::function<void(const std::string& /*command*/, const std::string& /*params*/, Swift::Message::ref)> callback);
+ void resetCommands();
+ void setRehashError(const std::string& error);
+
+ public:
+ boost::signal<void ()> onReset;
+ private:
+ bool roleIn(const Users::User::Role userRole, RoleList roles);
+ void handleQuitCommand(const std::string& command, const std::string& params, Swift::Message::ref message);
+ void handleHelpCommand(const std::string& command, const std::string& params, Swift::Message::ref message);
+ void handleJoinCommand(const std::string& /*command*/, const std::string& params, Swift::Message::ref message);
+ void handleJoinCommandSuccess(const Swift::JID& room, Swift::Message::ref message);
+ void handleJoinCommandFailure(const Swift::JID& room, const std::string& error, Swift::Message::ref message);
+ void handlePartCommand(const std::string& /*command*/, const std::string& params, Swift::Message::ref message);
+ void handleRehashCommand(const std::string& command, const std::string& params, Swift::Message::ref message);
+ private:
+ std::map<std::string, Command*> commands_;
+ Users* users_;
+ Swift::Client* client_;
+ Storage* storage_;
+ MUCs* mucs_;
+ std::string rehashError_;
+};
+
+
diff --git a/Swiftob/LuaCommands.cpp b/Swiftob/LuaCommands.cpp
new file mode 100644
index 0000000..7be818e
--- /dev/null
+++ b/Swiftob/LuaCommands.cpp
@@ -0,0 +1,377 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include "Swiftob/LuaCommands.h"
+
+#include <boost/bind.hpp>
+#include <vector>
+#include <algorithm>
+
+#include <Swiften/Client/Client.h>
+#include <Swiften/Network/TimerFactory.h>
+
+#include "Swiftob/Commands.h"
+
+#define LUA_COMMANDS "__Lua_Commands"
+#define STORAGE "__Storage"
+
+LuaCommands::LuaCommands(Commands* commands, const std::string& path, Client* client, TimerFactory* timerFactory, MUCs* mucs) : path_(path), scriptsPath_(boost::filesystem::path(path_) / "scripts") {
+ commands_ = commands;
+ client_ = client;
+ timerFactory_ = timerFactory;
+ mucs_ = mucs;
+ commands_->onReset.connect(boost::bind(&LuaCommands::registerCommands, this));
+ registerCommands();
+}
+
+void LuaCommands::registerCommands() {
+ std::cout << "Trying to load all scripts in " << scriptsPath_ << std::endl;
+ if (boost::filesystem::exists(scriptsPath_) && boost::filesystem::is_directory(scriptsPath_)) {
+ std::vector<boost::filesystem::path> files;
+ copy(boost::filesystem::directory_iterator(scriptsPath_), boost::filesystem::directory_iterator(), std::back_inserter(files));
+ foreach (boost::filesystem::path file, files) {
+ if (boost::filesystem::is_regular_file(file) && file.extension() == ".lua") {
+ loadScript(file);
+ }
+ }
+ }
+}
+
+static int l_register_command(lua_State *L) {
+ LuaCommands* commands = NULL;
+ lua_getfield(L, LUA_REGISTRYINDEX, LUA_COMMANDS);
+ commands = static_cast<LuaCommands*>(lua_touserdata(L, -1));
+ lua_pop(L, 1);
+ if (!lua_isfunction(L, 4)) {
+ return luaL_error(L, "register_command callback parameter must be a function");
+ }
+ //luaL_ref callback(lua_to(L, 4));
+ lua_pushvalue(L, 4);
+ int callbackIndex = luaL_ref(L, LUA_REGISTRYINDEX);
+ lua_pop(L, 1);
+
+ if (!lua_isstring(L, 3)) {
+ return luaL_error(L, "register_command description parameter must be a string");
+ }
+ std::string description(lua_tostring(L, 3));
+ lua_pop(L, 1);
+
+ if (!lua_isstring(L, 2)) {
+ return luaL_error(L, "register_command allowed roles parameter must be a string");
+ }
+ std::string roleString(lua_tostring(L, 2));
+ lua_pop(L, 1);
+ Commands::RoleList roleList = Commands::Owner;
+ if (roleString == "Owner") {
+ roleList = Commands::Owner;
+ } else if (roleString == "Anyone") {
+ roleList = Commands::Anyone;
+ } else {
+ return luaL_error(L, "register_command allowed roles parameter has illegal value");
+ }
+ if (!lua_isstring(L, 1)) {
+ return luaL_error(L, "register_command command name parameter must be a string");
+ }
+ std::string name(lua_tostring(L, 1));
+ lua_pop(L, 1);
+ std::cout << "Registering lua command '" << name << "' for '" << roleString << "' with callback index " << callbackIndex << std::endl;
+ commands->getCommands()->registerCommand(name, roleList, description, boost::bind(&LuaCommands::handleLuaCommand, commands, callbackIndex, L, _1, _2, _3));
+
+ return 0;
+}
+
+static std::string luatable_asstring(lua_State *L, const char* key) {
+ lua_getfield(L, -1, key);
+ const char* valueChars = lua_tostring(L, -1);
+ std::string value(valueChars != NULL ? valueChars : "");
+ lua_pop(L, 1);
+ return value;
+}
+
+static int luatable_asint(lua_State *L, const char* key) {
+ lua_getfield(L, -1, key);
+ int value = lua_tointeger(L, -1);
+ lua_pop(L, 1);
+ return value;
+}
+
+static int luatable_asfunction(lua_State *L, const char* key) {
+ lua_getfield(L, -1, key);
+ int callbackIndex = luaL_ref(L, LUA_REGISTRYINDEX);
+ return callbackIndex;
+}
+
+static Message::ref messageFromTable(lua_State *L) {
+ Message::ref message(new Message());
+ message->setFrom(JID(luatable_asstring(L, "from")));
+ message->setBody(luatable_asstring(L, "body"));
+ Message::Type type = Message::Normal;
+ std::string typeString(luatable_asstring(L, "type"));
+ if (typeString == "normal") {
+ type = Message::Normal;
+ } else if (typeString == "chat") {
+ type = Message::Chat;
+ } else if (typeString == "groupchat") {
+ type = Message::Groupchat;
+ } else if (typeString == "error") {
+ type = Message::Error;
+ } else if (typeString == "headline") {
+ type = Message::Headline;
+ } else {
+ return Message::ref();
+ }
+ message->setType(type);
+ return message;
+}
+
+LuaCommands* LuaCommands::commandsFromLua(lua_State *L) {
+ LuaCommands* commands = NULL;
+ lua_getfield(L, LUA_REGISTRYINDEX, LUA_COMMANDS);
+ commands = static_cast<LuaCommands*>(lua_touserdata(L, -1));
+ lua_pop(L, 1);
+ return commands;
+}
+
+Storage* LuaCommands::storageFromLua(lua_State *L) {
+ Storage* storage = NULL;
+ lua_getfield(L, LUA_REGISTRYINDEX, STORAGE);
+ storage = static_cast<Storage*>(lua_touserdata(L, -1));
+ lua_pop(L, 1);
+ return storage;
+}
+
+static int l_reply_to(lua_State *L) {
+ LuaCommands* commands = LuaCommands::commandsFromLua(L);
+
+ if (!lua_isboolean(L, 3) && lua_gettop(L) > 2) {
+ return luaL_error(L, "reply_to parameter 3 must be boolean if present");
+ }
+ bool outOfMUC = lua_toboolean(L, 3);
+ if (lua_gettop(L) == 3) {
+ lua_pop(L, 1);
+ }
+
+ if (!lua_isstring(L, 2)) {
+ return luaL_error(L, "reply_to body parameter must be a string");
+ }
+ std::string body(lua_tostring(L, 2));
+ lua_pop(L, 1);
+
+ if (!lua_istable(L, 1)) {
+ return luaL_error(L, "reply_to message parameter must be a table");
+ }
+ lua_pushvalue(L, 1);
+ Message::ref message(messageFromTable(L));
+ if (!message) {
+ return luaL_error(L, "message parameter invalid");
+ }
+ commands->getCommands()->replyTo(message, body, outOfMUC);
+ lua_pop(L, 1);
+
+ return 0;
+}
+
+static int l_muc_input_to_jid(lua_State *L) {
+ LuaCommands* commands = LuaCommands::commandsFromLua(L);
+ return commands->muc_input_to_jid(L);
+}
+
+int LuaCommands::muc_input_to_jid(lua_State *L) {
+ if (!lua_isstring(L, 2)) {
+ return luaL_error(L, "must pass a string to muc_input_to_jid p2");
+ }
+ std::string source = lua_tostring(L, 2);
+ JID sourceJID(source);
+ lua_pop(L, 1);
+ if (!lua_isstring(L, 1)) {
+ return luaL_error(L, "must pass a string to muc_input_to_jid p1");
+ }
+ std::string input = lua_tostring(L, 1);
+ lua_pop(L, 1);
+ JID result(input);
+ if (mucs_->contains(sourceJID.toBare())) {
+ if (result.isBare() && result.getNode().empty()) {
+ if (mucs_->getMUC(sourceJID.toBare())->hasOccupant(input)) {
+ result = JID(sourceJID.getNode(), sourceJID.getDomain(), input);
+ }
+ }
+ }
+
+ lua_pushstring(L, result.isValid() ? result.toString().c_str() : "");
+ return 1;
+}
+
+void LuaCommands::handleSoftwareVersionResponse(boost::shared_ptr<SoftwareVersion> version, ErrorPayload::ref error, bool timeout, GetSoftwareVersionRequest::ref request, Timer::ref timer, lua_State* L, Callbacks callbacks) {
+ request->onResponse.disconnect_all_slots();
+ timer->onTick.disconnect_all_slots();
+ timer->stop();
+ int callback = callbacks.failure;
+ int stackCount = 0;
+ if (timeout) {
+ callback = callbacks.timeout;
+ } else if (version) {
+ callback = callbacks.success;
+ }
+ lua_rawgeti(L, LUA_REGISTRYINDEX, callback);
+ if (error) {
+ lua_pushstring(L, error->getText().empty() ? "No error text" : error->getText().c_str());
+ stackCount++;
+ }
+ else if (version) {
+ lua_createtable(L, 0, 3);
+ lua_pushstring(L, version->getName().c_str());
+ lua_setfield(L, -2, "name");
+ lua_pushstring(L, version->getVersion().c_str());
+ lua_setfield(L, -2, "version");
+ lua_pushstring(L, version->getOS().c_str());
+ lua_setfield(L, -2, "os");
+ stackCount++;
+ }
+ else {
+ lua_pushliteral(L, "Missing payload");
+ stackCount++;
+ }
+ int result = lua_pcall(L, stackCount, 0, 0);
+ if (result != 0) {
+ std::string error(lua_tostring(L, -1));
+ lua_pop(L, 1);
+ std::cout << error << std::endl;
+ callbacks.erase(L);
+ luaL_error(L, error.c_str());
+ } else {
+ callbacks.erase(L);
+ }
+
+}
+
+static int l_get_software_version(lua_State *L) {
+ LuaCommands* commands = LuaCommands::commandsFromLua(L);
+ return commands->get_software_version(L);
+}
+
+int LuaCommands::get_software_version(lua_State *L) {
+ if (!lua_istable(L, 1)) {
+ return luaL_error(L, "get_software_version requires a table parameter.");
+ }
+ lua_pushvalue(L, 1);
+ JID to(luatable_asstring(L, "to"));
+ if (!to.isValid()) {
+ return luaL_error(L, "invalid JID.");
+ }
+ int timeout = luatable_asint(L, "timeout");
+ if (timeout == 0) {
+ return luaL_error(L, "invalid timeout.");
+ }
+
+ int successCallback = luatable_asfunction(L, "success_callback");
+ int failureCallback = luatable_asfunction(L, "failure_callback");
+ int timeoutCallback = luatable_asfunction(L, "timeout_callback");
+ GetSoftwareVersionRequest::ref request = GetSoftwareVersionRequest::create(to, client_->getIQRouter());
+ Timer::ref timer = timerFactory_->createTimer(timeout * 1000);
+ Callbacks callbacks(successCallback, failureCallback, timeoutCallback);
+ request->onResponse.connect(boost::bind(&LuaCommands::handleSoftwareVersionResponse, this, _1, _2, false, request, timer, L, callbacks));
+ boost::shared_ptr<SoftwareVersion> fakePayload;
+ ErrorPayload::ref fakeError;
+ timer->onTick.connect(boost::bind(&LuaCommands::handleSoftwareVersionResponse, this, fakePayload, fakeError, true, request, timer, L, callbacks));
+ timer->start();
+ request->send();
+ return 1;
+}
+
+static int l_store_setting(lua_State *L) {
+ return LuaCommands::commandsFromLua(L)->store_setting(L);
+}
+
+static int l_get_setting(lua_State *L) {
+ return LuaCommands::commandsFromLua(L)->get_setting(L);
+}
+
+int LuaCommands::store_setting(lua_State *L) {
+ if (!lua_isstring(L, 2) || !lua_isstring(L, 1)) {
+ return luaL_error(L, "both setting and key must be strings");
+ }
+ std::string value(lua_tostring(L, 2));
+ std::string key(lua_tostring(L, 1));
+ lua_pop(L, 2);
+ storageFromLua(L)->saveSetting(key, value);
+ return 0;
+}
+
+int LuaCommands::get_setting(lua_State *L) {
+ if (!lua_isstring(L, 1)) {
+ return luaL_error(L, "key must be a string");
+ }
+ std::string key(lua_tostring(L, 1));
+ lua_pop(L, 1);
+ lua_pushstring(L, storageFromLua(L)->getSetting(key).c_str());
+ return 1;
+
+}
+
+void LuaCommands::handleLuaCommand(int callbackIndex, lua_State* L, const std::string& command, const std::string& params, Swift::Message::ref message) {
+ lua_rawgeti(L, LUA_REGISTRYINDEX, callbackIndex);
+ lua_pushstring(L, command.c_str());
+ lua_pushstring(L, params.c_str());
+ messageOntoStack(message, L);
+ int result = lua_pcall(L, 3, 0, 0);
+ if (result != 0) {
+ std::string error(lua_tostring(L, -1));
+ lua_pop(L, 1);
+ error = "Command '" + command + "' failed: " + error;
+ std::cout << error << std::endl;
+ commands_->replyTo(message, error, false);
+ }
+}
+
+void LuaCommands::messageOntoStack(Swift::Message::ref message, lua_State* L) {
+ lua_createtable(L, 0, 4);
+ std::string typeString;
+ switch (message->getType()) {
+ case Message::Chat : typeString = "chat";break;
+ case Message::Groupchat : typeString = "groupchat";break;
+ case Message::Normal : typeString = "normal";break;
+ case Message::Error : typeString = "error";break;
+ case Message::Headline : typeString = "headline";break;
+ }
+ lua_pushstring(L, typeString.c_str());
+ lua_setfield(L, -2, "type");
+ lua_pushstring(L, message->getFrom().toString().c_str());
+ lua_setfield(L, -2, "from");
+ lua_pushstring(L, message->getFrom().toBare().toString().c_str());
+ lua_setfield(L, -2, "frombare");
+ lua_pushstring(L, message->getTo().toString().c_str());
+ lua_setfield(L, -2, "to");
+ lua_pushstring(L, message->getBody().c_str());
+ lua_setfield(L, -2, "body");
+}
+
+void LuaCommands::loadScript(boost::filesystem::path filePath) {
+ std::cout << "Trying to load file from " << filePath << std::endl;
+ lua_State* lua = lua_open();
+ luaL_openlibs(lua);
+ lua_pushlightuserdata(lua, this);
+ lua_setfield(lua, LUA_REGISTRYINDEX, LUA_COMMANDS);
+ std::string filename(filePath.filename());
+ filename += ".storage";
+ boost::filesystem::path storagePath(boost::filesystem::path(path_) / filename);
+ Storage* storage = new Storage(storagePath);
+ lua_pushlightuserdata(lua, storage);
+ lua_setfield(lua, LUA_REGISTRYINDEX, STORAGE);
+ lua_register(lua, "swiftob_register_command", &l_register_command);
+ lua_register(lua, "swiftob_reply_to", &l_reply_to);
+ lua_register(lua, "swiftob_get_software_version", &l_get_software_version);
+ lua_register(lua, "swiftob_muc_input_to_jid", &l_muc_input_to_jid);
+ lua_register(lua, "swiftob_store_setting", &l_store_setting);
+ lua_register(lua, "swiftob_get_setting", &l_get_setting);
+ int fileLoaded = luaL_dofile(lua, filePath.string().c_str());
+ if (fileLoaded == 0 ) {
+ std::cout << "Loaded" << std::endl;
+ } else {
+ const char* error = lua_tostring(lua, -1);
+ std::cout << "Error: " << error << std::endl;
+ lua_pop(lua, -1);
+ }
+}
diff --git a/Swiftob/LuaCommands.h b/Swiftob/LuaCommands.h
new file mode 100644
index 0000000..dc8e36e
--- /dev/null
+++ b/Swiftob/LuaCommands.h
@@ -0,0 +1,93 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#pragma once
+
+#include <string>
+#include <vector>
+
+extern "C" {
+#include <lua.h>
+#include <lauxlib.h>
+#include <lualib.h>
+}
+
+#include <boost/filesystem/fstream.hpp>
+#include <boost/noncopyable.hpp>
+#include "Swiften/Network/NetworkFactories.h"
+#include <Swiften/Elements/SoftwareVersion.h>
+#include <Swiften/Queries/Requests/GetSoftwareVersionRequest.h>
+#include <Swiften/Network/Timer.h>
+
+#include "Swiftob/Commands.h"
+#include "Swiftob/Storage.h"
+
+using namespace Swift;
+/**
+ * Yes, there's an odd naming scheme going on here for methods.
+ * normalCamelCase methods are methods called from C++
+ * lower_case_methods are exposed to Lua through wrappers and
+ * l_lower_case_functions are functions directly exposed (often calling the lower_case_methods).
+ */
+class LuaCommands {
+ public:
+ class LuaCommand : public Commands::Command, boost::noncopyable {
+ public:
+ /** Takes ownership of lua and storage.*/
+ LuaCommand(Commands::RoleList allowedBy, const std::string& description, lua_State* lua, Storage* storage) : Command(allowedBy, description), lua_(lua), storage_(storage) {
+ }
+
+ virtual ~LuaCommand() {
+ lua_close(lua_);
+ delete storage_;
+ };
+
+ private:
+ lua_State* lua_;
+ Storage* storage_;
+ };
+
+ class Callbacks {
+ public:
+ Callbacks(int success, int failure, int timeout) : success(success), failure(failure), timeout(timeout) {};
+ int success;
+ int failure;
+ int timeout;
+ void erase(lua_State *L) {
+ lua_pushnil(L);
+ lua_rawseti(L, LUA_REGISTRYINDEX, success);
+ lua_pushnil(L);
+ lua_rawseti(L, LUA_REGISTRYINDEX, failure);
+ lua_pushnil(L);
+ lua_rawseti(L, LUA_REGISTRYINDEX, timeout);
+ }
+ };
+
+ LuaCommands(Commands* commands, const std::string& path, Client* client, TimerFactory* timerFactory, MUCs* mucs);
+ /* Public but isn't really part of the API */
+ void handleLuaCommand(int callbackIndex, lua_State* L, const std::string& command, const std::string& params, Message::ref message);
+ Commands* getCommands() {return commands_;}
+ int get_software_version(lua_State *L);
+ int muc_input_to_jid(lua_State *L);
+ int store_setting(lua_State *L);
+ int get_setting(lua_State *L);
+ static LuaCommands* commandsFromLua(lua_State *L);
+ static Storage* storageFromLua(lua_State *L);
+ private:
+ void registerCommands();
+ void loadScript(boost::filesystem::path filePath);
+ void messageOntoStack(Message::ref message, lua_State* L);
+ void handleSoftwareVersionResponse(boost::shared_ptr<SoftwareVersion> version, ErrorPayload::ref error, bool timeout, GetSoftwareVersionRequest::ref request, Timer::ref timer, lua_State* L, Callbacks callbacks);
+ private:
+ std::string path_;
+ boost::filesystem::path scriptsPath_;
+ Commands* commands_;
+ MUCs* mucs_;
+ Client* client_;
+ TimerFactory* timerFactory_;
+};
+
+
diff --git a/Swiftob/MUCs.cpp b/Swiftob/MUCs.cpp
new file mode 100644
index 0000000..55bf313
--- /dev/null
+++ b/Swiftob/MUCs.cpp
@@ -0,0 +1,116 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include "Swiftob/MUCs.h"
+
+#include <Swiften/Client/Client.h>
+#include <Swiften/MUC/MUC.h>
+#include <Swiften/MUC/MUCManager.h>
+#include <Swiften/Base/String.h>
+
+#include "Swiftob/Storage.h"
+
+#define MUC_LIST_SETTING "muc_list"
+
+typedef std::pair<JID, MUC::ref> JIDMUCPair;
+
+MUCs::MUCs(Client* client, Storage* storage) : defaultNick_("Kanchil+") {
+ client_ = client;
+ storage_ = storage;
+ client_->onConnected.connect(boost::bind(&MUCs::handleConnected, this));
+}
+
+void MUCs::handleConnected() {
+ foreach (std::string room, String::split(storage_->getSetting(MUC_LIST_SETTING), ' ')) {
+ join(JID(room), boost::bind(&MUCs::handleInitialJoinSuccess, this), boost::bind(&MUCs::handleInitialJoinFailure, this, _1));
+ }
+}
+
+void MUCs::handleInitialJoinSuccess() {
+
+}
+
+void MUCs::handleInitialJoinFailure(const std::string&) {
+
+}
+
+void MUCs::join(const JID& room, boost::signal<void (const std::string&)>::slot_type successCallback, boost::function<void(const std::string& /*reason*/)> failureCallback) {
+ if (contains(room)) {
+ failureCallback("Already in room");
+ }
+ mucs_[room] = client_->getMUCManager()->createMUC(room);
+ mucs_[room]->onJoinComplete.connect(successCallback);
+ mucs_[room]->onJoinFailed.connect(boost::bind(&MUCs::handleJoinFailed, this, room, _1, failureCallback));
+ mucs_[room]->joinWithContextSince(defaultNick_, boost::posix_time::microsec_clock::universal_time());
+ save();
+}
+
+void MUCs::part(const JID& room) {
+ if (!contains(room)) {
+ return;
+ }
+ mucs_[room]->part();
+}
+
+bool MUCs::contains(const JID& room) {
+ return mucs_.find(room) != mucs_.end();
+}
+
+void MUCs::handleJoinFailed(const JID& muc, ErrorPayload::ref error, boost::function<void(const std::string& /*reason*/)> failureCallback) {
+ std::string errorMessage("Couldn't join room");
+ if (error) {
+ switch (error->getCondition()) {
+ case ErrorPayload::Conflict:
+ // rejoinNick = nick_ + "_";
+ errorMessage += ": Nickname in use";
+ // errorMessage = str(format("Unable to enter this room as %1%, retrying as %2%") % nick_ % rejoinNick);
+ break;
+ case ErrorPayload::JIDMalformed:
+ errorMessage += ": ";
+ errorMessage += "No nickname specified";
+ break;
+ case ErrorPayload::NotAuthorized:
+ errorMessage += ": ";
+ errorMessage += "A password needed";
+ break;
+ case ErrorPayload::RegistrationRequired:
+ errorMessage += ": ";
+ errorMessage += "Only members may enter";
+ break;
+ case ErrorPayload::Forbidden:
+ errorMessage += ": ";
+ errorMessage += "You are banned from the room";
+ break;
+ case ErrorPayload::ServiceUnavailable:
+ errorMessage += ": ";
+ errorMessage += "The room is full";
+ break;
+ case ErrorPayload::ItemNotFound:
+ errorMessage += ": ";
+ errorMessage += "The room does not exist";
+ break;
+ default:
+ break;
+ }
+ if (!error->getText().empty()) {
+ errorMessage += " - " + error->getText();
+ }
+ }
+ mucs_.erase(muc);
+ failureCallback(errorMessage);
+}
+
+void MUCs::save() {
+ std::string concat;
+ foreach (JIDMUCPair pair, mucs_) {
+ concat += pair.first.toString() + " ";
+ }
+ storage_->saveSetting(MUC_LIST_SETTING, concat);
+}
+
+MUC::ref MUCs::getMUC(const JID& room) {
+ return (mucs_.find(room) != mucs_.end()) ? mucs_[room] : MUC::ref();
+}
diff --git a/Swiftob/MUCs.h b/Swiftob/MUCs.h
new file mode 100644
index 0000000..e727ec2
--- /dev/null
+++ b/Swiftob/MUCs.h
@@ -0,0 +1,47 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#pragma once
+
+#include <vector>
+#include <string>
+
+#include <Swiften/JID/JID.h>
+#include <Swiften/Elements/Message.h>
+#include <Swiften/MUC/MUC.h>
+#include <Swiften/MUC/MUCRegistry.h>
+
+namespace Swift {
+class Client;
+class MUC;
+}
+
+class Storage;
+
+using namespace Swift;
+
+class MUCs {
+ public:
+ MUCs(Client* client, Storage* storage);
+ void join(const JID& room, boost::signal<void (const std::string&)>::slot_type successCallback, boost::function<void(const std::string& /*reason*/)> failureCallback);
+ void part(const JID& room);
+ bool contains(const JID& room);
+ MUC::ref getMUC(const JID& room);
+ private:
+ void handleConnected();
+ void handleJoinFailed(const JID& room, ErrorPayload::ref error, boost::function<void(const std::string& /*reason*/)> failureCallback);
+ void handleInitialJoinSuccess();
+ void handleInitialJoinFailure(const std::string&);
+ void save();
+ private:
+ MUCRegistry registry_;
+ std::map<JID, MUC::ref> mucs_;
+ Client* client_;
+ Storage* storage_;
+ std::string defaultNick_;
+};
+
+
diff --git a/Swiftob/SConscript b/Swiftob/SConscript
new file mode 100644
index 0000000..2139822
--- /dev/null
+++ b/Swiftob/SConscript
@@ -0,0 +1,26 @@
+Import("env")
+
+
+if env["SCONS_STAGE"] == "build":
+ myenv = env.Clone()
+ myenv.MergeFlags(myenv["SWIFTEN_FLAGS"])
+ myenv.MergeFlags(myenv["LIBIDN_FLAGS"])
+ myenv.MergeFlags(myenv["BOOST_FLAGS"])
+ myenv.MergeFlags(myenv["ZLIB_FLAGS"])
+ myenv.MergeFlags(myenv["OPENSSL_FLAGS"])
+ myenv.MergeFlags(myenv.get("SQLITE_FLAGS", {}))
+ myenv.MergeFlags(myenv.get("LIBXML_FLAGS", ""))
+ myenv.MergeFlags(myenv.get("EXPAT_FLAGS", ""))
+ myenv.MergeFlags(myenv["PLATFORM_FLAGS"])
+ myenv.MergeFlags(myenv["LUA_FLAGS"])
+ sources = [
+ "linit.c", # This is horrible!
+ "Swiftob.cpp",
+ "Users.cpp",
+ "Commands.cpp",
+ "MUCs.cpp",
+ "Storage.cpp",
+ "LuaCommands.cpp",
+ "main.cpp"
+ ]
+ swiftob = myenv.Program("swiftob", sources) \ No newline at end of file
diff --git a/Swiftob/Storage.cpp b/Swiftob/Storage.cpp
new file mode 100644
index 0000000..0cf16d7
--- /dev/null
+++ b/Swiftob/Storage.cpp
@@ -0,0 +1,47 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include "Swiftob/Storage.h"
+
+#include <Swiften/Base/String.h>
+#include <Swiften/Base/ByteArray.h>
+#include <Swiften/Base/foreach.h>
+
+typedef std::pair<std::string, std::string> Strings;
+
+Storage::Storage(const std::string& path) : settingsPath_(boost::filesystem::path(path)) {
+ load();
+}
+
+Storage::Storage(const boost::filesystem::path& path) : settingsPath_(path) {
+ load();
+}
+
+void Storage::load() {
+ if (boost::filesystem::exists(settingsPath_)) {
+ Swift::ByteArray data;
+ data.readFromFile(settingsPath_.string());
+ foreach (std::string line, Swift::String::split(data.toString(), '\n')) {
+ std::pair<std::string, std::string> pair = Swift::String::getSplittedAtFirst(line, '\t');
+ settings_[pair.first] = pair.second;
+ }
+ }
+}
+
+void Storage::saveSetting(const std::string& setting, const std::string& value) {
+ settings_[setting] = value;
+ std::string settingsString;
+ foreach(Strings pair, settings_) {
+ settingsString += pair.first + '\t' + pair.second + '\n';
+ }
+ boost::filesystem::ofstream file(settingsPath_);
+ file << settingsString;
+ file.close();
+}
+
+std::string Storage::getSetting(const std::string& setting) {
+ return settings_[setting];
+}
diff --git a/Swiftob/Storage.h b/Swiftob/Storage.h
new file mode 100644
index 0000000..c4181d8
--- /dev/null
+++ b/Swiftob/Storage.h
@@ -0,0 +1,28 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#pragma once
+
+#include <map>
+#include <string>
+
+#include <boost/filesystem/fstream.hpp>
+#include <boost/filesystem/path.hpp>
+
+class Storage {
+ public:
+ Storage(const std::string& path);
+ Storage(const boost::filesystem::path& path);
+ void saveSetting(const std::string& setting, const std::string& value);
+ std::string getSetting(const std::string& setting);
+ private:
+ void load();
+ std::map<std::string, std::string> settings_;
+ std::string path_;
+ boost::filesystem::path settingsPath_;
+};
+
+
diff --git a/Swiftob/Swiftob.cpp b/Swiftob/Swiftob.cpp
new file mode 100644
index 0000000..6c6dad0
--- /dev/null
+++ b/Swiftob/Swiftob.cpp
@@ -0,0 +1,117 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include "Swiftob/Swiftob.h"
+
+#include <string>
+#include <iostream>
+#include <boost/bind.hpp>
+
+#include <Swiften/JID/JID.h>
+#include <Swiften/Base/String.h>
+#include <Swiften/Presence/PresenceSender.h>
+
+#include "Swiftob/Users.h"
+#include "Swiftob/Storage.h"
+
+
+po::options_description Swiftob::getOptionsDescription() {
+ po::options_description result("Options");
+ result.add_options()
+ ("path", po::value<std::string>(), "Configuration folder")
+ ("help", "produce help message")
+ ("init", "Reset everything (Really, everything, be careful, you only want to use this on first run).")
+ ("jid", po::value<std::string>(), "JID to use")
+ ("password", po::value<std::string>(), "password")
+ ("initial-owner", po::value<std::string>(), "Initial bot owner (JID)")
+ ;
+ return result;
+}
+
+Swiftob::Swiftob(const po::variables_map& options) : options_(options), networkFactories_(&eventLoop_), quitting_(false) {
+ std::string path;
+ path = options["path"].as<std::string>();
+ client_ = new Swift::Client(Swift::JID(options["jid"].as<std::string>()), options["password"].as<std::string>(), &networkFactories_);
+ storage_ = new Storage(boost::filesystem::path(path) / "settings.txt");
+ mucs_ = new MUCs(client_, storage_);
+ users_ = new Users(client_, mucs_);
+ commands_ = new Commands(users_, client_, storage_, mucs_);
+ lua_ = new LuaCommands(commands_, path, client_, networkFactories_.getTimerFactory(), mucs_);
+ client_->onConnected.connect(boost::bind(&Swiftob::handleConnected, this));
+ client_->onDisconnected.connect(boost::bind(&Swiftob::handleDisconnected, this, _1));
+ client_->onMessageReceived.connect(boost::bind(&Swiftob::handleMessageReceived, this, _1));
+ if (options_.count("init") > 0) {
+
+ } else if (options_.count("jid") > 0 || options_.count("password") > 0 || options_.count("initial-owner") == 0) {
+ std::cout << "Ignoring initial config options without --initial" << std::endl;
+ }
+ client_->setAlwaysTrustCertificates();
+ client_->setSoftwareVersion("Swiftob", "pregit", "");
+ client_->connect();
+ eventLoop_.run();
+}
+
+void Swiftob::handleConnected() {
+ std::cout << "Connected" << std::endl;
+ if (options_.count("init") > 0) {}{ /* FIXME: Not ready for persistence yet*/
+ users_->clearAll();
+ users_->addUser(Users::User(Swift::JID(options_["initial-owner"].as<std::string>()), Users::User::Owner));
+ }
+ Swift::Presence::ref presence(new Swift::Presence());
+ presence->setStatus("Online and botty");
+ client_->getPresenceSender()->sendPresence(presence);
+}
+
+void Swiftob::handleDisconnected(const boost::optional<Swift::ClientError>& /*error*/) {
+ std::cout << "Disconnected" << std::endl;
+ /* FIXME: check if last connect was more than a minute ago. If so, go ahead and connect, if not then wait a minute before connecting.*/
+ if (quitting_) {
+ eventLoop_.stop();
+ } else {
+ client_->connect();
+ }
+}
+
+void Swiftob::handleMessageReceived(Swift::Message::ref message) {
+ Swift::Message::Type type = message->getType();
+ if (type == Swift::Message::Error || type == Swift::Message::Headline) {
+ std::cout << "Ignoring typed message" << std::endl;
+ return;
+ }
+ std::string body = message->getBody();
+ std::cout << "Got message with body " << body << std::endl;
+ if (body.size() == 0) {
+ std::cout << "Not handling empty body" << std::endl;
+ return;
+ }
+ /*Convert body into !command if it's not a MUC, and it misses the bang*/
+ std::string bangBody(body);
+ if (type != Swift::Message::Groupchat && body[0] != '!') {
+ bangBody = "!" + body;
+ }
+ std::cout << "After banging, body is " << bangBody << std::endl;
+ std::pair<std::string, std::string> split = Swift::String::getSplittedAtFirst(bangBody, ' ');
+ std::string commandName(split.first);
+ commandName = Swift::String::getSplittedAtFirst(commandName, '!').second;
+ /*FIXME: remove leading bang in commandName*/
+ if (commands_->hasCommand(commandName)) {
+ std::cout << "Matched command " << commandName << std::endl;
+ commands_->runCommand(commandName, split.second, message);
+ }
+}
+
+Swiftob::~Swiftob() {
+ delete lua_;
+ delete commands_;
+ delete storage_;
+ delete users_;
+ delete client_;
+}
+
+int Swiftob::exec() {
+ return 0;
+}
+
diff --git a/Swiftob/Swiftob.h b/Swiftob/Swiftob.h
new file mode 100644
index 0000000..45061d0
--- /dev/null
+++ b/Swiftob/Swiftob.h
@@ -0,0 +1,49 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#pragma once
+
+#include <boost/program_options/variables_map.hpp>
+#include <boost/program_options/options_description.hpp>
+#include <boost/shared_ptr.hpp>
+
+#include <string>
+
+#include <Swiften/Base/Platform.h>
+#include <Swiften/EventLoop/SimpleEventLoop.h>
+#include <Swiften/Base/boost_bsignals.h>
+#include <Swiften/Client/Client.h>
+#include <Swiften/Network/BoostNetworkFactories.h>
+
+#include "Swiftob/Commands.h"
+#include "Swiftob/LuaCommands.h"
+
+namespace po = boost::program_options;
+
+class Users;
+class Storage;
+class Swiftob {
+ public:
+ Swiftob(const po::variables_map& options);
+ static po::options_description getOptionsDescription();
+ int exec();
+ ~Swiftob();
+ private:
+ void handleConnected();
+ void handleDisconnected(const boost::optional<Swift::ClientError>&);
+ void handleMessageReceived(Swift::Message::ref);
+ private:
+ const po::variables_map options_;
+ Swift::SimpleEventLoop eventLoop_;
+ Swift::BoostNetworkFactories networkFactories_;
+ Commands* commands_;
+ LuaCommands* lua_;
+ Storage* storage_;
+ MUCs* mucs_;
+ bool quitting_;
+ Users* users_;
+ Swift::Client* client_;
+};
diff --git a/Swiftob/Users.cpp b/Swiftob/Users.cpp
new file mode 100644
index 0000000..55ba4eb
--- /dev/null
+++ b/Swiftob/Users.cpp
@@ -0,0 +1,45 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include "Swiftob/Users.h"
+
+#include <iostream>
+
+#include <Swiften/Client/Client.h>
+
+#include "Swiftob/MUCs.h"
+
+Users::Users(Client* client, MUCs* mucs) {
+ client_ = client;
+ mucs_ = mucs;
+}
+
+/* TODO: Store in roster */
+void Users::clearAll() {
+ users_.clear();
+}
+
+void Users::addUser(const User& user) {
+ users_.push_back(user);
+}
+
+Users::User::Role Users::getRoleForSender(Message::ref message) {
+ JID jid = message->getFrom();
+ MUC::ref muc = mucs_->getMUC(message->getFrom().toBare());
+ if (muc && muc->hasOccupant(message->getFrom().getResource())) {
+ MUCOccupant occupant = muc->getOccupant(message->getFrom().getResource());
+ if (occupant.getRealJID()) {
+ jid = occupant.getRealJID().get();
+ }
+ }
+ foreach (User user, users_) {
+ if (user.getJID().equals(jid.toBare(), JID::WithoutResource)) {
+ return user.getRole();
+ }
+ }
+ return User::Unknown;
+}
+
diff --git a/Swiftob/Users.h b/Swiftob/Users.h
new file mode 100644
index 0000000..0acc330
--- /dev/null
+++ b/Swiftob/Users.h
@@ -0,0 +1,49 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#pragma once
+
+#include <vector>
+#include <string>
+
+#include <Swiften/JID/JID.h>
+#include <Swiften/Elements/Message.h>
+
+namespace Swift {
+class Client;
+}
+
+class MUCs;
+
+using namespace Swift;
+
+class Users {
+ public:
+ class User {
+ public:
+ /* If you add a role here, edit the role lists in Commands.cpp*/
+ enum Role {Unknown, Owner};
+ User(const JID& jid, Role role) : jid_(jid), role_(role) {}
+ Role getRole() {return role_;}
+ Swift::JID getJID() {return jid_;}
+ private:
+ Swift::JID jid_;
+ Role role_;
+ };
+
+ public:
+ Users(Client* client, MUCs* mucs);
+ void clearAll();
+ void addUser(const User& user);
+ User::Role getRoleForSender(Message::ref message);
+
+ private:
+ std::vector<User> users_;
+ Client* client_;
+ MUCs* mucs_;
+};
+
+
diff --git a/Swiftob/linit.c b/Swiftob/linit.c
new file mode 100644
index 0000000..13c5b09
--- /dev/null
+++ b/Swiftob/linit.c
@@ -0,0 +1 @@
+#include "../3rdParty/Lua/src/linit.c"
diff --git a/Swiftob/main.cpp b/Swiftob/main.cpp
new file mode 100644
index 0000000..9908b45
--- /dev/null
+++ b/Swiftob/main.cpp
@@ -0,0 +1,37 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include <boost/program_options/options_description.hpp>
+#include <boost/program_options/variables_map.hpp>
+#include <boost/program_options.hpp>
+#include <boost/version.hpp>
+#include <iostream>
+
+#include "Swiftob.h"
+
+int main(int argc, char* argv[]) {
+ boost::program_options::options_description desc = Swiftob::getOptionsDescription();
+ boost::program_options::variables_map vm;
+ try {
+ boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), vm);
+ } catch (boost::program_options::unknown_option option) {
+#if BOOST_VERSION >= 104200
+ std::cout << "Ignoring unknown option " << option.get_option_name() << " but continuing." << std::endl;
+#else
+ std::cout << "Error: " << option.what() << " (continuing)" << std::endl;
+#endif
+ }
+ boost::program_options::notify(vm);
+ if (vm.count("help") > 0) {
+ std::cout << desc << "\n";
+ return 1;
+ }
+
+ Swiftob bot(vm);
+ int result = bot.exec();
+
+ return result;
+}
diff --git a/Swiftob/scripts/agenda.lua b/Swiftob/scripts/agenda.lua
new file mode 100644
index 0000000..897b89c
--- /dev/null
+++ b/Swiftob/scripts/agenda.lua
@@ -0,0 +1,94 @@
+agendas = {}
+currents = {}
+
+function full_agenda(from)
+ fullagenda = {}
+ fullagenda[1] = "Roll call"
+ fullagenda[2] = "Agenda bashing"
+ for i, v in ipairs(agendas[from]) do
+ table.insert(fullagenda, v)
+ end
+ table.insert(fullagenda, "Date of next meeting")
+ table.insert(fullagenda, "Any other business")
+ return fullagenda
+end
+
+function agenda_full_command(command, params, message)
+ from = message['frombare']
+ ensure_loaded(from)
+ agenda = agendas[from]
+ fullagenda = full_agenda(from)
+ reply = ""
+ for i, v in ipairs(fullagenda) do
+ reply = reply..i..") "..v.."\n"
+ end
+ reply = reply.."Fini"
+ swiftob_reply_to(message, reply)
+end
+
+function agenda_append_command(command, params, message)
+ from = message['frombare']
+ agenda_append(from, params)
+ agenda_save(from)
+ swiftob_reply_to(message, "Done.")
+end
+
+function agenda_up_command(command, params, message)
+ from = message['frombare']
+ ensure_loaded(from)
+ up = tonumber(params)
+ if up == nil then up = 1 end
+ currents[from] = currents[from] + up
+ if currents[from] <= 0 then currents[from] = 1 end
+ item = full_agenda(from)[currents[from]]
+ if item == nil then item = "Fini." end
+ reply = currents[from]..") "..item
+ swiftob_reply_to(message, reply)
+end
+
+
+function agenda_clear_command(command, params, message)
+ from = message['frombare']
+ agendas[from] = {}
+ agenda_save(from)
+ swiftob_reply_to(message, "Done.")
+end
+
+function agenda_save(from)
+ agenda = agendas[from]
+ swiftob_store_setting("count@@@"..from, #agenda)
+ for i, v in ipairs(agenda) do
+ swiftob_store_setting(i.."@@@"..from, v)
+ end
+end
+
+function ensure_loaded(from)
+ if agendas[from] == nil then
+ agenda_load(from)
+ end
+end
+
+function agenda_load(from)
+ agendas[from] = {}
+ currents[from] = 0
+ num_items = tonumber(swiftob_get_setting("count@@@"..from))
+ if num_items == nil then num_items = 0 end
+ for i = 1, num_items do
+ agenda_append(from, swiftob_get_setting(i.."@@@"..from))
+ end
+end
+
+function agenda_append(from, item)
+ ensure_loaded(from)
+ agenda = agendas[from]
+ table.insert(agenda, item)
+ agendas[from] = agenda
+end
+
+swiftob_register_command("agenda", "Anyone", "print the full agenda", agenda_full_command)
+swiftob_register_command("agendaappend", "Owner", "append an item to the agenda", agenda_append_command)
+swiftob_register_command("agendaclear", "Owner", "clear the agenda", agenda_clear_command)
+swiftob_register_command("agendaup", "Owner", "Moves the current counter by n, and returns the current agenda item", agenda_up_command)
+
+
+
diff --git a/Swiftob/scripts/echo.lua b/Swiftob/scripts/echo.lua
new file mode 100644
index 0000000..7adc2b3
--- /dev/null
+++ b/Swiftob/scripts/echo.lua
@@ -0,0 +1,5 @@
+function echo_message(command, params, message)
+ swiftob_reply_to(message, params)
+end
+
+swiftob_register_command("echo", "Anyone", "What did you say?", echo_message) \ No newline at end of file
diff --git a/Swiftob/scripts/eval.lua b/Swiftob/scripts/eval.lua
new file mode 100644
index 0000000..c9840dc
--- /dev/null
+++ b/Swiftob/scripts/eval.lua
@@ -0,0 +1,14 @@
+
+function eval_command(command, params, message)
+ assert(loadstring(params))()
+ swiftob_reply_to(message, "done")
+end
+
+function evalr_command(command, params, message)
+ result = assert(loadstring(params))()
+ swiftob_reply_to(message, "" .. result)
+end
+
+swiftob_register_command("eval", "Owner", "Evaluate an expression", eval_command)
+swiftob_register_command("evalr", "Owner", "Evaluate an expression and return the result", evalr_command)
+
diff --git a/Swiftob/scripts/version.lua b/Swiftob/scripts/version.lua
new file mode 100644
index 0000000..40b4e4d
--- /dev/null
+++ b/Swiftob/scripts/version.lua
@@ -0,0 +1,32 @@
+function friendly_version(version)
+ result = version['name']
+ if version['version'] ~= nil and version['version'] ~= "" then
+ result = result.." version "..version['version']
+ end
+ if version['os'] ~= nil and version['os'] ~= "" then
+ result = result .." on "..version['os']
+ end
+ return result
+end
+
+function version_command(command, params, message)
+ jid = swiftob_muc_input_to_jid(params, message['from'])
+ if jid == nil then
+
+ else
+ swiftob_get_software_version({
+ to=jid,
+ timeout=10,
+ success_callback=function(result)
+ swiftob_reply_to(message, params.." is running "..friendly_version(result))
+ end,
+ failure_callback=function(error)
+ swiftob_reply_to(message, "Error getting version from "..params..": "..error)
+ end,
+ timeout_callback=function()
+ swiftob_reply_to(message, "Timeout waiting for version from "..params)
+ end})
+ end
+end
+
+swiftob_register_command("version", "Anyone", "Ask for someone's version", version_command) \ No newline at end of file