]> Sergey Matveev's repositories - public-inbox.git/blobdiff - lib/PublicInbox/LeiMirror.pm
lei add-external --mirror: deduce paths for PSGI mount prefixes
[public-inbox.git] / lib / PublicInbox / LeiMirror.pm
index f8ca1ee55d7b4e585306356739dcae527a5944f9..fca11ccff037cae4d5b1f4b37e609c36733629fa 100644 (file)
@@ -12,15 +12,21 @@ use PublicInbox::Spawn qw(popen_rd spawn);
 sub do_finish_mirror { # dwaitpid callback
        my ($arg, $pid) = @_;
        my ($mrr, $lei) = @$arg;
-       if ($? == 0 && unlink("$mrr->{dst}/mirror.done")) {
+       my $f = "$mrr->{dst}/mirror.done";
+       if ($?) {
+               $lei->child_error($?);
+       } elsif (!unlink($f)) {
+               $lei->err("unlink($f): $!");
+       } else {
                $lei->add_external_finish($mrr->{dst});
+               $lei->qerr("# mirrored $mrr->{src} => $mrr->{dst}");
        }
        $lei->dclose;
 }
 
-sub mirror_done { # EOF callback for main daemon
+sub _lei_wq_eof { # EOF callback for main daemon
        my ($lei) = @_;
-       my $mrr = delete $lei->{mrr} or return;
+       my $mrr = delete $lei->{wq1} or return $lei->fail;
        $mrr->wq_wait_old(\&do_finish_mirror, $lei);
 }
 
@@ -32,7 +38,7 @@ sub try_scrape {
        my $curl = $self->{curl} //= PublicInbox::LeiCurl->new($lei) or return;
        my $cmd = $curl->for_uri($lei, $uri, '--compressed');
        my $opt = { 0 => $lei->{0}, 2 => $lei->{2} };
-       my $fh = popen_rd($cmd, $lei->{env}, $opt);
+       my $fh = popen_rd($cmd, undef, $opt);
        my $html = do { local $/; <$fh> } // die "read(curl $uri): $!";
        close($fh) or return $lei->child_error($?, "@$cmd failed");
 
@@ -104,7 +110,7 @@ sub _try_config {
        }
        return $lei->err("# @$cmd failed (non-fatal)") if $cerr;
        rename($f, $ce) or return $lei->err("link($f, $ce): $! (non-fatal)");
-       my $cfg = PublicInbox::Config::git_config_dump($f);
+       my $cfg = PublicInbox::Config->git_config_dump($f);
        my $ibx = $self->{ibx} = {};
        for my $sec (grep(/\Apublicinbox\./, @{$cfg->{-section_order}})) {
                for (qw(address newsgroup nntpmirror)) {
@@ -142,7 +148,7 @@ sub run_reap {
        my ($lei, $cmd, $opt) = @_;
        $lei->qerr("# @$cmd");
        $opt->{pgid} = 0;
-       my $pid = spawn($cmd, $lei->{env}, $opt);
+       my $pid = spawn($cmd, undef, $opt);
        my $reap = PublicInbox::OnDestroy->new($lei->can('sigint_reap'), $pid);
        my $err = waitpid($pid, 0) == $pid ? undef : "waitpid @$cmd: $!";
        @$reap = (); # cancel reap
@@ -194,6 +200,19 @@ failed to extract epoch number from $src
        index_cloned_inbox($self, 2);
 }
 
+# PSGI mount prefixes and manifest.js.gz prefixes don't always align...
+sub deduce_epochs ($$) {
+       my ($m, $path) = @_;
+       my ($v1_bare, @v2_epochs);
+       my $path_pfx = '';
+       do {
+               $v1_bare = $m->{$path};
+               @v2_epochs = grep(m!\A\Q$path\E/git/[0-9]+\.git\z!, keys %$m);
+       } while (!defined($v1_bare) && !@v2_epochs &&
+               $path =~ s!\A(/[^/]+)/!/! and $path_pfx .= $1);
+       ($path_pfx, $v1_bare, @v2_epochs);
+}
+
 sub try_manifest {
        my ($self) = @_;
        my $uri = URI->new($self->{src});
@@ -205,7 +224,7 @@ sub try_manifest {
        my $cmd = $curl->for_uri($lei, $uri);
        $lei->qerr("# @$cmd");
        my $opt = { 0 => $lei->{0}, 2 => $lei->{2} };
-       my ($fh, $pid) = popen_rd($cmd, $lei->{env}, $opt);
+       my ($fh, $pid) = popen_rd($cmd, undef, $opt);
        my $reap = PublicInbox::OnDestroy->new($lei->can('sigint_reap'), $pid);
        my $gz = do { local $/; <$fh> } // die "read(curl $uri): $!";
        close $fh;
@@ -223,8 +242,7 @@ sub try_manifest {
        die "$uri: error decoding `$js': $@" if $@;
        ref($m) eq 'HASH' or die "$uri unknown type: ".ref($m);
 
-       my $v1_bare = $m->{$path};
-       my @v2_epochs = grep(m!\A\Q$path\E/git/[0-9]+\.git\z!, keys %$m);
+       my ($path_pfx, $v1_bare, @v2_epochs) = deduce_epochs($m, $path);
        if (@v2_epochs) {
                # It may be possible to have v1 + v2 in parallel someday:
                $lei->err(<<EOM) if defined $v1_bare;
@@ -232,14 +250,16 @@ sub try_manifest {
 # @v2_epochs
 # ignoring $v1_bare (use --inbox-version=1 to force v1 instead)
 EOM
-               @v2_epochs = map { $uri->path($_); $uri->clone } @v2_epochs;
+               @v2_epochs = map {
+                       $uri->path($path_pfx.$_);
+                       $uri->clone
+               } @v2_epochs;
                clone_v2($self, \@v2_epochs);
-       } elsif ($v1_bare) {
+       } elsif (defined $v1_bare) {
                clone_v1($self);
-       } elsif (my @maybe = grep(m!\Q$path\E!, keys %$m)) {
-               die "E: confused by <$uri>, possible matches:\n@maybe";
        } else {
-               die "E: confused by <$uri>";
+               die "E: confused by <$uri>, possible matches:\n\t",
+                       join(', ', sort keys %$m), "\n";
        }
 }
 
@@ -262,14 +282,12 @@ sub do_mirror { # via wq_io_do
                return start_clone_url($self) if $self->{src} =~ m!://!;
                die "TODO: cloning local directories not supported, yet";
        };
-       return $lei->fail($@) if $@;
-       $lei->qerr("# mirrored $self->{src} => $self->{dst}");
+       $lei->fail($@) if $@;
 }
 
 sub start {
        my ($cls, $lei, $src, $dst) = @_;
-       my $self = bless { lei => $lei, src => $src, dst => $dst }, $cls;
-       $lei->{mrr} = $self;
+       my $self = bless { src => $src, dst => $dst }, $cls;
        if ($src =~ m!https?://!) {
                require URI;
                require PublicInbox::LeiCurl;
@@ -278,17 +296,16 @@ sub start {
        require PublicInbox::Inbox;
        require PublicInbox::Admin;
        require PublicInbox::InboxWritable;
-       my $op = $lei->workers_start($self, 'lei_mirror', 1, {
-               '' => [ \&mirror_done, $lei ]
-       });
+       my ($op_c, $ops) = $lei->workers_start($self, 1);
+       $lei->{wq1} = $self;
        $self->wq_io_do('do_mirror', []);
        $self->wq_close(1);
-       while ($op && $op->{sock}) { $op->event_step }
+       $lei->wait_wq_events($op_c, $ops);
 }
 
 sub ipc_atfork_child {
        my ($self) = @_;
-       $self->{lei}->lei_atfork_child;
+       $self->{lei}->_lei_atfork_child;
        $SIG{TERM} = sub { exit(128 + 15) }; # trigger OnDestroy $reap
        $self->SUPER::ipc_atfork_child;
 }