wwwstream: oneshot sets content-length

[public-inbox.git] / lib / PublicInbox / WwwStream.pm

diff --git a/lib/PublicInbox/WwwStream.pm b/lib/PublicInbox/WwwStream.pm
index 2dd8b157ddc7334d49a95fdd97fc7ea47563f59c..985e02622048b4a2ada72510aa84d4ab6273d7be 100644 (file)
--- a/lib/PublicInbox/WwwStream.pm
+++ b/lib/PublicInbox/WwwStream.pm
@@ -9,6 +9,7 @@
 package PublicInbox::WwwStream;
 use strict;
 use warnings;
+use bytes (); # length
 use PublicInbox::Hval qw(ascii_html prurl);
 our $TOR_URL = 'https://www.torproject.org/';
 our $CODE_URL = 'https://public-inbox.org/public-inbox.git';
@@ -28,7 +29,7 @@ sub new {
 
        bless {
                nr => 0,
-               cb => $cb || \&close,
+               cb => $cb,
                ctx => $ctx,
                base_url => base_url($ctx),
        }, $class;
@@ -170,13 +171,18 @@ sub getline {
 }
 
 sub oneshot {
-       my ($ctx, $code, $strref) = @_;
+       my ($ctx, $code, $sref) = @_;
        my $self = bless {
                ctx => $ctx,
                base_url => base_url($ctx),
        }, __PACKAGE__;
-       [ $code, [ 'Content-Type', 'text/html; charset=UTF-8' ],
-               [ _html_top($self), $$strref, _html_end($self) ] ]
+       my @x = (_html_top($self), $sref ? $$sref : (), _html_end($self));
+       my $len = 0;
+       $len += bytes::length($_) for @x;
+       [ $code, [
+               'Content-Type' => 'text/html; charset=UTF-8',
+               'Content-Length' => $len
+       ], \@x ];
 }
 
 1;