ljr/livejournal/bin/ljsysban.pl

#!/usr/bin/perl
#

use strict;
use Getopt::Long;

# parse input options
my ($list, $add, $modify, $banid, $status, $bandate, $banuntil, $what, $value, $note);
exit 1 unless GetOptions('list' => \$list,
                         'add' => \$add,
                         'modify' => \$modify,
                         'banid=s' => \$banid,
                         'status=s' => \$status,
                         'bandate=s' => \$bandate,
                         'banuntil=s' => \$banuntil,
                         'what=s' => \$what,
                         'value=s' => \$value,
                         'note=s' => \$note,
                         );

# did they give valid input?
my $an_opt = ($what || $value || $status || $bandate || $banuntil || $note);
unless (($list   && (($banid && ! $an_opt) || (! $banid && $an_opt)) ||
         ($add    && $what && $value) ||
         ($modify && $banid && $an_opt))) {

            die "Usage: ljsysban.pl [opts]\n\n" .
                "  --list   { <--banid=?> | or one of:\n" .
                "             [--what=? --status=? --bandate=datetime --banuntil=datetime\n" .
                "              --value=? --note=?]\n" .
                "           }\n\n" .
                "  --add    <--what=? --value=?\n" . 
                "             [--status=? --bandate=datetime --banuntil=datetime --note=?]>\n\n" .
                "  --modify <--banid=?>\n" . 
                "             [--status=? --bandate=datetime --banuntil=datetime --value=? --note=?]\n\n";
        }

# now load in the beast
require "$ENV{'LJHOME'}/cgi-bin/ljlib.pl";
my $dbh = LJ::get_db_writer();

# list bands
if ($list) {
    my $where;
    if ($banid) {
        $where = "banid=" . $dbh->quote($banid);
    } else {
        my @where = ();
        push @where, ("what=" . $dbh->quote($what)) if $what;
        push @where, ("value=" . $dbh->quote($value)) if $value;
        push @where, ("status=" . $dbh->quote($status)) if $status;
        push @where, ("bandate=" . $dbh->quote($bandate)) if $bandate;
        push @where, ("banuntil=" . $dbh->quote($banuntil)) if $banuntil;
        push @where, ("note=" . $dbh->quote($note)) if $note;
        $where = join(" AND ", @where);
    }

    my $sth = $dbh->prepare("SELECT * FROM sysban WHERE $where ORDER BY bandate ASC");
    $sth->execute;
    my $ct;
    while (my $ban = $sth->fetchrow_hashref) {
        print "> banid: $ban->{'banid'}, status: $ban->{'status'}, ";
        print "bandate: " . ($ban->{'bandate'} ? $ban->{'bandate'} : "BOT") . ", ";
        print "banuntil: " . ($ban->{'banuntil'} ? $ban->{'banuntil'} : "EOT") . "\n";
        print "> what: $ban->{'what'}, value: $ban->{'value'}\n";
        print "> note: $ban->{'note'}\n" if $ban->{'note'};
        print "\n";
        $ct++;
    }
    print "\n\tNO MATCHES\n\n" unless $ct;

    exit;
}

# add new ban
if ($add) {

    $status = ($status eq 'expired' ? 'expired' : 'active');

    $dbh->do("INSERT INTO sysban (status, what, value, note, bandate, banuntil)" .
             "VALUES (?, ?, ?, ?, " . 
             ($bandate ? $dbh->quote($bandate) : 'NOW()') . ", " . 
             ($banuntil ? $dbh->quote($banuntil) : 'NULL') . ")",
             undef, $status, $what, $value, $note);
    die $dbh->errstr if $dbh->err;
    my $insertid = $dbh->{'mysql_insertid'};

    if ($what eq 'ip') {
        LJ::procnotify_add("ban_ip", { 'ip' => $value,
                                       'exptime' => LJ::mysqldate_to_time($banuntil) });
        LJ::MemCache::delete("sysban:ip");
    }
    if ($what eq 'uniq') {
        LJ::procnotify_add("ban_uniq", { 'uniq' => $value,
                                         'exptime' => LJ::mysqldate_to_time($banuntil) });
        LJ::MemCache::delete("sysban:uniq");
    }

    # log in statushistory
    LJ::statushistory_add(0, 0, 'sysban_add',
                          "banid=$insertid; status=$status; " .
                          "bandate=" . ($bandate || LJ::mysql_time()) . "; " .
                          "banuntil=" . ($banuntil || 'NULL') . "; " .
                          "what=$what; value=$value; " .
                          "note=$note;");

    print "CREATED: banid=$insertid\n";
    exit;
}

