!752

!1002672 // pircbot library

import org.jibble.pircbot.*;

static boolean useBouncer = false;

static S server = useBouncer ? "second.tinybrain.de" : "irc.freenode.net";
static L<S> channels = litlist("##linux", "#pircbot");
static S name = "Lookie";
static int reconnectTimeout1 = 60; // seconds after which to send a ping to server
static int reconnectTimeout2 = 80; // seconds after which we try to reconnect

static Map<S, PersistentLog<Map>> bulk = new TreeMap; // key = channel (with "#")

static MyBot bot;

p {
  thread "IRC Log-In" {
    bot = new MyBot;
    bot.setVerbose(true);
    print("PircBot connecting to " + server);
    bot.connect(server);
    print("PircBot connect issued");
    
    autoReconnect(bot);
  }
}

static class MyBot extends PircBot {
  long lastReceived;
  
  MyBot() {
    setName(name);
    //setRealName(name); // newer version of PircBot only
    setLogin(name);
    setVersion(name);
    setFinger("");
    setAutoNickChange(true);
  }
  
  public void handleLine(String line) {
    getLog("freenode").add(litmap("time", smartNow(), "line", line));
    super.handleLine(line);
    lastReceived = now();
  }
  
  public void onConnect() {
    print("PircBot onConnect useBouncer=" + useBouncer);
    if (useBouncer) {
      S bouncerLogin = loadSecretTextFileMandatory("bouncer-login").trim();
      print("Logging in to bouncer");
      bot.sendMessage("root", bouncerLogin.trim());
      
      // session should already be there, so resume it
      print("Resuming bouncer session");
      bot.sendMessage("root", "connect freenode");
    }
    
    for (S channel : channels)
      bot.joinChannel(channel);
  }
}

static void autoReconnect(final MyBot bot) {
  thread "PircBot Auto-Reconnect" {
    while (licensed()) {
      sleepSeconds(10);
      
      if (now() >= bot.lastReceived + reconnectTimeout2*1000)
        try {
          print("PircBot: Trying to reconnect...");
          hardClose(bot);
          
          sleepSeconds(1); // allow for input thread to halt
          
          bot.reconnect();
        } catch (Exception e) {
          print("Reconnect fail: " + e);
          // Couldn't reconnect!
        }
      else if (now() >= bot.lastReceived + reconnectTimeout1*1000)
        bot.sendRawLine("TIME"); // send something
    }
  }
}

static Socket getSocket(PircBot bot) {
  ret (Socket) get(get(bot, "_inputThread"), "_socket");
}

answer {
  if "irc log size *"
    exceptionToUser { ret lstr(getLog(m.unq(0))); }
}

static synchronized PersistentLog<Map> getLog(S channelName) {
  checkChannelName(channelName);
  PersistentLog log = bulk.get(channelName);
  if (log == null)
    bulk.put(channelName, log = new PersistentLog(channelName + ".log"));
  ret log;
}

static S checkChannelName(S s) {
  assertTrue(nempty(s));
  for (int i = 0; i < l(s); i++)
    if (!(
      "#!".indexOf(s.charAt(i)) >= 0 || Character.isLetterOrDigit(s.charAt(i))))
      fail(s);
  ret s;
}

static void hardClose(PircBot bot) {
  if (bot == null) ret;
  
  try {
    //bot.disconnect();
    getSocket(bot).close();
  }
  catch (Exception e) { print(str(e)); }
}

static void cleanMeUp() {
  hardClose(bot);
}

static long smartNow_last;

static long smartNow() {
  ret smartNow_last = max(smartNow_last + 1, now());
}

Began life as a copy of #1002676