File Coverage

blib/lib/Plack/Middleware/AMF.pm

Criterion	Covered	Total	%
statement	10	12	83.3
branch			n/a
condition			n/a
subroutine	4	4	100.0
pod			n/a
total	14	16	87.5

package Plack::Middleware::AMF;

use warnings;
use strict;

our $VERSION = '0.02';

use parent "Plack::Middleware";

use Data::AMF::Remoting;
use Plack::Request;
use Plack::Util;
use Plack::Util::Accessor qw/path headers_handler message_handler/;
use UNIVERSAL::require;

sub prepare_app {
	my $self = shift;

	unless (defined $self->headers_handler) {
		$self->headers_handler(\&_default_headers_handler);
	}

	unless (defined $self->message_handler) {
		$self->message_handler(\&_default_message_handler);
	}

	if (ref $self->headers_handler ne 'CODE') {
		die 'headers_handler should be a code reference';
	}

	if (ref $self->message_handler ne 'CODE') {
		die 'message_handler should be a code reference';
	}
}

sub call {
	my $self = shift;
	my $env  = shift;

	my $res = $self->_handle_amf($env);

	return $res if $res;

	return $self->app->($env);
}

sub _handle_amf {
	my ($self, $env) = @_;

	my $path_match = $self->path or return;
	my $path = $env->{PATH_INFO};

	for ($path) {
		my $matched = 'CODE' eq ref $path_match ? $path_match->($_) : $_ =~ $path_match;
		return unless $matched;
	}

	my $req = Plack::Request->new($env);
	my $res = $req->new_response(200);

	my $remoting = Data::AMF::Remoting->new(
		source              => $req->raw_body,
		headers_did_process => $self->headers_handler,
		message_did_process => $self->message_handler,
	);
	$remoting->run if $remoting->{source};

	$res->content_type('application/x-amf');
	$res->body($remoting->data);

	return $res->finalize;
}

sub _default_headers_handler {}

sub _default_message_handler {
	my $message = shift;

	my ($controller_name, $action_name) = split '\.', $message->target_uri;

	$controller_name->require or die $@;

	my $controller = $controller_name->new;
	my $action = $controller->can($action_name);

	if (defined $action) {
		return $controller->$action(@{ $message->value });
	}
}

1;
__END__