/*       +------------------------------------+
 *       | Inspire Internet Relay Chat Daemon
 *       +------------------------------------+
 *
 *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
 * See: http://wiki.inspircd.org/Credits
 *
 * This program is free but copyrighted software; see
 *            the file COPYING for details.
 *
 * ---------------------------------------------------
 */

#include "inspircd.h"
#include <stdarg.h>

/* $ModDesc: Allows for opered clients to join channels without being seen, similar to unreal 3.1 +I mode */

class InvisibleMode : public ModeHandler
{
 public:
	InvisibleMode(InspIRCd* Instance, Module* Creator) : ModeHandler(Creator, 'Q', PARAM_NONE, MODETYPE_USER)
	{
		oper = true;
	}

	~InvisibleMode()
	{
	}

	ModeAction OnModeChange(User* source, User* dest, Channel* channel, std::string &parameter, bool adding)
	{
		if (dest->IsModeSet('Q') != adding)
		{
			dest->SetMode('Q', adding);

			/* Fix for bug #379 reported by stealth. On +/-Q make m_watch think the user has signed on/off */
			Module* m = ServerInstance->Modules->Find("m_watch.so");

			/* This must come before setting/unsetting the handler */
			if (m && adding)
				m->OnUserQuit(dest, "Connection closed", "Connection closed");

			/* This has to come after setting/unsetting the handler */
			if (m && !adding)
				m->OnPostConnect(dest);

			/* User appears to vanish or appear from nowhere */
			for (UCListIter f = dest->chans.begin(); f != dest->chans.end(); f++)
			{
				const UserMembList *ulist = (*f)->GetUsers();
				char tb[MAXBUF];

				snprintf(tb,MAXBUF,":%s %s %s", dest->GetFullHost().c_str(), adding ? "PART" : "JOIN", (*f)->name.c_str());
				std::string out = tb;
				std::string n = this->ServerInstance->Modes->ModeString(dest, (*f));

				for (UserMembCIter i = ulist->begin(); i != ulist->end(); i++)
				{
					/* User only appears to vanish for non-opers */
					if (IS_LOCAL(i->first) && !IS_OPER(i->first))
					{
						i->first->Write(out);
						if (!n.empty() && !adding)
							i->first->WriteServ("MODE %s +%s", (*f)->name.c_str(), n.c_str());
					}
				}
			}

			ServerInstance->SNO->WriteToSnoMask('a', "\2NOTICE\2: Oper %s has become %svisible (%cQ)", dest->GetFullHost().c_str(), adding ? "in" : "", adding ? '+' : '-');
			return MODEACTION_ALLOW;
		}
		else
		{
			return MODEACTION_DENY;
		}
	}
};

class InvisibleDeOper : public ModeWatcher
{
 public:
	InvisibleDeOper(InspIRCd* Instance) : ModeWatcher(Instance, 'o', MODETYPE_USER)
	{
	}

	bool BeforeMode(User* source, User* dest, Channel* channel, std::string &param, bool adding, ModeType type)
	{
		/* Users who are opers and have +Q get their +Q removed when they deoper */
		if ((!adding) && (dest->IsModeSet('Q')))
		{
			std::vector<std::string> newmodes;
			newmodes.push_back(dest->nick);
			newmodes.push_back("-Q");
			ServerInstance->Modes->Process(newmodes, source);
		}
		return true;
	}
};


class ModuleInvisible : public Module
{
 private:
	InvisibleMode qm;
	InvisibleDeOper ido;
 public:
	ModuleInvisible(InspIRCd* Me)
		: Module(Me), qm(Me, this), ido(Me)
	{
		if (!ServerInstance->Modes->AddMode(&qm))
			throw ModuleException("Could not add new modes!");
		if (!ServerInstance->Modes->AddModeWatcher(&ido))
			throw ModuleException("Could not add new mode watcher on usermode +o!");

		/* Yeah i know people can take this out. I'm not about to obfuscate code just to be a pain in the ass. */
		ServerInstance->Users->ServerNoticeAll("*** m_invisible.so has just been loaded on this network. For more information, please visit http://inspircd.org/wiki/Modules/invisible");
		Implementation eventlist[] = {
			I_OnUserPreMessage, I_OnUserPreNotice, I_OnUserJoin, I_OnUserPart, I_OnUserQuit,
			I_OnHostCycle, I_OnSendWhoLine, I_OnNamesListItem
		};
		ServerInstance->Modules->Attach(eventlist, this, 8);
	};

