diff options
Diffstat (limited to 'lib/PublicInbox/Fetch.pm')
-rw-r--r-- | lib/PublicInbox/Fetch.pm | 224 |
1 files changed, 224 insertions, 0 deletions
diff --git a/lib/PublicInbox/Fetch.pm b/lib/PublicInbox/Fetch.pm new file mode 100644 index 00000000..b0f1437c --- /dev/null +++ b/lib/PublicInbox/Fetch.pm @@ -0,0 +1,224 @@ +# Copyright (C) all contributors <meta@public-inbox.org> +# License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt> +# Wrapper to "git fetch" remote public-inboxes +package PublicInbox::Fetch; +use v5.12; +use parent qw(PublicInbox::IPC); +use URI (); +use PublicInbox::Spawn qw(popen_rd run_qx run_wait); +use PublicInbox::Admin; +use PublicInbox::LEI; +use PublicInbox::LeiCurl; +use PublicInbox::LeiMirror; +use PublicInbox::SHA qw(sha_all); +use File::Temp (); + +sub new { bless {}, __PACKAGE__ } + +sub remote_url ($$) { + my ($lei, $dir) = @_; + my $rn = $lei->{opt}->{'try-remote'} // [ 'origin', '_grokmirror' ]; + for my $r (@$rn) { + my $cmd = [ qw(git config), "remote.$r.url" ]; + my $url = run_qx($cmd, undef, { -C => $dir, 2 => $lei->{2} }); + next if $?; + $url =~ s!/*\n!!s; + return $url; + } + undef +} + +# PSGI mount prefixes and manifest.js.gz prefixes don't always align... +# TODO: remove, handle multi-inbox fetch +sub deduce_epochs ($$) { + my ($m, $path) = @_; + my ($v1_ent, @v2_epochs); + my $path_pfx = ''; + $path =~ s!/+\z!!; + do { + $v1_ent = $m->{$path}; + @v2_epochs = grep(m!\A\Q$path\E/git/[0-9]+\.git\z!, keys %$m); + } while (!defined($v1_ent) && !@v2_epochs && + $path =~ s!\A(/[^/]+)/!/! and $path_pfx .= $1); + ($path_pfx, $v1_ent ? $path : undef, @v2_epochs); +} + +sub do_manifest ($$$) { + my ($lei, $dir, $ibx_uri) = @_; + my $muri = URI->new("$ibx_uri/manifest.js.gz"); + my $ft = File::Temp->new(TEMPLATE => 'm-XXXX', + UNLINK => 1, DIR => $dir, SUFFIX => '.tmp'); + my $mf = "$dir/manifest.js.gz"; + my $m0; # current manifest.js.gz contents + if (open my $fh, '<', $mf) { + $m0 = eval { + PublicInbox::LeiMirror::decode_manifest($fh, $mf, $mf) + }; + warn($@) if $@; + } + my ($bn) = ($ft->filename =~ m!/([^/]+)\z!); + my $curl_cmd = $lei->{curl}->for_uri($lei, $muri, qw(-R -o), $bn); + my $opt = { -C => $dir }; + $opt->{$_} = $lei->{$_} for (0..2); + my $cerr = PublicInbox::LeiMirror::run_reap($lei, $curl_cmd, $opt); + if ($cerr) { + return [ 404, $muri ] if ($cerr >> 8) == 22; # 404 Missing + $lei->child_error($cerr, "@$curl_cmd failed"); + return; + } + my $m1 = eval { + PublicInbox::LeiMirror::decode_manifest($ft, $ft, $muri); + } or return [ 404, $muri ]; + my $mdiff = { %$m1 }; + + # filter out unchanged entries. We check modified, too, since + # fingerprints are SHA-1, so there's a teeny chance they'll collide + while (my ($k, $v0) = each %{$m0 // {}}) { + my $cur = $m1->{$k} // next; + my $f0 = $v0->{fingerprint} // next; + my $f1 = $cur->{fingerprint} // next; + my $t0 = $v0->{modified} // next; + my $t1 = $cur->{modified} // next; + delete($mdiff->{$k}) if $f0 eq $f1 && $t0 == $t1; + } + unless (keys %$mdiff) { + $lei->child_error(127 << 8) if $lei->{opt}->{'exit-code'}; + return; + } + my (undef, $v1_path, @v2_epochs) = + deduce_epochs($mdiff, $ibx_uri->path); + [ 200, $muri, $v1_path, \@v2_epochs, $ft, $mf, $m1 ]; +} + +sub get_fingerprint2 { + my ($git_dir) = @_; + my $rd = popen_rd([qw(git show-ref)], undef, { -C => $git_dir }); + sha_all(256, $rd)->digest; # ignore show-ref errors +} + +sub writable_dir ($) { + my ($dir) = @_; + return unless -d $dir && -w _; + my @st = stat($dir); + $st[2] & 0222; # any writable bits set? (in case of root) +} + +sub do_fetch { # main entry point + my ($cls, $lei, $cd) = @_; + my $ibx_ver; + $lei->{curl} //= PublicInbox::LeiCurl->new($lei) or return; + my $dir = PublicInbox::Admin::resolve_inboxdir($cd, \$ibx_ver); + my ($ibx_uri, @git_dir, @epochs, $mg, @new_epoch, $skip); + if ($ibx_ver == 1) { + my $url = remote_url($lei, $dir) // + die "E: $dir missing remote.*.url\n"; + $ibx_uri = URI->new($url); + } else { # v2: + require PublicInbox::MultiGit; + $mg = PublicInbox::MultiGit->new($dir, 'all.git', 'git'); + @epochs = $mg->git_epochs; + my ($git_url, $epoch); + for my $nr (@epochs) { # try newest epoch, first + my $edir = "$dir/git/$nr.git"; + if (!writable_dir($edir)) { + $skip->{$nr} = 1; + next; + } + next if defined $git_url; + if (defined(my $url = remote_url($lei, $edir))) { + $git_url = $url; + $epoch = $nr; + } else { + warn "W: $edir missing remote.*.url\n"; + my $o = { -C => $edir }; + $o->{1} = $o->{2} = $lei->{2}; + run_wait([qw(git config -l)], undef, $o) and + $lei->child_error($?); + } + } + @epochs = grep { !$skip->{$_} } @epochs if $skip; + $skip //= {}; # makes code below easier + $git_url or die "Unable to determine git URL\n"; + my $inbox_url = $git_url; + $inbox_url =~ s!/git/$epoch(?:\.git)?/?\z!! or + $inbox_url =~ s!/$epoch(?:\.git)?/?\z!! or die <<EOM; +Unable to infer inbox URL from <$git_url> +EOM + $ibx_uri = URI->new($inbox_url); + } + PublicInbox::LeiMirror::write_makefile($dir, $ibx_ver); + $lei->qerr("# inbox URL: $ibx_uri/"); + my $res = do_manifest($lei, $dir, $ibx_uri) or return; + my ($code, $muri, $v1_path, $v2_epochs, $ft, $mf, $m1) = @$res; + if ($code == 404) { + # any pre-manifest.js.gz instances running? Just fetch all + # existing ones and unconditionally try cloning the next + $v2_epochs = [ map { "$dir/git/$_.git" } @epochs ]; + if (@epochs) { + my $n = $epochs[-1] + 1; + push @$v2_epochs, "$dir/git/$n.git" if !$skip->{$n}; + } + } else { + $code == 200 or die "BUG unexpected code $code\n"; + } + my $mculled; + if ($ibx_ver == 2) { + defined($v1_path) and warn <<EOM; +E: got v1 `$v1_path' when expecting v2 epoch(s) in <$muri>, WTF? +EOM + @git_dir = map { "$dir/git/$_.git" } sort { $a <=> $b } map { + my ($nr) = (m!/([0-9]+)\.git\z!g); + $skip->{$nr} ? () : $nr; + } @$v2_epochs; + if ($m1 && scalar keys %$skip) { + my $re = join('|', keys %$skip); + my @del = grep(m!/git/$re\.git\z!, keys %$m1); + delete @$m1{@del}; + $mculled = 1; + } + } else { + $git_dir[0] = $dir; + } + # n.b. this expects all epochs are from the same host + my $torsocks = $lei->{curl}->torsocks($lei, $muri); + my $fp2 = $lei->{opt}->{'exit-code'} ? [] : undef; + my $xit = 127; + for my $d (@git_dir) { + my $cmd; + my $opt = {}; # for spawn + if (-d $d) { + $fp2->[0] = get_fingerprint2($d) if $fp2; + $cmd = [ @$torsocks, 'git', "--git-dir=$d", + PublicInbox::LeiMirror::fetch_args($lei, $opt)]; + } else { + my $e_uri = $ibx_uri->clone; + my ($epath) = ($d =~ m!(/git/[0-9]+\.git)\z!); + defined($epath) or + die "BUG: $d is not an epoch to clone\n"; + $e_uri->path($ibx_uri->path.$epath); + $cmd = [ @$torsocks, + PublicInbox::LeiMirror::clone_cmd($lei, $opt), + $$e_uri, $d]; + push @new_epoch, substr($epath, 5, -4) + 0; + $xit = 0; + } + my $cerr = PublicInbox::LeiMirror::run_reap($lei, $cmd, $opt); + # do not bail on clone failure if we didn't have a manifest + if ($cerr && ($code == 200 || -d $d)) { + $lei->child_error($cerr, "@$cmd failed"); + return; + } + if ($fp2 && $xit) { + $fp2->[1] = get_fingerprint2($d); + $xit = 0 if $fp2->[0] ne $fp2->[1]; + } + } + for my $i (@new_epoch) { $mg->epoch_cfg_set($i) } + if ($ft) { + PublicInbox::LeiMirror::dump_manifest($m1 => $ft) if $mculled; + PublicInbox::LeiMirror::ft_rename($ft, $mf, 0666); + } + $lei->child_error($xit << 8) if $fp2 && $xit; +} + +1; |