root/lang/perl/Morris/trunk/lib/Morris/Plugin/Channel/Reputation.pm @ 31817

# $Id$

package Morris::Plugin::Channel::Reputation;
use Moose;
use DBI;

with 'Morris::Plugin';

__PACKAGE__->meta->make_immutable;

no Moose;

sub register {
    my ($self, $conn) = @_;
    $conn->register_hook( 'channel.public', sub { $self->handle_message(@_) } );
}

sub handle_message {
    my ($self, $args) = @_;

    my $channel = $args->{message}->channel;
    my $message = $args->{message}->message;
    my $dbh;

    while ( $message =~ m{([\w_-]+)(--|\+\+)}g ) {
        my $who = $1;
        my $action = $2;
        my $add = $action eq '--' ? -1 : 1;
        my ($pluses, $minuses, $score) = (0, 0, 0);

        $dbh ||= $self->get_dbh();

        my $sth = $dbh->prepare("SELECT score, pluses, minuses FROM reputation WHERE nickname = ?");
        my $rv;
        if ( ($rv = $sth->execute($who)) > 0) {
            ($pluses, $minuses, $score) = $sth->fetchrow_array();
        }
        $sth->finish;

        if ($rv > 0) {
            $dbh->do("UPDATE reputation SET score = score + ?, pluses = pluses + ?, minuses = minuses + ? WHERE nickname = ?", undef, $add, $add > 0 ? (1, 0) : (0, 1), $who);
        } else {
            $dbh->do("INSERT INTO reputation (score, pluses, minuses, nickname) VALUES( ?, ? )", undef, $add, $add > 0 ? (1, 0) : (0, 1), $who);
        }

        $self->connection->irc_notice({
            channel => $channel,
            message => "$who: " . ($score + $add) . " ($pluses++, $minuses++)"
        });
    }
}

sub get_dbh {
    my $dbh = DBI->connect('dbi:SQLite:dbname=/service/morris/reputation.db');

    $dbh->do(<<EOSQL);
CRETE TABLE IF NOT EXISTS reputation (
    id INTEGER AUTO_INCREMENT PRIMARY KEY,
    score INTEGER NOT NULL DEFAULT 0,
    pluses INTEGER NOT NULL DEFAULT 0,
    minuses INTEGER NOT NULL DEFAULT 0,
    nickname TEXT NOT NULL,
    UNIQUE KEY (nickname)
EOSQL
    return $dbh;
}

1;