IRCBot.cpp

/********************************************************************************
 Copyright (C) 2014 Append Huang <Append@gmail.com>

 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
********************************************************************************/

#include <iostream>
#include <cwchar>
#include <string>
#include "constants.h"
#include "Log.h"
#include "wstringfunc.h"
#include "SimpleSocket.h"
#include "SimpleThread.h"
#include "IRCMsgThread.h"
#include "IRCBot.h"



/********************
* BASIC IRC Actions *
********************/    
   
bool IRCBot::connectTask(std::wstring server, std::wstring port, std::wstring nickname, std::wstring login, std::wstring password){ //true for connect
    // Procedure for Connecting to the IRC server.
	// return false for default, true only if successfully connected.
		iStatus=BOT_CONNECTING;
		//int iResult;
		if ( TheSocket->InitializeSocket(server,port) ) // Initialize Winsock //0 for success, 1 for error 
			onDebugMsg(L"IRCBot.connectTask: Error in socket initialization");
			
		if ( TheSocket->ConnectSocket() ) // Connect to server //0 for success, 1 for error
			onDebugMsg(L"IRCBot.connectTask: Error in establishing socket connection");

		else { //Successfully Connected
			wchar_t buffer[256]=L"";
			swprintf_s(buffer,256,L"Connected to %ls(%ls):%ls",server.c_str(),TheSocket->SocketIP().c_str(),port.c_str());
            onSysMsg(L"%ls",buffer);
			                
            // Log on to the server, must sent before creating thread
			std::wstring msg=L"";
            if(password!=L""){
				msg+=L"PASS " + password + L"\r\n";
                log(L"PASS ******", LogType(SEND));
            }
            if(nickname!=(L"")){
                msg+=L"NICK " + nickname + L"\r\n";
                log(L"NICK " + nickname, LogType(SEND));
            }
            if(login!=(L"")){
                msg+=L"USER " + login + L"r\n";
                log(L"USER " + login, LogType(SEND));
            	}
			std::string narrowmsg = to_utf8(msg);
            TheSocket->send_data((char*)narrowmsg.c_str(),(unsigned int)narrowmsg.length()); //Sending Message without queue.
            loginSuccessful = false; //Not known yet; Initialize for further checking.
			
            //setup input and output thread 
            threadRunning = true; 
			ircmsgThread->iStatus=IRC_CLOSED;
            ircmsgThread->StartThread();
            return true; //True for connect
        }
    return false;
}    

void IRCBot::IRC_join(std::wstring channel) { ircmsgThread->sendRaw(L"JOIN "+channel); }
void IRCBot::IRC_chat(std::wstring channel, std::wstring message) { ircmsgThread->sendRaw(L"PRIVMSG " + channel + L" :" + message); } 

IRCBot::IRCBot(){
	iStatus=BOT_CLOSED;
	threadRunning = false; //not for reconnect Task
	loginSuccessful = false;
	
	TheSocket = new SimpleSocket;
	
	ircmsgThread = new IRCMsgThread;
	ircmsgThread->SetIRCBot(this);								
	ircmsgThread->SetSocket(TheSocket);
	ircmsgThread->iStatus=IRC_CLOSED;
	//Timer Settings;
//	aliveCheckTask = new SimpleTimer ; 
//	aliveCheckTask->TickFunc=std::bind(&IRCBot::AliveCheckTask,this);
//	aliveCheckTask->SetTickTime(5000);            

//	reconnectTask = new SimpleTimer ;
//	reconnectTask->TickFunc=std::bind(&IRCBot::ReconnectTask,this);
//	reconnectTask->SetTickTime(5000); 

	lastIRCServer = L"";
	lastIRCPort = L"";
	lastIRCServerPass = L"";
	lastIRCNickname = L"";
	lastIRCLogin = L"";
	lastIRCChannel = L"";
}

IRCBot::~IRCBot(){
	close();
	delete ircmsgThread;
	delete TheSocket ;
}

