diff options
| -rw-r--r-- | include/inspircd.h | 1 | ||||
| -rw-r--r-- | include/numericbuilder.h | 119 |
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()) + { + } +}; |