about summary refs log tree commit homepage
path: root/lib/PublicInbox/GitCatFile.pm
diff options
context:
space:
mode:
Diffstat (limited to 'lib/PublicInbox/GitCatFile.pm')
-rw-r--r--lib/PublicInbox/GitCatFile.pm73
1 files changed, 73 insertions, 0 deletions
diff --git a/lib/PublicInbox/GitCatFile.pm b/lib/PublicInbox/GitCatFile.pm
new file mode 100644
index 00000000..bdad2b56
--- /dev/null
+++ b/lib/PublicInbox/GitCatFile.pm
@@ -0,0 +1,73 @@
+# Copyright (C) 2014, Eric Wong <normalperson@yhbt.net> and all contributors
+# License: GPLv2 or later (https://www.gnu.org/licenses/gpl-2.0.txt)
+# This is based on code in Git.pm which is GPLv2, but modified to avoid
+# dependence on environment variables for compatibility with mod_perl.
+# There are also API changes to simplify our usage and data set.
+package PublicInbox::GitCatFile;
+use strict;
+use warnings;
+use IPC::Open2 qw(open2);
+
+sub new {
+        my ($class, $git_dir) = @_;
+        bless { git_dir => $git_dir }, $class;
+}
+
+sub _cat_file_begin {
+        my ($self) = @_;
+        return if $self->{pid};
+        my ($in, $out);
+        my $pid = open2($in, $out, 'git', '--git-dir', $self->{git_dir},
+                        'cat-file', '--batch');
+
+        $self->{pid} = $pid;
+        $self->{in} = $in;
+        $self->{out} = $out;
+}
+
+sub cat_file {
+        my ($self, $object) = @_;
+
+        $self->_cat_file_begin;
+        print { $self->{out} } $object, "\n" or die "write error: $!\n";
+
+        my $in = $self->{in};
+        my $head = <$in>;
+        $head =~ / missing$/ and return undef;
+        $head =~ /^[0-9a-f]{40} \S+ (\d+)$/ or
+                die "Unexpected result from git cat-file: $head\n";
+
+        my $size = $1;
+        my $bytes_left = $size;
+        my $buf;
+        my $rv = '';
+
+        while ($bytes_left) {
+                my $read = read($in, $buf, $bytes_left);
+                defined($read) or die "read pipe failed: $!\n";
+                $rv .= $buf;
+                $bytes_left -= $read;
+        }
+
+        my $read = read($in, $buf, 1);
+        defined($read) or die "read pipe failed: $!\n";
+        if ($read != 1 || $buf ne "\n") {
+                die "newline missing after blob\n";
+        }
+        \$rv;
+}
+
+sub DESTROY {
+        my ($self) = @_;
+        my $pid = $self->{pid} or return;
+        $self->{pid} = undef;
+        foreach my $f (qw(in out)) {
+                my $fh = $self->{$f};
+                defined $fh or next;
+                close $fh;
+                $self->{$f} = undef;
+        }
+        waitpid $pid, 0;
+}
+
+1;