1 # Copyright (C) 2014-2015 all contributors <meta@public-inbox.org>
2 # License: GPLv2 or later (https://www.gnu.org/licenses/gpl-2.0.txt)
4 # Used to read files from a git repository without excessive forking.
5 # Used in our web interfaces as well as our -nntpd server.
6 # This is based on code in Git.pm which is GPLv2, but modified to avoid
7 # dependence on environment variables for compatibility with mod_perl.
8 # There are also API changes to simplify our usage and data set.
9 package PublicInbox::Git;
14 use PublicInbox::Spawn qw(spawn popen_rd);
17 my ($class, $git_dir) = @_;
18 bless { git_dir => $git_dir }, $class
22 my ($self, $batch, $in, $out, $pid) = @_;
23 return if $self->{$pid};
24 my ($in_r, $in_w, $out_r, $out_w);
26 pipe($in_r, $in_w) or fail($self, "pipe failed: $!");
27 pipe($out_r, $out_w) or fail($self, "pipe failed: $!");
29 my @cmd = ('git', "--git-dir=$self->{git_dir}", qw(cat-file), $batch);
30 my $redir = { 0 => fileno($out_r), 1 => fileno($in_w) };
31 $self->{$pid} = spawn(\@cmd, undef, $redir);
32 close $out_r or fail($self, "close failed: $!");
33 close $in_w or fail($self, "close failed: $!");
35 $self->{$out} = $out_w;
40 my ($self, $obj, $ref) = @_;
42 $self->_bidi_pipe(qw(--batch in out pid));
43 $self->{out}->print($obj, "\n") or fail($self, "write error: $!");
46 my $head = $in->getline;
47 $head =~ / missing$/ and return undef;
48 $head =~ /^[0-9a-f]{40} \S+ (\d+)$/ or
49 fail($self, "Unexpected result from git cat-file: $head");
52 my $ref_type = $ref ? ref($ref) : '';
56 $$ref = $size if ($ref_type eq 'SCALAR');
59 if ($ref_type eq 'CODE') {
60 $rv = eval { $ref->($in, \$left) };
65 my $r = read($in, my $x, $left > $max ? $max : $left);
66 defined($r) or fail($self, "read failed: $!");
67 $r == 0 and fail($self, 'exited unexpectedly');
74 my $r = read($in, $buf, $left, $offset);
75 defined($r) or fail($self, "read failed: $!");
76 $r == 0 and fail($self, 'exited unexpectedly');
83 my $r = read($in, my $buf, 1);
84 defined($r) or fail($self, "read failed: $!");
85 fail($self, 'newline missing after blob') if ($r != 1 || $buf ne "\n");
86 die $cb_err if $cb_err;
92 my ($self, $obj) = @_;
93 $self->_bidi_pipe(qw(--batch-check in_c out_c pid_c));
94 $self->{out_c}->print($obj, "\n") or fail($self, "write error: $!");
95 chomp(my $line = $self->{in_c}->getline);
96 my ($hex, $type, $size) = split(' ', $line);
97 return if $type eq 'missing';
102 my ($self, $in, $out, $pid) = @_;
103 my $p = $self->{$pid} or return;
104 $self->{$pid} = undef;
105 foreach my $f ($in, $out) {
106 my $fh = $self->{$f};
115 my ($self, $msg) = @_;
121 my ($self, @cmd) = @_;
122 @cmd = ('git', "--git-dir=$self->{git_dir}", @cmd);
128 _destroy($self, qw(in out pid));
129 _destroy($self, qw(in_c out_c pid_c));
132 sub DESTROY { cleanup(@_) }