File Coverage

blib/lib/Nephia/Plugin/SocketIO.pm

Criterion	Covered	Total	%
statement	27	60	45.0
branch	0	6	0.0
condition			n/a
subroutine	9	17	52.9
pod	2	3	66.6
total	38	86	44.1

package Nephia::Plugin::SocketIO;
use 5.008005;
use strict;
use warnings;
use parent 'Nephia::Plugin';
use PocketIO;
use Plack::Builder;
use HTML::Escape ();
use Sub::Recursive;
use Nephia::Plugin::SocketIO::Assets;

our $VERSION = "0.01";

sub new {
    my ($class, %opts) = @_;
    my $self = $class->SUPER::new(%opts);
    $self->app->builder_chain->append('SocketIO' => $class->can('_wrap_app'));
    return $self;
}

sub exports { qw/socketio/ }

sub socketio {
    my ($self, $context) = @_;
    return sub ($&) {
        my ($event, $code) = @_;
        $self->app->{events}{$event} = sub {
            my ($socket, $var) = @_;
            $code->($socket, _escape_html_recursive($var)); ### avoid XSS by _escape_html_recursive
        };
    };
}

sub _wrap_app {
    my ($app, $coderef) = @_;
    builder {
        mount '/socket.io.js' => sub {
            [200, ['Content-Type' => 'text/javascript'], [Nephia::Plugin::SocketIO::Assets->get('socket.io.js')]];
        };
        mount '/socket.io' => PocketIO->new(handler => sub {
            my $socket = shift;
            for my $event (keys %{$app->{events}}) {
                $socket->on($event => $app->{events}{$event});
            }
            $socket->send({buffer => []});
        });
        mount '/' => builder {
            enable 'SimpleContentFilter', filter => sub{
                s|()|$1\n|i;
            };
            $coderef;
        };
    };
}

sub _escape_html_recursive {
    my $v = shift;
    return unless $v;
    my $work = recursive {
        my $val = shift;
        ref($val) eq 'ARRAY' ? [map {$REC->($_)} @$val] :
        ref($val) eq 'HASH'  ? +{map {($_, $REC->($val->{$_}))} keys %$val} :
        HTML::Escape::escape_html($val) ;
    };
    $work->($v);
}

1;
__END__