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