do not support streaming interface, at least for now. Fixes a bug whe…

…re Content-Length gets a wrong value when body does not have </html>
miyagawa · Dec 17, 2009 · f1b23b5 · f1b23b5
1 parent 7817433
commit f1b23b5
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/lib/Plack/Middleware/Debug.pm b/lib/Plack/Middleware/Debug.pm
@@ -144,15 +144,18 @@ sub call {
                     panels   => $self->panels,
                     BASE_URL => $env->{SCRIPT_NAME},
                 };
-                my $content = $self->renderer->($vars);
-                if (my $cl = $headers->get('Content-Length')) {
-                    $headers->set('Content-Length' => $cl + length $content);
-                }
-                return sub {
-                    my $chunk = shift;
-                    return unless defined $chunk;
-                    $chunk =~ s!(?=</body>)!$content!i;
-                    return $chunk;
+
+                if ($res->[2]) {
+                    my $content = $self->renderer->($vars);
+                    my @body;
+                    Plack::Util::foreach($res->[2], sub {
+                        my $chunk = shift;
+                        $chunk =~ s!(?=</body>)!$content!i;
+                        push @body, $chunk;
+                    });
+
+                    $headers->set('Content-Length' => length(join '', @body));
+                    $res->[2] = \@body;
                 };
             }
             $res;