/*

* Copyright (c) 2014 Kevin Smith and Remko Tronçon

* Licensed under the GNU General Public License.

* See the COPYING file for more information.

*/

#include <boost/lambda/lambda.hpp>

#include <boost/lambda/bind.hpp>

#include <boost/assign/list_of.hpp>

#include <iostream>

#include <Sluift/SluiftComponent.h>

#include <Swiften/JID/JID.h>

#include <Swiften/Elements/SoftwareVersion.h>

#include <Swiften/Elements/Message.h>

#include <Swiften/Elements/Presence.h>

#include <Swiften/Elements/RawXMLPayload.h>

#include <Swiften/Elements/RosterItemPayload.h>

#include <Swiften/Elements/RosterPayload.h>

#include <Swiften/Elements/DiscoInfo.h>

#include <Swiften/Elements/MAMQuery.h>

#include <Swiften/Queries/GenericRequest.h>

#include <Swiften/Presence/PresenceSender.h>

#include <Swiften/Roster/XMPPRoster.h>

#include <Swiften/Roster/SetRosterRequest.h>

#include <Swiften/Presence/SubscriptionManager.h>

#include <Swiften/Roster/XMPPRosterItem.h>

#include <Swiften/Queries/IQRouter.h>

#include <Swiften/Queries/Requests/GetSoftwareVersionRequest.h>

#include <Sluift/Lua/FunctionRegistration.h>

#include <Swiften/Base/foreach.h>

#include <Swiften/Base/IDGenerator.h>

#include <Sluift/Lua/Check.h>

#include <Sluift/Lua/Value.h>

#include <Sluift/Lua/Exception.h>

#include <Sluift/Lua/LuaUtils.h>

#include <Sluift/globals.h>

#include <Sluift/ElementConvertors/StanzaConvertor.h>

#include <Sluift/ElementConvertors/IQConvertor.h>

#include <Sluift/ElementConvertors/PresenceConvertor.h>

#include <Sluift/ElementConvertors/MessageConvertor.h>

using namespace Swift;

namespace lambda = boost::lambda;

static inline SluiftComponent* getComponent(lua_State* L) {

return *Lua::checkUserData<SluiftComponent>(L, 1);

}

static inline int getGlobalTimeout(lua_State* L) {

lua_rawgeti(L, LUA_REGISTRYINDEX, Sluift::globals.moduleLibIndex);

lua_getfield(L, -1, "timeout");

int result = boost::numeric_cast<int>(lua_tointeger(L, -1));

lua_pop(L, 2);

return result;

}

static void addPayloadsToTable(lua_State* L, const std::vector<boost::shared_ptr<Payload> >& payloads) {

if (!payloads.empty()) {

lua_createtable(L, boost::numeric_cast<int>(payloads.size()), 0);

for (size_t i = 0; i < payloads.size(); ++i) {

Sluift::globals.elementConvertor.convertToLua(L, payloads[i]);

lua_rawseti(L, -2, boost::numeric_cast<int>(i+1));

}

Lua::registerGetByTypeIndex(L, -1);

lua_setfield(L, -2, "payloads");

}

}

static boost::shared_ptr<Payload> getPayload(lua_State* L, int index) {

if (lua_type(L, index) == LUA_TTABLE) {

return boost::dynamic_pointer_cast<Payload>(Sluift::globals.elementConvertor.convertFromLua(L, index));

}

else if (lua_type(L, index) == LUA_TSTRING) {

return boost::make_shared<RawXMLPayload>(Lua::checkString(L, index));

}

else {

return boost::shared_ptr<Payload>();

}

}

static std::vector< boost::shared_ptr<Payload> > getPayloadsFromTable(lua_State* L, int index) {

index = Lua::absoluteOffset(L, index);

std::vector< boost::shared_ptr<Payload> > result;

lua_getfield(L, index, "payloads");

if (lua_istable(L, -1)) {

for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) {

boost::shared_ptr<Payload> payload = getPayload(L, -1);

if (payload) {

result.push_back(payload);

}

}

}

lua_pop(L, 1);

return result;

}

