]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/LeiOverview.pm
lei: support remote externals
[public-inbox.git] / lib / PublicInbox / LeiOverview.pm
index c0b423f63e6b5774a0a1fead865dff033610cc6f..49538a609c577a4cab5b7a2e3327040d6f131d41 100644 (file)
@@ -38,6 +38,19 @@ sub ovv_out_lk_cancel ($) {
                unlink(delete($self->{lock_path}));
 }
 
+sub detect_fmt ($$) {
+       my ($lei, $dst) = @_;
+       if ($dst =~ m!\A([:/]+://)!) {
+               $lei->fail("$1 support not implemented, yet\n");
+       } elsif (!-e $dst || -d _) {
+               'maildir'; # the default TODO: MH?
+       } elsif (-f _ || -p _) {
+               $lei->fail("unable to determine mbox family of $dst\n");
+       } else {
+               $lei->fail("unable to determine format of $dst\n");
+       }
+}
+
 sub new {
        my ($class, $lei) = @_;
        my $opt = $lei->{opt};
@@ -54,7 +67,7 @@ sub new {
 
        }
        $fmt //= 'json' if $dst eq '/dev/stdout';
-       $fmt //= 'maildir';
+       $fmt //= detect_fmt($lei, $dst) or return;
 
        if (index($dst, '://') < 0) { # not a URL, so assume path
                 $dst = File::Spec->canonpath($dst);
@@ -82,7 +95,8 @@ sub new {
        if (!$json) {
                # default to the cheapest sort since MUA usually resorts
                $lei->{opt}->{'sort'} //= 'docid' if $dst ne '/dev/stdout';
-               $lei->{l2m} = PublicInbox::LeiToMail->new($lei);
+               $lei->{l2m} = eval { PublicInbox::LeiToMail->new($lei) };
+               return $lei->fail($@) if $@;
        }
        $lei->{dedupe} //= PublicInbox::LeiDedupe->new($lei);
        $self;
@@ -99,12 +113,13 @@ sub ovv_begin {
 # called once by parent (via PublicInbox::EOFpipe)
 sub ovv_end {
        my ($self, $lei) = @_;
+       my $out = $lei->{1} or return;
        if ($self->{fmt} eq 'json') {
                # JSON doesn't allow trailing commas, and preventing
                # trailing commas is a PITA when parallelizing outputs
-               print { $lei->{1} } "null]\n";
+               print $out "null]\n";
        } elsif ($self->{fmt} eq 'concatjson') {
-               print { $lei->{1} } "\n";
+               print $out "\n";
        }
 }
 
@@ -120,11 +135,9 @@ sub _unbless_smsg {
        delete @$smsg{qw(lines bytes num tid)};
        $smsg->{rt} = _iso8601(delete $smsg->{ts}); # JMAP receivedAt
        $smsg->{dt} = _iso8601(delete $smsg->{ds}); # JMAP UTCDate
-       $smsg->{relevance} = get_pct($mitem) if $mitem;
-
+       $smsg->{pct} = get_pct($mitem) if $mitem;
        if (my $r = delete $smsg->{references}) {
-               $smsg->{refs} = [
-                               map { "<$_>" } ($r =~ m/$MID_EXTRACT/go) ];
+               $smsg->{refs} = [ map { "<$_>" } ($r =~ m/$MID_EXTRACT/go) ];
        }
        if (my $m = delete($smsg->{mid})) {
                $smsg->{'m'} = "<$m>";
@@ -140,6 +153,16 @@ sub _unbless_smsg {
 
 sub ovv_atexit_child {
        my ($self, $lei) = @_;
+       if (my $l2m = delete $lei->{l2m}) {
+               # gracefully stop lei2mail processes after all
+               # ->write_mail work is complete
+               delete $l2m->{-wq_s1};
+               if (my $rd = delete $l2m->{each_smsg_done}) {
+                       read($rd, my $buf, 1); # wait for EOF
+               }
+       }
+       # order matters, git->{-tmp}->DESTROY must not fire until
+       # {each_smsg_done} hits EOF above
        if (my $git = delete $self->{git}) {
                $git->async_wait_all;
        }
@@ -178,8 +201,6 @@ sub _json_pretty {
 
 sub ovv_each_smsg_cb { # runs in wq worker usually
        my ($self, $lei, $ibxish) = @_;
-       $lei->{ovv_buf} = \(my $buf = '');
-       delete(@$self{qw(lock_path tmp_lk_id)}) unless $lei->{-parallel};
        my $json;
        $lei->{1}->autoflush(1);
        if (my $pkg = $self->{json}) {
@@ -187,18 +208,47 @@ sub ovv_each_smsg_cb { # runs in wq worker usually
                $json->utf8->canonical;
                $json->ascii(1) if $lei->{opt}->{ascii};
        }
-       if (my $l2m = $lei->{l2m}) {
+       my $l2m = $lei->{l2m};
+       if ($l2m && $ibxish->can('scheme')) { # remote https?:// mboxrd
+               delete $l2m->{-wq_s1};
+               my $g2m = $l2m->can('git_to_mail');
+               my $wcb = $l2m->write_cb($lei);
+               sub {
+                       my ($smsg, undef, $eml) = @_; # no mitem in $_[1]
+                       $wcb->(undef, $smsg, $eml);
+               };
+       } elsif ($l2m && $l2m->{-wq_s1}) {
+               my ($lei_ipc, @io) = $lei->atfork_parent_wq($l2m);
+               # n.b. $io[0] = qry_status_wr, $io[1] = mbox|stdout,
+               # $io[4] becomes a notification pipe that triggers EOF
+               # in this wq worker when all outstanding ->write_mail
+               # calls are complete
+               die "BUG: \$io[4] $io[4] unexpected" if $io[4];
+               pipe($l2m->{each_smsg_done}, $io[4]) or die "pipe: $!";
+               fcntl($io[4], 1031, 4096) if $^O eq 'linux';
+               delete @$lei_ipc{qw(l2m opt mset_opt cmd)};
+               my $git = $ibxish->git; # (LeiXSearch|Inbox|ExtSearch)->git
+               $self->{git} = $git;
+               my $git_dir = $git->{git_dir};
+               sub {
+                       my ($smsg, $mitem) = @_;
+                       $smsg->{pct} = get_pct($mitem) if $mitem;
+                       $l2m->wq_do('write_mail', \@io, $git_dir, $smsg,
+                                       $lei_ipc);
+               }
+       } elsif ($l2m) {
                my $wcb = $l2m->write_cb($lei);
                my $git = $ibxish->git; # (LeiXSearch|Inbox|ExtSearch)->git
                $self->{git} = $git; # for ovv_atexit_child
                my $g2m = $l2m->can('git_to_mail');
                sub {
                        my ($smsg, $mitem) = @_;
-                       my $kw = []; # TODO get from mitem
-                       $git->cat_async($smsg->{blob}, $g2m, [ $wcb, $kw ]);
+                       $smsg->{pct} = get_pct($mitem) if $mitem;
+                       $git->cat_async($smsg->{blob}, $g2m, [ $wcb, $smsg ]);
                };
        } elsif ($self->{fmt} =~ /\A(concat)?json\z/ && $lei->{opt}->{pretty}) {
                my $EOR = ($1//'') eq 'concat' ? "\n}" : "\n},";
+               $lei->{ovv_buf} = \(my $buf = '');
                sub { # DIY prettiness :P
                        my ($smsg, $mitem) = @_;
                        $smsg = _unbless_smsg($smsg, $mitem);
@@ -221,9 +271,9 @@ sub ovv_each_smsg_cb { # runs in wq worker usually
                }
        } elsif ($json) {
                my $ORS = $self->{fmt} eq 'json' ? ",\n" : "\n"; # JSONL
+               $lei->{ovv_buf} = \(my $buf = '');
                sub {
                        my ($smsg, $mitem) = @_;
-                       delete @$smsg{qw(tid num)};
                        $buf .= $json->encode(_unbless_smsg(@_)) . $ORS;
                        if (length($buf) > 65536) {
                                my $lk = $self->lock_for_scope;