	~ModuleInvisible()
	{
		ServerInstance->Modes->DelMode(&qm);
		ServerInstance->Modes->DelModeWatcher(&ido);
	};

	Version GetVersion();
	void OnUserJoin(Membership* memb, bool sync, bool created, CUList& excepts);
	void OnUserPart(Membership* memb, std::string &partmessage, CUList& excepts);
	void OnUserQuit(User* user, const std::string &reason, const std::string &oper_message);
	ModResult OnHostCycle(User* user);
	ModResult OnUserPreNotice(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list);
	ModResult OnUserPreMessage(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list);
	void OnSendWhoLine(User* source, User* user, Channel* channel, std::string& line);
	void OnNamesListItem(User* issuer, Membership* memb, std::string &prefixes, std::string &nick);
};

Version ModuleInvisible::GetVersion()
{
	return Version("Allows opers to join channels invisibly", VF_COMMON | VF_VENDOR);
}

static void BuildExcept(Membership* memb, CUList& excepts)
{
	const UserMembList* users = memb->chan->GetUsers();
	for(UserMembCIter i = users->begin(); i != users->end(); i++)
	{
		// hide from all local non-opers
		if (IS_LOCAL(i->first) && !IS_OPER(i->first))
			excepts.insert(i->first);
	}
}

void ModuleInvisible::OnUserJoin(Membership* memb, bool sync, bool created, CUList& excepts)
{
	if (memb->user->IsModeSet('Q'))
	{
		BuildExcept(memb, excepts);
		ServerInstance->SNO->WriteToSnoMask('a', "\2NOTICE\2: Oper %s has joined %s invisibly (+Q)",
			memb->user->GetFullHost().c_str(), memb->chan->name.c_str());
	}
}

void ModuleInvisible::OnUserPart(Membership* memb, std::string &partmessage, CUList& excepts)
{
	if (memb->user->IsModeSet('Q'))
	{
		BuildExcept(memb, excepts);
	}
}

void ModuleInvisible::OnUserQuit(User* user, const std::string &reason, const std::string &oper_message)
{
	if (user->IsModeSet('Q'))
	{
		Command* parthandler = ServerInstance->Parser->GetHandler("PART");
		std::vector<std::string> to_leave;
		if (parthandler)
		{
			for (UCListIter f = user->chans.begin(); f != user->chans.end(); f++)
					to_leave.push_back((*f)->name);
			/* We cant do this neatly in one loop, as we are modifying the map we are iterating */
			for (std::vector<std::string>::iterator n = to_leave.begin(); n != to_leave.end(); n++)
			{
				std::vector<std::string> parameters;
				parameters.push_back(*n);
				/* This triggers our OnUserPart, above, making the PART silent */
				parthandler->Handle(parameters, user);
			}
		}
	}
}

ModResult ModuleInvisible::OnHostCycle(User* user)
{
	return user->IsModeSet('Q') ? MOD_RES_DENY : MOD_RES_PASSTHRU;
}

/* No privmsg response when hiding - submitted by Eric at neowin */
ModResult ModuleInvisible::OnUserPreNotice(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
{
	if ((target_type == TYPE_USER) && (IS_LOCAL(user)))
	{
		User* target = (User*)dest;
		if(target->IsModeSet('Q') && !IS_OPER(user))
		{
			user->WriteNumeric(401, "%s %s :No such nick/channel",user->nick.c_str(), target->nick.c_str());
			return MOD_RES_DENY;
		}
	}
	return MOD_RES_PASSTHRU;
}

ModResult ModuleInvisible::OnUserPreMessage(User* user,void* dest,int target_type, std::string &text, char status, CUList &exempt_list)
{
	return OnUserPreNotice(user, dest, target_type, text, status, exempt_list);
}

void ModuleInvisible::OnSendWhoLine(User* source, User* user, Channel* channel, std::string& line)
{
	if (user->IsModeSet('Q') && !IS_OPER(source))
		line.clear();
}

void ModuleInvisible::OnNamesListItem(User* issuer, Membership* memb, std::string &prefixes, std::string &nick)
{
	if (memb->user->IsModeSet('Q') && !IS_OPER(issuer))
		nick.clear();
}

MODULE_INIT(ModuleInvisible)