]>
Commit | Line | Data |
---|---|---|
2da56f0d CP |
1 | /* Copyright (C) Chris Porter 2005 */ |
2 | /* ALL RIGHTS RESERVED. */ | |
3 | /* Don't put this into the SVN repo. */ | |
4 | ||
5 | #include "../localuser/localuser.h" | |
6 | #include "../localuser/localuserchannel.h" | |
7 | #include "../core/schedule.h" | |
026b1c90 | 8 | #include "../lib/irc_string.h" |
2da56f0d CP |
9 | |
10 | #include "lua.h" | |
11 | #include "luabot.h" | |
12 | ||
3522f515 CP |
13 | #include <stdarg.h> |
14 | ||
2da56f0d | 15 | nick *lua_nick = NULL; |
b86d82f2 | 16 | void *myureconnect = NULL, *myublip = NULL, *myutick = NULL; |
2da56f0d CP |
17 | |
18 | void lua_bothandler(nick *target, int type, void **args); | |
19 | ||
20 | void lua_onnewnick(int hooknum, void *arg); | |
21 | void lua_blip(void *arg); | |
b86d82f2 | 22 | void lua_tick(void *arg); |
2da56f0d CP |
23 | void lua_onkick(int hooknum, void *arg); |
24 | void lua_ontopic(int hooknum, void *arg); | |
25 | void lua_onauth(int hooknum, void *arg); | |
26 | void lua_ondisconnect(int hooknum, void *arg); | |
27 | void lua_onmode(int hooknum, void *arg); | |
28 | void lua_onop(int hooknum, void *arg); | |
29 | void lua_onquit(int hooknum, void *arg); | |
30 | void lua_onrename(int hooknum, void *arg); | |
31 | void lua_onconnect(int hooknum, void *arg); | |
32 | void lua_onjoin(int hooknum, void *arg); | |
e8b79634 | 33 | void lua_onpart(int hooknum, void *arg); |
2da56f0d CP |
34 | |
35 | void lua_registerevents(void) { | |
36 | registerhook(HOOK_NICK_NEWNICK, &lua_onnewnick); | |
37 | registerhook(HOOK_IRC_DISCON, &lua_ondisconnect); | |
38 | registerhook(HOOK_IRC_PRE_DISCON, &lua_ondisconnect); | |
39 | registerhook(HOOK_NICK_ACCOUNT, &lua_onauth); | |
40 | registerhook(HOOK_CHANNEL_TOPIC, &lua_ontopic); | |
41 | registerhook(HOOK_CHANNEL_KICK, &lua_onkick); | |
42 | registerhook(HOOK_CHANNEL_OPPED, &lua_onop); | |
43 | registerhook(HOOK_CHANNEL_DEOPPED, &lua_onop); | |
44 | registerhook(HOOK_NICK_LOSTNICK, &lua_onquit); | |
45 | registerhook(HOOK_NICK_RENAME, &lua_onrename); | |
46 | registerhook(HOOK_IRC_CONNECTED, &lua_onconnect); | |
47 | registerhook(HOOK_SERVER_END_OF_BURST, &lua_onconnect); | |
48 | registerhook(HOOK_CHANNEL_JOIN, &lua_onjoin); | |
e8b79634 | 49 | registerhook(HOOK_CHANNEL_PART, &lua_onpart); |
6361ce98 | 50 | registerhook(HOOK_CHANNEL_CREATE, &lua_onjoin); |
2da56f0d CP |
51 | } |
52 | ||
53 | void lua_deregisterevents(void) { | |
e8b79634 | 54 | deregisterhook(HOOK_CHANNEL_PART, &lua_onpart); |
6361ce98 | 55 | deregisterhook(HOOK_CHANNEL_CREATE, &lua_onjoin); |
2da56f0d CP |
56 | deregisterhook(HOOK_CHANNEL_JOIN, &lua_onjoin); |
57 | deregisterhook(HOOK_SERVER_END_OF_BURST, &lua_onconnect); | |
58 | deregisterhook(HOOK_IRC_CONNECTED, &lua_onconnect); | |
59 | deregisterhook(HOOK_NICK_RENAME, &lua_onrename); | |
60 | deregisterhook(HOOK_NICK_LOSTNICK, &lua_onquit); | |
61 | deregisterhook(HOOK_CHANNEL_DEOPPED, &lua_onop); | |
62 | deregisterhook(HOOK_CHANNEL_OPPED, &lua_onop); | |
63 | deregisterhook(HOOK_CHANNEL_KICK, &lua_onkick); | |
64 | deregisterhook(HOOK_CHANNEL_TOPIC, &lua_ontopic); | |
65 | deregisterhook(HOOK_NICK_ACCOUNT, &lua_onauth); | |
66 | deregisterhook(HOOK_IRC_PRE_DISCON, &lua_ondisconnect); | |
67 | deregisterhook(HOOK_IRC_DISCON, &lua_ondisconnect); | |
68 | deregisterhook(HOOK_NICK_NEWNICK, &lua_onnewnick); | |
69 | } | |
70 | ||
71 | void lua_startbot(void *arg) { | |
72 | channel *cp; | |
73 | ||
74 | myureconnect = NULL; | |
75 | ||
0225bed3 | 76 | lua_nick = registerlocaluser("U", "lua", "quakenet.department.of.corrections", LUA_FULLVERSION, "U", UMODE_ACCOUNT | UMODE_DEAF | UMODE_OPER | UMODE_SERVICE, &lua_bothandler); |
2da56f0d CP |
77 | if(!lua_nick) { |
78 | myureconnect = scheduleoneshot(time(NULL) + 1, &lua_startbot, NULL); | |
79 | return; | |
80 | } | |
81 | ||
82 | cp = findchannel(LUA_OPERCHAN); | |
0225bed3 | 83 | if(cp && localjoinchannel(lua_nick, cp)) { |
2da56f0d | 84 | localgetops(lua_nick, cp); |
0225bed3 CP |
85 | } else { |
86 | localcreatechannel(lua_nick, LUA_OPERCHAN); | |
87 | } | |
2da56f0d CP |
88 | |
89 | cp = findchannel(LUA_PUKECHAN); | |
0225bed3 | 90 | if(cp && localjoinchannel(lua_nick, cp)) { |
2da56f0d | 91 | localgetops(lua_nick, cp); |
0225bed3 CP |
92 | } else { |
93 | localcreatechannel(lua_nick, LUA_PUKECHAN); | |
94 | } | |
2da56f0d CP |
95 | |
96 | myublip = schedulerecurring(time(NULL) + 1, 0, 60, &lua_blip, NULL); | |
b86d82f2 | 97 | myutick = schedulerecurring(time(NULL) + 1, 0, 1, &lua_tick, NULL); |
2da56f0d CP |
98 | |
99 | lua_registerevents(); | |
100 | } | |
101 | ||
102 | void lua_destroybot(void) { | |
b86d82f2 CP |
103 | if(myutick) |
104 | deleteschedule(myutick, &lua_tick, NULL); | |
105 | ||
2da56f0d CP |
106 | if(myublip) |
107 | deleteschedule(myublip, &lua_blip, NULL); | |
108 | ||
109 | lua_deregisterevents(); | |
110 | ||
111 | if (myureconnect) | |
112 | deleteschedule(myureconnect, &lua_startbot, NULL); | |
113 | ||
114 | if(lua_nick) | |
115 | deregisterlocaluser(lua_nick, NULL); | |
116 | } | |
117 | ||
0225bed3 | 118 | int _lua_vpcall(lua_State *l, void *function, int mode, const char *sig, ...) { |
3522f515 CP |
119 | va_list va; |
120 | int narg = 0, nres, top = lua_gettop(l); | |
121 | ||
122 | lua_getglobal(l, "scripterror"); | |
0225bed3 CP |
123 | if(mode == LUA_CHARMODE) { |
124 | lua_getglobal(l, (const char *)function); | |
125 | } else { | |
126 | lua_rawgeti(l, LUA_REGISTRYINDEX, (int)function); | |
127 | } | |
128 | ||
3522f515 CP |
129 | if(!lua_isfunction(l, -1)) { |
130 | lua_settop(l, top); | |
131 | return 1; | |
132 | } | |
133 | ||
134 | va_start(va, sig); | |
135 | ||
136 | while(*sig) { | |
137 | switch(*sig++) { | |
138 | case 'i': | |
139 | lua_pushint(l, va_arg(va, int)); | |
140 | break; | |
141 | case 'l': | |
142 | lua_pushlong(l, va_arg(va, long)); | |
143 | break; | |
144 | case 's': | |
145 | lua_pushstring(l, va_arg(va, char *)); | |
146 | break; | |
147 | case 'S': | |
148 | lua_pushstring(l, ((sstring *)(va_arg(va, sstring *)))->content); | |
149 | break; | |
150 | case 'N': | |
151 | { | |
152 | nick *np = va_arg(va, nick *); | |
153 | LUA_PUSHNICK(l, np); | |
154 | break; | |
155 | } | |
156 | case 'C': | |
157 | { | |
158 | channel *cp = va_arg(va, channel *); | |
159 | LUA_PUSHCHAN(l, cp); | |
160 | break; | |
161 | } | |
162 | case '0': | |
163 | lua_pushnil(l); | |
164 | break; | |
165 | case '>': | |
166 | goto endwhile; | |
167 | ||
168 | default: | |
169 | Error("lua", ERR_ERROR, "Unable to parse vpcall signature (%c)", *(sig - 1)); | |
170 | } | |
171 | narg++; | |
172 | } | |
173 | ||
174 | endwhile: | |
175 | ||
176 | nres = strlen(sig); | |
177 | ||
ea15be33 CP |
178 | if(lua_debugpcall(l, (mode==LUA_CHARMODE)?function:"some_handler", narg, nres, top + 1)) { |
179 | Error("lua", ERR_ERROR, "Error pcalling %s: %s.", (mode==LUA_CHARMODE)?function:"some_handler", lua_tostring(l, -1)); | |
180 | } else { | |
181 | nres = -nres; | |
182 | while(*sig) { | |
183 | switch(*sig++) { | |
184 | default: | |
185 | Error("lua", ERR_ERROR, "Unable to parse vpcall return structure (%c)", *(sig - 1)); | |
186 | } | |
187 | nres++; | |
3522f515 | 188 | } |
3522f515 CP |
189 | } |
190 | ||
191 | lua_settop(l, top); | |
192 | va_end(va); | |
193 | return 0; | |
194 | } | |
195 | ||
2da56f0d CP |
196 | void lua_bothandler(nick *target, int type, void **args) { |
197 | nick *np; | |
198 | char *p; | |
3522f515 | 199 | int le; |
2da56f0d CP |
200 | |
201 | switch(type) { | |
202 | case LU_PRIVMSG: | |
203 | np = (nick *)args[0]; | |
204 | p = (char *)args[1]; | |
205 | ||
206 | if(!np || !p) | |
207 | return; | |
208 | ||
1f57d0e2 | 209 | lua_avpcall("irc_onmsg", "ls", np->numeric, p); |
2da56f0d CP |
210 | |
211 | break; | |
212 | case LU_PRIVNOTICE: | |
213 | np = (nick *)args[0]; | |
214 | p = (char *)args[1]; | |
215 | ||
216 | if(!np || !p) | |
217 | return; | |
218 | ||
219 | le = strlen(p); | |
220 | if((le > 1) && (p[0] == '\001')) { | |
221 | if(p[le - 1] == '\001') | |
222 | p[le - 1] = '\000'; | |
223 | ||
1f57d0e2 | 224 | lua_avpcall("irc_onctcp", "ls", np->numeric, p + 1); |
2da56f0d CP |
225 | |
226 | } | |
227 | ||
228 | break; | |
229 | ||
230 | case LU_KILLED: | |
231 | if(myublip) { | |
232 | myublip = NULL; | |
233 | deleteschedule(myublip, &lua_blip, NULL); | |
234 | } | |
235 | ||
b86d82f2 CP |
236 | if(myutick) { |
237 | myutick = NULL; | |
238 | deleteschedule(myutick, &lua_tick, NULL); | |
239 | } | |
240 | ||
2da56f0d CP |
241 | lua_nick = NULL; |
242 | myureconnect = scheduleoneshot(time(NULL) + 1, &lua_startbot, NULL); | |
243 | ||
244 | break; | |
245 | } | |
246 | } | |
247 | ||
248 | void lua_blip(void *arg) { | |
3522f515 | 249 | lua_avpcall("onblip", ""); |
b86d82f2 CP |
250 | } |
251 | ||
252 | void lua_tick(void *arg) { | |
3522f515 | 253 | lua_avpcall("ontick", ""); |
2da56f0d CP |
254 | } |
255 | ||
256 | void lua_onnewnick(int hooknum, void *arg) { | |
257 | nick *np = (nick *)arg; | |
2da56f0d CP |
258 | |
259 | if(!np) | |
260 | return; | |
261 | ||
1f57d0e2 | 262 | lua_avpcall("irc_onnewnick", "l", np->numeric); |
2da56f0d CP |
263 | } |
264 | ||
265 | void lua_onkick(int hooknum, void *arg) { | |
266 | void **arglist = (void **)arg; | |
267 | chanindex *ci = ((channel *)arglist[0])->index; | |
268 | nick *kicked = arglist[1]; | |
269 | nick *kicker = arglist[2]; | |
270 | char *message = (char *)arglist[3]; | |
2da56f0d CP |
271 | |
272 | if(!kicker || IsOper(kicker) || IsService(kicker) || IsXOper(kicker)) /* bloody Cruicky */ | |
273 | return; | |
274 | ||
1f57d0e2 | 275 | if(kicker) { |
ebdfcaa9 | 276 | lua_avpcall("irc_onkick", "Slls", ci->name, kicked->numeric, kicker->numeric, message); |
1f57d0e2 CP |
277 | } else { |
278 | lua_avpcall("irc_onkick", "Sl0s", ci->name, kicked->numeric, message); | |
279 | } | |
2da56f0d CP |
280 | } |
281 | ||
282 | void lua_ontopic(int hooknum, void *arg) { | |
283 | void **arglist = (void **)arg; | |
284 | channel *cp=(channel*)arglist[0]; | |
285 | nick *np = (nick *)arglist[1]; | |
2da56f0d CP |
286 | |
287 | if(!np || IsOper(np) || IsService(np) || IsXOper(np)) | |
288 | return; | |
289 | if(!cp || !cp->topic) | |
290 | return; | |
291 | ||
1f57d0e2 | 292 | lua_avpcall("irc_ontopic", "SlS", cp->index->name, np->numeric, cp->topic); |
2da56f0d CP |
293 | } |
294 | ||
295 | void lua_onop(int hooknum, void *arg) { | |
296 | void **arglist = (void **)arg; | |
297 | chanindex *ci = ((channel *)arglist[0])->index; | |
298 | nick *np = arglist[1]; | |
299 | nick *target = arglist[2]; | |
2da56f0d CP |
300 | |
301 | if(!target) | |
302 | return; | |
303 | ||
3522f515 | 304 | if(np) { |
1f57d0e2 | 305 | lua_avpcall(hooknum == HOOK_CHANNEL_OPPED?"irc_onop":"irc_ondeop", "Sll", ci->name, np->numeric, target->numeric); |
3522f515 | 306 | } else { |
1f57d0e2 | 307 | lua_avpcall(hooknum == HOOK_CHANNEL_OPPED?"irc_onop":"irc_ondeop", "S0l", ci->name, target->numeric); |
3522f515 | 308 | } |
2da56f0d CP |
309 | } |
310 | ||
311 | void lua_onjoin(int hooknum, void *arg) { | |
312 | void **arglist = (void **)arg; | |
313 | chanindex *ci = ((channel *)arglist[0])->index; | |
314 | nick *np = arglist[1]; | |
2da56f0d CP |
315 | |
316 | if(!ci || !np) | |
317 | return; | |
318 | ||
1f57d0e2 | 319 | lua_avpcall("irc_onjoin", "Sl", ci->name, np->numeric); |
2da56f0d CP |
320 | } |
321 | ||
e8b79634 CP |
322 | void lua_onpart(int hooknum, void *arg) { |
323 | void **arglist = (void **)arg; | |
324 | chanindex *ci = ((channel *)arglist[0])->index; | |
325 | nick *np = arglist[1]; | |
326 | ||
327 | if(!ci || !np) | |
328 | return; | |
329 | ||
1f57d0e2 | 330 | lua_avpcall("irc_onpart", "Sl", ci->name, np->numeric); |
e8b79634 CP |
331 | } |
332 | ||
2da56f0d CP |
333 | void lua_onrename(int hooknum, void *arg) { |
334 | nick *np = (void *)arg; | |
2da56f0d CP |
335 | |
336 | if(!np) | |
337 | return; | |
338 | ||
1f57d0e2 | 339 | lua_avpcall("irc_onrename", "l", np->numeric); |
2da56f0d CP |
340 | } |
341 | ||
342 | void lua_onquit(int hooknum, void *arg) { | |
343 | nick *np = (nick *)arg; | |
2da56f0d CP |
344 | |
345 | if(!np) | |
346 | return; | |
347 | ||
1f57d0e2 | 348 | lua_avpcall("irc_onquit", "l", np->numeric); |
2da56f0d CP |
349 | } |
350 | ||
351 | void lua_onauth(int hooknum, void *arg) { | |
352 | nick *np = (nick *)arg; | |
2da56f0d CP |
353 | |
354 | if(!np) | |
355 | return; | |
356 | ||
1f57d0e2 | 357 | lua_avpcall("irc_onauth", "l", np->numeric); |
2da56f0d CP |
358 | } |
359 | ||
360 | void lua_ondisconnect(int hooknum, void *arg) { | |
3522f515 | 361 | lua_avpcall(hooknum == HOOK_IRC_DISCON?"irc_ondisconnect":"irc_onpredisconnect", ""); |
2da56f0d CP |
362 | } |
363 | ||
364 | void lua_onconnect(int hooknum, void *arg) { | |
3522f515 | 365 | lua_avpcall(hooknum == HOOK_IRC_CONNECTED?"irc_onconnect":"irc_onendofburst", ""); |
2da56f0d CP |
366 | } |
367 | ||
368 | void lua_onload(lua_State *l) { | |
3522f515 | 369 | lua_vpcall(l, "onload", ""); |
2da56f0d CP |
370 | } |
371 | ||
372 | void lua_onunload(lua_State *l) { | |
3522f515 | 373 | lua_vpcall(l, "onunload", ""); |
2da56f0d CP |
374 | } |
375 | ||
376 | int lua_channelmessage(channel *cp, char *message, ...) { | |
377 | char buf[512]; | |
378 | va_list va; | |
379 | ||
380 | va_start(va, message); | |
381 | vsnprintf(buf, sizeof(buf), message, va); | |
382 | va_end(va); | |
383 | ||
384 | sendmessagetochannel(lua_nick, cp, "%s", buf); | |
385 | ||
386 | return 0; | |
387 | } | |
388 | ||
389 | int lua_message(nick *np, char *message, ...) { | |
390 | char buf[512]; | |
391 | va_list va; | |
392 | ||
393 | va_start(va, message); | |
394 | vsnprintf(buf, sizeof(buf), message, va); | |
395 | va_end(va); | |
396 | ||
397 | sendmessagetouser(lua_nick, np, "%s", buf); | |
398 | ||
399 | return 0; | |
400 | } | |
401 | ||
402 | int lua_notice(nick *np, char *message, ...) { | |
403 | char buf[512]; | |
404 | va_list va; | |
405 | ||
406 | va_start(va, message); | |
407 | vsnprintf(buf, sizeof(buf), message, va); | |
408 | va_end(va); | |
409 | ||
410 | sendnoticetouser(lua_nick, np, "%s", buf); | |
411 | ||
412 | return 0; | |
413 | } |