# modify existing ban
if ($modify) {

    # load selected ban
    my $ban = $dbh->selectrow_hashref("SELECT * FROM sysban WHERE banid=?", undef, $banid);
    die $dbh->errstr if $dbh->err;

    my @set = ();

    # ip/uniq ban and we're going to change the value
    if (($value && $value ne $ban->{'value'}) ||
        $banuntil && $banuntil ne $ban->{'banuntil'} || 
        ($status && $status ne $ban->{'status'} && $status eq 'expired')) {

        if ($ban->{'what'} eq 'ip') {
            LJ::procnotify_add("unban_ip", { 'ip' => $value || $ban->{'value'}});
            LJ::MemCache::delete("sysban:ip");
        }
        
        if ($ban->{'what'} eq 'uniq') {
            LJ::procnotify_add("unban_uniq", { 'uniq' => $value || $ban->{'value'} });
            LJ::MemCache::delete("sysban:uniq");
        }
    }
        
    # what - must have a value
    if ($what && $what ne $ban->{'what'}) {
        $ban->{'what'} = $what;
        push @set, "what=" . $dbh->quote($ban->{'what'});
    }

    # ip/uniq ban and we are going to change the value
    if (($value && $value ne $ban->{'value'}) ||
        $banuntil && $banuntil ne $ban->{'banuntil'} || 
        ($status && $status ne $ban->{'status'} && $status eq 'active')) {

        my $new_banuntil = $banuntil || $ban->{'banuntil'};

        if ($ban->{'what'} eq 'ip') {
            LJ::procnotify_add("ban_ip", { 'ip' => $value || $ban->{'value'},
                                           'exptime' => LJ::mysqldate_to_time($new_banuntil) });
            LJ::MemCache::delete("sysban:ip");
        }

        if ($ban->{'what'} eq 'uniq') {
            LJ::procnotify_add("ban_uniq", { 'uniq' => $value || $ban->{'value'},
                                             'exptime' => LJ::mysqldate_to_time($new_banuntil) });
            LJ::MemCache::delete("sysban:uniq");
        }
    }

    # value - must have a value
    if ($value && $value ne $ban->{'value'}) {
        $ban->{'value'} = $value;
        push @set, "value=" . $dbh->quote($ban->{'value'});
    }

    # status - must have a value
    if ($status && $status ne $ban->{'status'}) {
        $ban->{'status'} = ($status eq 'expired' ? 'expired' : 'active');
        push @set, "status=" . $dbh->quote($ban->{'status'});
    }

    # banuntil
    if ($banuntil && $banuntil ne $ban->{'banuntil'}) {
        $ban->{'banuntil'} = ($banuntil && $banuntil ne 'NULL') ? $banuntil : 0;
        push @set, "banuntil=" . ($ban->{'banuntil'} ? $dbh->quote($ban->{'banuntil'}) : 'NULL');
    }

    # bandate - must have a value
    if ($bandate && $bandate ne $ban->{'bandate'}) {
        $ban->{'bandate'} = $bandate;
        push @set, "bandate=" . $dbh->quote($ban->{'bandate'});
    }

    # note - can be changed to blank
    if (defined $note && $note ne $ban->{'note'}) {
        $ban->{'note'} = $note;
        push @set, "note=" . $dbh->quote($ban->{'note'});
    }

    # do update
    $dbh->do("UPDATE sysban SET " . join(", ", @set) . " WHERE banid=?", undef, $ban->{'banid'});

    # log in statushistory
    my $msg; map { $msg .= " " if $msg;
                   $msg .= "$_=$ban->{$_};" }  qw(banid status bandate banuntil what value note);
    LJ::statushistory_add(0, 0, 'sysban_mod', $msg);

    print "MODIFIED: banid=$banid\n";
    exit;
}