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_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   void **harg = (void **)arg;
 364   nick *np = harg[0];
 365   char *oldnick = harg[1];
 366
 367   if(!np)
 368     return;
 369
 370   lua_avpcall("irc_onrename", "ls", np->numeric, oldnick);
 371 }
 372
 373 void lua_onquit(int hooknum, void *arg) {
 374   nick *np = (nick *)arg;
 375
 376   if(!np)
 377     return;
 378
 379   lua_avpcall("irc_onquit", "l", np->numeric);
 380 }
 381
 382 void lua_onauth(int hooknum, void *arg) {
 383   nick *np = (nick *)arg;
 384
 385   if(!np)
 386     return;
 387
 388   lua_avpcall("irc_onauth", "l", np->numeric);
 389 }
 390
 391 void lua_ondisconnect(int hooknum, void *arg) {
 392   lua_avpcall(hooknum == HOOK_IRC_DISCON?"irc_ondisconnect":"irc_onpredisconnect", "");
 393 }
 394
 395 void lua_onconnect(int hooknum, void *arg) {
 396   lua_avpcall(hooknum == HOOK_IRC_CONNECTED?"irc_onconnect":"irc_onendofburst", "");
 397 }
 398
 399 void lua_onload(lua_State *l) {
 400   lua_vpcall(l, "onload", "");
 401 }
 402
 403 void lua_onunload(lua_State *l) {
 404   lua_vpcall(l, "onunload", "");
 405 }
 406
 407 int lua_channelmessage(channel *cp, char *message, ...) {
 408   char buf[512];
 409   va_list va;
 410
 411   va_start(va, message);
 412   vsnprintf(buf, sizeof(buf), message, va);
 413   va_end(va);
 414
 415   sendmessagetochannel(lua_nick, cp, "%s", buf);
 416
 417   return 0;
 418 }
 419
 420 int lua_message(nick *np, char *message, ...) {
 421   char buf[512];
 422   va_list va;
 423
 424   va_start(va, message);
 425   vsnprintf(buf, sizeof(buf), message, va);
 426   va_end(va);
 427
 428   sendmessagetouser(lua_nick, np, "%s", buf);
 429
 430   return 0;
 431 }
 432
 433 int lua_notice(nick *np, char *message, ...) {
 434   char buf[512];
 435   va_list va;
 436
 437   va_start(va, message);
 438   vsnprintf(buf, sizeof(buf), message, va);
 439   va_end(va);
 440
 441   sendnoticetouser(lua_nick, np, "%s", buf);
 442
 443   return 0;
 444 }
 445
 446 char *printallmodes(channel *cp) {
 447   static char buf[1024];
 448   char buf2[12];
 449
 450   if(IsLimit(cp))
 451     snprintf(buf2, sizeof(buf2), "%d", cp->limit);
 452
 453   snprintf(buf, sizeof(buf), "%s %s%s%s", printflags(cp->flags, cmodeflags), IsLimit(cp)?buf2:"", IsLimit(cp)?" ":"", IsKey(cp)?cp->key->content:"");
 454
 455   return buf;
 456 }
 457
 458 static char *printlimitedmodes(channel *cp, flag_t before) {
 459   static char buf[1024];
 460
 461   snprintf(buf, sizeof(buf), "%s", printflags(cp->flags, cmodeflags));
 462
 463   return buf;
 464 }
 465
 466 void lua_onmode(int hooknum, void *arg) {
 467   void **arglist = (void **)arg;
 468   channel *cp = (channel *)arglist[0];
 469   chanindex *ci = cp->index;
 470   nick *np = arglist[1];
 471   flag_t beforeflags = (flag_t)(long)arglist[3];
 472
 473   if(np) {
 474     lua_avpcall("irc_onmode", "Slss", ci->name, np->numeric, printallmodes(cp), printlimitedmodes(cp, beforeflags));
 475   } else {
 476     lua_avpcall("irc_onmode", "S0ss", ci->name, printallmodes(cp), printlimitedmodes(cp, beforeflags));
 477   }
 478 }
 479
 480