]> Sergey Matveev's repositories - public-inbox.git/commitdiff
lei <q|up>: writes to Maildirs and IMAP use mail-sync
authorEric Wong <e@80x24.org>
Sun, 2 May 2021 06:05:42 +0000 (06:05 +0000)
committerEric Wong <e@80x24.org>
Mon, 3 May 2021 18:45:27 +0000 (18:45 +0000)
This will allow keyword updates from other folders to propagate
to folders where search results may be duplicated.

lib/PublicInbox/LEI.pm
lib/PublicInbox/LeiQuery.pm
lib/PublicInbox/LeiStore.pm
lib/PublicInbox/LeiToMail.pm

index d5c6bd525621be299bfe1d362b65f1f9426a0e3e..599cfab25346042685f40f7c0cd00144df8a40cd 100644 (file)
@@ -145,7 +145,7 @@ our %CMD = ( # sorted in order of importance/use:
        qw(save output|mfolder|o=s format|f=s dedupe|d=s threads|t+
        sort|s=s reverse|r offset=i pretty jobs|j=s globoff|g augment|a
        import-before! lock=s@ rsyncable alert=s@ mua=s verbose|v+
-       color!), @c_opt, opt_dash('limit|n=i', '[0-9]+') ],
+       color! mail-sync!), @c_opt, opt_dash('limit|n=i', '[0-9]+') ],
 
 'up' => [ 'OUTPUT|--all', 'update saved search',
        qw(jobs|j=s lock=s@ alert=s@ mua=s verbose|v+ all:s), @c_opt ],
index efe328ccc3945a0af99c721af9549ed15eb67e11..1999a5344885f22799b666459579d04d88aa9831 100644 (file)
@@ -29,10 +29,9 @@ sub _start_query { # used by "lei q" and "lei up"
                return $self->fail("`$mj' writer jobs must be >= 1");
        }
        my $l2m = $self->{l2m};
