]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/IMAP.pm
imap: support sequence number FETCH
[public-inbox.git] / lib / PublicInbox / IMAP.pm
index 4292c564f01423aa0bee1f74a620974c0963a570..2aa7ab346c1fb1462cdc3391906d513047e84796 100644 (file)
@@ -348,11 +348,34 @@ sub fetch_body ($;$) {
        join('', @hold);
 }
 
+sub dummy_message ($$) {
+       my ($seqno, $ibx) = @_;
+       my $ret = <<EOF;
+From: nobody\@localhost\r
+To: nobody\@localhost\r
+Date: Thu, 01 Jan 1970 00:00:00 +0000\r
+Message-ID: <dummy-$seqno\@$ibx->{newsgroup}>\r
+Subject: dummy message #$seqno\r
+\r
+You're seeing this message because your IMAP client didn't use UIDs.\r
+The message which used to use this sequence number was likely spam\r
+and removed by the administrator.\r
+EOF
+       \$ret;
+}
+
 sub uid_fetch_cb { # called by git->cat_async
        my ($bref, $oid, $type, $size, $fetch_m_arg) = @_;
        my ($self, undef, $ibx, undef, undef, $msgs, $want) = @$fetch_m_arg;
        my $smsg = shift @$msgs or die 'BUG: no smsg';
-       $smsg->{blob} eq $oid or die "BUG: $smsg->{blob} != $oid";
+       if (!defined($oid)) {
+               # it's possible to have TOCTOU if an admin runs
+               # public-inbox-(edit|purge), just move onto the next message
+               return unless defined $want->{-seqno};
+               $bref = dummy_message($smsg->{num}, $ibx);
+       } else {
+               $smsg->{blob} eq $oid or die "BUG: $smsg->{blob} != $oid";
+       }
        $$bref =~ s/(?<!\r)\n/\r\n/sg; # make strict clients happy
 
        # fixup old bug from import (pre-a0c07cba0e5d8b6a)
@@ -390,7 +413,7 @@ sub uid_fetch_cb { # called by git->cat_async
                $self->msg_more(' BODYSTRUCTURE '.fetch_body($eml, 1));
        $want->{BODY} and
                $self->msg_more(' BODY '.fetch_body($eml));
-       if (my $partial = delete $want->{-partial}) {
+       if (my $partial = $want->{-partial}) {
                partial_emit($self, $partial, $eml);
        }
        $self->msg_more(")\r\n");
@@ -569,8 +592,8 @@ sub partial_prepare ($$$) {
 
 sub partial_emit ($$$) {
        my ($self, $partial, $eml) = @_;
-       for my $k (sort keys %$partial) {
-               my ($cb, @args) = @{$partial->{$k}};
+       for (@$partial) {
+               my ($k, $cb, @args) = @$_;
                my ($offset, $len) = splice(@args, -2);
                # $cb is partial_body|partial_hdr_get|partial_hdr_not
                my $str = $cb->($eml, @args) // '';
@@ -591,23 +614,30 @@ sub partial_emit ($$$) {
        }
 }
 
-sub cmd_uid_fetch ($$$;@) {
-       my ($self, $tag, $range, @want) = @_;
+sub fetch_common ($$$$) {
+       my ($self, $tag, $range, $want) = @_;
        my $ibx = $self->{ibx} or return "$tag BAD No mailbox selected\r\n";
-       if ($want[0] =~ s/\A\(//s) {
-               $want[-1] =~ s/\)\z//s or return "$tag BAD no rparen\r\n";
+       if ($want->[0] =~ s/\A\(//s) {
+               $want->[-1] =~ s/\)\z//s or return "$tag BAD no rparen\r\n";
        }
        my (%partial, %want);
-       while (defined(my $att = shift @want)) {
+       while (defined(my $att = shift @$want)) {
                $att = uc($att);
                my $x = $FETCH_ATT{$att};
                if ($x) {
                        %want = (%want, %$x);
-               } elsif (!partial_prepare(\%partial, \@want, $att)) {
+               } elsif (!partial_prepare(\%partial, $want, $att)) {
                        return "$tag BAD param: $att\r\n";
                }
        }
-       $want{-partial} = \%partial if scalar keys %partial;
+
+       # stabilize partial order for consistency and ease-of-debugging:
+       if (scalar keys %partial) {
+               $want{-partial} = [ map {
+                       [ $_, @{$partial{$_}} ]
+               } sort keys %partial ];
+       }
+
        my ($beg, $end);
        my $msgs = [];
        if ($range =~ /\A([0-9]+):([0-9]+)\z/s) {
@@ -622,8 +652,54 @@ sub cmd_uid_fetch ($$$;@) {
        } else {
                return "$tag BAD fetch range\r\n";
        }
-       long_response($self, \&uid_fetch_m, $tag, $ibx,
-                               \$beg, $end, $msgs, \%want);
+       [ $tag, $ibx, \$beg, $end, $msgs, \%want ];
+}
+
+sub cmd_uid_fetch ($$$;@) {
+       my ($self, $tag, $range, @want) = @_;
+       my $args = fetch_common($self, $tag, $range, \@want);
+       ref($args) eq 'ARRAY' ?
+               long_response($self, \&uid_fetch_m, @$args) :
+               $args; # error
+}
+
+sub seq_fetch_m { # long_response
+       my ($self, $tag, $ibx, $beg, $end, $msgs, $want) = @_;
+       if (!@$msgs) { # refill
+               @$msgs = @{$ibx->over->query_xover($$beg, $end)};
+               if (!@$msgs) {
+                       $self->write(\"$tag OK Fetch done\r\n");
+                       return;
+               }
+               $$beg = $msgs->[-1]->{num} + 1;
+       }
+       my $seq = $want->{-seqno}++;
+       my $cur_num = $msgs->[0]->{num};
+       if ($cur_num == $seq) { # as expected
+               my $git = $ibx->git;
+               $git->cat_async_begin; # TODO: actually make async
+               $git->cat_async($msgs->[0]->{blob}, \&uid_fetch_cb, \@_);
+               $git->cat_async_wait;
+       } elsif ($cur_num > $seq) {
+               # send dummy messages until $seq catches up to $cur_num
+               my $smsg = bless { num => $seq, ts => 0 }, 'PublicInbox::Smsg';
+               unshift @$msgs, $smsg;
+               my $bref = dummy_message($seq, $ibx);
+               uid_fetch_cb($bref, undef, undef, undef, \@_);
+       } else { # should not happen
+               die "BUG: cur_num=$cur_num < seq=$seq";
+       }
+       1; # more messages on the way
+}
+
+sub cmd_fetch ($$$;@) {
+       my ($self, $tag, $range, @want) = @_;
+       my $args = fetch_common($self, $tag, $range, \@want);
+       ref($args) eq 'ARRAY' ? do {
+               my $want = $args->[-1];
+               $want->{-seqno} = ${$args->[2]}; # $$beg
+               long_response($self, \&seq_fetch_m, @$args)
+       } : $args; # error
 }
 
 sub uid_search_all { # long_response