X-Git-Url: http://git.annexia.org/?p=xavierbot.git;a=blobdiff_plain;f=xavierbot.pl;h=d2294772af835f5f1a8628f05b9cd6ec03782c4d;hp=bfdd9bfb8b01806d72f13684b9f1c5c2a32f5bad;hb=e510484ab8b113b43cf95a18820ad3a0629edf4b;hpb=5e6674263be334ebb33bfda3412b5c3682968d85

diff --git a/xavierbot.pl b/xavierbot.pl
index bfdd9bf..d229477 100755
--- a/xavierbot.pl
+++ b/xavierbot.pl
@@ -1,11 +1,13 @@
-#!/usr/bin/perl -wT
+#!/usr/bin/perl -w
 # xavierbot : an OCaml interpreter IRC bot.
 # By Richard W.M. Jones <rich@annexia.org>.
 # This code is in the Public Domain.
-# $Id: xavierbot.pl,v 1.7 2007/06/29 07:55:47 rjones Exp $
+# $Id: xavierbot.pl,v 1.14 2010/04/04 19:38:40 rjones Exp $
 
 use strict;
 use POE qw(Component::IRC Wheel::Run);
+use Getopt::Long;
+use POE::Component::IRC::Plugin::NickServID;
 
 #----------------------------------------------------------------------
 # Start of configuration.
@@ -13,13 +15,33 @@ use POE qw(Component::IRC Wheel::Run);
 my $nick = "xavierbot";
 my $ircname = "Xavierbot";		# Printable name.
 my $server = "chat.freenode.net";
-#my $server = "devserv.devel.redhat.com";
 my $port = 6667;
 my $channel = "#ocaml";
 
+my $nickservpw = "123456"; # or undef
+
 # End of configuration.
 #----------------------------------------------------------------------
 
+# Command line args can override configuration.
+GetOptions ("nick=s" => \$nick,
+	    "ircname=s" => \$ircname,
+	    "server=s" => \$server,
+	    "port=i" => \$port,
+	    "channel=s" => \$channel)
+    or die "$0: GetOptions: $!";
+
+# Simple flood protection.  This counts number of lines received from
+# the toplevel, and is reset when we send a line.  If this exceeds
+# some value, then we just eat lines.
+# XXX This ought to count characters, not lines.
+my $flood_lim = 0;
+
+# Are we awake or sleeping?
+my $sleeping = 0;
+
+#----------------------------------------------------------------------
+
 $ENV{PATH} = "/usr/bin:/bin";
 
 POE::Session->create (
@@ -43,6 +65,11 @@ sub _start
 	 port => $port,
 	 ) or die "POE::Component::IRC->spawn failed: $!";
 
+    $irc->plugin_add('NickServID',
+                     POE::Component::IRC::Plugin::NickServID->new(
+                         Password => $nickservpw,
+                     ));
+
     my $ocaml = start_toplevel ();
 
     $kernel->sig(CHLD => qw(got_sigchld));
@@ -75,24 +102,46 @@ sub irc_public
     my $nick = (split /!/, $who)[0];
     my $channel = $where->[0];
 
+    my @usage =
+	(
+	 "expr ;;  evaluate expr in toplevel and print result",
+	 "help     help message",
+	 "restart  restart the toplevel",
+	 "sleep    go to sleep",
+	 "wake     wake me up from sleep",
+	 );
+
     print "got: $what\n";
-    if (my ($stmt) = $what =~ /^\s*([^#].*;;)\s*$/) {
-	$heap->{ocaml}->put ("$stmt\n");
-    }
     # XXX How to interpolate $nick into the patterns?
-    elsif ($what =~ /^\s*xavierbot\b.*\bhelp\b/) {
+    if ($what =~ /^\s*xavierbot\b.*\bhelp\b/) {
 	my $nick = (split /!/, $who)[0];
 	$kernel->post ($sender => privmsg => $channel =>
-		       "$nick: expr ;;      evaluate expr in OCaml toplevel");
-	$kernel->post ($sender => privmsg => $channel =>
-		       "$nick: help         help message");
-	$kernel->post ($sender => privmsg => $channel =>
-		       "$nick: restart      restart the toplevel");
+	    "hello $nick, I am xavierbot 0.8, an OCaml toplevel");
+	$kernel->post ($sender => privmsg => $channel => $_)
+	    foreach (@usage);
     }
     elsif ($what =~ /^\s*xavierbot\b.*\brestart\b/) {
+	$sleeping = 0;
 	print STDOUT "got instruction to restart ...\n";
 	restart_toplevel ($heap->{ocaml});
     }
+    elsif (!$sleeping) {
+	if (my ($stmt) = $what =~ m/^\s*([^\#].*;;)\s*$/) {
+	    $heap->{ocaml}->put ("$stmt\n");
+	    $flood_lim = 0;
+	}
+	elsif ($what =~ /^\s*xavierbot\b.*\b(sleep|shut|quiet)\b/) {
+	    $sleeping = 1;
+	    $kernel->post ($sender => privmsg => $channel =>
+			   "xavierbot goes to sleep (do 'xavierbot wake' to wake)");
+	}
+    } else { # sleeping
+	if ($what =~ /^\s*xavierbot\b.*\bwake\b/) {
+	    $sleeping = 0;
+	    $kernel->post ($sender => privmsg => $channel =>
+			   "xavierbot wakes up");
+	}
+    }
     undef;
 }
 
@@ -120,7 +169,10 @@ sub got_stdout
 {
     my ($kernel,$heap, $input, $wheel_id) = @_[KERNEL,HEAP,ARG0,ARG1];
     print "Child said: $input\n";
-    $kernel->post ($heap->{irc} => privmsg => $channel => "$input");
+    if ($flood_lim < 10) {
+	$kernel->post ($heap->{irc} => privmsg => $channel => "$input");
+    }
+    $flood_lim++;
 }
 
 # Got a SIGCHLD, so start the bot up again.