File Coverage

blib/lib/SocketIO/Emitter.pm

Criterion	Covered	Total	%
statement	18	67	26.8
branch	0	12	0.0
condition	0	6	0.0
subroutine	6	17	35.2
pod	0	11	0.0
total	24	113	21.2

package SocketIO::Emitter;
use strict;
use warnings;
our $VERSION = 0.04;

use Redis;
use Data::MessagePack;

use Moo;
use namespace::clean;

has redis  => ( is => 'rw');
has key    => ( is => 'rw');
has prefix => ( is => 'rw');
has rooms  => ( is => 'rw', default => sub {[]} );
has flags  => ( is => 'rw', default => sub {{}} );
has messagepack => ( is => 'rw', default => sub { Data::MessagePack->new() } );

my $UID = 'emitter';
my $EVENT = 2;
my $BINARY_EVENT = 5;

sub BUILD {
     my $self = shift;
     # redis
     my $redis = $self->redis || Redis->new();
     $self->redis($redis);
     # prefix
     $self->prefix($self->key ? $self->key : 'socket.io');
}

sub json      { $_[0]->flags->{json}      = 1; $_[0]; }
sub volatile  { $_[0]->flags->{volatile}  = 1; $_[0]; }
sub broadcast { $_[0]->flags->{broadcast} = 1; $_[0]; }

sub in {
    my ($self, $room) = @_;
    push @{$self->rooms}, $room
        unless grep { $_ eq $room } @{$self->rooms};
    $self;
}

sub to {
    my ($self, $room) = @_;
    $self->in($room);
    $self;
}

sub of {
    my ($self, $nsp) = @_;
    $self->flags->{nsp} = $nsp;
    $self;
}

sub emit {
    my ($self, @args) = @_;

    $self->flags->{nsp} = '/' unless exists $self->{flags}->{nsp};
    my $chn = $self->prefix . '#' . $self->flags->{nsp} . '#';

    my $pack_data = $self->pack(@args);
    my $packed = $self->messagepack->utf8->pack($pack_data);

    if (scalar @{ $self->rooms }) {
        for my $room (@{ $self->rooms }) {
            my $chn_room = $chn . $room . '#';
            $self->redis->publish($chn_room, $packed);
        }
    } else {
        $self->redis->publish($chn, $packed);
    }

    # clear
    $self->clear;

    $self;
}

sub pack {
    my ($self, @args) = @_;

    my %packet;
    $packet{type} = ($self->include_binary(@args)) ? $BINARY_EVENT : $EVENT;
    $packet{data} = \@args;
    $packet{nsp}  = delete $self->flags->{'nsp'};

    return [$UID, \%packet, { rooms => $self->rooms, flags => $self->flags }];
}

sub clear {
    my ($self) = @_;

    $self->rooms([]);
    $self->flags({});
}

sub include_binary {
    my ($self, @args) = @_;
    for(@args) {
        return 1 if $_ && /[[:^ascii:]]/;
    }
    return;
}

1;

__END__