-       if ($l2m && ($opt->{'import-remote'} //= 1) |
-                               # we use \1 (a ref) to distinguish between
-                               # user-supplied and default value
-                               (($opt->{'import-before'} //= \1) ? 1 : 0)) {
+       # we use \1 (a ref) to distinguish between default vs. user-supplied
+       if ($l2m && grep { $opt->{$_} //= \1 } (qw(mail-sync import-remote
+                                                       import-before))) {
                $self->_lei_store(1)->write_prepare($self);
        }
        $l2m and $l2m->{-wq_nr_workers} = $mj // do {
index 8af740fdf6e9b3e30ef251764c941354ef5a3785..29362b2e12ef9e1a3ede8509afbc10a592a48fcd 100644 (file)
@@ -193,15 +193,15 @@ sub remove_eml_vmd {
        \@docids;
 }
 
-sub set_sync_info ($$$) {
-       my ($self, $oidhex, $sync_info) = @_;
+sub set_sync_info {
+       my ($self, $oidhex, $folder, $id) = @_;
        ($self->{lms} //= do {
                require PublicInbox::LeiMailSync;
                my $f = "$self->{priv_eidx}->{topdir}/mail_sync.sqlite3";
                my $lms = PublicInbox::LeiMailSync->new($f);
                $lms->lms_begin;
                $lms;
-       })->set_src($oidhex, @$sync_info);
+       })->set_src($oidhex, $folder, $id);
 }
 
 sub add_eml {
@@ -212,7 +212,7 @@ sub add_eml {
        my $smsg = bless { -oidx => $oidx }, 'PublicInbox::Smsg';
        my $im_mark = $im->add($eml, undef, $smsg);
        if ($vmd && $vmd->{sync_info}) {
-               set_sync_info($self, $smsg->{blob}, $vmd->{sync_info});
+               set_sync_info($self, $smsg->{blob}, @{$vmd->{sync_info}});
        }
        $im_mark or return; # duplicate blob returns undef
 
index ab4de3780d042e3df0456416856ce84aab985b74..71acf952c046fc0c984b2205110723afba4df84f 100644 (file)
@@ -242,7 +242,7 @@ sub _buf2maildir {
        my $kw = $smsg->{kw} // [];
        my $sfx = join('', sort(map { $kw2char{$_} // () } @$kw));
        my $rand = ''; # chosen by die roll :P
-       my ($tmp, $fh, $final, $ok);
+       my ($tmp, $fh, $base, $ok);
        my $common = $smsg->{blob} // _rand;
        if (defined(my $pct = $smsg->{pct})) { $common .= "=$pct" }
        do {
@@ -257,11 +257,12 @@ sub _buf2maildir {
                $dst .= 'cur/';
                $rand = '';
                do {
-                       $final = $dst.$rand.$common.':2,'.$sfx;
-               } while (!($ok = link($tmp, $final)) && $!{EEXIST} &&
+                       $base = $rand.$common.':2,'.$sfx
+               } while (!($ok = link($tmp, $dst.$base)) && $!{EEXIST} &&
                        ($rand = _rand.','));
-               die "link($tmp, $final): $!" unless $ok;
+               die "link($tmp, $dst$base): $!" unless $ok;
                unlink($tmp) or warn "W: failed to unlink $tmp: $!\n";
+               \$base;
        } else {
                my $err = "Error writing $smsg->{blob} to $dst: $!\n";
                $_[0] = undef; # clobber dst
@@ -276,13 +277,16 @@ sub _maildir_write_cb ($$) {
        $dedupe->prepare_dedupe if $dedupe;
        my $dst = $lei->{ovv}->{dst};
        my $lse = $lei->{lse}; # may be undef
+       my $sto = $lei->{opt}->{'mail-sync'} ? $lei->{sto} : undef;
+       my $out = $sto ? 'maildir:'.$lei->rel2abs($dst) : undef;
        sub { # for git_to_mail
                my ($bref, $smsg, $eml) = @_;
                $dst // return $lei->fail; # dst may be undef-ed in last run
                return if $dedupe && $dedupe->is_dup($eml //
                                                PublicInbox::Eml->new($$bref));
                $lse->xsmsg_vmd($smsg) if $lse;
-               _buf2maildir($dst, $bref // \($eml->as_string), $smsg);
+               my $n = _buf2maildir($dst, $bref // \($eml->as_string), $smsg);
+               $sto->ipc_do('set_sync_info', $smsg->{blob}, $out, $n) if $sto;
                ++$lei->{-nr_write};
        }
 }
@@ -291,21 +295,27 @@ sub _imap_write_cb ($$) {
        my ($self, $lei) = @_;
        my $dedupe = $lei->{dedupe};
        $dedupe->prepare_dedupe if $dedupe;
-       my $imap_append = $lei->{net}->can('imap_append');
+       my $append = $lei->{net}->can('imap_append');
        my $mic = $lei->{net}->mic_get($self->{uri});
        my $folder = $self->{uri}->mailbox;
        my $lse = $lei->{lse}; # may be undef
+       my $sto = $lei->{opt}->{'mail-sync'} ? $lei->{sto} : undef;
+       my $out = $lei->{ovv}->{dst};
        sub { # for git_to_mail
                my ($bref, $smsg, $eml) = @_;
                $mic // return $lei->fail; # mic may be undef-ed in last run
                return if $dedupe && $dedupe->is_dup($eml //
                                                PublicInbox::Eml->new($$bref));
                $lse->xsmsg_vmd($smsg) if $lse;
-               eval { $imap_append->($mic, $folder, $bref, $smsg, $eml) };
+               my $uid = eval { $append->($mic, $folder, $bref, $smsg, $eml) };
                if (my $err = $@) {
                        undef $mic;
                        die $err;
                }
+               # imap_append returns UID if IMAP server has UIDPLUS extension
+               ($sto && $uid =~ /\A[0-9]+\z/) and
+                       $sto->ipc_do('set_sync_info',
+                                       $smsg->{blob}, $out, $uid + 0);
                ++$lei->{-nr_write};
        }
 }