mirror of
https://github.com/pragma-/pbot.git
synced 2024-12-25 12:12:34 +01:00
188 lines
5.7 KiB
Perl
188 lines
5.7 KiB
Perl
# File: IgnoreList.pm
|
|
# Author: pragma_
|
|
#
|
|
# Purpose: Manages ignore list.
|
|
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package PBot::IgnoreList;
|
|
|
|
use parent 'PBot::Class';
|
|
|
|
use warnings; use strict;
|
|
use feature 'unicode_strings';
|
|
|
|
use PBot::IgnoreListCommands;
|
|
use Time::HiRes qw(gettimeofday);
|
|
|
|
sub initialize {
|
|
my ($self, %conf) = @_;
|
|
$self->{filename} = $conf{filename};
|
|
|
|
$self->{ignore_list} = {};
|
|
$self->{ignore_flood_counter} = {};
|
|
$self->{last_timestamp} = {};
|
|
|
|
$self->{commands} = PBot::IgnoreListCommands->new(pbot => $self->{pbot});
|
|
$self->load_ignores();
|
|
|
|
$self->{pbot}->{timer}->register(sub { $self->check_ignore_timeouts }, 10);
|
|
}
|
|
|
|
sub add {
|
|
my $self = shift;
|
|
my ($hostmask, $channel, $length) = @_;
|
|
|
|
if ($length < 0) { $self->{ignore_list}->{$hostmask}->{$channel} = -1; }
|
|
else { $self->{ignore_list}->{$hostmask}->{$channel} = gettimeofday + $length; }
|
|
|
|
$self->save_ignores();
|
|
}
|
|
|
|
sub remove {
|
|
my $self = shift;
|
|
my ($hostmask, $channel) = @_;
|
|
|
|
delete $self->{ignore_list}->{$hostmask}->{$channel};
|
|
|
|
if (not keys %{$self->{ignore_list}->{$hostmask}}) { delete $self->{ignore_list}->{$hostmask}; }
|
|
|
|
$self->save_ignores();
|
|
}
|
|
|
|
sub clear_ignores {
|
|
my $self = shift;
|
|
$self->{ignore_list} = {};
|
|
}
|
|
|
|
sub load_ignores {
|
|
my $self = shift;
|
|
my $filename;
|
|
|
|
if (@_) { $filename = shift; }
|
|
else { $filename = $self->{filename}; }
|
|
|
|
if (not defined $filename) {
|
|
Carp::carp "No ignorelist path specified -- skipping loading of ignorelist";
|
|
return;
|
|
}
|
|
|
|
$self->{pbot}->{logger}->log("Loading ignorelist from $filename ...\n");
|
|
|
|
open(FILE, "< $filename") or Carp::croak "Couldn't open $filename: $!\n";
|
|
my @contents = <FILE>;
|
|
close(FILE);
|
|
|
|
my $i = 0;
|
|
|
|
foreach my $line (@contents) {
|
|
chomp $line;
|
|
$i++;
|
|
|
|
my ($hostmask, $channel, $length) = split(/\s+/, $line);
|
|
|
|
if (not defined $hostmask || not defined $channel || not defined $length) { Carp::croak "Syntax error around line $i of $filename\n"; }
|
|
|
|
if (exists ${$self->{ignore_list}}{$hostmask}{$channel}) { Carp::croak "Duplicate ignore [$hostmask][$channel] found in $filename around line $i\n"; }
|
|
|
|
$self->{ignore_list}->{$hostmask}->{$channel} = $length;
|
|
}
|
|
|
|
$self->{pbot}->{logger}->log(" $i entries in ignorelist\n");
|
|
}
|
|
|
|
sub save_ignores {
|
|
my $self = shift;
|
|
my $filename;
|
|
|
|
if (@_) { $filename = shift; }
|
|
else { $filename = $self->{filename}; }
|
|
|
|
if (not defined $filename) {
|
|
Carp::carp "No ignorelist path specified -- skipping saving of ignorelist\n";
|
|
return;
|
|
}
|
|
|
|
open(FILE, "> $filename") or die "Couldn't open $filename: $!\n";
|
|
|
|
foreach my $hostmask (keys %{$self->{ignore_list}}) {
|
|
foreach my $channel (keys %{$self->{ignore_list}->{$hostmask}}) {
|
|
my $length = $self->{ignore_list}->{$hostmask}->{$channel};
|
|
print FILE "$hostmask $channel $length\n";
|
|
}
|
|
}
|
|
|
|
close(FILE);
|
|
}
|
|
|
|
sub check_ignore {
|
|
my $self = shift;
|
|
my ($nick, $user, $host, $channel, $silent) = @_;
|
|
my $pbot = $self->{pbot};
|
|
$channel = lc $channel;
|
|
|
|
my $hostmask = "$nick!$user\@$host";
|
|
|
|
my $now = gettimeofday;
|
|
|
|
if (defined $channel) { # do not execute following if text is coming from STDIN ($channel undef)
|
|
if ($channel =~ /^#/) { $self->{ignore_flood_counter}->{$channel}++; }
|
|
|
|
if (not exists $self->{last_timestamp}->{$channel}) { $self->{last_timestamp}->{$channel} = $now; }
|
|
elsif ($now - $self->{last_timestamp}->{$channel} >= 30) {
|
|
$self->{last_timestamp}->{$channel} = $now;
|
|
if (exists $self->{ignore_flood_counter}->{$channel} and $self->{ignore_flood_counter}->{$channel} > 0) { $self->{ignore_flood_counter}->{$channel} = 0; }
|
|
}
|
|
|
|
=cut
|
|
if (exists $self->{ignore_flood_counter}->{$channel} and $self->{ignore_flood_counter}->{$channel} > 5) {
|
|
$self->{commands}->ignore_user("", "floodcontrol", "", "", ".* $channel 300");
|
|
$self->{ignore_flood_counter}->{$channel} = 0;
|
|
if ($channel =~ /^#/) {
|
|
$pbot->{conn}->me($channel, "has been overwhelmed.");
|
|
$pbot->{conn}->me($channel, "lies down and falls asleep.");
|
|
return 1;
|
|
}
|
|
}
|
|
=cut
|
|
|
|
}
|
|
|
|
foreach my $ignored (keys %{$self->{ignore_list}}) {
|
|
foreach my $ignored_channel (keys %{$self->{ignore_list}->{$ignored}}) {
|
|
my $ignored_channel_escaped = quotemeta $ignored_channel;
|
|
my $ignored_escaped = quotemeta $ignored;
|
|
|
|
$ignored_channel_escaped =~ s/\\(\.|\*)/$1/g;
|
|
$ignored_escaped =~ s/\\(\.|\*)/$1/g;
|
|
|
|
if (($channel =~ /$ignored_channel_escaped/i) && ($hostmask =~ /$ignored_escaped/i)) {
|
|
$self->{pbot}->{logger}->log("$nick!$user\@$host message ignored in channel $channel (matches [$ignored] host and [$ignored_channel] channel)\n") unless $silent;
|
|
return 1;
|
|
}
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
sub check_ignore_timeouts {
|
|
my $self = shift;
|
|
my $now = gettimeofday();
|
|
|
|
foreach my $hostmask (keys %{$self->{ignore_list}}) {
|
|
foreach my $channel (keys %{$self->{ignore_list}->{$hostmask}}) {
|
|
next if ($self->{ignore_list}->{$hostmask}->{$channel} == -1); #permanent ignore
|
|
|
|
if ($self->{ignore_list}->{$hostmask}->{$channel} < $now) {
|
|
$self->{pbot}->{logger}->log("Unignoring $hostmask in channel $channel.\n");
|
|
$self->remove($hostmask, $channel);
|
|
if ($hostmask eq ".*") { $self->{pbot}->{conn}->me($channel, "awakens."); }
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
1;
|