nntp: NEWGROUPS uses long_response

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

diff --git a/lib/PublicInbox/NNTPD.pm b/lib/PublicInbox/NNTPD.pm
index 50d022be4ae2142ca3bcb39c378c70c8ab349f04..33bc5fda7a9a936e0fc95c28a14d91113d3bda31 100644 (file)
--- a/lib/PublicInbox/NNTPD.pm
+++ b/lib/PublicInbox/NNTPD.pm
@@ -1,4 +1,4 @@
-# Copyright (C) 2016 all contributors <meta@public-inbox.org>
+# Copyright (C) 2016-2020 all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 
 # represents an NNTPD (currently a singleton),
@@ -6,42 +6,67 @@
 package PublicInbox::NNTPD;
 use strict;
 use warnings;
-require PublicInbox::Config;
+use Sys::Hostname;
+use PublicInbox::Config;
+use PublicInbox::InboxIdle;
 
 sub new {
        my ($class) = @_;
+       my $pi_config = PublicInbox::Config->new;
+       my $name = $pi_config->{'publicinbox.nntpserver'};
+       if (!defined($name) or $name eq '') {
+               $name = hostname;
+       } elsif (ref($name) eq 'ARRAY') {
+               $name = $name->[0];
+       }
+
        bless {
                groups => {},
                err => \*STDERR,
                out => \*STDOUT,
-               grouplist => [],
+               pi_config => $pi_config,
+               servername => $name,
+               greet => \"201 $name ready - post via email\r\n",
+               # accept_tls => { SSL_server => 1, ..., SSL_reuse_ctx => ... }
+               # idler => PublicInbox::InboxIdle
        }, $class;
 }
 
-sub refresh_groups () {
-       my ($self) = @_;
-       my $pi_config = PublicInbox::Config->new;
-       my $new = {};
-       my @list;
-       foreach my $k (keys %$pi_config) {
-               $k =~ /\Apublicinbox\.([^\.]+)\.mainrepo\z/ or next;
-               my $name = $1;
-               my $git_dir = $pi_config->{$k};
-               my $ngname = $pi_config->{"publicinbox.$name.newsgroup"};
-               next unless defined $ngname;
-               next if ($ngname eq ''); # disabled
-               my $ng = $pi_config->lookup_newsgroup($ngname) or next;
+sub refresh_groups {
+       my ($self, $sig) = @_;
+       my $pi_config = $sig ? PublicInbox::Config->new : $self->{pi_config};
+       my $groups = $pi_config->{-by_newsgroup}; # filled during each_inbox
+       $pi_config->each_inbox(sub {
+               my ($ibx) = @_;
+               my $ngname = $ibx->{newsgroup} or return;
+               if (ref $ngname) {
+                       warn 'multiple newsgroups not supported: '.
+                               join(', ', @$ngname). "\n";
+               # Newsgroup name needs to be compatible with RFC 3977
+               # wildmat-exact and RFC 3501 (IMAP) ATOM-CHAR.
+               # Leave out a few chars likely to cause problems or conflicts:
+               # '|', '<', '>', ';', '#', '$', '&',
+               } elsif ($ngname =~ m![^A-Za-z0-9/_\.\-\~\@\+\=:]!) {
+                       warn "newsgroup name invalid: `$ngname'\n";
+                       delete $groups->{$ngname};
+               } elsif ($ibx->nntp_usable) {
+                       # Only valid if msgmap and search works
 
-               # Only valid if msgmap and search works
-               if ($ng->nntp_usable) {
-                       $new->{$ngname} = $ng;
-                       push @list, $ng;
+                       # preload to avoid fragmentation:
+                       $ibx->description;
+                       $ibx->base_url;
+               } else {
+                       delete $groups->{$ngname};
                }
-       }
-       @list = sort { $a->{newsgroup} cmp $b->{newsgroup} } @list;
-       $self->{grouplist} = \@list;
+       });
+       $self->{groupnames} = [ sort(keys %$groups) ];
+       $self->{pi_config} = $pi_config;
        # this will destroy old groups that got deleted
-       %{$self->{groups}} = %$new;
+       $self->{groups} = $groups;
+}
+
+sub idler_start {
+       $_[0]->{idler} //= PublicInbox::InboxIdle->new($_[0]->{pi_config});
 }
 
 1;