/***************************************************************************
 *   Copyright (C) 2004 by Janos Horvath                                   *
 *   bourne@freemail.hu                                                    *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU Library 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 Library 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 "cpilua.h"
#include <verlihub/stringutils.h>
#include <verlihub/cbanlist.h>
#include <verlihub/cserverdc.h>
#include <dirent.h>
#include <string>
#include <config.h>

using namespace nStringUtils;

cpiLua::cpiLua() : mConsole(this), mQuery(NULL)
{
	mName = LUA_PI_IDENTIFIER;
	mVersion = VERSION;
}

cpiLua::~cpiLua()
{
	mQuery->Clear();
	delete mQuery;
	this->Empty();
}

void cpiLua::OnLoad(cServerDC *server)
{
	cVHPlugin::OnLoad(server);
	mQuery = new cQuery(server->mMySQL);
	mScriptDir = mServer->mConfigBaseDir + "/scripts/";
	AutoLoad();
}

bool cpiLua::RegisterAll()
{
	RegisterCallBack("VH_OnNewConn");
	RegisterCallBack("VH_OnCloseConn");
	RegisterCallBack("VH_OnParsedMsgChat");
	RegisterCallBack("VH_OnParsedMsgPM");
	RegisterCallBack("VH_OnParsedMsgSearch");
	RegisterCallBack("VH_OnParsedMsgConnectToMe");
	RegisterCallBack("VH_OnParsedMsgRevConnectToMe");
	RegisterCallBack("VH_OnParsedMsgSR");
	RegisterCallBack("VH_OnParsedMsgMyINFO");
	RegisterCallBack("VH_OnParsedMsgValidateNick");
	RegisterCallBack("VH_OnParsedMsgAny");
	RegisterCallBack("VH_OnParsedMsgSupport");
	RegisterCallBack("VH_OnParsedMsgMyPass");
	RegisterCallBack("VH_OnUnknownMsg");
	RegisterCallBack("VH_OnOperatorCommand");
	RegisterCallBack("VH_OnOperatorKicks");
	RegisterCallBack("VH_OnOperatorDrops");
	RegisterCallBack("VH_OnValidateTag");
	RegisterCallBack("VH_OnUserCommand");
	RegisterCallBack("VH_OnUserLogin");
	RegisterCallBack("VH_OnUserLogout");
	RegisterCallBack("VH_OnTimer");
	RegisterCallBack("VH_OnNewReg");
	RegisterCallBack("VH_OnNewBan");
}

bool cpiLua::AutoLoad()
{
	if(Log(0)) LogStream() << "Open dir: " << mScriptDir << endl;
	string pathname, filename;
	DIR *dir = opendir(mScriptDir.c_str());
	if(!dir)
	{
		if(Log(1)) LogStream() << "Open dir error" << endl;
		return false;
	}
	struct dirent *dent = NULL;
	
	while(NULL != (dent=readdir(dir)))
	{
		filename = dent->d_name;
		if((filename.size() > 4) && (StrCompare(filename,filename.size()-4,4,".lua")==0))
		{
			pathname = mScriptDir + filename;
			cLuaInterpreter *ip = new cLuaInterpreter(pathname);
			if(ip)
			{
				if(ip->Init())
				{
					AddData(ip);
					if(Log(1)) LogStream() << "Success loading and parsing LUA script: " << filename << endl;
				}
				else
				{
					if(Log(1)) LogStream() << "Failed loading or parsing LUA script: " << filename << endl;
					delete ip;
				}
			}
		}
	}
	
	closedir(dir);
	return true;
}

bool cpiLua::CallAll(char * fncname, char * args[])
{
	bool result, ret = true;
	if(Size())
	{
		tvLuaInterpreter::iterator it;
		for(it = mLua.begin(); it != mLua.end(); ++it)
		{
			result = (*it)->CallFunction(fncname, args);
			if(!result)
				ret = false;
		}
	}
	return ret;
}

bool cpiLua::OnNewConn(cConnDC *conn)
{
	if(conn != NULL)
	{
		char * args[] = {	(char *)conn->AddrIP().c_str(),
					NULL };
		
		return CallAll("VH_OnNewConn", args);
	}
	return true;
}

bool cpiLua::OnCloseConn(cConnDC *conn)
{
	if(conn != NULL)
	{
		char * args[] = {	(char *)conn->AddrIP().c_str(),
					NULL };
		
		return CallAll("VH_OnCloseConn", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgChat(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->ChunkString(eCH_CH_MSG).c_str(),
					NULL }; // eCH_CH_ALL, eCH_CH_NICK, eCH_CH_MSG
		
		return CallAll("VH_OnParsedMsgChat", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgPM(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->ChunkString(eCH_PM_MSG).c_str(),
					(char *) msg->ChunkString(eCH_PM_TO).c_str(),
					NULL }; // eCH_PM_ALL, eCH_PM_TO, eCH_PM_FROM, eCH_PM_CHMSG, eCH_PM_NICK, eCH_PM_MSG
		
		return CallAll("VH_OnParsedMsgPM", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgSupport(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->mStr.c_str(),
					NULL };
		
		return CallAll("VH_OnParsedMsgSupport", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgMyPass(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->ChunkString(eCH_1_ALL).c_str(),
					NULL }; // eCH_1_ALL, eCH_1_PARAM
		
		return CallAll("VH_OnParsedMsgMyPass", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgRevConnectToMe(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->ChunkString(eCH_RC_OTHER).c_str(),
					NULL };
		
		return CallAll("VH_OnParsedMsgRevConnectToMe", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgConnectToMe(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->ChunkString(eCH_CM_NICK).c_str(),
					(char *)msg->ChunkString(eCH_CM_IP).c_str(),
					(char *)msg->ChunkString(eCH_CM_PORT).c_str(),
					NULL }; // eCH_CM_NICK, eCH_CM_ACTIVE, eCH_CM_IP, eCH_CM_PORT
		
		return CallAll("VH_OnParsedMsgConnectToMe", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgSearch(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->ChunkString(eCH_AS_ALL).c_str(),
					NULL }; // active: eCH_AS_ALL, eCH_AS_ADDR, eCH_AS_IP, eCH_AS_PORT, eCH_AS_QUERY; passive: eCH_PS_ALL, eCH_PS_NICK, eCH_PS_QUERY
		
		return CallAll("VH_OnParsedMsgSearch", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgSR(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->ChunkString(eCH_SR_ALL).c_str(),
					NULL }; // eCH_SR_ALL, eCH_SR_FROM, eCH_SR_PATH, eCH_SR_SIZE, eCH_SR_SLOTS, eCH_SR_SL_FR, eCH_SR_SL_TO, eCH_SR_HUBINFO, eCH_SR_TO
		
		return CallAll("VH_OnParsedMsgSR", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgMyINFO(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->ChunkString(eCH_MI_ALL).c_str(),
					NULL }; // eCH_MI_ALL, eCH_MI_DEST, eCH_MI_NICK, eCH_MI_INFO, eCH_MI_DESC, eCH_MI_SPEED, eCH_MI_MAIL, eCH_MI_SIZE
		
		return CallAll("VH_OnParsedMsgMyINFO", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgValidateNick(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)msg->ChunkString(eCH_1_ALL).c_str(),
					NULL }; // eCH_1_ALL, eCH_1_PARAM
		
		return CallAll("VH_OnParsedMsgValidateNick", args);
	}
	return true;
}

bool cpiLua::OnParsedMsgAny(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(), // send nick instead of IP
					(char *)msg->mStr.c_str(),
					NULL };
	
		return CallAll("VH_OnParsedMsgAny", args);
	}
	return true;
}

bool cpiLua::OnUnknownMsg(cConnDC *conn, cMessageDC *msg)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (msg != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)msg->mStr.c_str(),
					NULL };
		
		return CallAll("VH_OnUnknownMsg", args);
	}
	return true;
}

bool cpiLua::OnOperatorCommand(cConnDC *conn, string *command)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (command != NULL))
	{
		if(mConsole.DoCommand(*command, conn)) return false;
	
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)command->c_str(),
					NULL };
		
		return CallAll("VH_OnOperatorCommand", args);
	}
	return true;
}

bool cpiLua::OnOperatorKicks(cUser *OP, cUser *user, string *reason)
{
	if((OP != NULL) && (user !=NULL) && (reason != NULL))
	{
		char * args[] = {	(char *)OP->mNick.c_str(),
					(char *)user->mNick.c_str(),
					(char *)reason->c_str(),
					NULL };
		
		return CallAll("VH_OnOperatorKicks", args);
	}
	return true;
}

bool cpiLua::OnOperatorDrops(cUser *OP, cUser *user)
{
	if((OP != NULL) && (user != NULL))
	{
		char * args[] = {	(char *)OP->mNick.c_str(),
					(char *)user->mNick.c_str(),
					NULL };
		
		return CallAll("VH_OnOperatorDrops", args);
	}
	return true;
}

bool cpiLua::OnUserCommand(cConnDC *conn, string *command)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (command != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)command->c_str(),
					NULL };
		
		return CallAll("VH_OnUserCommand", args);
	}
	return true;
}

bool cpiLua::OnValidateTag(cConnDC *conn, cDCTag *tag)
{
	if((conn != NULL) && (conn->mpUser != NULL) && (tag != NULL))
	{
		char * args[] = {	(char *)conn->mpUser->mNick.c_str(),
					(char *)tag->mTag.c_str(), // string mTag, string mTagBody, double mClientVersion
					NULL };
		
		return CallAll("VH_OnValidateTag", args);
	}
	return true;
}

bool cpiLua::OnUserLogin(cUser *user)
{
	if(user != NULL)
	{
		char * args[] = {	(char *)user->mNick.c_str(),
					NULL };
		
		return CallAll("VH_OnUserLogin", args);
	}
	return true;
}

bool cpiLua::OnUserLogout(cUser *user)
{
	if(user != NULL)
	{
		char * args[] = {	(char *)user->mNick.c_str(),
					NULL };
		
		return CallAll("VH_OnUserLogout", args);
	}
	return true;
}

bool cpiLua::OnTimer()
{
	char * args[] = { NULL };
	return CallAll("VH_OnTimer", args);
}

bool cpiLua::OnNewReg(cRegUserInfo *reginfo)
{
//string mRegOp, string mNick, string mPasswd
	if(reginfo != NULL)
	{
		char * args[] = {	(char *)reginfo->mRegOp.c_str(),
					(char *)reginfo->mNick.c_str(),
					(char *)reginfo->mClass,
					NULL };
					
		return CallAll("VH_OnNewReg", args);
	}
	return true;
}

bool cpiLua::OnNewBan(cBan *ban)
{
//long mDateStart, long mDateEnd, int mType, unsigned long mRangeMin, unsigned long mRangeMax, string mMail, __int64 mShare, string mHost, string mNick, string mIP
	if(ban != NULL)
	{
		char * args[] = {	(char *)ban->mNickOp.c_str(),
					(char *)ban->mIP.c_str(),
					(char *)ban->mNick.c_str(),
					(char *)ban->mReason.c_str(),
					NULL };
					
		return CallAll("VH_OnNewBan", args);
	}
	return true;
}

REGISTER_PLUGIN(cpiLua);