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