lib/PublicInbox/Fetch.pm - pub/scm/infra/public-inbox - Git at Google

 # 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 ();
 use PublicInbox::Git qw(git_exe);

 sub new { bless {}, __PACKAGE__ }

 sub remote_url ($$) {
 	my ($lei, $dir) = @_;
 	my $rn = $lei->{opt}->{'try-remote'} // [ 'origin', '_grokmirror' ];
 	for my $r (@$rn) {
 		my $cmd = [ git_exe, '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([git_exe, '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([git_exe, qw(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_exe, "--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;
	# 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 ();
	use PublicInbox::Git qw(git_exe);

	sub new { bless {}, __PACKAGE__ }

	sub remote_url ($$) {
	my ($lei, $dir) = @_;
	my $rn = $lei->{opt}->{'try-remote'} // [ 'origin', '_grokmirror' ];
	for my $r (@$rn) {
	my $cmd = [ git_exe, '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([git_exe, '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([git_exe, qw(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_exe, "--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;