Switched cs_set to the subcommand system and added cs_saset

1 files changed, 116 insertions, 0 deletions

diff --git a/src/core/cs_set_restricted.cpp b/src/core/cs_set_restricted.cpp
new file mode 100644
index 000000000..7e0cc42b3
--- /dev/null
+++ b/src/core/cs_set_restricted.cpp
@@ -0,0 +1,116 @@
+/* ChanServ core functions
+ *
+ * (C) 2003-2010 Anope Team
+ * Contact us at team@anope.org
+ *
+ * Please read COPYING and README for further details.
+ *
+ * Based on the original code of Epona by Lara.
+ * Based on the original code of Services by Andy Church.
+ *
+ * $Id$
+ *
+ */
+/*************************************************************************/
+
+#include "module.h"
+
+class CommandCSSetRestricted : public Command
+{
+ public:
+	CommandCSSetRestricted(const ci::string &cname, const std::string &cpermission = "") : Command(cname, 2, 2, cpermission)
+	{
+	}
+
+	CommandReturn Execute(User *u, const std::vector<ci::string> &params)
+	{
+		ChannelInfo *ci = cs_findchan(params[0]);
+		assert(ci);
+
+		if (params[1] == "ON")
+		{
+			ci->SetFlag(CI_RESTRICTED);
+			if (ci->levels[CA_NOJOIN] < 0)
+				ci->levels[CA_NOJOIN] = 0;
+			notice_lang(Config.s_ChanServ, u, CHAN_SET_RESTRICTED_ON, ci->name.c_str());
+		}
+		else if (params[1] == "OFF")
+		{
+			ci->UnsetFlag(CI_RESTRICTED);
+			if (ci->levels[CA_NOJOIN] >= 0)
+				ci->levels[CA_NOJOIN] = -2;
+			notice_lang(Config.s_ChanServ, u, CHAN_SET_RESTRICTED_OFF, ci->name.c_str());
+		}
+		else
+			this->OnSyntaxError(u, "RESTRICTED");
+
+		return MOD_CONT;
+	}
+
+	bool OnHelp(User *u, const ci::string &)
+	{
+		notice_help(Config.s_ChanServ, u, CHAN_HELP_SET_RESTRICTED, "SET");
+		return true;
+	}
+
+	void OnSyntaxError(User *u, const ci::string &)
+	{
+		syntax_error(Config.s_ChanServ, u, "SET RESTRICTED", CHAN_SET_RESTRICTED_SYNTAX);
+	}
+
+	void OnServHelp(User *u)
+	{
+		notice_lang(Config.s_ChanServ, u, CHAN_HELP_CMD_SET_RESTRICTED);
+	}
+};
+
+class CommandCSSASetRestricted : public CommandCSSetRestricted
+{
+ public:
+	CommandCSSASetRestricted(const ci::string &cname) : CommandCSSetRestricted(cname, "chanserv/saset/restricted")
+	{
+	}
+
+	bool OnHelp(User *u, const ci::string &)
+	{
+		notice_help(Config.s_ChanServ, u, CHAN_HELP_SET_RESTRICTED, "SASET");
+		return true;
+	}
+
+	void OnSyntaxError(User *u, const ci::string &)
+	{
+		syntax_error(Config.s_ChanServ, u, "SASET RESTRICTED", CHAN_SASET_RESTRICTED_SYNTAX);
+	}
+};
+
+class CSSetRestricted : public Module
+{
+ public:
+	CSSetRestricted(const std::string &modname, const std::string &creator) : Module(modname, creator)
+	{
+		this->SetAuthor("Anope");
+		this->SetVersion("$Id$");
+		this->SetType(CORE);
+
+		Command *c = FindCommand(ChanServ, "SET");
+		if (c)
+			c->AddSubcommand(new CommandCSSetRestricted("RESTRICTED"));
+
+		c = FindCommand(ChanServ, "SASET");
+		if (c)
+			c->AddSubcommand(new CommandCSSASetRestricted("RESTRICTED"));
+	}
+
+	~CSSetRestricted()
+	{
+		Command *c = FindCommand(ChanServ, "SET");
+		if (c)
+			c->DelSubcommand("RESTRICTED");
+
+		c = FindCommand(ChanServ, "SASET");
+		if (c)
+			c->DelSubcommand("RESTRICTED");
+	}
+};
+
+MODULE_INIT(CSSetRestricted)