/*
 * InspIRCd -- Internet Relay Chat Daemon
 *
 *   Copyright (C) 2017-2023 Sadie Powell <sadie@witchery.services>
 *   Copyright (C) 2012-2013 Attila Molnar <attilamolnar@hush.com>
 *   Copyright (C) 2012 Robby <robby@chatbelgie.be>
 *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
 *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
 *   Copyright (C) 2008 Dennis Friis <peavey@inspircd.org>
 *   Copyright (C) 2007 Craig Edwards <brain@inspircd.org>
 *   Copyright (C) 2006 Oliver Lupton <om@inspircd.org>
 *
 * This file is part of InspIRCd.  InspIRCd 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, version 2.
 *
 * 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, see <http://www.gnu.org/licenses/>.
 */


#include "inspircd.h"
#include "listmode.h"
#include "modules/extban.h"
#include "modules/isupport.h"

enum
{
	// From RFC 2812.
	RPL_EXCEPTLIST = 348,
	RPL_ENDOFEXCEPTLIST = 349
};

class BanException final
	: public ListModeBase
{
private:
	ExtBan::ManagerRef extbanmgr;

public:
	BanException(Module* Creator)
		: ListModeBase(Creator, "banexception", 'e', RPL_EXCEPTLIST, RPL_ENDOFEXCEPTLIST)
		, extbanmgr(Creator)
	{
		syntax = "<mask>";
	}

	bool CompareEntry(const std::string& entry, const std::string& value) const override
	{
		if (extbanmgr)
		{
			auto res = extbanmgr->CompareEntry(this, entry, value);
			if (res != ExtBan::Comparison::NOT_AN_EXTBAN)
				return res == ExtBan::Comparison::MATCH;
		}
		return irc::equals(entry, value);
	}

	bool ValidateParam(LocalUser* user, Channel* channel, std::string& parameter) override
	{
		if (!extbanmgr || !extbanmgr->Canonicalize(parameter))
			ModeParser::CleanMask(parameter);
		return true;
	}
};

class ModuleBanException final
	: public Module
	, public ExtBan::EventListener
	, public ISupport::EventListener
{
private:
	BanException be;

public:
	ModuleBanException()
		: Module(VF_VENDOR, "Adds channel mode e (banexception) which allows channel operators to exempt user masks from channel mode b (ban).")
		, ExtBan::EventListener(this)
		, ISupport::EventListener(this)
		, be(this)
	{
	}

	void OnBuildISupport(ISupport::TokenMap& tokens) override
	{
		tokens["EXCEPTS"] = ConvToStr(be.GetModeChar());
	}

	ModResult OnExtBanCheck(User* user, Channel* chan, ExtBan::Base* extban, bool full) override
	{
		ListModeBase::ModeList* list = be.GetList(chan);
		if (!list)
			return MOD_RES_PASSTHRU;

		for (const auto& ban : *list)
		{
			bool inverted;
			std::string name;
			std::string value;
			if (!ExtBan::Parse(ban.mask, name, value, inverted))
				continue;

			if (name.size() == 1)
			{
				// It is an extban but not this extban.
				if (name[0] != extban->GetLetter())
					continue;
			}
			else
			{
				// It is an extban but not this extban.
				if (!irc::equals(name, extban->GetName()))
					continue;
			}

			return extban->IsMatch(user, chan, value, full) != inverted ? MOD_RES_ALLOW : MOD_RES_PASSTHRU;
		}
		return MOD_RES_PASSTHRU;
	}

	ModResult OnCheckChannelBan(User* user, Channel* chan, bool full) override
	{
		ListModeBase::ModeList* list = be.GetList(chan);
		if (!list)
		{
			// No list, proceed normally
			return MOD_RES_PASSTHRU;
		}

		for (const auto& entry : *list)
		{
			if (chan->CheckBan(user, entry.mask, full))
			{
				// They match an entry on the list, so let them in.
				return MOD_RES_ALLOW;
			}
		}
		return MOD_RES_PASSTHRU;
	}

	void ReadConfig(ConfigStatus& status) override
	{
		be.DoRehash();
	}
};

MODULE_INIT(ModuleBanException)