IRCBotStatus IRCBot::AliveCheckTask() {
	switch (ircmsgThread->iStatus) { 
		case IRC_DISCONNECTED: 	//IRCMsgThread: Disconnect Unexpectly
			onAccidentDisconnection();
			break;
		case IRC_WRONGLOGIN:	//IRCMsgThread: Wrong Login Information
			onLoginFailed();
			break;
		case IRC_CLOSED:		//StopThread
			break;
		case IRC_NORMAL:		//Still Running
			ircmsgThread->sendRaw(L"PING");
			break;
	} 
	if(iStatus==BOT_CONNECTED) LoginCheckTask();
	return iStatus;
}

void IRCBot::LoginCheckTask() {
	if(!loginSuccessful){
		onSysMsg(L"Incorrect login information");
		onLoginFailed();
	}
}

bool IRCBot::isConnected(){
    if(TheSocket->isSocketInvalid()) return false;
    else if( TheSocket->isConnected() && ircmsgThread->iStatus==IRC_NORMAL ) return true;
    else return false;
}

void IRCBot::reconnect(){
	bool bConnect=connectTask(lastIRCServer,lastIRCPort,lastIRCNickname,lastIRCLogin,lastIRCServerPass);
    if(bConnect) { //success 
        onConnectSuccess();
	} else {
		onSysMsg(L"Reconnect Failed; wait for next Tick");
		iStatus=BOT_NEEDRECONNECT;
	}
}

void IRCBot::connect(std::wstring server, std::wstring port, std::wstring nickname, std::wstring login, std::wstring password, std::wstring channel){
	if(iStatus!=BOT_CLOSED) close();
    lastIRCServer = server;
    lastIRCPort = port;
    lastIRCServerPass = password;
    lastIRCNickname = nickname;
    lastIRCLogin = login;
	lastIRCChannel = channel;
    iStatus=BOT_NEEDRECONNECT;	//reconnect();
}

void IRCBot::close(){
		iStatus=BOT_CLOSING;
        if(threadRunning) {
				threadRunning = false;
				//close threads
				if(ircmsgThread->iStatus==IRC_NORMAL) {
					onSysMsg(TEXT("IRCBot.close: Closing IRCBot-ircmsgThread"));
					ircmsgThread->sendRaw(L"QUIT :Leaving");
					ircmsgThread->interruptSignal();
					Sleep(100);
					TheSocket->CloseSocket();
					ircmsgThread->StopThread();
				}
				onSysMsg(L"IRCBot.close: All threads closed, IRCBot Stopped");
		}
		iStatus=BOT_CLOSED;
}		


void IRCBot::onLoginFailed(){ //Call from LoginCheckTask
	onSysMsg(L"Failed to Login");
	close();
}

void IRCBot::onLoginSuccess(){ //Call from IRCMsgThread
	onSysMsg(L"Login Successfully");
    ircmsgThread->sendRaw(L"CAP REQ :twitch.tv/tags");
	IRC_join(lastIRCChannel);
}
  
void IRCBot::onAccidentDisconnection(){ //the connection is closed by accident
	onSysMsg(L"Accidentally disconnected from server: Reconnecting...");
	close();
	iStatus=BOT_NEEDRECONNECT;
}

void IRCBot::onConnectSuccess(){
	onSysMsg(L"Connect Tasks Successfully Finished");
	//Sleep(500);
	//setup disconnection checking timer
	//if(ircmsgThread->iStatus!=IRC_NORMAL)	 { 
		//ircmsgThread will drop very quickly if login failed
		//if connection dropped, the bot status should still be CONNECTING.
	//	AliveCheckTask(); //Determine the message
	//}
	//else{ //ircmsgThread is running
	Sleep(500); 
	iStatus=BOT_CONNECTED;	
	//LoginCheckTask(); //Check for once
	//}
}

bool IRCBot::receiveMsg(TircMsg &ircmsg) {return ircmsgThread->receiveMsg(ircmsg);}
bool IRCBot::QueueEmpty() {return ircmsgThread->QueueEmpty();}
IRCBotStatus IRCBot::CheckIRCBotStatus(){ return iStatus;}