mirror of
https://github.com/pragma-/pbot.git
synced 2024-11-08 21:19:35 +01:00
339 lines
8.9 KiB
Perl
339 lines
8.9 KiB
Perl
# File: PBot.pm
|
|
# Author: pragma_
|
|
#
|
|
# Purpose: IRC Bot (3rd generation)
|
|
|
|
package PBot::PBot;
|
|
|
|
use strict;
|
|
use warnings;
|
|
|
|
BEGIN {
|
|
use vars qw($VERSION);
|
|
$VERSION = "0.6.0-beta";
|
|
}
|
|
|
|
# unbuffer stdout
|
|
STDOUT->autoflush(1);
|
|
|
|
use Carp ();
|
|
use PBot::Logger;
|
|
|
|
use PBot::StdinReader;
|
|
|
|
use Net::IRC;
|
|
use PBot::IRCHandlers;
|
|
|
|
use PBot::AntiFlood;
|
|
use PBot::Interpreter;
|
|
use PBot::Commands;
|
|
|
|
use PBot::ChanOps;
|
|
use PBot::Channels;
|
|
|
|
use PBot::Quotegrabs;
|
|
#use PBot::QuotegrabCommands;
|
|
|
|
use PBot::Factoids;
|
|
use PBot::FactoidCommands;
|
|
|
|
use PBot::BotAdmins;
|
|
use PBot::BotAdminCommands;
|
|
|
|
use PBot::IgnoreList;
|
|
use PBot::IgnoreListCommands;
|
|
|
|
use PBot::Timer;
|
|
|
|
sub new {
|
|
if(ref($_[1]) eq 'HASH') {
|
|
Carp::croak("Options to Logger should be key/value pairs, not hash reference");
|
|
}
|
|
|
|
my ($class, %conf) = @_;
|
|
|
|
my $self = bless {}, $class;
|
|
$self->initialize(%conf);
|
|
return $self;
|
|
}
|
|
|
|
sub initialize {
|
|
my ($self, %conf) = @_;
|
|
|
|
my $log_file = delete $conf{log_file};
|
|
my $channels_file = delete $conf{channels_file};
|
|
my $admins_file = delete $conf{admins_file};
|
|
|
|
my $botnick = delete $conf{botnick};
|
|
my $identify_password = delete $conf{identify_password};
|
|
|
|
my $ircserver = delete $conf{ircserver};
|
|
my $max_msg_len = delete $conf{max_msg_len};
|
|
my $MAX_FLOOD_MESSAGES = delete $conf{MAX_FLOOD_MESSAGES};
|
|
my $MAX_NICK_MESSAGES = delete $conf{MAX_NICK_MESSAGES};
|
|
|
|
my $factoids_file = delete $conf{factoids_file};
|
|
my $export_factoids_timeout = delete $conf{export_factoids_timeout};
|
|
my $export_factoids_path = delete $conf{export_factoids_path};
|
|
my $export_factoids_site = delete $conf{export_factoids_site};
|
|
my $module_dir = delete $conf{module_dir};
|
|
|
|
my $quotegrabs_file = delete $conf{quotegrabs_file};
|
|
my $export_quotegrabs_timeout = delete $conf{export_quotegrabs_timeout};
|
|
my $export_quotegrabs_path = delete $conf{export_quotegrabs_path};
|
|
my $export_quotegrabs_site = delete $conf{export_quotegrabs_site};
|
|
|
|
$ircserver = "irc.freenode.net" unless defined $ircserver;
|
|
$botnick = "pbot2" unless defined $botnick;
|
|
$identify_password = "" unless defined $identify_password;
|
|
|
|
$max_msg_len = 460 unless defined $max_msg_len;
|
|
$MAX_FLOOD_MESSAGES = 4 unless defined $MAX_FLOOD_MESSAGES;
|
|
$MAX_NICK_MESSAGES = 8 unless defined $MAX_NICK_MESSAGES;
|
|
|
|
$self->{botnick} = $botnick;
|
|
$self->{identify_password} = $identify_password;
|
|
|
|
$self->{max_msg_len} = $max_msg_len;
|
|
$self->{MAX_FLOOD_MESSAGES} = $MAX_FLOOD_MESSAGES;
|
|
$self->{MAX_NICK_MESSAGES} = $MAX_NICK_MESSAGES;
|
|
$self->{FLOOD_CHAT} = 0;
|
|
|
|
my $logger = PBot::Logger->new(log_file => $log_file);
|
|
$self->{logger} = $logger;
|
|
|
|
$self->{admins} = PBot::BotAdmins->new(
|
|
pbot => $self,
|
|
filename => $admins_file,
|
|
);
|
|
|
|
$self->admins->load_admins();
|
|
$self->admins->add_admin('*', "$botnick!stdin\@localhost", 50, 'admin');
|
|
$self->admins->login('*', "$botnick!stdin\@localhost", "admin");
|
|
|
|
$self->{factoids} = PBot::Factoids->new(
|
|
pbot => $self,
|
|
filename => $factoids_file,
|
|
export_path => $export_factoids_path,
|
|
export_site => $export_factoids_site,
|
|
);
|
|
|
|
$self->factoids->add_factoid('text', '.*', $botnick, 'version', "/say pbot2 version $VERSION");
|
|
$self->factoids->load_factoids() if defined $factoids_file;
|
|
|
|
$self->module_dir($module_dir);
|
|
|
|
$self->{commands} = PBot::Commands->new(pbot => $self);
|
|
|
|
$self->{antiflood} = PBot::AntiFlood->new(pbot => $self);
|
|
$self->{ignorelist} = PBot::IgnoreList->new(pbot => $self);
|
|
|
|
$self->interpreter(PBot::Interpreter->new(pbot => $self));
|
|
$self->interpreter->register(sub { return $self->commands->interpreter(@_); });
|
|
$self->interpreter->register(sub { return $self->factoids->interpreter(@_); });
|
|
|
|
$self->{botadmincmds} = PBot::BotAdminCommands->new(pbot => $self);
|
|
$self->{factoidcmds} = PBot::FactoidCommands->new(pbot => $self);
|
|
$self->{ignorelistcmds} = PBot::IgnoreListCommands->new(pbot => $self);
|
|
|
|
$self->{irc} = Net::IRC->new();
|
|
$self->{ircserver} = $ircserver;
|
|
$self->{irchandlers} = PBot::IRCHandlers->new(pbot => $self);
|
|
|
|
$self->{channels} = PBot::Channels->new(pbot => $self, filename => $channels_file);
|
|
$self->channels->load_channels() if defined $channels_file;
|
|
|
|
$self->{chanops} = PBot::ChanOps->new(pbot => $self);
|
|
|
|
$self->{timer} = PBot::Timer->new(timeout => 10);
|
|
$self->timer->register(sub { $self->factoids->export_factoids }, $export_factoids_timeout) if defined $export_factoids_path;
|
|
# $self->timer->register(sub { $self->quotegrabs->export_quotegrabs }, $export_quotegrabs_timeout) if defined $export_quotegrabs_path;
|
|
$self->timer->start();
|
|
}
|
|
|
|
# TODO: add disconnect subroutine
|
|
|
|
sub connect {
|
|
my ($self, $server) = @_;
|
|
|
|
$server = $self->ircserver if not defined $server;
|
|
|
|
if($self->{connected}) {
|
|
# TODO: disconnect, clean-up, etc
|
|
}
|
|
|
|
$self->logger->log("Connecting to $server ...\n");
|
|
|
|
$self->conn($self->irc->newconn(
|
|
Nick => $self->{botnick},
|
|
Username => $self->{username},
|
|
Ircname => $self->{ircname},
|
|
Server => $server,
|
|
Port => $self->{port}))
|
|
or Carp::croak "$0: Can't connect to IRC server.\n";
|
|
|
|
$self->{connected} = 1;
|
|
|
|
#set up default handlers for the IRC engine
|
|
$self->conn->add_handler([ 251,252,253,254,302,255 ], sub { $self->irchandlers->on_init(@_) });
|
|
$self->conn->add_handler(376 , sub { $self->irchandlers->on_connect(@_) });
|
|
$self->conn->add_handler('disconnect' , sub { $self->irchandlers->on_disconnect(@_) });
|
|
$self->conn->add_handler('notice' , sub { $self->irchandlers->on_notice(@_) });
|
|
$self->conn->add_handler('caction' , sub { $self->irchandlers->on_action(@_) });
|
|
$self->conn->add_handler('public' , sub { $self->irchandlers->on_public(@_) });
|
|
$self->conn->add_handler('msg' , sub { $self->irchandlers->on_msg(@_) });
|
|
$self->conn->add_handler('mode' , sub { $self->irchandlers->on_mode(@_) });
|
|
$self->conn->add_handler('part' , sub { $self->irchandlers->on_departure(@_) });
|
|
$self->conn->add_handler('join' , sub { $self->irchandlers->on_join(@_) });
|
|
$self->conn->add_handler('quit' , sub { $self->irchandlers->on_departure(@_) });
|
|
}
|
|
|
|
#main loop
|
|
sub do_one_loop {
|
|
my $self = shift;
|
|
|
|
# process IRC events
|
|
$self->irc->do_one_loop();
|
|
|
|
# process STDIN events
|
|
$self->check_stdin();
|
|
}
|
|
|
|
sub start {
|
|
my $self = shift;
|
|
|
|
if(not defined $self->{connected} or $self->{connected} == 0) {
|
|
$self->connect();
|
|
}
|
|
|
|
while(1) {
|
|
$self->do_one_loop();
|
|
}
|
|
}
|
|
|
|
sub check_stdin {
|
|
my $self = shift;
|
|
|
|
my $input = PBot::StdinReader::check_stdin();
|
|
|
|
return if not defined $input;
|
|
|
|
$self->logger->log("---------------------------------------------\n");
|
|
$self->logger->log("Read '$input' from STDIN\n");
|
|
|
|
my ($from, $text);
|
|
|
|
if($input =~ m/^~([^ ]+)\s+(.*)/) {
|
|
$from = $1;
|
|
$text = "!$2";
|
|
} else {
|
|
$from = undef;
|
|
$text = "!$input";
|
|
}
|
|
|
|
return $self->interpreter->process_line($from, $self->{botnick}, "stdin", "localhost", $text);
|
|
}
|
|
|
|
sub irc {
|
|
my $self = shift;
|
|
return $self->{irc};
|
|
}
|
|
|
|
sub logger {
|
|
my $self = shift;
|
|
if(@_) { $self->{logger} = shift; }
|
|
return $self->{logger};
|
|
}
|
|
|
|
sub channels {
|
|
my $self = shift;
|
|
if(@_) { $self->{channels} = shift; }
|
|
return $self->{channels};
|
|
}
|
|
|
|
sub factoids {
|
|
my $self = shift;
|
|
if(@_) { $self->{factoids} = shift; }
|
|
return $self->{factoids};
|
|
}
|
|
|
|
sub timer {
|
|
my $self = shift;
|
|
if(@_) { $self->{timer} = shift; }
|
|
return $self->{timer};
|
|
}
|
|
|
|
sub conn {
|
|
my $self = shift;
|
|
if(@_) { $self->{conn} = shift; }
|
|
return $self->{conn};
|
|
}
|
|
|
|
sub irchandlers {
|
|
my $self = shift;
|
|
if(@_) { $self->{irchandlers} = shift; }
|
|
return $self->{irchandlers};
|
|
}
|
|
|
|
sub interpreter {
|
|
my $self = shift;
|
|
if(@_) { $self->{interpreter} = shift; }
|
|
return $self->{interpreter};
|
|
}
|
|
|
|
sub admins {
|
|
my $self = shift;
|
|
if(@_) { $self->{admins} = shift; }
|
|
return $self->{admins};
|
|
}
|
|
|
|
sub commands {
|
|
my $self = shift;
|
|
if(@_) { $self->{commands} = shift; }
|
|
return $self->{commands};
|
|
}
|
|
|
|
sub botnick {
|
|
my $self = shift;
|
|
if(@_) { $self->{botnick} = shift; }
|
|
return $self->{botnick};
|
|
}
|
|
|
|
sub identify_password {
|
|
my $self = shift;
|
|
if(@_) { $self->{identify_password} = shift; }
|
|
return $self->{identify_password};
|
|
}
|
|
|
|
sub max_msg_len {
|
|
my $self = shift;
|
|
if(@_) { $self->{max_msg_len} = shift; }
|
|
return $self->{max_msg_len};
|
|
}
|
|
|
|
sub module_dir {
|
|
my $self = shift;
|
|
if(@_) { $self->{module_dir} = shift; }
|
|
return $self->{module_dir};
|
|
}
|
|
|
|
sub ignorelist {
|
|
my $self = shift;
|
|
if(@_) { $self->{ignorelist} = shift; }
|
|
return $self->{ignorelist};
|
|
}
|
|
|
|
sub antiflood {
|
|
my $self = shift;
|
|
if(@_) { $self->{antiflood} = shift; }
|
|
return $self->{antiflood};
|
|
}
|
|
|
|
sub ircserver {
|
|
my $self = shift;
|
|
if(@_) { $self->{ircserver} = shift; }
|
|
return $self->{ircserver};
|
|
}
|
|
|
|
1;
|