imap+nntp: share COMPRESS implementation

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

diff --git a/lib/PublicInbox/HTTPD.pm b/lib/PublicInbox/HTTPD.pm
index 02f424c67138edca6af65835b1f2feb328d3a72d..715e453841be3fd4fe0b2a1d389d299a8a24ab15 100644 (file)
--- a/lib/PublicInbox/HTTPD.pm
+++ b/lib/PublicInbox/HTTPD.pm
@@ -1,14 +1,15 @@
-# Copyright (C) 2016-2021 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>
 
 # wraps a listen socket for HTTP and links it to the PSGI app in
 # public-inbox-httpd
 package PublicInbox::HTTPD;
+use v5.10.1;
 use strict;
-use warnings;
-use Plack::Util;
+use Plack::Util ();
+use Plack::Builder;
+use PublicInbox::HTTP;
 use PublicInbox::HTTPD::Async;
-use PublicInbox::Daemon;
 
 sub pi_httpd_async { PublicInbox::HTTPD::Async->new(@_) }
 
@@ -41,10 +42,41 @@ sub new {
                # detect when to use async paths for slow blobs
                'pi-httpd.async' => \&pi_httpd_async
        );
-       bless {
-               app => $app,
-               env => \%env
-       }, $class;
+       bless { app => $app, env => \%env }, $class;
+}
+
+my %httpds; # per-listen-FD mapping for HTTPD->{env}->{SERVER_<NAME|PORT>}
+my $default_app; # ugh...
+
+sub refresh {
+       if (@main::ARGV) {
+               eval { $default_app = Plack::Util::load_psgi(@ARGV) };
+               if ($@) {
+                       die $@,
+"$0 runs in /, command-line paths must be absolute\n";
+               }
+       } else {
+               require PublicInbox::WWW;
+               my $www = PublicInbox::WWW->new;
+               $www->preload;
+               $default_app = builder {
+                       eval { enable 'ReverseProxy' };
+                       $@ and warn <<EOM;
+Plack::Middleware::ReverseProxy missing,
+URL generation for redirects may be wrong if behind a reverse proxy
+EOM
+                       enable 'Head';
+                       sub { $www->call(@_) };
+               };
+       }
+       %httpds = (); # invalidate cache
+}
+
+sub post_accept { # Listener->{post_accept}
+       my ($client, $addr, $srv) = @_; # $_[3] - tls_wrap (unused)
+       my $httpd = $httpds{fileno($srv)} //=
+                               __PACKAGE__->new($srv, $default_app, $client);
+       PublicInbox::HTTP->new($client, $addr, $httpd),
 }
 
 1;