// Transport includes

#include "transport/config.h"

#include "transport/networkplugin.h"

#include "transport/logging.h"

// Yahoo2

#include <yahoo2.h>

#include <yahoo2_callbacks.h>

#include <stdio.h>

#include <stdarg.h>

#include <stdlib.h>

#include "yahoohandler.h"

#include "yahoolocalaccount.h"

#include "httpfetch.h"

// Swiften

#include "Swiften/Swiften.h"

#include "Swiften/Network/TLSConnectionFactory.h"

#include "Swiften/TLS/OpenSSL/OpenSSLContextFactory.h"

// for signal handler

#include "unistd.h"

#include "signal.h"

#include "sys/wait.h"

#include "sys/signal.h"

// Boost

#include <boost/algorithm/string.hpp>

using namespace boost::filesystem;

using namespace boost::program_options;

using namespace Transport;

class YahooHandler;

class YahooLocalAccount;

static std::string *currently_read_data;

static YahooLocalAccount *currently_writting_account;

YahooHandler::YahooHandler(YahooLocalAccount *account, int conn_tag, int handler_tag, void *data, yahoo_input_condition cond) :

	handler_tag(handler_tag), conn_tag(conn_tag), data(data), cond(cond), remove_later(false), account(account) {}

YahooHandler::~YahooHandler() {}

void YahooHandler::ready(std::string *buffer) {

	if (cond == YAHOO_INPUT_WRITE) {

		YahooLocalAccount *old = currently_writting_account;

		currently_writting_account = account;

		yahoo_write_ready(account->id, (void *) conn_tag, data);

		currently_writting_account = old;

	}

	else {

		if (!buffer) {

			return;

		}

		// yahoo_read_ready calls ext_yahoo_read(...) in a loop, so we just have to choose proper buffer from which will

		// that method read. We do that by static currently_read_data pointer.

		currently_read_data = buffer;

		// libyahoo2 reads data per 1024 bytes, so if we still have some data after the first ext_yahoo_read call,

		// we have to call yahoo_read_ready again...

		do {

			yahoo_read_ready(account->id, (void *) conn_tag, data);

		} while (currently_read_data->size() != 0);

	}

}

typedef struct {

	std::string yahoo_id;

	std::string name;

	int status;

	int away;

	std::string msg;

	std::string group;

	yc.ext_yahoo_conf_message = ext_yahoo_conf_message;

	yc.ext_yahoo_chat_cat_xml = ext_yahoo_chat_cat_xml;

	yc.ext_yahoo_chat_join = ext_yahoo_chat_join;

	yc.ext_yahoo_chat_userjoin = ext_yahoo_chat_userjoin;

	yc.ext_yahoo_chat_userleave = ext_yahoo_chat_userleave;

	yc.ext_yahoo_chat_message = ext_yahoo_chat_message;

	yc.ext_yahoo_chat_yahoologout = ext_yahoo_chat_yahoologout;

	yc.ext_yahoo_chat_yahooerror = ext_yahoo_chat_yahooerror;

	yc.ext_yahoo_got_webcam_image = ext_yahoo_got_webcam_image;

	yc.ext_yahoo_webcam_invite = ext_yahoo_webcam_invite;

	yc.ext_yahoo_webcam_invite_reply = ext_yahoo_webcam_invite_reply;

	yc.ext_yahoo_webcam_closed = ext_yahoo_webcam_closed;

	yc.ext_yahoo_webcam_viewer = ext_yahoo_webcam_viewer;

	yc.ext_yahoo_webcam_data_request = ext_yahoo_webcam_data_request;

	yc.ext_yahoo_got_file = ext_yahoo_got_file;

	yc.ext_yahoo_got_ft_data = ext_yahoo_got_ft_data;

	yc.ext_yahoo_get_ip_addr = ext_yahoo_get_ip_addr;

	yc.ext_yahoo_file_transfer_done = ext_yahoo_file_transfer_done;

	yc.ext_yahoo_contact_added = ext_yahoo_contact_added;

	yc.ext_yahoo_rejected = ext_yahoo_rejected;

	yc.ext_yahoo_typing_notify = ext_yahoo_typing_notify;

	yc.ext_yahoo_game_notify = ext_yahoo_game_notify;

	yc.ext_yahoo_mail_notify = ext_yahoo_mail_notify;

	yc.ext_yahoo_got_search_result = ext_yahoo_got_search_result;

	yc.ext_yahoo_system_message = ext_yahoo_system_message;

	yc.ext_yahoo_error = ext_yahoo_error;

	yc.ext_yahoo_log = ext_yahoo_log;

	yc.ext_yahoo_add_handler = ext_yahoo_add_handler;

	yc.ext_yahoo_remove_handler = ext_yahoo_remove_handler;

	yc.ext_yahoo_connect = ext_yahoo_connect;

	yc.ext_yahoo_connect_async = ext_yahoo_connect_async;

	yc.ext_yahoo_read = ext_yahoo_read;

	yc.ext_yahoo_write = ext_yahoo_write;

	yc.ext_yahoo_close = ext_yahoo_close;

	yc.ext_yahoo_got_buddyicon = ext_yahoo_got_buddyicon;

	yc.ext_yahoo_got_buddyicon_checksum = ext_yahoo_got_buddyicon_checksum;

	yc.ext_yahoo_buddyicon_uploaded = ext_yahoo_buddyicon_uploaded;

	yc.ext_yahoo_got_buddyicon_request = ext_yahoo_got_buddyicon_request;

	yc.ext_yahoo_got_ping = ext_yahoo_got_ping;

	yc.ext_yahoo_got_buddy_change_group = ext_yahoo_got_buddy_change_group;

	yahoo_register_callbacks(&yc);

}

int main (int argc, char* argv[]) {

	std::string host;

	int port;

	if (signal(SIGCHLD, spectrum_sigchld_handler) == SIG_ERR) {

		std::cout << "SIGCHLD handler can't be set\n";

		return -1;

	}

	boost::program_options::options_description desc("Usage: spectrum [OPTIONS] <config_file.cfg>\nAllowed options");

	desc.add_options()

		("host,h", value<std::string>(&host), "host")

		("port,p", value<int>(&port), "port")

		;

	try

	{

		boost::program_options::variables_map vm;

		boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), vm);

		boost::program_options::notify(vm);

	}

	catch (std::runtime_error& e)

	{

		std::cout << desc << "\n";

		exit(1);

	}

	catch (...)

	{

		std::cout << desc << "\n";

		exit(1);

	}

	if (argc < 5) {

		return 1;

	}

	Config config;

	if (!config.load(argv[5])) {

		std::cerr << "Can't open " << argv[1] << " configuration file.\n";

		return 1;

	}

	Logging::initBackendLogging(&config);

	register_callbacks();

	yahoo_set_log_level(YAHOO_LOG_DEBUG);

	Swift::SimpleEventLoop eventLoop;

	loop_ = &eventLoop;

	np = new YahooPlugin(&config, &eventLoop, host, port);

	loop_->run();

	return 0;

}