#include "TwitterPlugin.h"

#include "Requests/StatusUpdateRequest.h"

#include "Requests/DirectMessageRequest.h"

#include "Requests/TimelineRequest.h"

#include "Requests/FetchFriends.h"

#include "Requests/HelpMessageRequest.h"

#include "Requests/PINExchangeProcess.h"

#include "Requests/OAuthFlow.h"

#include "Requests/CreateFriendRequest.h"

#include "Requests/DestroyFriendRequest.h"

#include "Requests/RetweetRequest.h"

#include "Requests/ProfileImageRequest.h"

#include "Swiften/StringCodecs/Hexify.h"

DEFINE_LOGGER(logger, "Twitter Backend");

TwitterPlugin *np = NULL;

Swift::SimpleEventLoop *loop_; // Event Loop

const std::string OLD_APP_KEY = "PCWAdQpyyR12ezp2fVwEhw";

const std::string OLD_APP_SECRET = "EveLmCXJIg2R7BTCpm6OWV8YyX49nI0pxnYXh7JMvDg";

#define abs(x) ((x)<0?-(x):(x))

#define SHA(x) (Swift::Hexify::hexify(Swift::SHA1::getHash(Swift::createByteArray((x)))))

//Compares two +ve intergers 'a' and 'b' represented as strings

static int cmp(std::string a, std::string b)

{

	int diff = abs((int)a.size() - (int)b.size());

	if(a.size() < b.size()) a = std::string(diff,'0') + a;

	else b = std::string(diff,'0') + b;

	if(a == b) return 0;

	if(a < b) return -1;

	return 1;

}

TwitterPlugin::TwitterPlugin(Config *config, Swift::SimpleEventLoop *loop, StorageBackend *storagebackend, const std::string &host, int port) : NetworkPlugin()

{

	this->config = config;

	this->storagebackend = storagebackend;

	this->m_firstPing = true;

	if (CONFIG_HAS_KEY(config, "twitter.consumer_key") == false) {

		consumerKey = "5mFePMiJi0KpeURONkelg";

	}

	else {

		consumerKey = CONFIG_STRING(config, "twitter.consumer_key");

	}

	if (CONFIG_HAS_KEY(config, "twitter.consumer_secret") == false) {

		consumerSecret = "YFZCDJwRhbkccXEnaYr1waCQejTJcOY8F7l5Wim3FA";

	}

	else {

		consumerSecret = CONFIG_STRING(config, "twitter.consumer_secret");

	}

	if (consumerSecret.empty() || consumerKey.empty()) {

		LOG4CXX_ERROR(logger, "Consumer key and Consumer secret can't be empty.");

		exit(1);

	}

	adminLegacyName = "twitter.com";

	adminChatRoom = "#twitter";

	adminNickName = "twitter";

	adminAlias = "twitter";

	OAUTH_KEY = "twitter_oauth_token";

	OAUTH_SECRET = "twitter_oauth_secret";

	MODE = "mode";

	m_factories = new Swift::BoostNetworkFactories(loop);

	m_conn = m_factories->getConnectionFactory()->createConnection();

	m_conn->onDataRead.connect(boost::bind(&TwitterPlugin::_handleDataRead, this, _1));

	m_conn->connect(Swift::HostAddressPort(SWIFT_HOSTADDRESS(host), port));

	tp = new ThreadPool(loop_, 10);

	LOG4CXX_INFO(logger, "Fetch timeout is set to " << CONFIG_INT_DEFAULTED(config, "twitter.fetch_timeout", 90000));

	tweet_timer = m_factories->getTimerFactory()->createTimer(CONFIG_INT_DEFAULTED(config, "twitter.fetch_timeout", 90000));

	message_timer = m_factories->getTimerFactory()->createTimer(CONFIG_INT_DEFAULTED(config, "twitter.fetch_timeout", 90000));

	tweet_timer->onTick.connect(boost::bind(&TwitterPlugin::pollForTweets, this));

	tweet_timer->start();

	message_timer->start();

#if HAVE_SWIFTEN_3

		cryptoProvider = SWIFTEN_SHRPTR_NAMESPACE::shared_ptr<Swift::CryptoProvider>(Swift::PlatformCryptoProvider::create());

#endif

	LOG4CXX_INFO(logger, "Starting the plugin.");

}

