handle scalar refs content as well

travisbeck · Aug 28, 2011 · 3e50de8 · 3e50de8
1 parent ced8577
commit 3e50de8
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/lib/Dancer.pm b/lib/Dancer.pm
@@ -375,14 +375,18 @@ sub _send_file {
                 exists $options{'around_cb'}
                     and return $options{'around_cb'}->($content);
 
-                while ( my $line = <$content> ) {
-                    exists $options{'before_cb'}
-                        and $options{'before_cb'}->($line);
-
-                    $writer->write($line);
-
-                    exists $options{'after_cb'}
-                        and $options{'after_cb'}->($line);
+                if ( ref $content ) {
+                    while ( my $line = <$content> ) {
+                        exists $options{'before_cb'}
+                            and $options{'before_cb'}->($line);
+
+                        $writer->write($line);
+
+                        exists $options{'after_cb'}
+                            and $options{'after_cb'}->($line);
+                    }
+                } else {
+                    $writer->write($content);
                 }
             };
         } );