File Coverage

blib/lib/AnyEvent/Digg/Stream.pm

Criterion	Covered	Total	%
statement	18	60	30.0
branch	0	14	0.0
condition	0	14	0.0
subroutine	6	16	37.5
pod	1	1	100.0
total	25	105	23.8

package AnyEvent::Digg::Stream;

use strict;
use warnings;

use AnyEvent::HTTP;
use Carp qw(croak);
use JSON;
use URI;

our $VERSION = '0.05';
$VERSION = eval $VERSION;

sub new {
    my ($class, %params) = @_;

    my $on_error      = $params{on_error}      || sub { croak @_ };
    my $on_disconnect = $params{on_disconnect} || sub { croak 'Disconnected' };
    my $on_event      = $params{on_event}      || sub { };
    my $on_comment    = $params{on_comment}    || sub { };
    my $on_digg       = $params{on_digg}       || sub { };
    my $on_submission = $params{on_submission} || sub { };

    my %events = (
        digg       => $on_digg,
        comment    => $on_comment,
        submission => $on_submission,
    );

    my @events;
    for my $event (keys %{{map { $_=>1 } @{$params{events}} }}) {
        $on_error->("Unknown event type: $event\n") unless $events{$event};
        push @events, $event;
    }

    my $uri = URI->new('http://services.digg.com/2.0/stream');
    $uri->query_form(
        format => 'json',
        3 == @events ? () : @events ? (types => join ',', @events) : (),
    );

    my $json = JSON->new->utf8;

    my $conn = http_request(
        GET     => $uri,
        headers => {
            'User-Agent' => __PACKAGE__ . "/$VERSION",
            Referer      => undef,
        },
        on_header => sub {
            my ($headers) = @_;
            if ('200' ne $headers->{Status}) {
                $on_error->("$headers->{Status}: $headers->{Reason}");
                return;
            }
            if ('application/json' ne $headers->{'content-type'}) {
                $on_error->(
                    "Unexpected Content-Type: $headers->{'content-type'}"
                );
                return;
            }
            return 1;
        },
        on_body => sub {
            my ($content) = @_;
            if (my $data = eval { $json->incr_parse($content) }) {
                $on_event->($data);
                ($events{$data->{type}} || sub {})->($data);
            }
            elsif ($@) {
                $json->incr_skip;
            }
            return 1;
        },
        sub { $on_disconnect->() },
    );

    return $conn;
}


1;

__END__