]> Sergey Matveev's repositories - public-inbox.git/blob - lib/PublicInbox/MboxGz.pm
ds: inline set_cloexec
[public-inbox.git] / lib / PublicInbox / MboxGz.pm
1 # Copyright (C) 2015-2021 all contributors <meta@public-inbox.org>
2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
3 package PublicInbox::MboxGz;
4 use strict;
5 use parent 'PublicInbox::GzipFilter';
6 use PublicInbox::Eml;
7 use PublicInbox::Hval qw/to_filename/;
8 use PublicInbox::Mbox;
9 *msg_hdr = \&PublicInbox::Mbox::msg_hdr;
10 *msg_body = \&PublicInbox::Mbox::msg_body;
11
12 sub async_next ($) {
13         my ($http) = @_; # PublicInbox::HTTP
14         my $ctx = $http->{forward} or return;
15         eval {
16                 $ctx->{smsg} = $ctx->{cb}->($ctx) or return $ctx->close;
17                 $ctx->smsg_blob($ctx->{smsg});
18         };
19         warn "E: $@" if $@;
20 }
21
22 sub mbox_gz {
23         my ($self, $cb, $fn) = @_;
24         $self->{cb} = $cb;
25         $self->{base_url} = $self->{ibx}->base_url($self->{env});
26         $self->{gz} = PublicInbox::GzipFilter::gzip_or_die();
27         $fn = to_filename($fn // '') // 'no-subject';
28         # http://www.iana.org/assignments/media-types/application/gzip
29         bless $self, __PACKAGE__;
30         my $res_hdr = [ 'Content-Type' => 'application/gzip',
31                 'Content-Disposition' => "inline; filename=$fn.mbox.gz" ];
32         $self->psgi_response(200, $res_hdr);
33 }
34
35 # called by Plack::Util::foreach or similar (generic PSGI)
36 sub getline {
37         my ($self) = @_;
38         my $cb = $self->{cb} or return;
39         while (my $smsg = $cb->($self)) {
40                 my $eml = $self->{ibx}->smsg_eml($smsg) or next;
41                 $self->zmore(msg_hdr($self, $eml));
42                 return $self->translate(msg_body($eml));
43         }
44         # signal that we're done and can return undef next call:
45         delete $self->{cb};
46         $self->zflush;
47 }
48
49 no warnings 'once';
50 *async_eml = \&PublicInbox::Mbox::async_eml;
51 1;