File Coverage

blib/lib/Net/Google/Code/Issue/Attachment.pm

Criterion	Covered	Total	%
statement	64	65	98.4
branch	16	26	61.5
condition	2	5	40.0
subroutine	9	9	100.0
pod	2	2	100.0
total	93	107	86.9

package Net::Google::Code::Issue::Attachment;
use Any::Moose;
with 'Net::Google::Code::Role::Fetchable', 'Net::Google::Code::Role::HTMLTree';
use Scalar::Util qw/blessed/;
use MIME::Types;
use File::MMagic;

has 'name'         => ( isa => 'Str', is => 'rw' );
has 'url'          => ( isa => 'Str', is => 'rw' );
has 'size'         => ( isa => 'Str', is => 'rw' );
has 'id'           => ( isa => 'Int', is => 'rw' );
has content => (
    isa     => 'Str',
    is      => 'rw',
    lazy    => 1,
    default => sub { ($_[0]->_load)[0] },
);

has content_type => (
    isa     => 'Str',
    is      => 'rw',
    lazy    => 1,
    default => sub { ($_[0]->_load)[1] },
);

sub parse {
    my $self = shift;
    my $tree    = shift;
    my $need_delete = not blessed $tree;
    $tree = $self->html_tree( html => $tree ) unless blessed $tree;
    my $tr = $tree->find_by_tag_name('tr');

    my $b    = $tr->find_by_tag_name('b');    # name lives here
    if ($b) {
        my $name = $b->content_array_ref->[0];
        $name =~ s/^\s+//;
        $name =~ s/\s+$//;
        $self->name($name);

        # google code doesn't parse download's content type at all, we need to
        # figure it out by ourselves
        my $content_type = $self->_mime_type;
        if ( $content_type ) {
            $self->content_type( $content_type );
        }
    }

    my @tds = $tr->find_by_tag_name('td');
    if (@tds) {
        $self->url( $tds[0]->find_by_tag_name('a')->attr('href') );
        if ( $self->url =~ /aid=(-?\d+)/ ) {
            my $id = $1;
            $self->id( $id );
        }

        if ( $tds[1] ) {
            my $size = $tds[1]->content_array_ref->[2];
            if ( $size && $size =~ /([\d.]+)\s*(\w+)/ ) {
                $self->size("$1 $2");
            }
            else {
                warn 'failed to parse size' unless $size;
            }
        }
    }

    $tree->delete if $need_delete;
    return 1;
}

sub parse_attachments {
    my $self    = shift;
    my $element = shift;
    my $need_delete = not blessed $element;
    $element = $self->html_tree( html => $element ) unless blessed $element;

    my @attachments;

    my @items = $element->find_by_tag_name('tr');
    while ( scalar @items ) {
        my $tr = shift @items;
        my $a   = Net::Google::Code::Issue::Attachment->new;

        if ( $a->parse( $tr ) ) {
            push @attachments, $a;
        }
    }
    $element->delete if $need_delete;
    return @attachments;
}

sub _load {
    my $self    = shift;
    #XXX weird happens if the previous fetch is also an attachment,
    # which will make the following fetch a Bad Request.
    $self->fetch( 'http://code.google.com' );

    my $content = $self->fetch( $self->url );

    # in case MIME::Types failed to get, let File::MMagic rescue!
    my $content_type =
         $self->_mime_type
      || File::MMagic->new->checktype_contents($content)
      || 'application/octet-stream';
    $self->content( $content );
    $self->content_type( $content_type );
    return $content, $content_type;
}

sub _mime_type {
    my $self      = shift;
    my $mime_type = MIME::Types->new->mimeTypeOf( $self->name );
    return $mime_type ? $mime_type->type : undef;
}

no Any::Moose;
__PACKAGE__->meta->make_immutable;

1;

__END__