SLUIFT_LUA_FUNCTION(Component, async_connect) {

SluiftComponent* component = getComponent(L);

std::string host;

int port = 0;

if (lua_istable(L, 2)) {

if (boost::optional<std::string> hostString = Lua::getStringField(L, 2, "host")) {

host = *hostString;

}

if (boost::optional<int> portInt = Lua::getIntField(L, 2, "port")) {

port = *portInt;

}

}

component->connect(host, port);

return 0;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Component, set_trace_enabled,

"Enable/disable tracing of the data sent/received.\n\n.",

"self\n"

"enable a boolean specifying whether to enable/disable tracing",

""

) {

getComponent(L)->setTraceEnabled(lua_toboolean(L, 1));

return 0;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Component, wait_connected,

"Block until the component is connected.\n\nThis is useful after an `async_connect`.",

"self",

""

) {

getComponent(L)->waitConnected(getGlobalTimeout(L));

return 0;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Component, is_connected,

"Checks whether this component is still connected.\n\nReturns a boolean.",

"self\n",

""

) {

lua_pushboolean(L, getComponent(L)->isConnected());

return 1;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Component, disconnect,

"Disconnect from the server",

"self\n",

""

) {

Sluift::globals.eventLoop.runOnce();

getComponent(L)->disconnect();

return 0;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Component, set_version,

"Sets the published version of this component.",

"self",

"name the name of the component software\n"

"version the version identifier of this component\n"

"os the OS this component is running on\n"

) {

Sluift::globals.eventLoop.runOnce();

SluiftComponent* component = getComponent(L);

if (boost::shared_ptr<SoftwareVersion> version = boost::dynamic_pointer_cast<SoftwareVersion>(Sluift::globals.elementConvertor.convertFromLuaUntyped(L, 2, "software_version"))) {

component->setSoftwareVersion(version->getName(), version->getVersion(), version->getOS());

}

return 0;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Component, send_message,

"Send a message.",

"self\n"

"to the JID to send the message to\n"

"body the body of the message. Can alternatively be specified using the `body` option\n",

"to the JID to send the message to\n"

"body the body of the message\n"

"subject the subject of the MUC room to set\n"

"type the type of message to send (`normal`, `chat`, `error`, `groupchat`, `headline`)\n"

"payloads payloads to add to the message\n"

) {

Sluift::globals.eventLoop.runOnce();

JID to;

boost::optional<std::string> from;

boost::optional<std::string> body;

boost::optional<std::string> subject;

std::vector<boost::shared_ptr<Payload> > payloads;

int index = 2;

Message::Type type = Message::Chat;

if (lua_isstring(L, index)) {

to = std::string(lua_tostring(L, index));

++index;

if (lua_isstring(L, index)) {

body = lua_tostring(L, index);

++index;

}

}

if (lua_istable(L, index)) {

if (boost::optional<std::string> value = Lua::getStringField(L, index, "to")) {

to = *value;

}

if (boost::optional<std::string> value = Lua::getStringField(L, index, "from")) {

from = value;

}

if (boost::optional<std::string> value = Lua::getStringField(L, index, "body")) {

body = value;

}

if (boost::optional<std::string> value = Lua::getStringField(L, index, "type")) {

type = MessageConvertor::convertMessageTypeFromString(*value);

}

if (boost::optional<std::string> value = Lua::getStringField(L, index, "subject")) {

subject = value;

}

payloads = getPayloadsFromTable(L, index);

}

if (!to.isValid()) {

throw Lua::Exception("Missing 'to'");

}

if ((!body || body->empty()) && !subject && payloads.empty()) {

throw Lua::Exception("Missing any of 'body', 'subject' or 'payloads'");

}

Message::ref message = boost::make_shared<Message>();

message->setTo(to);

if (from && !from->empty()) {

message->setFrom(*from);

}

if (body && !body->empty()) {

message->setBody(*body);

}

if (subject) {

message->setSubject(*subject);

}

message->addPayloads(payloads.begin(), payloads.end());

message->setType(type);

getComponent(L)->getComponent()->sendMessage(message);

return 0;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Component, send_presence,

"Send presence.",

"self\n"

"body the text of the presence. Can alternatively be specified using the `status` option\n",

"to the JID to send the message to\n"

"from the JID to send the message from\n"

"status the text of the presence\n"

"priority the priority of the presence\n"

"type the type of message to send (`available`, `error`, `probe`, `subscribe`, `subscribed`, `unavailable`, `unsubscribe`, `unsubscribed`)\n"

"payloads payloads to add to the presence\n"

) {

Sluift::globals.eventLoop.runOnce();

boost::shared_ptr<Presence> presence = boost::make_shared<Presence>();

int index = 2;

if (lua_isstring(L, index)) {

presence->setStatus(lua_tostring(L, index));

++index;

}

if (lua_istable(L, index)) {

if (boost::optional<std::string> value = Lua::getStringField(L, index, "to")) {

presence->setTo(*value);

}

if (boost::optional<std::string> value = Lua::getStringField(L, index, "from")) {

presence->setFrom(*value);

}

if (boost::optional<std::string> value = Lua::getStringField(L, index, "status")) {

presence->setStatus(*value);

}

if (boost::optional<int> value = Lua::getIntField(L, index, "priority")) {

presence->setPriority(*value);

}

if (boost::optional<std::string> value = Lua::getStringField(L, index, "type")) {

presence->setType(PresenceConvertor::convertPresenceTypeFromString(*value));

}

std::vector< boost::shared_ptr<Payload> > payloads = getPayloadsFromTable(L, index);

presence->addPayloads(payloads.begin(), payloads.end());

}

getComponent(L)->getComponent()->sendPresence(presence);

lua_pushvalue(L, 1);

return 0;

}

