lua/luabot.c

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