mirror of
https://github.com/pragma-/pbot.git
synced 2024-12-23 19:22:40 +01:00
Add countertrigger command to Counter plugin
This commit is contained in:
Pragmatic Software
parent
d8c2084587
commit
d251d0ac7a
@ -4,6 +4,9 @@ package PBot::Plugins::Counter;
|
||||
use warnings;
|
||||
use strict;
|
||||
|
||||
use feature 'switch';
|
||||
no if $] >= 5.018, warnings => "experimental::smartmatch";
|
||||
|
||||
use Carp ();
|
||||
use DBI;
|
||||
use Time::Duration qw/duration/;
|
||||
@ -22,11 +25,14 @@ sub initialize {
|
||||
|
||||
$self->{pbot} = delete $conf{pbot} // Carp::croak("Missing pbot reference to " . __FILE__);
|
||||
|
||||
$self->{pbot}->{commands}->register(sub { $self->counteradd(@_) }, 'counteradd', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->counterdel(@_) }, 'counterdel', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->counterreset(@_) }, 'counterreset', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->countershow(@_) }, 'countershow', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->counterlist(@_) }, 'counterlist', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->counteradd(@_) }, 'counteradd', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->counterdel(@_) }, 'counterdel', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->counterreset(@_) }, 'counterreset', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->countershow(@_) }, 'countershow', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->counterlist(@_) }, 'counterlist', 0);
|
||||
$self->{pbot}->{commands}->register(sub { $self->countertrigger(@_) }, 'countertrigger', 0);
|
||||
|
||||
$self->{pbot}->{event_dispatcher}->register_handler('irc.public', sub { $self->on_public(@_) });
|
||||
|
||||
$self->{filename} = $self->{pbot}->{registry}->get_value('general', 'data_dir') . '/counters.sqlite3';
|
||||
|
||||
@ -41,6 +47,7 @@ sub unload {
|
||||
$self->{pbot}->{commands}->unregister('counterreset');
|
||||
$self->{pbot}->{commands}->unregister('countershow');
|
||||
$self->{pbot}->{commands}->unregister('counterlist');
|
||||
$self->{pbot}->{commands}->unregister('countertrigger');
|
||||
}
|
||||
|
||||
sub create_database {
|
||||
@ -56,6 +63,14 @@ CREATE TABLE IF NOT EXISTS Counters (
|
||||
description TEXT,
|
||||
timestamp NUMERIC
|
||||
)
|
||||
SQL
|
||||
|
||||
$self->{dbh}->do(<<SQL);
|
||||
CREATE TABLE IF NOT EXISTS Triggers (
|
||||
channel TEXT,
|
||||
trigger TEXT,
|
||||
target TEXT,
|
||||
)
|
||||
SQL
|
||||
|
||||
$self->{dbh}->disconnect;
|
||||
@ -91,12 +106,19 @@ sub add_counter {
|
||||
return 0;
|
||||
}
|
||||
|
||||
my $sth = $self->{dbh}->prepare('INSERT INTO Counters (channel, name, description, timestamp) VALUES (?, ?, ?, ?)');
|
||||
$sth->bind_param(1, lc $channel);
|
||||
$sth->bind_param(2, lc $name);
|
||||
$sth->bind_param(3, $description);
|
||||
$sth->bind_param(4, scalar gettimeofday);
|
||||
$sth->execute();
|
||||
eval {
|
||||
my $sth = $self->{dbh}->prepare('INSERT INTO Counters (channel, name, description, timestamp) VALUES (?, ?, ?, ?)');
|
||||
$sth->bind_param(1, lc $channel);
|
||||
$sth->bind_param(2, lc $name);
|
||||
$sth->bind_param(3, $description);
|
||||
$sth->bind_param(4, scalar gettimeofday);
|
||||
$sth->execute();
|
||||
};
|
||||
|
||||
if ($@) {
|
||||
$self->{pbot}->{logger}->log("Add counter failed: $@");
|
||||
return 0;
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
@ -109,11 +131,18 @@ sub reset_counter {
|
||||
return (undef, undef);
|
||||
}
|
||||
|
||||
my $sth = $self->{dbh}->prepare('UPDATE Counters SET timestamp = ? WHERE channel = ? AND name = ?');
|
||||
$sth->bind_param(1, scalar gettimeofday);
|
||||
$sth->bind_param(2, lc $channel);
|
||||
$sth->bind_param(3, lc $name);
|
||||
$sth->execute();
|
||||
eval {
|
||||
my $sth = $self->{dbh}->prepare('UPDATE Counters SET timestamp = ? WHERE channel = ? AND name = ?');
|
||||
$sth->bind_param(1, scalar gettimeofday);
|
||||
$sth->bind_param(2, lc $channel);
|
||||
$sth->bind_param(3, lc $name);
|
||||
$sth->execute();
|
||||
};
|
||||
|
||||
if ($@) {
|
||||
$self->{pbot}->{logger}->log("Reset counter failed: $@");
|
||||
return (undef, undef);
|
||||
}
|
||||
|
||||
return ($description, $timestamp);
|
||||
}
|
||||
@ -126,10 +155,18 @@ sub delete_counter {
|
||||
return 0;
|
||||
}
|
||||
|
||||
my $sth = $self->{dbh}->prepare('DELETE FROM Counters WHERE channel = ? AND name = ?');
|
||||
$sth->bind_param(1, lc $channel);
|
||||
$sth->bind_param(2, lc $name);
|
||||
$sth->execute();
|
||||
eval {
|
||||
my $sth = $self->{dbh}->prepare('DELETE FROM Counters WHERE channel = ? AND name = ?');
|
||||
$sth->bind_param(1, lc $channel);
|
||||
$sth->bind_param(2, lc $name);
|
||||
$sth->execute();
|
||||
};
|
||||
|
||||
|
||||
if ($@) {
|
||||
$self->{pbot}->{logger}->log("Delete counter failed: $@");
|
||||
return 0;
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
@ -145,7 +182,7 @@ sub list_counters {
|
||||
};
|
||||
|
||||
if ($@) {
|
||||
$self->{pbot}->{logger}->log("Get counter failed: $@");
|
||||
$self->{pbot}->{logger}->log("List counters failed: $@");
|
||||
}
|
||||
|
||||
return map { $_->[0] } @$counters;
|
||||
@ -171,6 +208,83 @@ sub get_counter {
|
||||
return ($description, $time);
|
||||
}
|
||||
|
||||
sub add_trigger {
|
||||
my ($self, $channel, $trigger, $target) = @_;
|
||||
|
||||
my $exists = $self->get_trigger($channel, $trigger);
|
||||
if (defined $exists) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
eval {
|
||||
my $sth = $self->{dbh}->prepare('INSERT INTO Triggers (channel, trigger, target) VALUES (?, ?, ?)');
|
||||
$sth->bind_param(1, lc $channel);
|
||||
$sth->bind_param(2, lc $trigger);
|
||||
$sth->bind_param(3, lc $target);
|
||||
$sth->execute();
|
||||
};
|
||||
|
||||
if ($@) {
|
||||
$self->{pbot}->{logger}->log("Add trigger failed: $@");
|
||||
return 0;
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
sub delete_trigger {
|
||||
my ($self, $channel, $trigger) = @_;
|
||||
|
||||
my $target = $self->get_trigger($channel, $trigger);
|
||||
if (not defined $target) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
my $sth = $self->{dbh}->prepare('DELETE FROM Triggers WHERE channel = ? AND trigger = ?');
|
||||
$sth->bind_param(1, lc $channel);
|
||||
$sth->bind_param(2, lc $trigger);
|
||||
$sth->execute();
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
sub list_triggers {
|
||||
my ($self, $channel) = @_;
|
||||
|
||||
my $triggers = eval {
|
||||
my $sth = $self->{dbh}->prepare('SELECT trigger, target FROM Triggers WHERE channel = ?');
|
||||
$sth->bind_param(1, lc $channel);
|
||||
$sth->execute();
|
||||
return $sth->fetchall_arrayref({});
|
||||
};
|
||||
|
||||
if ($@) {
|
||||
$self->{pbot}->{logger}->log("List triggers failed: $@");
|
||||
}
|
||||
|
||||
return @$triggers;
|
||||
}
|
||||
|
||||
sub get_trigger {
|
||||
my ($self, $channel, $trigger) = @_;
|
||||
|
||||
my $target = eval {
|
||||
my $sth = $self->{dbh}->prepare('SELECT target FROM Triggers WHERE channel = ? AND trigger = ?');
|
||||
$sth->bind_param(1, lc $channel);
|
||||
$sth->bind_param(2, lc $trigger);
|
||||
$sth->execute();
|
||||
my $row = $sth->fetchrow_hashref();
|
||||
return $row->{target};
|
||||
};
|
||||
|
||||
if ($@) {
|
||||
$self->{pbot}->{logger}->log("Get trigger failed: $@");
|
||||
return undef;
|
||||
}
|
||||
|
||||
return $target;
|
||||
}
|
||||
|
||||
sub counteradd {
|
||||
my ($self, $from, $nick, $user, $host, $arguments) = @_;
|
||||
|
||||
@ -263,7 +377,7 @@ sub counterreset {
|
||||
my ($description, $timestamp) = $self->reset_counter($channel, $name);
|
||||
if (defined $description) {
|
||||
my $ago = duration gettimeofday - $timestamp;
|
||||
$result = "It had been $ago since $description.";
|
||||
$result = "It had been $ago since $description. Not any more.";
|
||||
} else {
|
||||
$result = "No such counter.";
|
||||
}
|
||||
@ -343,4 +457,159 @@ sub counterlist {
|
||||
return $result;
|
||||
}
|
||||
|
||||
sub countertrigger {
|
||||
my ($self, $from, $nick, $user, $host, $arguments) = @_;
|
||||
|
||||
if (not $self->dbi_begin) {
|
||||
return "Internal error.";
|
||||
}
|
||||
|
||||
my $command;
|
||||
($command, $arguments) = split / /, $arguments, 2;
|
||||
|
||||
my ($channel, $result);
|
||||
|
||||
given ($command) {
|
||||
when ('list') {
|
||||
if ($from =~ m/^#/) {
|
||||
$channel = $from;
|
||||
} else {
|
||||
($channel) = split / /, $arguments, 1;
|
||||
if ($channel !~ m/^#/) {
|
||||
$self->dbi_end;
|
||||
return "Usage from private message: countertrigger list <channel>";
|
||||
}
|
||||
}
|
||||
|
||||
my @triggers = $self->list_triggers($channel);
|
||||
|
||||
if (not @triggers) {
|
||||
$result = "No counter triggers set for $channel.";
|
||||
} else {
|
||||
$result = "Triggers for $channel: ";
|
||||
my $comma = '';
|
||||
foreach my $trigger (@triggers) {
|
||||
$result .= "$comma$trigger->{trigger} -> $trigger->{target}";
|
||||
$comma = ', ';
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
when ('add') {
|
||||
if ($from =~ m/^#/) {
|
||||
$channel = $from;
|
||||
} else {
|
||||
($channel, $arguments) = split / /, $arguments, 2;
|
||||
if ($channel !~ m/^#/) {
|
||||
$self->dbi_end;
|
||||
return "Usage from private message: countertrigger add <channel> <regex> <target>";
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
my ($trigger, $target) = split / /, $arguments, 2;
|
||||
|
||||
if (not defined $trigger or not defined $target) {
|
||||
if ($from !~ m/^#/) {
|
||||
$result = "Usage from private message: countertrigger add <channel> <regex> <target>";
|
||||
} else {
|
||||
$result = "Usage: countertrigger add <regex> <target>";
|
||||
}
|
||||
$self->dbi_end;
|
||||
return $result;
|
||||
}
|
||||
|
||||
my $exists = $self->get_trigger($channel, $trigger);
|
||||
|
||||
if (defined $exists) {
|
||||
$self->dbi_end;
|
||||
return "Trigger already exists.";
|
||||
}
|
||||
|
||||
if ($self->add_trigger($channel, $trigger, $target)) {
|
||||
$result = "Trigger added.";
|
||||
} else {
|
||||
$result = "Failed to add trigger.";
|
||||
}
|
||||
}
|
||||
|
||||
when ('delete') {
|
||||
if ($from =~ m/^#/) {
|
||||
$channel = $from;
|
||||
} else {
|
||||
($channel, $arguments) = split / /, $arguments, 2;
|
||||
if ($channel !~ m/^#/) {
|
||||
$self->dbi_end;
|
||||
return "Usage from private message: countertrigger delete <channel> <regex>";
|
||||
}
|
||||
}
|
||||
|
||||
my ($trigger) = split / /, $arguments, 1;
|
||||
|
||||
if (not defined $trigger) {
|
||||
if ($from !~ m/^#/) {
|
||||
$result = "Usage from private message: countertrigger delete <channel> <regex>";
|
||||
} else {
|
||||
$result = "Usage: countertrigger delete <regex>";
|
||||
}
|
||||
$self->dbi_end;
|
||||
return $result;
|
||||
}
|
||||
|
||||
my $target = $self->get_trigger($channel, $trigger);
|
||||
|
||||
if (not defined $target) {
|
||||
$result = "No such trigger.";
|
||||
} else {
|
||||
$self->delete_trigger($channel, $trigger);
|
||||
$result = "Trigger deleted.";
|
||||
}
|
||||
}
|
||||
|
||||
default {
|
||||
$result = "Usage: countertrigger <list/add/delete> [arguments]";
|
||||
}
|
||||
}
|
||||
|
||||
$self->dbi_end;
|
||||
return $result;
|
||||
}
|
||||
|
||||
sub on_public {
|
||||
my ($self, $event_type, $event) = @_;
|
||||
my ($nick, $user, $host, $msg) = ($event->{event}->nick, $event->{event}->user, $event->{event}->host, $event->{event}->args);
|
||||
my $channel = $event->{event}->{to}[0];
|
||||
|
||||
if (not $self->dbi_begin) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
my @triggers = $self->list_triggers($channel);
|
||||
|
||||
my $message = "$nick!$user\@$host $msg";
|
||||
|
||||
eval {
|
||||
foreach my $trigger (@triggers) {
|
||||
if ($message =~ m/$trigger->{trigger}/i) {
|
||||
my ($desc, $timestamp) = $self->reset_counter($channel, $trigger->{target});
|
||||
|
||||
if (defined $desc) {
|
||||
if (gettimeofday - $timestamp >= 60 * 60) {
|
||||
my $ago = duration gettimeofday - $timestamp;
|
||||
$event->{conn}->privmsg($channel, "It had been $ago since $desc. Not any more.");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
if ($@) {
|
||||
$self->{pbot}->{logger}->log("Error in counter trigger: $@");
|
||||
}
|
||||
|
||||
$self->dbi_end;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
1;
|
||||
|
||||
Loading…
Reference in New Issue
Block a user