mirror of https://github.com/pragma-/pbot.git
52 lines
1.2 KiB
Perl
Executable File
52 lines
1.2 KiB
Perl
Executable File
#!/usr/bin/env perl
|
|
|
|
# Add hostmask field to Messages table of Message History database
|
|
|
|
use warnings; use strict;
|
|
|
|
BEGIN {
|
|
use File::Basename;
|
|
my $location = -l __FILE__ ? dirname readlink __FILE__ : dirname __FILE__;
|
|
unshift @INC, $location;
|
|
}
|
|
|
|
my ($data_dir, $version, $last_update) = @ARGV;
|
|
|
|
print "Updating message history database... version: $version, last_update: $last_update, data_dir: $data_dir\n";
|
|
|
|
if (not -e "$data_dir/message_history.sqlite3") {
|
|
print "$data_dir/message_history.sqlite3 does not exist; skipping update.\n";
|
|
exit 0;
|
|
}
|
|
|
|
use DBI;
|
|
|
|
my $dbh = DBI->connect("dbi:SQLite:dbname=$data_dir/message_history.sqlite3", "", "", {RaiseError => 1, PrintError => 0, AutoInactiveDestroy => 1, sqlite_unicode => 1})
|
|
or die $DBI::errstr;
|
|
|
|
eval {
|
|
my %existing = ();
|
|
foreach my $col (@{$dbh->selectall_arrayref("PRAGMA TABLE_INFO(Messages)")}) {
|
|
$existing{$col->[1]} = $col->[2];
|
|
}
|
|
|
|
$dbh->begin_work;
|
|
|
|
if (not exists $existing{'hostmask'}) {
|
|
$dbh->do('ALTER TABLE Messages ADD COLUMN hostmask TEXT COLLATE NOCASE');
|
|
}
|
|
|
|
$dbh->commit;
|
|
};
|
|
|
|
if ($@) {
|
|
print "Error updating: $@";
|
|
$dbh->rollback;
|
|
exit 1;
|
|
}
|
|
|
|
$dbh->disconnect;
|
|
|
|
|
|
exit 0;
|