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