mirror of
https://github.com/pragma-/pbot.git
synced 2024-11-22 20:09:43 +01:00
Added support for quotegrabs: !grab, !getq, and !rq
This commit is contained in:
Pragmatic Software
parent
e706505921
commit
e4b521e1bc
196
pbot2.pl
196
pbot2.pl
@ -8,13 +8,14 @@
|
||||
# Version History:
|
||||
########################
|
||||
|
||||
my $VERSION = "0.4.1";
|
||||
my $VERSION = "0.4.2";
|
||||
|
||||
########################
|
||||
# todo! add support for admin management - needs support for adding/removing/saving!
|
||||
# todo! multi-channel support pathetic (note 12/08/09, fixed multi-channel for anti-flood and for ignore)
|
||||
# todo! most of this crap needs to be refactored (note 11/23/09, refactored execute_module)
|
||||
#
|
||||
# 0.4.2 (12/09/09): added support for quotegrabs: !grab, !getq, and !rq
|
||||
# 0.4.1 (12/08/09): improved anti-flood system to be significantly more accurate and per-channel
|
||||
# added per-nick-per-channel message history using %flood_watch
|
||||
# add per-channel support to ignore system
|
||||
@ -127,12 +128,13 @@ $SIG{ALRM} = \&sig_alarm_handler;
|
||||
my $home = $ENV{HOME};
|
||||
my $channels_file = "$home/pbot2/channels";
|
||||
my $commands_file = "$home/pbot2/commands";
|
||||
my $quotegrabs_file = "$home/pbot2/quotegrabs";
|
||||
my $admins_file = "$home/pbot2/admins";
|
||||
my $module_dir = "$home/pbot2/modules";
|
||||
my $ircserver = 'irc.freenode.net';
|
||||
my $botnick = 'candide';
|
||||
my $altbotnick = 'candide_';
|
||||
my $identify_password = 'helios';
|
||||
my $identify_password = 'habernat21';
|
||||
my $export_factoids_timeout = 300; # every 5 minutes
|
||||
my $export_factoids_time = gettimeofday + $export_factoids_timeout;
|
||||
my $export_factoids_path = "$home/htdocs/candide/factoids.html";
|
||||
@ -144,6 +146,8 @@ my $MAX_NICK_MESSAGES = 8;
|
||||
my $FLOOD_CHAT = 0;
|
||||
my $FLOOD_JOIN = 1; # currently unused -- todo?
|
||||
|
||||
my %flood_watch = ();
|
||||
|
||||
# set some defaults ...
|
||||
my $max_msg_len = 460;
|
||||
my %commands = ( version => {
|
||||
@ -155,6 +159,14 @@ my %commands = ( version => {
|
||||
ref_user => "nobody" }
|
||||
);
|
||||
|
||||
my @quotegrabs = ({
|
||||
nick => "candide",
|
||||
text => "Who's a bot?",
|
||||
channel => "#pbot2",
|
||||
grabbed_by => "pragma_",
|
||||
timestamp => 0
|
||||
});
|
||||
|
||||
my %admins = ( pragma_ => {
|
||||
password => '*',
|
||||
level => 50,
|
||||
@ -169,6 +181,7 @@ my %channels = ();
|
||||
|
||||
#... and load the rest
|
||||
load_channels();
|
||||
load_quotegrabs();
|
||||
load_commands();
|
||||
|
||||
sub plog;
|
||||
@ -187,6 +200,9 @@ my %internal_commands = (
|
||||
add => { sub => \&add_text, level=> 0 },
|
||||
regex => { sub => \&add_regex, level=> 0 },
|
||||
learn => { sub => \&add_text, level=> 0 },
|
||||
grab => { sub => \"egrab, level=> 0 },
|
||||
getq => { sub => \&show_quotegrab, level=> 0 },
|
||||
rq => { sub => \&show_random_quotegrab, level=> 0 },
|
||||
info => { sub => \&info, level=> 0 },
|
||||
show => { sub => \&show, level=> 0 },
|
||||
histogram => { sub => \&histogram, level=> 0 },
|
||||
@ -222,6 +238,7 @@ my %internal_commands = (
|
||||
$conn->add_handler([ 251,252,253,254,302,255 ], \&on_init);
|
||||
$conn->add_handler(376 , \&on_connect );
|
||||
$conn->add_handler('disconnect' , \&on_disconnect );
|
||||
$conn->add_handler('caction' , \&on_action );
|
||||
$conn->add_handler('public' , \&on_public );
|
||||
$conn->add_handler('msg' , \&on_msg );
|
||||
$conn->add_handler('mode' , \&on_mode );
|
||||
@ -312,6 +329,26 @@ sub list {
|
||||
return "/msg $nick Usage: list <modules|factoids|commands|admins>";
|
||||
}
|
||||
|
||||
if($arguments =~/^messages\s+(.*?)\s+(.*)$/) {
|
||||
my $nick_search = $1;
|
||||
my $channel = $2;
|
||||
|
||||
if(not exists $flood_watch{$nick}) {
|
||||
return "/msg $nick No messages for $nick_search yet.";
|
||||
}
|
||||
|
||||
if(not exists $flood_watch{$nick}{$channel}) {
|
||||
return "/msg $nick No messages for $nick_search in $channel yet.";
|
||||
}
|
||||
|
||||
my @messages = @{ $flood_watch{$nick}{$channel}{messages} };
|
||||
|
||||
for(my $i = 0; $i <= $#messages; $i++) {
|
||||
$conn->privmsg($nick, "" . ($i + 1) . ": " . $messages[$i]->{msg} . "\n");
|
||||
}
|
||||
return "";
|
||||
}
|
||||
|
||||
if($arguments =~ /^modules$/i) {
|
||||
$text = "Loaded modules: ";
|
||||
foreach my $command (sort keys %commands) {
|
||||
@ -1430,12 +1467,12 @@ sub save_channels {
|
||||
}
|
||||
|
||||
sub load_commands {
|
||||
plog "Loading commands from $commands_file ...\n";
|
||||
|
||||
open(FILE, "< $commands_file") or die "Couldn't open $commands_file: $!\n";
|
||||
my @contents = <FILE>;
|
||||
close(FILE);
|
||||
|
||||
plog "Loading commands from $commands_file ...\n";
|
||||
|
||||
my $i = 0;
|
||||
foreach my $line (@contents) {
|
||||
chomp $line;
|
||||
@ -1502,8 +1539,6 @@ sub load_admins {
|
||||
sub save_admins {
|
||||
}
|
||||
|
||||
my %flood_watch = ();
|
||||
|
||||
sub check_flood {
|
||||
my ($nick, $host, $channel, $max, $mode, $msg) = @_;
|
||||
my $now = gettimeofday;
|
||||
@ -1567,6 +1602,149 @@ sub check_flood {
|
||||
}
|
||||
}
|
||||
|
||||
sub quotegrab {
|
||||
my ($from, $nick, $host, $arguments) = @_;
|
||||
|
||||
if(not defined $arguments) {
|
||||
return "Usage: !grab <nick> [history] -- where [history] is an optional argument that is an integer number of recent messages; e.g., to grab the 3rd most recent message for nick, use !grab nick 3";
|
||||
}
|
||||
|
||||
my ($grab_nick, $grab_history) = split(/\s+/, $arguments, 2);
|
||||
|
||||
$grab_history = 1 if not defined $grab_history;
|
||||
|
||||
if($grab_history < 1 || $grab_history > $MAX_NICK_MESSAGES) {
|
||||
return "/msg $nick Please choose a history between 1 and $MAX_NICK_MESSAGES";
|
||||
}
|
||||
|
||||
if(not exists $flood_watch{$grab_nick}) {
|
||||
return "No message history for $grab_nick.";
|
||||
}
|
||||
|
||||
if(not exists $flood_watch{$grab_nick}{$from}) {
|
||||
return "No message history for $grab_nick in $from.";
|
||||
}
|
||||
|
||||
my @messages = @{ $flood_watch{$grab_nick}{$from}{messages} };
|
||||
|
||||
$grab_history--;
|
||||
|
||||
if($grab_history > $#messages) {
|
||||
return "$grab_nick has only " . ($#messages + 1) . " messages in the history.";
|
||||
}
|
||||
|
||||
$grab_history = $#messages - $grab_history;
|
||||
|
||||
plog "$nick ($from) grabbed <$grab_nick> $messages[$grab_history]->{msg}\n";
|
||||
|
||||
my $quotegrab = {};
|
||||
$quotegrab->{nick} = $grab_nick;
|
||||
$quotegrab->{channel} = $from;
|
||||
$quotegrab->{timestamp} = $messages[$grab_history]->{timestamp};
|
||||
$quotegrab->{grabbed_by} = $nick;
|
||||
$quotegrab->{text} = $messages[$grab_history]->{msg};
|
||||
push @quotegrabs, $quotegrab;
|
||||
save_quotegrabs();
|
||||
my $msg = $messages[$grab_history]->{msg};
|
||||
$msg =~ s/(.{8}).*/$1.../;
|
||||
return "Quote grabbed: " . ($#quotegrabs + 1) . ": <$grab_nick> $msg";
|
||||
}
|
||||
|
||||
sub show_quotegrab {
|
||||
my ($from, $nick, $host, $arguments) = @_;
|
||||
|
||||
if($arguments < 1 || $arguments > $#quotegrabs + 1) {
|
||||
return "/msg $nick Valid range for !getq is 1 - " . ($#quotegrabs + 1);
|
||||
}
|
||||
|
||||
my $quotegrab = $quotegrabs[$arguments - 1];
|
||||
return "$arguments: <$quotegrab->{nick}> $quotegrab->{text}";
|
||||
}
|
||||
|
||||
sub show_random_quotegrab {
|
||||
my ($from, $nick, $host, $arguments) = @_;
|
||||
my @quotes = ();
|
||||
my $nick_search = ".*";
|
||||
my $channel_search = $from;
|
||||
my $channel_search_quoted = quotemeta($channel_search);
|
||||
|
||||
if(defined $arguments) {
|
||||
($nick_search, $channel_search) = split(/\s+/, $arguments, 2);
|
||||
if(not defined $channel_search) {
|
||||
$channel_search = $from;
|
||||
$channel_search_quoted = quotemeta($channel_search);
|
||||
}
|
||||
}
|
||||
|
||||
eval {
|
||||
for(my $i = 0; $i <= $#quotegrabs; $i++) {
|
||||
my $hash = $quotegrabs[$i];
|
||||
if($hash->{channel} =~ /$channel_search_quoted/i && $hash->{nick} =~ /$nick_search/) {
|
||||
$hash->{id} = $i + 1;
|
||||
push @quotes, $hash;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
if($@) {
|
||||
plog "Error in show_random_quotegrab parameters: $@\n";
|
||||
return "/msg $nick Error: $@"
|
||||
}
|
||||
|
||||
if($#quotes < 0) {
|
||||
if($nick_search eq ".*") {
|
||||
return "No quotes grabbed for $channel_search yet. Use !grab to grab a quote.";
|
||||
} else {
|
||||
return "No quotes grabbed for $nick_search in $channel_search yet. Use !grab to grab a quote.";
|
||||
}
|
||||
}
|
||||
|
||||
my $quotegrab = $quotes[int rand($#quotes + 1)];
|
||||
return "$quotegrab->{id}: <$quotegrab->{nick}> $quotegrab->{text}";
|
||||
}
|
||||
|
||||
sub load_quotegrabs {
|
||||
plog "Loading quotegrabs from $quotegrabs_file ...\n";
|
||||
|
||||
open(FILE, "< $quotegrabs_file") or die "Couldn't open $quotegrabs_file: $!\n";
|
||||
my @contents = <FILE>;
|
||||
close(FILE);
|
||||
|
||||
my $i = 0;
|
||||
foreach my $line (@contents) {
|
||||
chomp $line;
|
||||
$i++;
|
||||
my ($nick, $channel, $timestamp, $grabbed_by, $text) = split(/\s+/, $line, 5);
|
||||
if(not defined $nick || not defined $channel || not defined $timestamp
|
||||
|| not defined $grabbed_by || not defined $text) {
|
||||
die "Syntax error around line $i of $quotegrabs_file\n";
|
||||
}
|
||||
|
||||
my $quotegrab = {};
|
||||
$quotegrab->{nick} = $nick;
|
||||
$quotegrab->{channel} = $channel;
|
||||
$quotegrab->{timestamp} = $timestamp;
|
||||
$quotegrab->{grabbed_by} = $grabbed_by;
|
||||
$quotegrab->{text} = $text;
|
||||
push @quotegrabs, $quotegrab;
|
||||
}
|
||||
plog " $i quotegrabs loaded.\n";
|
||||
plog "Done.\n";
|
||||
}
|
||||
|
||||
sub save_quotegrabs {
|
||||
open(FILE, "> $quotegrabs_file") or die "Couldn't open $quotegrabs_file: $!\n";
|
||||
|
||||
for(my $i = 0; $i <= $#quotegrabs; $i++) {
|
||||
my $quotegrab = $quotegrabs[$i];
|
||||
next if $quotegrab->{timestamp} == 0;
|
||||
print FILE "$quotegrab->{nick} $quotegrab->{channel} $quotegrab->{timestamp} $quotegrab->{grabbed_by} $quotegrab->{text}\n";
|
||||
}
|
||||
|
||||
close(FILE);
|
||||
system("cp $quotegrabs_file $quotegrabs_file.bak");
|
||||
}
|
||||
|
||||
sub quiet_nick_timed {
|
||||
my ($nick, $channel, $length) = @_;
|
||||
|
||||
@ -1822,6 +2000,12 @@ sub on_msg {
|
||||
on_public($conn, $event);
|
||||
}
|
||||
|
||||
sub on_action {
|
||||
my ($conn, $event) = @_;
|
||||
|
||||
on_public($conn, $event);
|
||||
}
|
||||
|
||||
sub on_mode {
|
||||
my ($conn, $event) = @_;
|
||||
my ($nick, $host) = ($event->nick, $event->host);
|
||||
|
||||
Loading…
Reference in New Issue
Block a user