/*

* All rights reserved.

* 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/SluiftClient.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/Disco/ClientDiscoManager.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 <Swiften/TLS/PKCS12Certificate.h>

#include <Sluift/Lua/FunctionRegistration.h>

SLUIFT_LUA_FUNCTION_WITH_HELP(

Client, set_version,

"Sets the published version of this client.",

"self",

"name the name of the client software\n"

"version the version identifier of this client\n"

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

) {

Sluift::globals.eventLoop.runOnce();

SluiftClient* client = getClient(L);

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

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

}

return 0;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Client, get_contacts,

"Returns a table of all the contacts in the contact list.",

"self\n",

""

) {

Sluift::globals.eventLoop.runOnce();

SluiftClient* client = getClient(L);

Lua::Table contactsTable;

std::string subscription;

switch(item.getSubscription()) {

case RosterItemPayload::None: subscription = "none"; break;

case RosterItemPayload::To: subscription = "to"; break;

case RosterItemPayload::From: subscription = "from"; break;

case RosterItemPayload::Both: subscription = "both"; break;

case RosterItemPayload::Remove: subscription = "remove"; break;

}

Lua::Table itemTable = boost::assign::map_list_of

("jid", boost::make_shared<Lua::Value>(item.getJID().toString()))

("name", boost::make_shared<Lua::Value>(item.getName()))

("subscription", boost::make_shared<Lua::Value>(subscription))

("groups", boost::make_shared<Lua::Value>(std::vector<Lua::Value>(item.getGroups().begin(), item.getGroups().end())));

contactsTable[item.getJID().toString()] = boost::make_shared<Lua::Value>(itemTable);

}

pushValue(L, contactsTable);

Lua::registerTableToString(L, -1);

return 1;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Client, 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"

const char* rawJID = lua_tostring(L, -1);

if (rawJID) {

item.setJID(std::string(rawJID));

}

lua_getfield(L, 2, "name");

const char* rawName = lua_tostring(L, -1);

if (rawName) {

item.setName(rawName);

}

lua_getfield(L, 2, "groups");

if (!lua_isnil(L, -1)) {

if (lua_type(L, -1) == LUA_TTABLE) {

for (size_t i = 1; i <= lua_objlen(L, -1); ++i) {

lua_rawgeti(L, -1, boost::numeric_cast<int>(i));

const char* rawGroup = lua_tostring(L, -1);

if (rawGroup) {

item.addGroup(rawGroup);

}

lua_pop(L, 1);

}

}

else {

throw Lua::Exception("Groups should be a table");

}

}

}

else {

item.setJID(Lua::checkString(L, 2));

}

if (!client->getClient()->getRoster()->containsJID(item.getJID())) {

RosterPayload::ref roster = boost::make_shared<RosterPayload>();

roster->addItem(item);

Sluift::Response response = client->sendVoidRequest(

SetRosterRequest::create(roster, client->getClient()->getIQRouter()), timeout);

if (response.error) {

return response.convertToLuaResult(L);

}

}

client->getClient()->getSubscriptionManager()->requestSubscription(item.getJID());

lua_pushboolean(L, true);

return 1;

}

SLUIFT_LUA_FUNCTION_WITH_HELP(

Client, remove_contact,

"Remove a contact from the contact list.",

"self\n"

"jid the JID of the contact to remove\n",

""

) {

Sluift::globals.eventLoop.runOnce();

SluiftClient* client = getClient(L);

JID jid(Lua::checkString(L, 2));

int timeout = getGlobalTimeout(L);

RosterPayload::ref roster = boost::make_shared<RosterPayload>();

roster->addItem(RosterItemPayload(JID(Lua::checkString(L, 2)), "", RosterItemPayload::Remove));