]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/IMAPD.pm
imap+nntp: share COMPRESS implementation
[public-inbox.git] / lib / PublicInbox / IMAPD.pm
index a10fabffb1fbfa0cce2007635d93badd07b91c4c..b24097a2453c10783aa334484ba3920d329d85b5 100644 (file)
-# Copyright (C) 2020 all contributors <meta@public-inbox.org>
+# Copyright (C) all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 
 # represents an IMAPD (currently a singleton),
 # see script/public-inbox-imapd for how it is used
 package PublicInbox::IMAPD;
 use strict;
-use parent qw(PublicInbox::NNTPD);
+use v5.10.1;
+use PublicInbox::Config;
+use PublicInbox::ConfigIter;
 use PublicInbox::InboxIdle;
 use PublicInbox::IMAP;
-# *UID_BLOCK = \&PublicInbox::IMAP::UID_BLOCK;
+use PublicInbox::DummyInbox;
+my $dummy = bless { uidvalidity => 0 }, 'PublicInbox::DummyInbox';
 
 sub new {
        my ($class) = @_;
        bless {
-               groups => {},
+               mailboxes => {},
                err => \*STDERR,
                out => \*STDOUT,
-               grouplist => [],
                # accept_tls => { SSL_server => 1, ..., SSL_reuse_ctx => ... }
-               # pi_config => PublicInbox::Config
+               # pi_cfg => PublicInbox::Config
                # idler => PublicInbox::InboxIdle
        }, $class;
 }
 
