1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
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())
{
}
};
|
