/*

 * Copyright (C) 2008-2009 J-P Nurmi jpnurmi@gmail.com

 *

 * This example is free, and not covered by LGPL license. There is no

 * restriction applied to their modification, redistribution, using and so on.

 * You can study them, modify them, use them in your own program - either

 * completely or partially. By using it you may give me some credits in your

 * program, but you don't have to.

 */

#include "session.h"

#include <QtCore>

#include <iostream>

#include <IrcCommand>

#include <IrcMessage>

#include "ircnetworkplugin.h"

#define FROM_UTF8(WHAT) QString::fromUtf8((WHAT).c_str(), (WHAT).size())

#define TO_UTF8(WHAT) std::string((WHAT).toUtf8().data(), (WHAT).toUtf8().size())

#include "transport/logging.h"

DEFINE_LOGGER(logger, "IRCSession");

static bool sentList;

MyIrcSession::MyIrcSession(const std::string &user, IRCNetworkPlugin *np, const std::string &suffix, QObject* parent) : IrcSession(parent)

{

	this->np = np;

	this->user = user;

	this->suffix = suffix;

	m_connected = false;

	rooms = 0;

	connect(this, SIGNAL(disconnected()), SLOT(on_disconnected()));

	connect(this, SIGNAL(socketError(QAbstractSocket::SocketError)), SLOT(on_socketError(QAbstractSocket::SocketError)));

	connect(this, SIGNAL(connected()), SLOT(on_connected()));

	connect(this, SIGNAL(messageReceived(IrcMessage*)), this, SLOT(onMessageReceived(IrcMessage*)));

	m_awayTimer = new QTimer(this);

	connect(m_awayTimer, SIGNAL(timeout()), this, SLOT(awayTimeout()));

}

MyIrcSession::~MyIrcSession() {

	delete m_awayTimer;

}

void MyIrcSession::on_connected() {

	m_connected = true;

	if (suffix.empty()) {

		np->handleConnected(user);

// 		if (!sentList) {

// 			sendCommand(IrcCommand::createList("", ""));

// 			sentList = true;

// 		}

	}

// 	sendCommand(IrcCommand::createCapability("REQ", QStringList("away-notify")));

	for(AutoJoinMap::iterator it = m_autoJoin.begin(); it != m_autoJoin.end(); it++) {

		sendCommand(IrcCommand::createJoin(FROM_UTF8(it->second->getChannel()), FROM_UTF8(it->second->getPassword())));

	}

	if (getIdentify().find(" ") != std::string::npos) {

		std::string to = getIdentify().substr(0, getIdentify().find(" "));

		std::string what = getIdentify().substr(getIdentify().find(" ") + 1);

		sendCommand(IrcCommand::createMessage(FROM_UTF8(to), FROM_UTF8(what)));

	}

}

void MyIrcSession::on_socketError(QAbstractSocket::SocketError error) {

	on_disconnected();

}

void MyIrcSession::on_disconnected() {

	if (suffix.empty()) {

		np->handleDisconnected(user, 0, "");

		np->tryNextServer();

	}

	m_connected = false;

}

bool MyIrcSession::correctNickname(std::string &nickname) {

	bool flags = 0;

	switch(nickname.at(0)) {

		case '@': nickname = nickname.substr(1); flags = 1; break;

		case '+': nickname = nickname.substr(1); break;

		default: break;

	}