TwitterPlugin::~TwitterPlugin()

{

	delete storagebackend;

	std::set<std::string>::iterator it;

	for(it = onlineUsers.begin() ; it != onlineUsers.end() ; it++) delete userdb[*it].sessions;

	delete tp;

}

// Send data to NetworkPlugin server

void TwitterPlugin::sendData(const std::string &string)

{

	m_conn->write(Swift::createSafeByteArray(string));

}

// Receive date from the NetworkPlugin server and invoke the appropirate payload handler (implement in the NetworkPlugin class)

void TwitterPlugin::_handleDataRead(SWIFTEN_SHRPTR_NAMESPACE::shared_ptr<Swift::SafeByteArray> data)

{

	if (m_firstPing) {

		m_firstPing = false;

		// Users can join the network without registering if we allow

		// one user to connect multiple IRC networks.

		NetworkPlugin::PluginConfig cfg;

		cfg.setNeedPassword(false);

		sendConfig(cfg);

	}

	std::string d(data->begin(), data->end());

	handleDataRead(d);

}

// User trying to login into his twitter account

void TwitterPlugin::handleLoginRequest(const std::string &user, const std::string &legacyName, const std::string &password)

{

	if(userdb.count(user) && (userdb[user].connectionState == NEW ||

										userdb[user].connectionState == CONNECTED ||

										userdb[user].connectionState == WAITING_FOR_PIN)) {

		LOG4CXX_INFO(logger, std::string("A session corresponding to ") + user + std::string(" is already active"));

		return;

	}

	LOG4CXX_INFO(logger, std::string("Received login request for ") + user);

	initUserSession(user, legacyName, password);

	handleConnected(user);

	LOG4CXX_INFO(logger, "SPECTRUM 1 USER? - " << (userdb[user].spectrum1User? "true" : "false"));

	LOG4CXX_INFO(logger, user << ": Adding Buddy " << adminLegacyName << " " << adminAlias);

	handleBuddyChanged(user, adminLegacyName, adminAlias, std::vector<std::string>(), pbnetwork::STATUS_ONLINE);

	userdb[user].nickName = "";

	LOG4CXX_INFO(logger, "Querying database for usersettings of " << user);

	std::string key, secret;

	getUserOAuthKeyAndSecret(user, key, secret);

	if(key == "" || secret == "") {

		LOG4CXX_INFO(logger, "Intiating OAuth Flow for user " << user);

		setTwitterMode(user, 0);

		tp->runAsThread(new OAuthFlow(np, userdb[user].sessions, user, userdb[user].sessions->getTwitterUsername()));

	} else {

		LOG4CXX_INFO(logger, user << " is already registerd. Using the stored oauth key and secret");

		LOG4CXX_INFO(logger, key << " " << secret);

		pinExchangeComplete(user, key, secret);

	}

}

// User logging out

void TwitterPlugin::handleLogoutRequest(const std::string &user, const std::string &legacyName)

{

	if (userdb.count(user)) {

		delete userdb[user].sessions;

		userdb[user].sessions = NULL;

		userdb[user].connectionState = DISCONNECTED;

	}

	if(onlineUsers.count(user)) {

		onlineUsers.erase(user);

	}

}

// User joining a Chatroom

void TwitterPlugin::handleJoinRoomRequest(const std::string &user, const std::string &room, const std::string &nickname, const std::string &password)

{

	if(room == adminChatRoom) {

		LOG4CXX_INFO(logger, "Received Join Twitter room request for " << user << " '" << nickname << "'");