File Coverage

blib/lib/BusyBird/Filter/Twitter.pm

Criterion	Covered	Total	%
statement	74	74	100.0
branch	23	24	95.8
condition	2	3	66.6
subroutine	16	16	100.0
pod	8	8	100.0
total	123	125	98.4

package BusyBird::Filter::Twitter;
use strict;
use warnings;
use BusyBird::DateTime::Format;
use BusyBird::Util qw(split_with_entities vivifiable_as);
use BusyBird::Filter qw(filter_map);
use Exporter qw(import);

{
    my @names = qw(all search_status status_id unescape);
    our %EXPORT_TAGS = (
        transform => [map { "trans_twitter_$_" } @names],
        filter => [map { "filter_twitter_$_" } @names]
    );
}
BusyBird::Util::export_ok_all_tags;

my $DATETIME_FORMATTER = 'BusyBird::DateTime::Format';


my %_SEARCH_KEY_MAP = (
    id => 'from_user_id',
    id_str => 'from_user_id_str',
    screen_name => 'from_user',
    profile_image_url => 'profile_image_url',
);

sub trans_twitter_search_status {
    my ($status) = @_;
    if(exists($status->{created_at})) {
        $status->{created_at} = $DATETIME_FORMATTER->format_datetime(
            $DATETIME_FORMATTER->parse_datetime($status->{created_at})
        );
    }
    return $status if defined $status->{user};
    $status->{user} = {};
    foreach my $new_id_key (keys %_SEARCH_KEY_MAP) {
        my $orig_id_key = $_SEARCH_KEY_MAP{$new_id_key};
        $status->{user}{$new_id_key} = delete $status->{$orig_id_key} if exists $status->{$orig_id_key};
    }
    return $status;
}

my $FILTER_SEARCH = filter_map \&trans_twitter_search_status;

sub filter_twitter_search_status {
    return $FILTER_SEARCH;
}

sub trans_twitter_status_id {
    my ($status, $api_url) = @_;
    $api_url = "https://api.twitter.com/1.1/" if not defined $api_url;
    $api_url =~ s|/+$||;
    foreach my $key (qw(id id_str in_reply_to_status_id in_reply_to_status_id_str)) {
        next if not defined $status->{$key};
        if(vivifiable_as($status->{busybird}, "HASH")
               && vivifiable_as($status->{busybird}{original}, "HASH")) {
            $status->{busybird}{original}{$key} = $status->{$key};
        }
        $status->{$key} = "$api_url/statuses/show/" . $status->{$key} . ".json";
    }
    return $status;
}

sub filter_twitter_status_id {
    my ($api_url) = @_;
    return filter_map sub { trans_twitter_status_id($_[0], $api_url) };
}

sub trans_twitter_unescape {
    my ($status) = @_;
    if(ref($status->{retweeted_status}) eq "HASH") {
        trans_twitter_unescape($status->{retweeted_status});
    }
    if(!defined($status->{text})) {
        return $status;
    }
    my $segments = split_with_entities($status->{text}, $status->{entities});
    my $new_text = "";
    my %new_entities = ();
    if(ref($status->{entities}) eq 'HASH') {
        %new_entities = map { $_ => [] } keys %{$status->{entities}};
    }
    foreach my $segment (@$segments) {
        $segment->{text} =~ s/</
        $segment->{text} =~ s/>/>/g;
        $segment->{text} =~ s/"/"/g;
        $segment->{text} =~ s/&/&/g;
        if(ref($segment->{entity}) eq "HASH") {
            if(vivifiable_as($segment->{entity}{indices}, "ARRAY")) {
                $segment->{entity}{indices}[0] = length($new_text);
                $segment->{entity}{indices}[1] = $segment->{entity}{indices}[0] + length($segment->{text});
            }
            push(@{$new_entities{$segment->{type}}}, $segment->{entity});
        }
        $new_text .= $segment->{text};
    }
    $status->{text} = $new_text;
    if(defined($status->{entities})) {
        $status->{entities} = \%new_entities;
    }
    return $status;
}

my $FILTER_UNESCAPE = filter_map \&trans_twitter_unescape;

sub filter_twitter_unescape {
    return $FILTER_UNESCAPE;
}

sub trans_twitter_all {
    my ($status, $api_url) = @_;
    return trans_twitter_unescape(
        trans_twitter_status_id(
            trans_twitter_search_status($status),
            $api_url
        )
    );
}

sub filter_twitter_all {
    my ($api_url) = @_;
    return filter_map sub { trans_twitter_all($_[0], $api_url) };
}


1;
__END__