Add Numeric::Builder

2 files changed, 120 insertions, 0 deletions

diff --git a/include/inspircd.h b/include/inspircd.h
index 91b70fbd8..20a6508c9 100644
--- a/include/inspircd.h
+++ b/include/inspircd.h
@@ -664,4 +664,5 @@ inline void stdalgo::culldeleter::operator()(classbase* item)
 		ServerInstance->GlobalCulls.AddItem(item);
 }
 
+#include "numericbuilder.h"
 #include "modules/whois.h"
diff --git a/include/numericbuilder.h b/include/numericbuilder.h
new file mode 100644
index 000000000..9f4cfd7dd
--- /dev/null
+++ b/include/numericbuilder.h
@@ -0,0 +1,119 @@
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
+ *
+ *   Copyright (C) 2015 Attila Molnar <attilamolnar@hush.com>
+ *
+ * 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/>.
+ */
+
+
+#pragma once
+
+namespace Numeric
+{
+	class WriteNumericSink;
+
+	template <char Sep, bool SendEmpty, typename Sink>
+	class GenericBuilder;
+
+	template <char Sep = ',', bool SendEmpty = false>
+	class Builder;
+}
+
+class Numeric::WriteNumericSink
+{
+	LocalUser* const user;
+
+ public:
+	WriteNumericSink(LocalUser* u)
+		: user(u)
+	{
+	}
+
+	void operator()(unsigned int numeric, const std::string& params) const
+	{
+		user->WriteNumeric(numeric, params);
+	}
+};
+
+template <char Sep, bool SendEmpty, typename Sink>
+class Numeric::GenericBuilder
+{
+	Sink sink;
+	std::string data;
+	const unsigned int numeric;
+	const std::string::size_type max;
+	std::string::size_type beginpos;
+
+	bool HasRoom(const std::string::size_type additional) const
+	{
+		return (data.size() + additional <= max);
+	}
+
+ public:
+	GenericBuilder(Sink s, unsigned int num, bool addparam = true, size_t additionalsize = 0)
+		: sink(s)
+		, numeric(num)
+		, max(ServerInstance->Config->Limits.MaxLine - ServerInstance->Config->ServerName.size() - additionalsize - 9)
+	{
+		if (addparam)
+			data.push_back(':');
+		SaveBeginPos();
+	}
+
+	std::string& GetNumeric() { return data; }
+
+	void Add(const std::string& entry)
+	{
+		if (!HasRoom(entry.size()))
+			Flush();
+		data.append(entry).push_back(Sep);
+	}
+
+	void Add(const std::string& entry1, const std::string& entry2)
+	{
+		if (!HasRoom(entry1.size() + entry2.size()))
+			Flush();
+		data.append(entry1).append(entry2).push_back(Sep);
+	}
+
+	void Flush()
+	{
+		if (IsEmpty())
+		{
+			if (!SendEmpty)
+				return;
+		}
+		else
+		{
+			data.erase(data.size()-1);
+		}
+
+		sink(numeric, data);
+		if (data.size() > beginpos)
+			data.erase(beginpos);
+	}
+
+	bool IsEmpty() const { return (data.size() == beginpos); }
+	void SaveBeginPos() { beginpos = data.size(); }
+};
+
+template <char Sep, bool SendEmpty>
+class Numeric::Builder : public GenericBuilder<Sep, SendEmpty, WriteNumericSink>
+{
+ public:
+	Builder(LocalUser* user, unsigned int num, bool addparam = true, size_t additionalsize = 0)
+		: GenericBuilder<Sep, SendEmpty, WriteNumericSink>(WriteNumericSink(user), num, addparam, additionalsize + user->nick.size())
+	{
+	}
+};