static int sendQuery(lua_State* L, IQ::Type type) {

SluiftComponent* component = getComponent(L);

JID to;

if (boost::optional<std::string> toString = Lua::getStringField(L, 2, "to")) {

to = JID(*toString);

}

JID from;

if (boost::optional<std::string> fromString = Lua::getStringField(L, 2, "from")) {

from = JID(*fromString);

}

int timeout = getGlobalTimeout(L);

if (boost::optional<int> timeoutInt = Lua::getIntField(L, 2, "timeout")) {

timeout = *timeoutInt;

}

boost::shared_ptr<Payload> payload;

lua_getfield(L, 2, "query");

payload = getPayload(L, -1);

lua_pop(L, 1);

return component->sendRequest(

boost::make_shared< GenericRequest<Payload> >(type, from, to, payload, component->getComponent()->getIQRouter()), timeout).convertToLuaResult(L);

}

#define DISPATCH_PUBSUB_PAYLOAD(payloadType, container, response) \

else if (boost::shared_ptr<payloadType> p = boost::dynamic_pointer_cast<payloadType>(payload)) { \

return component->sendPubSubRequest(type, to, p, timeout).convertToLuaResult(L); \

}

SLUIFT_LUA_FUNCTION(Component, get) {

return sendQuery(L, IQ::Get);

}

SLUIFT_LUA_FUNCTION(Component, set) {

return sendQuery(L, IQ::Set);

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Component, send,

"Sends a raw string",

"self\n"

"data the string to send\n",

""

) {

Sluift::globals.eventLoop.runOnce();

getComponent(L)->getComponent()->sendData(std::string(Lua::checkString(L, 2)));

lua_pushvalue(L, 1);

return 0;

}

static void pushEvent(lua_State* L, const SluiftComponent::Event& event) {

switch (event.type) {

case SluiftComponent::Event::MessageType: {

Message::ref message = boost::dynamic_pointer_cast<Message>(event.stanza);

Lua::Table result = boost::assign::map_list_of

("type", boost::make_shared<Lua::Value>(std::string("message")))

("from", boost::make_shared<Lua::Value>(message->getFrom().toString()))

("to", boost::make_shared<Lua::Value>(message->getTo().toString()))

("body", boost::make_shared<Lua::Value>(message->getBody()))

("message_type", boost::make_shared<Lua::Value>(MessageConvertor::convertMessageTypeToString(message->getType())));

Lua::pushValue(L, result);

addPayloadsToTable(L, message->getPayloads());

Lua::registerTableToString(L, -1);

break;

}

case SluiftComponent::Event::PresenceType: {

Presence::ref presence = boost::dynamic_pointer_cast<Presence>(event.stanza);

Lua::Table result = boost::assign::map_list_of

("type", boost::make_shared<Lua::Value>(std::string("presence")))

("from", boost::make_shared<Lua::Value>(presence->getFrom().toString()))

("to", boost::make_shared<Lua::Value>(presence->getTo().toString()))

("status", boost::make_shared<Lua::Value>(presence->getStatus()))

("presence_type", boost::make_shared<Lua::Value>(PresenceConvertor::convertPresenceTypeToString(presence->getType())));

Lua::pushValue(L, result);

addPayloadsToTable(L, presence->getPayloads());

Lua::registerTableToString(L, -1);

break;

}

}

}

struct CallUnaryLuaPredicateOnEvent {

CallUnaryLuaPredicateOnEvent(lua_State* L, int index) : L(L), index(index) {

}

bool operator()(const SluiftComponent::Event& event) {

lua_pushvalue(L, index);

pushEvent(L, event);

if (lua_pcall(L, 1, 1, 0) != 0) {

throw Lua::Exception(lua_tostring(L, -1));

}

bool result = lua_toboolean(L, -1);

lua_pop(L, 1);

return result;

}

lua_State* L;

int index;

};

SLUIFT_LUA_FUNCTION(Component, get_next_event) {

Sluift::globals.eventLoop.runOnce();

SluiftComponent* component = getComponent(L);

int timeout = getGlobalTimeout(L);

boost::optional<SluiftComponent::Event::Type> type;

int condition = 0;

if (lua_istable(L, 2)) {

if (boost::optional<std::string> typeString = Lua::getStringField(L, 2, "type")) {

if (*typeString == "message") {

type = SluiftComponent::Event::MessageType;

}

else if (*typeString == "presence") {

type = SluiftComponent::Event::PresenceType;

}

}

if (boost::optional<int> timeoutInt = Lua::getIntField(L, 2, "timeout")) {

timeout = *timeoutInt;

}

lua_getfield(L, 2, "if");

if (lua_isfunction(L, -1)) {

condition = Lua::absoluteOffset(L, -1);

}

}

boost::optional<SluiftComponent::Event> event;

if (condition) {

event = component->getNextEvent(timeout, CallUnaryLuaPredicateOnEvent(L, condition));

}

else if (type) {

event = component->getNextEvent(

timeout, lambda::bind(&SluiftComponent::Event::type, lambda::_1) == *type);

}

else {

event = component->getNextEvent(timeout);

}

if (event) {

pushEvent(L, *event);

}

else {

lua_pushnil(L);

}

return 1;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Component, jid,

"Returns the JID of this component",

"self\n",

""

) {

SluiftComponent* component = getComponent(L);

lua_pushstring(L, component->getComponent()->getJID().toString().c_str());

return 1;

}

SLUIFT_LUA_FUNCTION(Component, __gc) {

SluiftComponent* component = getComponent(L);

delete component;

return 0;

}