Add boilderplate POD
[catagits/Catalyst-Engine-STOMP.git] / lib / Catalyst / Controller / MessageDriven.pm
index 10e645d..56883c0 100644 (file)
@@ -1,6 +1,9 @@
 package Catalyst::Controller::MessageDriven;
 use Moose;
-use YAML::XS qw/ LoadFile Dump /;
+use Data::Serializer;
+use Moose::Util::TypeConstraints;
+use MooseX::Types::Moose qw/Str/;
+use namespace::autoclean;
 
 BEGIN { extends 'Catalyst::Controller' }
 
@@ -14,10 +17,10 @@ Catalyst::Controller::MessageDriven
   use Moose;
   BEGIN { extends 'Catalyst::Controller::MessageDriven' }
 
-  sub some_action : Local { 
+  sub some_action : Local {
       my ($self, $c, $message) = @_;
 
-      # Handle message 
+      # Handle message
 
       # Reply with a minimal response message
       my $response = { type => 'testaction_response' };
@@ -28,67 +31,143 @@ Catalyst::Controller::MessageDriven
 
 A Catalyst controller base class for use with Catalyst::Engine::Stomp,
 which handles YAML-serialized messages. A top-level "type" key in the
-YAML determines the action dispatched to. 
+YAML determines the action dispatched to.
+
+=head1 METHODS
+
+=head2 begin
+
+Deserializes the request into C<< $c->stash->{request} >>
+
+=head2 default
+
+Dispatches to method named by the key C<< $c->stash->{request}->{type} >>
+
+=head2 end
+
+Serializes the response from C<< $c->stash->{response} >>
+
+=head1 CONFIGURATION
+
+In the configuration file add the following to set the value for a parameter
+
+  <MessageDriven>
+    type_key foo
+  </MessageDriven>
+
+=head2 type_key
+
+The hash key the module will try to pull out the received message to call
+within the controller. This defaults to 'type'.
+
+=head2 serializer
+
+The serializer used to serialiser/deserialise. See Data::Serializer to see
+what is available. Defaults to YAML. JSON is anotther that is available.
+
 
 =cut
 
-sub begin : Private { 
-       my ($self, $c) = @_;
-       
-       # Deserialize the request message
-       
-        my $message;
-       eval {
-               my $body = $c->request->body;
-               $message = LoadFile( "$body" );
-       };
-       if ($@) {
-               # can't reply - reply_to is embedded in the message
-               $c->error("exception in deserialize: $@");
-       }
-       else {
-               $c->stash->{request} = $message;
-       }
+class_type 'Data::Serializer';
+my $serializer_t = subtype 'Data::Serializer', where { 1 };
+coerce $serializer_t, from 'Str',
+    via { Data::Serializer->new( serializer => $_ ) };
+
+has serializer => (
+    isa => $serializer_t, is => 'ro', required => 1,
+    default => 'YAML', coerce => 1,
+);
+
+has type_key => (
+    is => 'ro', required =>1,
+    default => 'type',
+);
+
+
+sub begin : Private {
+    my ($self, $c) = @_;
+
+    # Deserialize the request message
+    my $message;
+    my $s = $self->serializer;
+    eval {
+        my $body = $c->request->body;
+        open my $IN, "$body" or die "can't open temp file $body";
+        $message = $s->raw_deserialize(do { local $/; <$IN> });
+    };
+    if ($@) {
+        # can't reply - reply_to is embedded in the message
+        $c->error("exception in deserialize: $@");
+    }
+    else {
+        $c->stash->{request} = $message;
+    }
 }
 
 sub end : Private {
-       my ($self, $c) = @_;
-
-       # Engine will send our reply based on the value of this header.
-       $c->response->headers->header( 'X-Reply-Address' => $c->stash->{request}->{reply_to} );
-
-       # Custom error handler - steal errors from catalyst and dump them into
-       # the stash, to get them serialized out as the reply.
-       if (scalar @{$c->error}) {
-               my $error = join "\n", @{$c->error};
-               $c->stash->{response} = { status => 'ERROR', error => $error };
-               $c->error(0); # clear errors, so our response isn't clobbered
-       }
-
-       # Serialize the response
-       my $output;
-       eval {
-               $output = Dump( $c->stash->{response} );
-       };
-       if ($@) {
-               my $error = "exception in serialize: $@";
-               $c->error($error);
-               $c->stash->{response} = { status => 'ERROR', error => $error };
-               $output = Dump( $c->stash->{response} );
-       }
-
-       $c->response->output( $output );
+    my ($self, $c) = @_;
+
+    # Engine will send our reply based on the value of this header.
+    $c->response->headers->header( 'X-Reply-Address' => $c->stash->{request}->{reply_to} );
+
+    # The wire response
+    my $output;
+
+    # Load a serializer
+    my $s = $self->serializer;
+
+    # Custom error handler - steal errors from catalyst and dump them into
+    # the stash, to get them serialized out as the reply.
+     if (scalar @{$c->error}) {
+        $c->log->error($_) for @{$c->error}; # Log errors in Catalyst
+        my $error = join "\n", @{$c->error};
+        $c->stash->{response} = { status => 'ERROR', error => $error };
+        $output = $s->serialize( $c->stash->{response} );
+        $c->clear_errors;
+        $c->response->status(400);
+     }
+
+    # Serialize the response
+    eval {
+        $output = $s->raw_serialize( $c->stash->{response} );
+    };
+    if ($@) {
+         my $error = "exception in serialize: $@";
+         $c->stash->{response} = { status => 'ERROR', error => $error };
+         $output = $s->serialize( $c->stash->{response} );
+        $c->response->status(400);
+    }
+
+    $c->response->output( $output );
 }
 
 sub default : Private {
-       my ($self, $c) = @_;
-
-       # Forward the request to the appropriate action, based on the
-       # message type.
-       my $action = $c->stash->{request}->{type};
-       $c->forward($action, [$c->stash->{request}]);
+    my ($self, $c) = @_;
+
+    # Forward the request to the appropriate action, based on the
+    # message type.
+    my $action = $c->stash->{request}->{ $self->type_key };
+    if (defined $action) {
+        $c->forward($action, [$c->stash->{request}]);
+    }
+    else {
+         $c->error('no message type specified');
+    }
 }
 
 __PACKAGE__->meta->make_immutable;
 
-1;
+=head1 AUTHOR AND CONTRIBUTORS
+
+See information in L<Catalyst::Engine::Stomp>
+
+=head1 LICENCE AND COPYRIGHT
+
+Copyright (C) 2009 Venda Ltd
+
+This library is free software; you can redistribute it and/or modify
+it under the same terms as Perl itself, either Perl version 5.8.8 or,
+at your option, any later version of Perl 5 you may have available.
+
+=cut
+