-sub refresh_inboxlist ($) {
-       my ($self) = @_;
-       my @names = map { $_->{newsgroup} } @{delete $self->{grouplist}};
-       my %ns; # "\Noselect \HasChildren"
-
-       if (my @uc = grep(/[A-Z]/, @names)) {
-               warn "Uppercase not allowed for IMAP newsgroup(s):\n",
-                       map { "\t$_\n" } @uc;
-               my %uc = map { $_ => 1 } @uc;
-               @names = grep { !$uc{$_} } @names;
-       }
-       for (@names) {
-               my $up = $_;
-               while ($up =~ s/\.[^\.]+\z//) {
-                       $ns{$up} = '\\Noselect \\HasChildren';
-               }
-       }
-       @names = map {;
-               my $at = delete($ns{$_}) ? '\\HasChildren' : '\\HasNoChildren';
-               qq[* LIST ($at) "." $_\r\n]
-       } @names;
-       push(@names, map { qq[* LIST ($ns{$_}) "." $_\r\n] } keys %ns);
-       @names = sort {
-               my ($xa) = ($a =~ / (\S+)\r\n/g);
-               my ($xb) = ($b =~ / (\S+)\r\n/g);
-               length($xa) <=> length($xb);
-       } @names;
-       $self->{inboxlist} = \@names;
-}
-
-sub imapd_refresh_ibx { # pi_config->each_inbox cb
+sub imapd_refresh_ibx { # pi_cfg->each_inbox cb
        my ($ibx, $imapd) = @_;
        my $ngname = $ibx->{newsgroup} or return;
-       if (ref $ngname) {
-               warn 'multiple newsgroups not supported: '.
-                       join(', ', @$ngname). "\n";
-       } elsif ($ngname =~ m![^a-z0-9/_\.\-\~\@\+\=:]! ||
-                $ngname =~ /\.[0-9]+-[0-9]+\z/) {
-               warn "mailbox name invalid: `$ngname'\n";
+
+       # We require lower-case since IMAP mailbox names are
+       # case-insensitive (but -nntpd matches INN in being
+       # case-sensitive
+       if ($ngname =~ m![^a-z0-9/_\.\-\~\@\+\=:]! ||
+                       # don't confuse with 50K slices
+                       $ngname =~ /\.[0-9]+\z/) {
+               warn "mailbox name invalid: newsgroup=`$ngname'\n";
+               return;
        }
+       $ibx->over or return;
+       $ibx->{over} = undef;
 
-       my $mm = $ibx->mm or return;
-       $ibx->{mm} = undef;
-       defined($ibx->{uidvalidity} = $mm->created_at) or return;
-       $imapd->{tmp_groups}->{$ngname} = $ibx;
+       # RFC 3501 2.3.1.1 -  "A good UIDVALIDITY value to use in
+       # this case is a 32-bit representation of the creation
+       # date/time of the mailbox"
+       eval { $ibx->uidvalidity };
+       my $mm = delete($ibx->{mm}) or return;
+       defined($ibx->{uidvalidity}) or return;
+       PublicInbox::IMAP::ensure_slices_exist($imapd, $ibx, $mm->max);
 
        # preload to avoid fragmentation:
        $ibx->description;
        $ibx->base_url;
-       # my $max = $mm->max // 0;
-       # my $uid_min = UID_BLOCK * int($max/UID_BLOCK) + 1;
+
+       # ensure dummies are selectable
+       my $dummies = $imapd->{dummies};
+       do {
+               $dummies->{$ngname} = $dummy;
+       } while ($ngname =~ s/\.[^\.]+\z//);
 }
 
 sub imapd_refresh_finalize {
-       my ($imapd, $pi_config) = @_;
-       $imapd->{groups} = delete $imapd->{tmp_groups};
-       $imapd->{grouplist} = [ values %{$imapd->{groups}} ];
-       refresh_inboxlist($imapd);
-       $imapd->{pi_config} = $pi_config;
+       my ($imapd, $pi_cfg) = @_;
+       my $mailboxes;
+       if (my $next = delete $imapd->{imapd_next}) {
+               $imapd->{mailboxes} = delete $next->{mailboxes};
+               $mailboxes = delete $next->{dummies};
+       } else {
+               $mailboxes = delete $imapd->{dummies};
+       }
+       %$mailboxes = (%$mailboxes, %{$imapd->{mailboxes}});
+       $imapd->{mailboxes} = $mailboxes;
+       $imapd->{mailboxlist} = [
+               map { $_->[2] }
+               sort { $a->[0] cmp $b->[0] || $a->[1] <=> $b->[1] }
+               map {
+                       my $u = $_; # capitalize "INBOX" for user-familiarity
+                       $u =~ s/\Ainbox(\.|\z)/INBOX$1/i;
+                       if ($mailboxes->{$_} == $dummy) {
+                               [ $u, -1,
+                                 qq[* LIST (\\HasChildren) "." $u\r\n]]
+                       } else {
+                               $u =~ /\A(.+)\.([0-9]+)\z/ or
+                                       die "BUG: `$u' has no slice digit(s)";
+                               [ $1, $2 + 0,
+                                 qq[* LIST (\\HasNoChildren) "." $u\r\n] ]
+                       }
+               } keys %$mailboxes
+       ];
+       $imapd->{pi_cfg} = $pi_cfg;
        if (my $idler = $imapd->{idler}) {
-               $idler->refresh($pi_config);
+               $idler->refresh($pi_cfg);
        }
 }
 
-sub imapd_refresh_step { # pi_config->iterate_start cb
-       my ($pi_config, $section, $imapd) = @_;
+sub imapd_refresh_step { # PublicInbox::ConfigIter cb
+       my ($pi_cfg, $section, $imapd) = @_;
        if (defined($section)) {
                return if $section !~ m!\Apublicinbox\.([^/]+)\z!;
-               my $ibx = $pi_config->lookup_name($1) or return;
-               imapd_refresh_ibx($ibx, $imapd);
-       } else { # "EOF"
-               imapd_refresh_finalize($imapd, $pi_config);
+               my $ibx = $pi_cfg->lookup_name($1) or return;
+               imapd_refresh_ibx($ibx, $imapd->{imapd_next});
+       } else { # undef == "EOF"
+               imapd_refresh_finalize($imapd, $pi_cfg);
        }
 }
 
 sub refresh_groups {
        my ($self, $sig) = @_;
-       my $pi_config = PublicInbox::Config->new;
-       $self->{tmp_groups} = {};
-       if (0 && $sig) { # SIGHUP
-               $pi_config->iterate_start(\&imapd_refresh_step, $self);
-               PublicInbox::DS::requeue($pi_config); # call event_step
-       } else { # initial start
-               $pi_config->each_inbox(\&imapd_refresh_ibx, $self);
-               imapd_refresh_finalize($self, $pi_config);
+       my $pi_cfg = PublicInbox::Config->new;
+       if ($sig) { # SIGHUP is handled through the event loop
+               $self->{imapd_next} = { dummies => {}, mailboxes => {} };
+               my $iter = PublicInbox::ConfigIter->new($pi_cfg,
+                                               \&imapd_refresh_step, $self);
+               $iter->event_step;
+       } else { # initial start is synchronous
+               $self->{dummies} = {};
+               $pi_cfg->each_inbox(\&imapd_refresh_ibx, $self);
+               imapd_refresh_finalize($self, $pi_cfg);
        }
 }
 
 sub idler_start {
-       $_[0]->{idler} //= PublicInbox::InboxIdle->new($_[0]->{pi_config});
+       $_[0]->{idler} //= PublicInbox::InboxIdle->new($_[0]->{pi_cfg});
 }
 
 1;