lei export-kw: new command to export keywords to Maildirs

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

diff --git a/lib/PublicInbox/LeiSearch.pm b/lib/PublicInbox/LeiSearch.pm
index cd28a700647bf62a37655108d9fb1467f776fbcc..9297d060eef8b5e02a27a0a7189b2ac845eb4fe8 100644 (file)
--- a/lib/PublicInbox/LeiSearch.pm
+++ b/lib/PublicInbox/LeiSearch.pm
@@ -27,6 +27,20 @@ sub msg_keywords {
        wantarray ? sort(keys(%$kw)) : $kw;
 }
 
+# returns undef if blob is unknown
+sub oid_keywords {
+       my ($self, $oidhex) = @_;
+       my @num = $self->over->blob_exists($oidhex) or return;
+       my $xdb = $self->xdb; # set {nshard};
+       my %kw;
+       for my $num (@num) { # there should only be one...
+               my $doc = $xdb->get_document(num2docid($self, $num));
+               my $x = xap_terms('K', $doc);
+               %kw = (%kw, %$x);
+       }
+       \%kw;
+}
+
 # lookup keywords+labels for external messages
 sub xsmsg_vmd {
        my ($self, $smsg, $want_label) = @_;
@@ -63,7 +77,9 @@ sub content_key ($) {
 }
 
 sub _cmp_1st { # git->cat_async callback
-       my ($bref, $oid, $type, $size, $cmp) = @_; # cmp: [chash, xoids, smsg]
+       my ($bref, $oid, $type, $size, $cmp) = @_;
+       # cmp: [chash, xoids, smsg, lms]
+       $bref //= $cmp->[3] ? $cmp->[3]->local_blob($oid, 1) : undef;
        if ($bref && content_hash(PublicInbox::Eml->new($bref)) eq $cmp->[0]) {
                $cmp->[1]->{$oid} = $cmp->[2]->{num};
        }
@@ -78,6 +94,8 @@ sub xoids_for {
        my @overs = ($self->over // $self->overs_all);
        my $git = $self->git;
        my $xoids = {};
+       # no lms when used via {ale}:
+       my $lms = $self->{-lms_ro} //= lms($self) if defined($self->{topdir});
        for my $mid (@$mids) {
                for my $o (@overs) {
                        my ($id, $prev);
@@ -85,7 +103,7 @@ sub xoids_for {
                                next if $cur->{bytes} == 0 ||
                                        $xoids->{$cur->{blob}};
                                $git->cat_async($cur->{blob}, \&_cmp_1st,
-                                               [ $chash, $xoids, $cur ]);
+                                               [$chash, $xoids, $cur, $lms]);
                                if ($min && scalar(keys %$xoids) >= $min) {
                                        $git->cat_async_wait;
                                        return $xoids;
@@ -144,4 +162,14 @@ sub lms {
        -f $f ? PublicInbox::LeiMailSync->new($f) : undef;
 }
 
+# allow SolverGit->resolve_patch to work with "lei index"
+sub smsg_eml {
+       my ($self, $smsg) = @_;
+       PublicInbox::Inbox::smsg_eml($self, $smsg) // do {
+               my $lms = lms($self);
+               my $bref = $lms ? $lms->local_blob($smsg->{blob}, 1) : undef;
+               $bref ? PublicInbox::Eml->new($bref) : undef;
+       };
+}
+
 1;