wwwstream::oneshot => html_oneshot

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

diff --git a/lib/PublicInbox/WwwStream.pm b/lib/PublicInbox/WwwStream.pm
index fceef745d0c4ec2291448a2ff131a68880030530..915a71ba0bd82df950ee498f03c9a2c078a78a79 100644 (file)
--- a/lib/PublicInbox/WwwStream.pm
+++ b/lib/PublicInbox/WwwStream.pm
@@ -9,6 +9,9 @@
 package PublicInbox::WwwStream;
 use strict;
 use warnings;
+use base qw(Exporter);
+our @EXPORT_OK = qw(html_oneshot);
+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';
@@ -169,15 +172,19 @@ sub getline {
        delete $self->{cb} ? _html_end($self) : undef;
 }
 
-sub oneshot {
-       my ($ctx, $code, $strref) = @_;
+sub html_oneshot ($$;$) {
+       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 ? $$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;