File Coverage

blib/lib/Plack/Middleware/SocketIO/Base.pm

Criterion	Covered	Total	%
statement	27	72	37.5
branch	0	8	0.0
condition			n/a
subroutine	9	22	40.9
pod	9	9	100.0
total	45	111	40.5

package Plack::Middleware::SocketIO::Base;

use strict;
use warnings;

use JSON   ();
use Encode ();
use Try::Tiny;
use Scalar::Util qw(weaken);

use Plack::Request;
use Plack::Middleware::SocketIO::Handle;
use Plack::Middleware::SocketIO::Resource;

sub new {
    my $class = shift;

    my $self = bless {@_}, $class;

    weaken $self->{env};
    $self->{req} = Plack::Request->new($self->{env});

    return $self;
}

sub req { shift->{req} }
sub env { shift->{req}->{env} }

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

    return $self->{resource} unless defined $resource;

    $self->{resource} = $resource;

    return $self;
}

sub add_connection {
    my $self = shift;

    return Plack::Middleware::SocketIO::Resource->instance->add_connection(
        type => $self->name,
        @_
    );
}

sub remove_connection {
    my $self = shift;
    my ($conn) = @_;

    Plack::Middleware::SocketIO::Resource->instance->remove_connection(
        $conn->id);

    return $self;
}

sub find_connection_by_id {
    my $self = shift;
    my ($id) = @_;

    return Plack::Middleware::SocketIO::Resource->instance->connection($id);
}

sub client_connected {
    my $self = shift;
    my ($conn) = @_;

    return if $conn->is_connected;

    $self->_log_client_connected($conn);

    $conn->connect;
}

sub client_disconnected {
    my $self = shift;
    my ($conn) = @_;

    $conn->disconnect;

    $self->_log_client_disconnected($conn);

    $self->remove_connection($conn);
}

sub _log_client_connected {
    my $self = shift;
    my ($conn) = @_;

    my $logger = $self->_get_logger;
    return unless $logger;

    $logger->(
        {   level   => 'debug',
            message => sprintf(
                "Client '%s' connected via '%s'",
                $conn->id, $conn->type
            )
        }
    );
}

sub _log_client_disconnected {
    my $self = shift;
    my ($conn) = @_;

    my $logger = $self->_get_logger;
    return unless $logger;

    $logger->(
        {   level   => 'debug',
            message => sprintf("Client '%s' disconnected", $conn->id)
        }
    );
}

sub _get_logger {
    my $self = shift;

    return $self->env->{'psgix.logger'};
}

sub _build_handle {
    my $self = shift;

    return Plack::Middleware::SocketIO::Handle->new(@_);
}

1;
__END__