diff options
| -rw-r--r-- | irssi-logger/irssi_logger.pl | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/irssi-logger/irssi_logger.pl b/irssi-logger/irssi_logger.pl new file mode 100644 index 0000000..0d92013 --- /dev/null +++ b/irssi-logger/irssi_logger.pl @@ -0,0 +1,137 @@ +use warnings; +use strict; +use DBI; +use Irssi; +use Irssi::Irc; +use POSIX qw/strftime/; + +use vars qw($VERSION %IRSSI); + +# Requirements: +# - postgresql +# - postgresql-contrib (pg_trgm) + +$VERSION = "1.0"; +%IRSSI = ( + authors => "Aaron Bieber", + contact => "deftly\@gmail.com", + name => "irssi_logger", + description => "Logs chats to a PostgreSQL database.", + license => "BSD", + url => "https://github.com/qbit/irssi_logger", + ); + +my $user = $ENV{LOGNAME} || $ENV{USER} || getpwuid($<); +my $dbh; + + +my $sql = qq~ +insert into logs (logdate, nick, log, channel) values (?, ?, ?, ?) +~; +my $search = qq~ +SELECT log, similarity(log, ?) AS sml + FROM logs + WHERE log % ? and + channel = ? + ORDER BY sml DESC, log + LIMIT 1; +~; +my $check_db = qq~ +select true as exists from pg_tables where tablename = 'logs'; +~; +my $create = qq~ +CREATE TABLE logs ( + id serial not null, + dateadded timestamp without time zone default now(), + logdate timestamp without time zone default now(), + nick text not null, + log text not null, + channel text not null +) +~; +my $init_pg_trgm = qq~ +CREATE extension pg_trgm +~; +my $create_trgm_idx = qq~ +CREATE index logs_trgm_idx on logs USING gist (log gist_trgm_ops); +~; +my $create_date_idx = qq~ +CREATE index logs_date_idx on logs (logdate) +~; +my $create_chan_idx = qq~ +CREATE index logs_nick_idx on logs (nick) +~; + +sub db_init { + my ($dbname, $dbuser, $dbpass) = @_; + # my $mdbh = DBI->connect("dbi:Pg:dbname=$dbname", $dbuser, $dbpass) || Irssi::print("Can't connect to postgres! " . DBI::errstr); + my $mdbh = DBI->connect("DBI:Pg:database=$dbname;host=192.168.1.109;port=30432", $dbuser, $dbpass) || Irssi::print("Can't connect to postgres! " . DBI::errstr); + + my $sth = $mdbh->prepare($check_db); + $sth->execute(); + my $row = $sth->fetchrow_hashref(); + + if (! $row->{exists}) { + Irssi::print("Creating database."); + $mdbh->do($create) || Irssi::print("Can't create db! " . DBI::errstr); + $mdbh->do($init_pg_trgm) || Irssi::print("Can't create extension " . DBI::errstr); + $mdbh->do($create_trgm_idx) || Irssi::print("Can't create trgm index " . DBI::errstr); + $mdbh->do($create_date_idx) || Irssi::print("Can't create date index " . DBI::errstr); + $mdbh->do($create_chan_idx) || irssi::print("Can't create chan index " . DBI::errstr); + } else { + Irssi::print("Database already exists."); + } + $sth->finish(); + $mdbh->disconnect(); + + return 1; +} + +sub connect_db { + my $dbname = Irssi::settings_get_str('il_dbname') || $user; + my $dbuser = Irssi::settings_get_str('il_dbuser') || $user; + my $dbpass = Irssi::settings_get_str('il_dbpass') || ""; + + db_init($dbname, $dbuser, $dbpass); + + Irssi::print("Connecting to the database"); + + # return DBI->connect("dbi:Pg:dbname=$dbname", $dbuser, $dbpass) || Irssi::print("Can't connect to db!" . DBI::errstr); + return DBI->connect("dbi:Pg:dbname=$dbname;host=192.168.1.109;port=30432", $dbuser, $dbpass) || Irssi::print("Can't connect to db!" . DBI::errstr); +} + +sub write_db { + my ($nick, $message, $target) = @_; + my @vals; + my $date = strftime("%Y-%m-%d %H:%M:%S", localtime); + + $dbh = connect_db() unless $dbh; + + push(@vals, $date); + push(@vals, $nick); + push(@vals, $message); + push(@vals, $target); + + defined or $_ = "" for @vals; + + $dbh->do($sql, undef, @vals) || Irssi::print("Can't log to DB! " . DBI::errstr); +} + +sub log_me { + my ($server, $message, $target) = @_; + write_db($server->{nick}, $message, $target); +} + +sub log { + my ($server, $message, $nick, $address, $target) = @_; + write_db($nick, $message, $target) +} + +Irssi::signal_add_last('message public', 'log'); +Irssi::signal_add_last('message own_public', 'log_me'); + +Irssi::settings_add_str('irssi_logger', 'il_dbname', $user); +Irssi::settings_add_str('irssi_logger', 'il_dbuser', $user); +Irssi::settings_add_str('irssi_logger', 'il_dbpass', ""); + +Irssi::print("irssi_logger loaded!"); |