]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/m_svshold.cpp
Some more to fix still, modules probably wont load correctly atm
[user/henk/code/inspircd.git] / src / modules / m_svshold.cpp
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
6  * See: http://www.inspircd.org/wiki/index.php/Credits
7  *
8  * This program is free but copyrighted software; see
9  *            the file COPYING for details.
10  *
11  * ---------------------------------------------------
12  */
13
14 #include "inspircd.h"
15
16 /* $ModDesc: Implements SVSHOLD. Like Q:Lines, but can only be added/removed by Services. */
17
18 /** Holds a SVSHold item
19  */
20 class SVSHold : public classbase
21 {
22 public:
23         std::string nickname;
24         std::string set_by;
25         time_t set_on;
26         long length;
27         std::string reason;
28
29         SVSHold()
30         {
31         }
32
33         SVSHold(const std::string &nn, const std::string &sb, const time_t so, const long ln, const std::string &rs) : nickname(nn), set_by(sb), set_on(so), length(ln), reason(rs)
34         {
35         }
36 };
37
38
39 bool SVSHoldComp(const SVSHold* ban1, const SVSHold* ban2);
40
41 typedef std::vector<SVSHold*> SVSHoldlist;
42 typedef std::map<irc::string, SVSHold*> SVSHoldMap;
43
44 /* SVSHolds is declared here, as our type is right above. Don't try move it. */
45 SVSHoldlist SVSHolds;
46 SVSHoldMap HoldMap;
47
48 /** Handle /SVSHold
49  */
50 class CommandSvshold : public Command
51 {
52  public:
53         CommandSvshold(InspIRCd* Me) : Command(Me, "SVSHOLD", 'o', 1)
54         {
55                 this->source = "m_svshold.so";
56                 this->syntax = "<nickname> [<duration> :<reason>]";
57                 TRANSLATE4(TR_NICK, TR_TEXT, TR_TEXT, TR_END);
58         }
59
60         CmdResult Handle(const char** parameters, int pcnt, User *user)
61         {
62                 /* syntax: svshold nickname time :reason goes here */
63                 /* 'time' is a human-readable timestring, like 2d3h2s. */
64
65                 if (!ServerInstance->ULine(user->server))
66                 {
67                         /* don't allow SVSHOLD from non-ulined clients */
68                         return CMD_FAILURE;
69                 }
70
71                 if (pcnt == 1)
72                 {
73                         SVSHoldMap::iterator n = HoldMap.find(parameters[0]);
74                         if (n != HoldMap.end())
75                         {
76                                 /* form: svshold nickname removes a hold. */
77                                 for (SVSHoldlist::iterator iter = SVSHolds.begin(); iter != SVSHolds.end(); iter++)
78                                 {
79                                         if (parameters[0] == assign((*iter)->nickname))
80                                         {
81                                                 unsigned long remaining = 0;
82                                                 if ((*iter)->length)
83                                                 {
84                                                         remaining = ((*iter)->set_on + (*iter)->length) - ServerInstance->Time();
85                                                         user->WriteServ( "386 %s %s :Removed SVSHOLD with %lu seconds left before expiry (%s)", user->nick, (*iter)->nickname.c_str(), remaining, (*iter)->reason.c_str());
86                                                 }
87                                                 else
88                                                 {
89                                                         user->WriteServ( "386 %s %s :Removed permanent SVSHOLD (%s)", user->nick, (*iter)->nickname.c_str(), (*iter)->reason.c_str());
90                                                 }
91                                                 SVSHolds.erase(iter);
92                                                 break;
93                                         }
94                                 }
95
96                                 HoldMap.erase(n);
97                                 delete n->second;
98                         }
99                 }
100                 else if (pcnt >= 2)
101                 {
102                         /* full form to add a SVSHold */
103                         if (ServerInstance->IsNick(parameters[0]))
104                         {
105                                 // parameters[0] = w00t
106                                 // parameters[1] = 1h3m2s
107                                 // parameters[2] = Registered nickname
108                                 
109                                 /* Already exists? */
110                                 if (HoldMap.find(parameters[0]) != HoldMap.end())
111                                 {
112                                         user->WriteServ( "385 %s %s :SVSHOLD already exists", user->nick, parameters[0]);
113                                         return CMD_FAILURE;
114                                 }
115
116                                 long length = ServerInstance->Duration(parameters[1]);
117                                 std::string reason = (pcnt > 2) ? parameters[2] : "No reason supplied";
118                                 
119                                 SVSHold* S = new SVSHold(parameters[0], user->nick, ServerInstance->Time(), length, reason);
120                                 SVSHolds.push_back(S);
121                                 HoldMap[parameters[0]] = S;
122
123                                 std::sort(SVSHolds.begin(), SVSHolds.end(), SVSHoldComp);
124
125                                 if(length > 0)
126                                 {
127                                         user->WriteServ( "385 %s %s :Added %lu second SVSHOLD (%s)", user->nick, parameters[0], length, reason.c_str());
128                                         ServerInstance->WriteOpers("*** %s added %lu second SVSHOLD on %s (%s)", user->nick, length, parameters[0], reason.c_str());
129                                 }
130                                 else
131                                 {
132                                         user->WriteServ( "385 %s %s :Added permanent SVSHOLD on %s (%s)", user->nick, parameters[0], parameters[0], reason.c_str());
133                                         ServerInstance->WriteOpers("*** %s added permanent SVSHOLD on %s (%s)", user->nick, parameters[0], reason.c_str());
134                                 }
135                         }
136                         else
137                         {
138                                 /* as this is primarily a Services command, do not provide an error */
139                                 return CMD_FAILURE;
140                         }
141                 }
142
143                 return CMD_SUCCESS;
144         }
145 };
146
147 bool SVSHoldComp(const SVSHold* ban1, const SVSHold* ban2)
148 {
149         return ((ban1->set_on + ban1->length) < (ban2->set_on + ban2->length));
150 }
151
152 class ModuleSVSHold : public Module
153 {
154         CommandSvshold *mycommand;
155         
156
157  public:
158         ModuleSVSHold(InspIRCd* Me) : Module(Me)
159         {
160                 mycommand = new CommandSvshold(Me);
161                 ServerInstance->AddCommand(mycommand);
162                 Implementation eventlist[] = { I_OnUserPreNick, I_OnSyncOtherMetaData, I_OnDecodeMetaData, I_OnStats };
163                 ServerInstance->Modules->Attach(eventlist, this, 4);
164         }
165
166         void Implements(char* List)
167         {
168                 List[I_OnUserPreNick] = List[I_OnSyncOtherMetaData] = List[I_OnDecodeMetaData] = List[I_OnStats] = 1;
169         }
170         
171         virtual int OnStats(char symbol, User* user, string_list &results)
172         {
173                 ExpireBans();
174         
175                 if(symbol == 'S')
176                 {
177                         for(SVSHoldlist::iterator iter = SVSHolds.begin(); iter != SVSHolds.end(); iter++)
178                         {
179                                 unsigned long remaining = ((*iter)->set_on + (*iter)->length) - ServerInstance->Time();
180                                 results.push_back(std::string(ServerInstance->Config->ServerName)+" 210 "+user->nick+" "+(*iter)->nickname.c_str()+" "+(*iter)->set_by+" "+ConvToStr((*iter)->set_on)+" "+ConvToStr((*iter)->length)+" "+ConvToStr(remaining)+" :"+(*iter)->reason);
181                         }
182                 }
183                 
184                 return 0;
185         }
186
187         virtual int OnUserPreNick(User *user, const std::string &newnick)
188         {
189                 ExpireBans();
190         
191                 /* check SVSHolds in here, and apply as necessary. */
192                 SVSHoldMap::iterator n = HoldMap.find(assign(newnick));
193                 if (n != HoldMap.end())
194                 {
195                         user->WriteServ( "432 %s %s :Reserved nickname: %s", user->nick, newnick.c_str(), n->second->reason.c_str());
196                         return 1;
197                 }
198                 return 0;
199         }
200         
201         virtual void OnSyncOtherMetaData(Module* proto, void* opaque, bool displayable)
202         {
203                 for(SVSHoldMap::iterator iter = HoldMap.begin(); iter != HoldMap.end(); iter++)
204                 {
205                         proto->ProtoSendMetaData(opaque, TYPE_OTHER, NULL, "SVSHold", EncodeSVSHold(iter->second));
206                 }
207         }
208
209         virtual void OnDecodeMetaData(int target_type, void* target, const std::string &extname, const std::string &extdata)
210         {
211                 if((target_type == TYPE_OTHER) && (extname == "SVSHold"))
212                 {
213                         SVSHold* S = DecodeSVSHold(extdata); /* NOTE: Allocates a new SVSHold* */
214                         if (HoldMap.find(assign(S->nickname)) == HoldMap.end())
215                         {
216                                 SVSHolds.push_back(S);
217                                 HoldMap[assign(S->nickname)] = S;
218                                 std::sort(SVSHolds.begin(), SVSHolds.end(), SVSHoldComp);
219                         }
220                         else
221                         {
222                                 delete S;
223                         }
224                 }
225         }
226
227         virtual ~ModuleSVSHold()
228         {
229         }
230         
231         virtual Version GetVersion()
232         {
233                 return Version(1, 1, 0, 1, VF_COMMON | VF_VENDOR, API_VERSION);
234         }
235
236         std::string EncodeSVSHold(const SVSHold* ban)
237         {
238                 std::ostringstream stream;      
239                 stream << ban->nickname << " " << ban->set_by << " " << ban->set_on << " " << ban->length << " :" << ban->reason;
240                 return stream.str();    
241         }
242
243         SVSHold* DecodeSVSHold(const std::string &data)
244         {
245                 SVSHold* res = new SVSHold();
246                 int set_on;
247                 irc::tokenstream tokens(data);
248                 tokens.GetToken(res->nickname);
249                 tokens.GetToken(res->set_by);
250                 tokens.GetToken(set_on);
251                 res->set_on = set_on;
252                 tokens.GetToken(res->length);
253                 tokens.GetToken(res->reason);
254                 return res;
255         }
256
257         void ExpireBans()
258         {
259                 SVSHoldlist::iterator iter,safeiter;
260                 for (iter = SVSHolds.begin(); iter != SVSHolds.end(); iter++)
261                 {
262                         /* 0 == permanent, don't mess with them! -- w00t */
263                         if ((*iter)->length != 0)
264                         {
265                                 if ((*iter)->set_on + (*iter)->length <= ServerInstance->Time())
266                                 {
267                                         ServerInstance->Log(DEBUG, "m_svshold.so: hold on %s expired, removing...", (*iter)->nickname.c_str());
268                                         ServerInstance->WriteOpers("*** %li second SVSHOLD on %s (%s) set %u seconds ago expired", (*iter)->length, (*iter)->nickname.c_str(), (*iter)->reason.c_str(), ServerInstance->Time() - (*iter)->set_on);
269                                         HoldMap.erase(assign((*iter)->nickname));
270                                         delete *iter;
271                                         safeiter = iter;
272                                         --iter;
273                                         SVSHolds.erase(safeiter);
274                                 }
275                         }
276                 }
277         }
278 };
279
280 MODULE_INIT(ModuleSVSHold)