lei export-kw: support exporting keywords to IMAP

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

diff --git a/lib/PublicInbox/NetWriter.pm b/lib/PublicInbox/NetWriter.pm
index 49ac02a6eedd6fae23cf6d3a5fbb3c93d1096233..8ec7f85c876b568cf6102cc9dbdf705203bb2d03 100644 (file)
--- a/lib/PublicInbox/NetWriter.pm
+++ b/lib/PublicInbox/NetWriter.pm
@@ -26,20 +26,26 @@ sub imap_append {
                die "APPEND $folder: $@";
 }
 
+# updates $uri with UIDVALIDITY
 sub mic_for_folder {
        my ($self, $uri) = @_;
-       if (!ref($uri)) {
-               my $u = PublicInbox::URIimap->new($uri);
-               $_[1] = $uri = $u;
-       }
        my $mic = $self->mic_get($uri) or die "E: not connected: $@";
        $mic->select($uri->mailbox) or return;
+       my $uidval;
+       for ($mic->Results) {
+               /^\* OK \[UIDVALIDITY ([0-9]+)\].*/ or next;
+               $uidval = $1;
+               last;
+       }
+       $uidval //= $mic->uidvalidity($uri->mailbox) or
+               die "E: failed to get uidvalidity from <$uri>: $@";
+       $uri->uidvalidity($uidval);
        $mic;
 }
 
 sub imap_delete_all {
-       my ($self, $url) = @_;
-       my $mic = mic_for_folder($self, my $uri = $url) or return;
+       my ($self, $uri) = @_;
+       my $mic = mic_for_folder($self, $uri) or return;
        my $sec = $self->can('uri_section')->($uri);
        local $0 = $uri->mailbox." $sec";
        if ($mic->delete_message('1:*')) {
@@ -48,22 +54,20 @@ sub imap_delete_all {
 }
 
 sub imap_delete_1 {
-       my ($self, $url, $uid, $delete_mic) = @_;
-       $$delete_mic //= mic_for_folder($self, my $uri = $url) or return;
+       my ($self, $uri, $uid, $delete_mic) = @_;
+       $$delete_mic //= mic_for_folder($self, $uri) or return;
        $$delete_mic->delete_message($uid);
 }
 
-sub imap_set_kw {
-       my ($self, $url, $uid, $kw) = @_;
-       my $mic = mic_for_folder($self, my $uri = $url) or return;
-       $mic->set_flag(kw2flags($kw), $uid);
+sub imap_add_kw {
+       my ($self, $mic, $uid, $kw) = @_;
+       $mic->store($uid, '+FLAGS.SILENT', '('.kw2flags($kw).')');
        $mic; # caller must ->expunge
 }
 
-sub imap_unset_kw {
-       my ($self, $url, $uid, $kw) = @_;
-       my $mic = mic_for_folder($self, my $uri = $url) or return;
-       $mic->unset_flag(kw2flags($kw), $uid);
+sub imap_set_kw {
+       my ($self, $mic, $uid, $kw) = @_;
+       $mic->store($uid, 'FLAGS.SILENT', '('.kw2flags($kw).')');
        $mic; # caller must ->expunge
 }