X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FCatalyst%2FAction%2FSerialize%2FJSONP.pm;h=4354ce2a062705b0c240aa7a77f5bdfcc07ce583;hb=9ffbd15626ce11a2c727169d5b0d382f4386d486;hp=00c9f84ab29b37c3c659bc7baa85f985d8b3c2e3;hpb=d0d292d4373d4af3455eac0b88366fef95f354be;p=catagits%2FCatalyst-Action-REST.git

diff --git a/lib/Catalyst/Action/Serialize/JSONP.pm b/lib/Catalyst/Action/Serialize/JSONP.pm
index 00c9f84..4354ce2 100644
--- a/lib/Catalyst/Action/Serialize/JSONP.pm
+++ b/lib/Catalyst/Action/Serialize/JSONP.pm
@@ -1,7 +1,9 @@
 package Catalyst::Action::Serialize::JSONP;
-use base 'Catalyst::Action::Serialize::JSON';
+use Moose;
+use namespace::autoclean;
+BEGIN {extends 'Catalyst::Action::Serialize::JSON'};
 
-sub execute {
+after 'execute' => sub { 
   my $self = shift;
   my ($controller, $c) = @_;
 
@@ -11,20 +13,15 @@ sub execute {
       $controller->{'callback_key'} 
     ) || 'callback';
 
-  if ($c->req->param($callback_key)) {
-    $self->{_jsonp_callback} = $c->req->param($callback_key);
-    $c->res->content_type('text/javascript');
+  my $callback_value = $c->req->param($callback_key);
+  if ($callback_value) {
+    if ($callback_value =~ /^\w+/) {
+      $c->res->content_type('text/javascript');
+      $c->res->output($callback_value.'('.$c->res->output().');');
+    } else { 
+      warn 'Callback: '.$callback_value.' will not generate valid Javascript. Falling back to JSON output';
+    }
   }
-  $self->next::method($controller, $c);
-}
-
-sub serialize {
-  my $self = shift;
-  my $json = $self->next::method(@_);
-  if ($self->{_jsonp_callback}) {
-    $json = $self->{_jsonp_callback}.'('.$json.');';
-  }
-  return $json;
-}
+};
 
 1;