File Coverage

blib/lib/POE/Component/IRC/Plugin/OutputToPastebin.pm

Criterion	Covered	Total	%
statement	15	62	24.1
branch	0	14	0.0
condition	0	3	0.0
subroutine	5	16	31.2
pod	1	5	20.0
total	21	100	21.0

package POE::Component::IRC::Plugin::OutputToPastebin;

use warnings;
use strict;

our $VERSION = '0.002';

use Carp;
use POE::Component::IRC::Plugin qw(:ALL);
use POE qw(Component::WWW::Pastebin::Bot::Pastebot::Create);

sub new {
    my $class = shift;
    croak "Must have even number of arguments to new()"
        if @_ & 1;

    my %args = @_;
    $args{ +lc } = delete $args{ $_ } for keys %args;
    %args = (
        max_tries   => 3,
        trigger     => '[irc_to_pastebin]',
        pastebins   => [ qw(http://p3m.org/pfn) ],
        timeout     => 20,
        debug       => 0,
        %args,
    );
    $args{_site_iterator} = 0;
    return bless \%args, $class;
}

sub PCI_register {
    my ( $self, $irc ) = @_;
    $self->{irc} = $irc;

    $irc->plugin_register( $self, 'USER', qw(privmsg notice) );
    $irc->{session_id} = POE::Session->create(
        object_states => [
            $self   => [ qw(_start  _shutdown  _paste_it  _pasted) ],
        ],
    )->ID;

    return 1;
}

sub _start {
    my ( $kernel, $self ) = @_[KERNEL, OBJECT];
    $self->{session_id} = $_[SESSION]->ID;
    $kernel->refcount_increment( $self->{session_id} => __PACKAGE__ );

    $self->{poco}
    = POE::Component::WWW::Pastebin::Bot::Pastebot::Create->spawn(
        obj_args => { timeout => $self->{timeout} },
        debug    => $self->{debug},
    );
}

sub U_privmsg { shift->_process( @_ ); }
sub U_notice  { shift->_process( @_ ); }

sub _process {
    my ( $self, $irc, $out_ref ) = @_;

    my ( $pre_text, $to_paste ) = split /\Q$self->{trigger}/, $$out_ref, 2;

    if ( defined $to_paste and length $to_paste ) {
        $poe_kernel->post(
            $self->{session_id} => _paste_it => $pre_text, $to_paste, 0
        );

        return PCI_EAT_ALL;
    }

    return PCI_EAT_NONE;
}

sub _paste_it {
    my ( $self, $pre_text, $to_paste, $try ) = @_[ OBJECT, ARG0..$#_];

    $self->{_site_iterator} = 0
        if $#{ $self->{pastebins} } < ++$self->{_site_iterator};

    $self->{poco}->paste( {
            site    => $self->{pastebins}[ $self->{_site_iterator} ],
            event   => '_pasted',
            content => $to_paste,
            _pre    => $pre_text,
            _try    => $try,
        }
    );
}

sub _post_irc_message {
    my ( $self, $message ) = @_;
    $poe_kernel->post( $self->{irc} => quote => $message );
}

sub _pasted {
    my ( $self, $in_ref ) = @_[OBJECT, ARG0];
    if ( $in_ref->{error} ) {
        $self->{debug} and carp "Paster error: $in_ref->{error}";

        if ( ++$in_ref->{_try} > $self->{max_tries} ) {
            $self->_post_irc_message("$in_ref->{_pre} [paster error]");
        }
        else {
            $poe_kernel->post(
                $self->{session_id} => _paste_it =>
                @$in_ref{ qw(_pre content _try) }
            );
        }
    }
    else {
        $self->{debug} and carp "Pasted, got uri $in_ref->{uri}";

        $self->_post_irc_message( "$in_ref->{_pre}$in_ref->{uri}" );
    }
}

sub PCI_unregister {
    my ( $self, $irc ) = @_;
    $poe_kernel->call( $self->{session_id} => '_shutdown' );
    delete $self->{irc};
    return 1;
}

sub _shutdown {
    my ( $kernel, $self ) = @_[KERNEL, OBJECT];
    $self->{poco}->shutdown;
    $kernel->alarm_remove_all;
    $kernel->refcount_decremenet( $self->{session_id} => __PACKAGE__ );
    return;
}


1;
__END__