File Coverage

blib/lib/Plack/App/EventSource.pm

Criterion	Covered	Total	%
statement	44	49	89.8
branch	8	12	66.6
condition			n/a
subroutine	8	8	100.0
pod	1	1	100.0
total	61	70	87.1

package Plack::App::EventSource;
use 5.008001;
use strict;
use warnings;

use parent 'Plack::Component';

our $VERSION = "0.03";

use Plack::Util::Accessor qw(handler_cb headers);
use Plack::App::EventSource::Connection;

sub call {
    my $self = shift;
    my ($env) = @_;

    return [405, [], ['Method not allowed']]
      unless $env->{REQUEST_METHOD} eq 'GET';

    return sub {
        my $respond = shift;

        my $writer = $respond->(
            [
                200,
                [
                    'Content-Type' => 'text/event-stream; charset=UTF-8',
                    'Cache-Control' =>
                      'no-store, no-cache, must-revalidate, max-age=0',
                    'Access-Control-Allow-Methods' => 'GET',
                    @{$self->headers || []}
                ]
            ]
        );

        my $connection; $connection = Plack::App::EventSource::Connection->new(
            push_cb => sub {
                my (@messages) = @_;

                foreach my $msg (@messages) {
                    if (ref $msg eq 'HASH') {
                        my $event = join "\x0d\x0a",
                            map { "$_: ".$msg->{$_} }
                            grep { defined $msg->{$_} }
                            qw(event id data retry);
                        eval { $writer->write("$event\x0d\x0a"); 1 }
                          or do { $connection->close; return };
                    }
                    else {
                        eval { $writer->write("data: $msg\x0d\x0a"); 1 } or do { $connection->close; return };
                    }
                }

                eval { $writer->write("\x0d\x0a"); 1 } or do {
                    $connection->close;
                };
            },
            close_cb => sub {
                eval { $writer->close };
            }
        );

        $self->{handler_cb}->($connection, $env);
    };
}

1;
__END__