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