perl/Git.pm: Honor SSH_ASKPASS as fallback if GIT_ASKPASS is not set
[git.git] / perl / Git.pm
1 =head1 NAME
2
3 Git - Perl interface to the Git version control system
4
5 =cut
6
7
8 package Git;
9
10 use 5.008;
11 use strict;
12
13
14 BEGIN {
15
16 our ($VERSION, @ISA, @EXPORT, @EXPORT_OK);
17
18 # Totally unstable API.
19 $VERSION = '0.01';
20
21
22 =head1 SYNOPSIS
23
24   use Git;
25
26   my $version = Git::command_oneline('version');
27
28   git_cmd_try { Git::command_noisy('update-server-info') }
29               '%s failed w/ code %d';
30
31   my $repo = Git->repository (Directory => '/srv/git/cogito.git');
32
33
34   my @revs = $repo->command('rev-list', '--since=last monday', '--all');
35
36   my ($fh, $c) = $repo->command_output_pipe('rev-list', '--since=last monday', '--all');
37   my $lastrev = <$fh>; chomp $lastrev;
38   $repo->command_close_pipe($fh, $c);
39
40   my $lastrev = $repo->command_oneline( [ 'rev-list', '--all' ],
41                                         STDERR => 0 );
42
43   my $sha1 = $repo->hash_and_insert_object('file.txt');
44   my $tempfile = tempfile();
45   my $size = $repo->cat_blob($sha1, $tempfile);
46
47 =cut
48
49
50 require Exporter;
51
52 @ISA = qw(Exporter);
53
54 @EXPORT = qw(git_cmd_try);
55
56 # Methods which can be called as standalone functions as well:
57 @EXPORT_OK = qw(command command_oneline command_noisy
58                 command_output_pipe command_input_pipe command_close_pipe
59                 command_bidi_pipe command_close_bidi_pipe
60                 version exec_path html_path hash_object git_cmd_try
61                 remote_refs prompt
62                 temp_acquire temp_release temp_reset temp_path);
63
64
65 =head1 DESCRIPTION
66
67 This module provides Perl scripts easy way to interface the Git version control
68 system. The modules have an easy and well-tested way to call arbitrary Git
69 commands; in the future, the interface will also provide specialized methods
70 for doing easily operations which are not totally trivial to do over
71 the generic command interface.
72
73 While some commands can be executed outside of any context (e.g. 'version'
74 or 'init'), most operations require a repository context, which in practice
75 means getting an instance of the Git object using the repository() constructor.
76 (In the future, we will also get a new_repository() constructor.) All commands
77 called as methods of the object are then executed in the context of the
78 repository.
79
80 Part of the "repository state" is also information about path to the attached
81 working copy (unless you work with a bare repository). You can also navigate
82 inside of the working copy using the C<wc_chdir()> method. (Note that
83 the repository object is self-contained and will not change working directory
84 of your process.)
85
86 TODO: In the future, we might also do
87
88         my $remoterepo = $repo->remote_repository (Name => 'cogito', Branch => 'master');
89         $remoterepo ||= Git->remote_repository ('http://git.or.cz/cogito.git/');
90         my @refs = $remoterepo->refs();
91
92 Currently, the module merely wraps calls to external Git tools. In the future,
93 it will provide a much faster way to interact with Git by linking directly
94 to libgit. This should be completely opaque to the user, though (performance
95 increase notwithstanding).
96
97 =cut
98
99
100 use Carp qw(carp croak); # but croak is bad - throw instead
101 use Error qw(:try);
102 use Cwd qw(abs_path cwd);
103 use IPC::Open2 qw(open2);
104 use Fcntl qw(SEEK_SET SEEK_CUR);
105 }
106
107
108 =head1 CONSTRUCTORS
109
110 =over 4
111
112 =item repository ( OPTIONS )
113
114 =item repository ( DIRECTORY )
115
116 =item repository ()
117
118 Construct a new repository object.
119 C<OPTIONS> are passed in a hash like fashion, using key and value pairs.
120 Possible options are:
121
122 B<Repository> - Path to the Git repository.
123
124 B<WorkingCopy> - Path to the associated working copy; not strictly required
125 as many commands will happily crunch on a bare repository.
126
127 B<WorkingSubdir> - Subdirectory in the working copy to work inside.
128 Just left undefined if you do not want to limit the scope of operations.
129
130 B<Directory> - Path to the Git working directory in its usual setup.
131 The C<.git> directory is searched in the directory and all the parent
132 directories; if found, C<WorkingCopy> is set to the directory containing
133 it and C<Repository> to the C<.git> directory itself. If no C<.git>
134 directory was found, the C<Directory> is assumed to be a bare repository,
135 C<Repository> is set to point at it and C<WorkingCopy> is left undefined.
136 If the C<$GIT_DIR> environment variable is set, things behave as expected
137 as well.
138
139 You should not use both C<Directory> and either of C<Repository> and
140 C<WorkingCopy> - the results of that are undefined.
141
142 Alternatively, a directory path may be passed as a single scalar argument
143 to the constructor; it is equivalent to setting only the C<Directory> option
144 field.
145
146 Calling the constructor with no options whatsoever is equivalent to
147 calling it with C<< Directory => '.' >>. In general, if you are building
148 a standard porcelain command, simply doing C<< Git->repository() >> should
149 do the right thing and setup the object to reflect exactly where the user
150 is right now.
151
152 =cut
153
154 sub repository {
155         my $class = shift;
156         my @args = @_;
157         my %opts = ();
158         my $self;
159
160         if (defined $args[0]) {
161                 if ($#args % 2 != 1) {
162                         # Not a hash.
163                         $#args == 0 or throw Error::Simple("bad usage");
164                         %opts = ( Directory => $args[0] );
165                 } else {
166                         %opts = @args;
167                 }
168         }
169
170         if (not defined $opts{Repository} and not defined $opts{WorkingCopy}
171                 and not defined $opts{Directory}) {
172                 $opts{Directory} = '.';
173         }
174
175         if (defined $opts{Directory}) {
176                 -d $opts{Directory} or throw Error::Simple("Directory not found: $opts{Directory} $!");
177
178                 my $search = Git->repository(WorkingCopy => $opts{Directory});
179                 my $dir;
180                 try {
181                         $dir = $search->command_oneline(['rev-parse', '--git-dir'],
182                                                         STDERR => 0);
183                 } catch Git::Error::Command with {
184                         $dir = undef;
185                 };
186
187                 if ($dir) {
188                         $dir =~ m#^/# or $dir = $opts{Directory} . '/' . $dir;
189                         $opts{Repository} = abs_path($dir);
190
191                         # If --git-dir went ok, this shouldn't die either.
192                         my $prefix = $search->command_oneline('rev-parse', '--show-prefix');
193                         $dir = abs_path($opts{Directory}) . '/';
194                         if ($prefix) {
195                                 if (substr($dir, -length($prefix)) ne $prefix) {
196                                         throw Error::Simple("rev-parse confused me - $dir does not have trailing $prefix");
197                                 }
198                                 substr($dir, -length($prefix)) = '';
199                         }
200                         $opts{WorkingCopy} = $dir;
201                         $opts{WorkingSubdir} = $prefix;
202
203                 } else {
204                         # A bare repository? Let's see...
205                         $dir = $opts{Directory};
206
207                         unless (-d "$dir/refs" and -d "$dir/objects" and -e "$dir/HEAD") {
208                                 # Mimic git-rev-parse --git-dir error message:
209                                 throw Error::Simple("fatal: Not a git repository: $dir");
210                         }
211                         my $search = Git->repository(Repository => $dir);
212                         try {
213                                 $search->command('symbolic-ref', 'HEAD');
214                         } catch Git::Error::Command with {
215                                 # Mimic git-rev-parse --git-dir error message:
216                                 throw Error::Simple("fatal: Not a git repository: $dir");
217                         }
218
219                         $opts{Repository} = abs_path($dir);
220                 }
221
222                 delete $opts{Directory};
223         }
224
225         $self = { opts => \%opts };
226         bless $self, $class;
227 }
228
229 =back
230
231 =head1 METHODS
232
233 =over 4
234
235 =item command ( COMMAND [, ARGUMENTS... ] )
236
237 =item command ( [ COMMAND, ARGUMENTS... ], { Opt => Val ... } )
238
239 Execute the given Git C<COMMAND> (specify it without the 'git-'
240 prefix), optionally with the specified extra C<ARGUMENTS>.
241
242 The second more elaborate form can be used if you want to further adjust
243 the command execution. Currently, only one option is supported:
244
245 B<STDERR> - How to deal with the command's error output. By default (C<undef>)
246 it is delivered to the caller's C<STDERR>. A false value (0 or '') will cause
247 it to be thrown away. If you want to process it, you can get it in a filehandle
248 you specify, but you must be extremely careful; if the error output is not
249 very short and you want to read it in the same process as where you called
250 C<command()>, you are set up for a nice deadlock!
251
252 The method can be called without any instance or on a specified Git repository
253 (in that case the command will be run in the repository context).
254
255 In scalar context, it returns all the command output in a single string
256 (verbatim).
257
258 In array context, it returns an array containing lines printed to the
259 command's stdout (without trailing newlines).
260
261 In both cases, the command's stdin and stderr are the same as the caller's.
262
263 =cut
264
265 sub command {
266         my ($fh, $ctx) = command_output_pipe(@_);
267
268         if (not defined wantarray) {
269                 # Nothing to pepper the possible exception with.
270                 _cmd_close($fh, $ctx);
271
272         } elsif (not wantarray) {
273                 local $/;
274                 my $text = <$fh>;
275                 try {
276                         _cmd_close($fh, $ctx);
277                 } catch Git::Error::Command with {
278                         # Pepper with the output:
279                         my $E = shift;
280                         $E->{'-outputref'} = \$text;
281                         throw $E;
282                 };
283                 return $text;
284
285         } else {
286                 my @lines = <$fh>;
287                 defined and chomp for @lines;
288                 try {
289                         _cmd_close($fh, $ctx);
290                 } catch Git::Error::Command with {
291                         my $E = shift;
292                         $E->{'-outputref'} = \@lines;
293                         throw $E;
294                 };
295                 return @lines;
296         }
297 }
298
299
300 =item command_oneline ( COMMAND [, ARGUMENTS... ] )
301
302 =item command_oneline ( [ COMMAND, ARGUMENTS... ], { Opt => Val ... } )
303
304 Execute the given C<COMMAND> in the same way as command()
305 does but always return a scalar string containing the first line
306 of the command's standard output.
307
308 =cut
309
310 sub command_oneline {
311         my ($fh, $ctx) = command_output_pipe(@_);
312
313         my $line = <$fh>;
314         defined $line and chomp $line;
315         try {
316                 _cmd_close($fh, $ctx);
317         } catch Git::Error::Command with {
318                 # Pepper with the output:
319                 my $E = shift;
320                 $E->{'-outputref'} = \$line;
321                 throw $E;
322         };
323         return $line;
324 }
325
326
327 =item command_output_pipe ( COMMAND [, ARGUMENTS... ] )
328
329 =item command_output_pipe ( [ COMMAND, ARGUMENTS... ], { Opt => Val ... } )
330
331 Execute the given C<COMMAND> in the same way as command()
332 does but return a pipe filehandle from which the command output can be
333 read.
334
335 The function can return C<($pipe, $ctx)> in array context.
336 See C<command_close_pipe()> for details.
337
338 =cut
339
340 sub command_output_pipe {
341         _command_common_pipe('-|', @_);
342 }
343
344
345 =item command_input_pipe ( COMMAND [, ARGUMENTS... ] )
346
347 =item command_input_pipe ( [ COMMAND, ARGUMENTS... ], { Opt => Val ... } )
348
349 Execute the given C<COMMAND> in the same way as command_output_pipe()
350 does but return an input pipe filehandle instead; the command output
351 is not captured.
352
353 The function can return C<($pipe, $ctx)> in array context.
354 See C<command_close_pipe()> for details.
355
356 =cut
357
358 sub command_input_pipe {
359         _command_common_pipe('|-', @_);
360 }
361
362
363 =item command_close_pipe ( PIPE [, CTX ] )
364
365 Close the C<PIPE> as returned from C<command_*_pipe()>, checking
366 whether the command finished successfully. The optional C<CTX> argument
367 is required if you want to see the command name in the error message,
368 and it is the second value returned by C<command_*_pipe()> when
369 called in array context. The call idiom is:
370
371         my ($fh, $ctx) = $r->command_output_pipe('status');
372         while (<$fh>) { ... }
373         $r->command_close_pipe($fh, $ctx);
374
375 Note that you should not rely on whatever actually is in C<CTX>;
376 currently it is simply the command name but in future the context might
377 have more complicated structure.
378
379 =cut
380
381 sub command_close_pipe {
382         my ($self, $fh, $ctx) = _maybe_self(@_);
383         $ctx ||= '<unknown>';
384         _cmd_close($fh, $ctx);
385 }
386
387 =item command_bidi_pipe ( COMMAND [, ARGUMENTS... ] )
388
389 Execute the given C<COMMAND> in the same way as command_output_pipe()
390 does but return both an input pipe filehandle and an output pipe filehandle.
391
392 The function will return return C<($pid, $pipe_in, $pipe_out, $ctx)>.
393 See C<command_close_bidi_pipe()> for details.
394
395 =cut
396
397 sub command_bidi_pipe {
398         my ($pid, $in, $out);
399         my ($self) = _maybe_self(@_);
400         local %ENV = %ENV;
401         my $cwd_save = undef;
402         if ($self) {
403                 shift;
404                 $cwd_save = cwd();
405                 _setup_git_cmd_env($self);
406         }
407         $pid = open2($in, $out, 'git', @_);
408         chdir($cwd_save) if $cwd_save;
409         return ($pid, $in, $out, join(' ', @_));
410 }
411
412 =item command_close_bidi_pipe ( PID, PIPE_IN, PIPE_OUT [, CTX] )
413
414 Close the C<PIPE_IN> and C<PIPE_OUT> as returned from C<command_bidi_pipe()>,
415 checking whether the command finished successfully. The optional C<CTX>
416 argument is required if you want to see the command name in the error message,
417 and it is the fourth value returned by C<command_bidi_pipe()>.  The call idiom
418 is:
419
420         my ($pid, $in, $out, $ctx) = $r->command_bidi_pipe('cat-file --batch-check');
421         print "000000000\n" $out;
422         while (<$in>) { ... }
423         $r->command_close_bidi_pipe($pid, $in, $out, $ctx);
424
425 Note that you should not rely on whatever actually is in C<CTX>;
426 currently it is simply the command name but in future the context might
427 have more complicated structure.
428
429 =cut
430
431 sub command_close_bidi_pipe {
432         local $?;
433         my ($pid, $in, $out, $ctx) = @_;
434         foreach my $fh ($in, $out) {
435                 unless (close $fh) {
436                         if ($!) {
437                                 carp "error closing pipe: $!";
438                         } elsif ($? >> 8) {
439                                 throw Git::Error::Command($ctx, $? >>8);
440                         }
441                 }
442         }
443
444         waitpid $pid, 0;
445
446         if ($? >> 8) {
447                 throw Git::Error::Command($ctx, $? >>8);
448         }
449 }
450
451
452 =item command_noisy ( COMMAND [, ARGUMENTS... ] )
453
454 Execute the given C<COMMAND> in the same way as command() does but do not
455 capture the command output - the standard output is not redirected and goes
456 to the standard output of the caller application.
457
458 While the method is called command_noisy(), you might want to as well use
459 it for the most silent Git commands which you know will never pollute your
460 stdout but you want to avoid the overhead of the pipe setup when calling them.
461
462 The function returns only after the command has finished running.
463
464 =cut
465
466 sub command_noisy {
467         my ($self, $cmd, @args) = _maybe_self(@_);
468         _check_valid_cmd($cmd);
469
470         my $pid = fork;
471         if (not defined $pid) {
472                 throw Error::Simple("fork failed: $!");
473         } elsif ($pid == 0) {
474                 _cmd_exec($self, $cmd, @args);
475         }
476         if (waitpid($pid, 0) > 0 and $?>>8 != 0) {
477                 throw Git::Error::Command(join(' ', $cmd, @args), $? >> 8);
478         }
479 }
480
481
482 =item version ()
483
484 Return the Git version in use.
485
486 =cut
487
488 sub version {
489         my $verstr = command_oneline('--version');
490         $verstr =~ s/^git version //;
491         $verstr;
492 }
493
494
495 =item exec_path ()
496
497 Return path to the Git sub-command executables (the same as
498 C<git --exec-path>). Useful mostly only internally.
499
500 =cut
501
502 sub exec_path { command_oneline('--exec-path') }
503
504
505 =item html_path ()
506
507 Return path to the Git html documentation (the same as
508 C<git --html-path>). Useful mostly only internally.
509
510 =cut
511
512 sub html_path { command_oneline('--html-path') }
513
514 =item prompt ( PROMPT )
515
516 Query user C<PROMPT> and return answer from user.
517
518 Honours GIT_ASKPASS and SSH_ASKPASS environment variables for querying
519 the user. If no *_ASKPASS variable is set or an error occoured,
520 the terminal is tried as a fallback.
521
522 =cut
523
524 sub prompt {
525         my ($prompt) = @_;
526         my $ret;
527         if (exists $ENV{'GIT_ASKPASS'}) {
528                 $ret = _prompt($ENV{'GIT_ASKPASS'}, $prompt);
529         }
530         if (!defined $ret && exists $ENV{'SSH_ASKPASS'}) {
531                 $ret = _prompt($ENV{'SSH_ASKPASS'}, $prompt);
532         }
533         if (!defined $ret) {
534                 print STDERR $prompt;
535                 STDERR->flush;
536                 require Term::ReadKey;
537                 Term::ReadKey::ReadMode('noecho');
538                 $ret = '';
539                 while (defined(my $key = Term::ReadKey::ReadKey(0))) {
540                         last if $key =~ /[\012\015]/; # \n\r
541                         $ret .= $key;
542                 }
543                 Term::ReadKey::ReadMode('restore');
544                 print STDERR "\n";
545                 STDERR->flush;
546         }
547         return $ret;
548 }
549
550 sub _prompt {
551         my ($askpass, $prompt) = @_;
552         return unless length $askpass;
553         my $ret;
554         open my $fh, "-|", $askpass, $prompt or return;
555         $ret = <$fh>;
556         $ret =~ s/[\015\012]//g; # strip \r\n, chomp does not work on all systems (i.e. windows) as expected
557         close ($fh);
558         return $ret;
559 }
560
561 =item repo_path ()
562
563 Return path to the git repository. Must be called on a repository instance.
564
565 =cut
566
567 sub repo_path { $_[0]->{opts}->{Repository} }
568
569
570 =item wc_path ()
571
572 Return path to the working copy. Must be called on a repository instance.
573
574 =cut
575
576 sub wc_path { $_[0]->{opts}->{WorkingCopy} }
577
578
579 =item wc_subdir ()
580
581 Return path to the subdirectory inside of a working copy. Must be called
582 on a repository instance.
583
584 =cut
585
586 sub wc_subdir { $_[0]->{opts}->{WorkingSubdir} ||= '' }
587
588
589 =item wc_chdir ( SUBDIR )
590
591 Change the working copy subdirectory to work within. The C<SUBDIR> is
592 relative to the working copy root directory (not the current subdirectory).
593 Must be called on a repository instance attached to a working copy
594 and the directory must exist.
595
596 =cut
597
598 sub wc_chdir {
599         my ($self, $subdir) = @_;
600         $self->wc_path()
601                 or throw Error::Simple("bare repository");
602
603         -d $self->wc_path().'/'.$subdir
604                 or throw Error::Simple("subdir not found: $subdir $!");
605         # Of course we will not "hold" the subdirectory so anyone
606         # can delete it now and we will never know. But at least we tried.
607
608         $self->{opts}->{WorkingSubdir} = $subdir;
609 }
610
611
612 =item config ( VARIABLE )
613
614 Retrieve the configuration C<VARIABLE> in the same manner as C<config>
615 does. In scalar context requires the variable to be set only one time
616 (exception is thrown otherwise), in array context returns allows the
617 variable to be set multiple times and returns all the values.
618
619 =cut
620
621 sub config {
622         return _config_common({}, @_);
623 }
624
625
626 =item config_bool ( VARIABLE )
627
628 Retrieve the bool configuration C<VARIABLE>. The return value
629 is usable as a boolean in perl (and C<undef> if it's not defined,
630 of course).
631
632 =cut
633
634 sub config_bool {
635         my $val = scalar _config_common({'kind' => '--bool'}, @_);
636
637         # Do not rewrite this as return (defined $val && $val eq 'true')
638         # as some callers do care what kind of falsehood they receive.
639         if (!defined $val) {
640                 return undef;
641         } else {
642                 return $val eq 'true';
643         }
644 }
645
646
647 =item config_path ( VARIABLE )
648
649 Retrieve the path configuration C<VARIABLE>. The return value
650 is an expanded path or C<undef> if it's not defined.
651
652 =cut
653
654 sub config_path {
655         return _config_common({'kind' => '--path'}, @_);
656 }
657
658
659 =item config_int ( VARIABLE )
660
661 Retrieve the integer configuration C<VARIABLE>. The return value
662 is simple decimal number.  An optional value suffix of 'k', 'm',
663 or 'g' in the config file will cause the value to be multiplied
664 by 1024, 1048576 (1024^2), or 1073741824 (1024^3) prior to output.
665 It would return C<undef> if configuration variable is not defined,
666
667 =cut
668
669 sub config_int {
670         return scalar _config_common({'kind' => '--int'}, @_);
671 }
672
673 # Common subroutine to implement bulk of what the config* family of methods
674 # do. This curently wraps command('config') so it is not so fast.
675 sub _config_common {
676         my ($opts) = shift @_;
677         my ($self, $var) = _maybe_self(@_);
678
679         try {
680                 my @cmd = ('config', $opts->{'kind'} ? $opts->{'kind'} : ());
681                 unshift @cmd, $self if $self;
682                 if (wantarray) {
683                         return command(@cmd, '--get-all', $var);
684                 } else {
685                         return command_oneline(@cmd, '--get', $var);
686                 }
687         } catch Git::Error::Command with {
688                 my $E = shift;
689                 if ($E->value() == 1) {
690                         # Key not found.
691                         return;
692                 } else {
693                         throw $E;
694                 }
695         };
696 }
697
698 =item get_colorbool ( NAME )
699
700 Finds if color should be used for NAMEd operation from the configuration,
701 and returns boolean (true for "use color", false for "do not use color").
702
703 =cut
704
705 sub get_colorbool {
706         my ($self, $var) = @_;
707         my $stdout_to_tty = (-t STDOUT) ? "true" : "false";
708         my $use_color = $self->command_oneline('config', '--get-colorbool',
709                                                $var, $stdout_to_tty);
710         return ($use_color eq 'true');
711 }
712
713 =item get_color ( SLOT, COLOR )
714
715 Finds color for SLOT from the configuration, while defaulting to COLOR,
716 and returns the ANSI color escape sequence:
717
718         print $repo->get_color("color.interactive.prompt", "underline blue white");
719         print "some text";
720         print $repo->get_color("", "normal");
721
722 =cut
723
724 sub get_color {
725         my ($self, $slot, $default) = @_;
726         my $color = $self->command_oneline('config', '--get-color', $slot, $default);
727         if (!defined $color) {
728                 $color = "";
729         }
730         return $color;
731 }
732
733 =item remote_refs ( REPOSITORY [, GROUPS [, REFGLOBS ] ] )
734
735 This function returns a hashref of refs stored in a given remote repository.
736 The hash is in the format C<refname =\> hash>. For tags, the C<refname> entry
737 contains the tag object while a C<refname^{}> entry gives the tagged objects.
738
739 C<REPOSITORY> has the same meaning as the appropriate C<git-ls-remote>
740 argument; either a URL or a remote name (if called on a repository instance).
741 C<GROUPS> is an optional arrayref that can contain 'tags' to return all the
742 tags and/or 'heads' to return all the heads. C<REFGLOB> is an optional array
743 of strings containing a shell-like glob to further limit the refs returned in
744 the hash; the meaning is again the same as the appropriate C<git-ls-remote>
745 argument.
746
747 This function may or may not be called on a repository instance. In the former
748 case, remote names as defined in the repository are recognized as repository
749 specifiers.
750
751 =cut
752
753 sub remote_refs {
754         my ($self, $repo, $groups, $refglobs) = _maybe_self(@_);
755         my @args;
756         if (ref $groups eq 'ARRAY') {
757                 foreach (@$groups) {
758                         if ($_ eq 'heads') {
759                                 push (@args, '--heads');
760                         } elsif ($_ eq 'tags') {
761                                 push (@args, '--tags');
762                         } else {
763                                 # Ignore unknown groups for future
764                                 # compatibility
765                         }
766                 }
767         }
768         push (@args, $repo);
769         if (ref $refglobs eq 'ARRAY') {
770                 push (@args, @$refglobs);
771         }
772
773         my @self = $self ? ($self) : (); # Ultra trickery
774         my ($fh, $ctx) = Git::command_output_pipe(@self, 'ls-remote', @args);
775         my %refs;
776         while (<$fh>) {
777                 chomp;
778                 my ($hash, $ref) = split(/\t/, $_, 2);
779                 $refs{$ref} = $hash;
780         }
781         Git::command_close_pipe(@self, $fh, $ctx);
782         return \%refs;
783 }
784
785
786 =item ident ( TYPE | IDENTSTR )
787
788 =item ident_person ( TYPE | IDENTSTR | IDENTARRAY )
789
790 This suite of functions retrieves and parses ident information, as stored
791 in the commit and tag objects or produced by C<var GIT_type_IDENT> (thus
792 C<TYPE> can be either I<author> or I<committer>; case is insignificant).
793
794 The C<ident> method retrieves the ident information from C<git var>
795 and either returns it as a scalar string or as an array with the fields parsed.
796 Alternatively, it can take a prepared ident string (e.g. from the commit
797 object) and just parse it.
798
799 C<ident_person> returns the person part of the ident - name and email;
800 it can take the same arguments as C<ident> or the array returned by C<ident>.
801
802 The synopsis is like:
803
804         my ($name, $email, $time_tz) = ident('author');
805         "$name <$email>" eq ident_person('author');
806         "$name <$email>" eq ident_person($name);
807         $time_tz =~ /^\d+ [+-]\d{4}$/;
808
809 =cut
810
811 sub ident {
812         my ($self, $type) = _maybe_self(@_);
813         my $identstr;
814         if (lc $type eq lc 'committer' or lc $type eq lc 'author') {
815                 my @cmd = ('var', 'GIT_'.uc($type).'_IDENT');
816                 unshift @cmd, $self if $self;
817                 $identstr = command_oneline(@cmd);
818         } else {
819                 $identstr = $type;
820         }
821         if (wantarray) {
822                 return $identstr =~ /^(.*) <(.*)> (\d+ [+-]\d{4})$/;
823         } else {
824                 return $identstr;
825         }
826 }
827
828 sub ident_person {
829         my ($self, @ident) = _maybe_self(@_);
830         $#ident == 0 and @ident = $self ? $self->ident($ident[0]) : ident($ident[0]);
831         return "$ident[0] <$ident[1]>";
832 }
833
834
835 =item hash_object ( TYPE, FILENAME )
836
837 Compute the SHA1 object id of the given C<FILENAME> considering it is
838 of the C<TYPE> object type (C<blob>, C<commit>, C<tree>).
839
840 The method can be called without any instance or on a specified Git repository,
841 it makes zero difference.
842
843 The function returns the SHA1 hash.
844
845 =cut
846
847 # TODO: Support for passing FILEHANDLE instead of FILENAME
848 sub hash_object {
849         my ($self, $type, $file) = _maybe_self(@_);
850         command_oneline('hash-object', '-t', $type, $file);
851 }
852
853
854 =item hash_and_insert_object ( FILENAME )
855
856 Compute the SHA1 object id of the given C<FILENAME> and add the object to the
857 object database.
858
859 The function returns the SHA1 hash.
860
861 =cut
862
863 # TODO: Support for passing FILEHANDLE instead of FILENAME
864 sub hash_and_insert_object {
865         my ($self, $filename) = @_;
866
867         carp "Bad filename \"$filename\"" if $filename =~ /[\r\n]/;
868
869         $self->_open_hash_and_insert_object_if_needed();
870         my ($in, $out) = ($self->{hash_object_in}, $self->{hash_object_out});
871
872         unless (print $out $filename, "\n") {
873                 $self->_close_hash_and_insert_object();
874                 throw Error::Simple("out pipe went bad");
875         }
876
877         chomp(my $hash = <$in>);
878         unless (defined($hash)) {
879                 $self->_close_hash_and_insert_object();
880                 throw Error::Simple("in pipe went bad");
881         }
882
883         return $hash;
884 }
885
886 sub _open_hash_and_insert_object_if_needed {
887         my ($self) = @_;
888
889         return if defined($self->{hash_object_pid});
890
891         ($self->{hash_object_pid}, $self->{hash_object_in},
892          $self->{hash_object_out}, $self->{hash_object_ctx}) =
893                 $self->command_bidi_pipe(qw(hash-object -w --stdin-paths --no-filters));
894 }
895
896 sub _close_hash_and_insert_object {
897         my ($self) = @_;
898
899         return unless defined($self->{hash_object_pid});
900
901         my @vars = map { 'hash_object_' . $_ } qw(pid in out ctx);
902
903         command_close_bidi_pipe(@$self{@vars});
904         delete @$self{@vars};
905 }
906
907 =item cat_blob ( SHA1, FILEHANDLE )
908
909 Prints the contents of the blob identified by C<SHA1> to C<FILEHANDLE> and
910 returns the number of bytes printed.
911
912 =cut
913
914 sub cat_blob {
915         my ($self, $sha1, $fh) = @_;
916
917         $self->_open_cat_blob_if_needed();
918         my ($in, $out) = ($self->{cat_blob_in}, $self->{cat_blob_out});
919
920         unless (print $out $sha1, "\n") {
921                 $self->_close_cat_blob();
922                 throw Error::Simple("out pipe went bad");
923         }
924
925         my $description = <$in>;
926         if ($description =~ / missing$/) {
927                 carp "$sha1 doesn't exist in the repository";
928                 return -1;
929         }
930
931         if ($description !~ /^[0-9a-fA-F]{40} \S+ (\d+)$/) {
932                 carp "Unexpected result returned from git cat-file";
933                 return -1;
934         }
935
936         my $size = $1;
937
938         my $blob;
939         my $bytesRead = 0;
940
941         while (1) {
942                 my $bytesLeft = $size - $bytesRead;
943                 last unless $bytesLeft;
944
945                 my $bytesToRead = $bytesLeft < 1024 ? $bytesLeft : 1024;
946                 my $read = read($in, $blob, $bytesToRead, $bytesRead);
947                 unless (defined($read)) {
948                         $self->_close_cat_blob();
949                         throw Error::Simple("in pipe went bad");
950                 }
951
952                 $bytesRead += $read;
953         }
954
955         # Skip past the trailing newline.
956         my $newline;
957         my $read = read($in, $newline, 1);
958         unless (defined($read)) {
959                 $self->_close_cat_blob();
960                 throw Error::Simple("in pipe went bad");
961         }
962         unless ($read == 1 && $newline eq "\n") {
963                 $self->_close_cat_blob();
964                 throw Error::Simple("didn't find newline after blob");
965         }
966
967         unless (print $fh $blob) {
968                 $self->_close_cat_blob();
969                 throw Error::Simple("couldn't write to passed in filehandle");
970         }
971
972         return $size;
973 }
974
975 sub _open_cat_blob_if_needed {
976         my ($self) = @_;
977
978         return if defined($self->{cat_blob_pid});
979
980         ($self->{cat_blob_pid}, $self->{cat_blob_in},
981          $self->{cat_blob_out}, $self->{cat_blob_ctx}) =
982                 $self->command_bidi_pipe(qw(cat-file --batch));
983 }
984
985 sub _close_cat_blob {
986         my ($self) = @_;
987
988         return unless defined($self->{cat_blob_pid});
989
990         my @vars = map { 'cat_blob_' . $_ } qw(pid in out ctx);
991
992         command_close_bidi_pipe(@$self{@vars});
993         delete @$self{@vars};
994 }
995
996
997 { # %TEMP_* Lexical Context
998
999 my (%TEMP_FILEMAP, %TEMP_FILES);
1000
1001 =item temp_acquire ( NAME )
1002
1003 Attempts to retreive the temporary file mapped to the string C<NAME>. If an
1004 associated temp file has not been created this session or was closed, it is
1005 created, cached, and set for autoflush and binmode.
1006
1007 Internally locks the file mapped to C<NAME>. This lock must be released with
1008 C<temp_release()> when the temp file is no longer needed. Subsequent attempts
1009 to retrieve temporary files mapped to the same C<NAME> while still locked will
1010 cause an error. This locking mechanism provides a weak guarantee and is not
1011 threadsafe. It does provide some error checking to help prevent temp file refs
1012 writing over one another.
1013
1014 In general, the L<File::Handle> returned should not be closed by consumers as
1015 it defeats the purpose of this caching mechanism. If you need to close the temp
1016 file handle, then you should use L<File::Temp> or another temp file faculty
1017 directly. If a handle is closed and then requested again, then a warning will
1018 issue.
1019
1020 =cut
1021
1022 sub temp_acquire {
1023         my $temp_fd = _temp_cache(@_);
1024
1025         $TEMP_FILES{$temp_fd}{locked} = 1;
1026         $temp_fd;
1027 }
1028
1029 =item temp_release ( NAME )
1030
1031 =item temp_release ( FILEHANDLE )
1032
1033 Releases a lock acquired through C<temp_acquire()>. Can be called either with
1034 the C<NAME> mapping used when acquiring the temp file or with the C<FILEHANDLE>
1035 referencing a locked temp file.
1036
1037 Warns if an attempt is made to release a file that is not locked.
1038
1039 The temp file will be truncated before being released. This can help to reduce
1040 disk I/O where the system is smart enough to detect the truncation while data
1041 is in the output buffers. Beware that after the temp file is released and
1042 truncated, any operations on that file may fail miserably until it is
1043 re-acquired. All contents are lost between each release and acquire mapped to
1044 the same string.
1045
1046 =cut
1047
1048 sub temp_release {
1049         my ($self, $temp_fd, $trunc) = _maybe_self(@_);
1050
1051         if (exists $TEMP_FILEMAP{$temp_fd}) {
1052                 $temp_fd = $TEMP_FILES{$temp_fd};
1053         }
1054         unless ($TEMP_FILES{$temp_fd}{locked}) {
1055                 carp "Attempt to release temp file '",
1056                         $temp_fd, "' that has not been locked";
1057         }
1058         temp_reset($temp_fd) if $trunc and $temp_fd->opened;
1059
1060         $TEMP_FILES{$temp_fd}{locked} = 0;
1061         undef;
1062 }
1063
1064 sub _temp_cache {
1065         my ($self, $name) = _maybe_self(@_);
1066
1067         _verify_require();
1068
1069         my $temp_fd = \$TEMP_FILEMAP{$name};
1070         if (defined $$temp_fd and $$temp_fd->opened) {
1071                 if ($TEMP_FILES{$$temp_fd}{locked}) {
1072                         throw Error::Simple("Temp file with moniker '" .
1073                                 $name . "' already in use");
1074                 }
1075         } else {
1076                 if (defined $$temp_fd) {
1077                         # then we're here because of a closed handle.
1078                         carp "Temp file '", $name,
1079                                 "' was closed. Opening replacement.";
1080                 }
1081                 my $fname;
1082
1083                 my $tmpdir;
1084                 if (defined $self) {
1085                         $tmpdir = $self->repo_path();
1086                 }
1087
1088                 ($$temp_fd, $fname) = File::Temp->tempfile(
1089                         'Git_XXXXXX', UNLINK => 1, DIR => $tmpdir,
1090                         ) or throw Error::Simple("couldn't open new temp file");
1091
1092                 $$temp_fd->autoflush;
1093                 binmode $$temp_fd;
1094                 $TEMP_FILES{$$temp_fd}{fname} = $fname;
1095         }
1096         $$temp_fd;
1097 }
1098
1099 sub _verify_require {
1100         eval { require File::Temp; require File::Spec; };
1101         $@ and throw Error::Simple($@);
1102 }
1103
1104 =item temp_reset ( FILEHANDLE )
1105
1106 Truncates and resets the position of the C<FILEHANDLE>.
1107
1108 =cut
1109
1110 sub temp_reset {
1111         my ($self, $temp_fd) = _maybe_self(@_);
1112
1113         truncate $temp_fd, 0
1114                 or throw Error::Simple("couldn't truncate file");
1115         sysseek($temp_fd, 0, SEEK_SET) and seek($temp_fd, 0, SEEK_SET)
1116                 or throw Error::Simple("couldn't seek to beginning of file");
1117         sysseek($temp_fd, 0, SEEK_CUR) == 0 and tell($temp_fd) == 0
1118                 or throw Error::Simple("expected file position to be reset");
1119 }
1120
1121 =item temp_path ( NAME )
1122
1123 =item temp_path ( FILEHANDLE )
1124
1125 Returns the filename associated with the given tempfile.
1126
1127 =cut
1128
1129 sub temp_path {
1130         my ($self, $temp_fd) = _maybe_self(@_);
1131
1132         if (exists $TEMP_FILEMAP{$temp_fd}) {
1133                 $temp_fd = $TEMP_FILEMAP{$temp_fd};
1134         }
1135         $TEMP_FILES{$temp_fd}{fname};
1136 }
1137
1138 sub END {
1139         unlink values %TEMP_FILEMAP if %TEMP_FILEMAP;
1140 }
1141
1142 } # %TEMP_* Lexical Context
1143
1144 =back
1145
1146 =head1 ERROR HANDLING
1147
1148 All functions are supposed to throw Perl exceptions in case of errors.
1149 See the L<Error> module on how to catch those. Most exceptions are mere
1150 L<Error::Simple> instances.
1151
1152 However, the C<command()>, C<command_oneline()> and C<command_noisy()>
1153 functions suite can throw C<Git::Error::Command> exceptions as well: those are
1154 thrown when the external command returns an error code and contain the error
1155 code as well as access to the captured command's output. The exception class
1156 provides the usual C<stringify> and C<value> (command's exit code) methods and
1157 in addition also a C<cmd_output> method that returns either an array or a
1158 string with the captured command output (depending on the original function
1159 call context; C<command_noisy()> returns C<undef>) and $<cmdline> which
1160 returns the command and its arguments (but without proper quoting).
1161
1162 Note that the C<command_*_pipe()> functions cannot throw this exception since
1163 it has no idea whether the command failed or not. You will only find out
1164 at the time you C<close> the pipe; if you want to have that automated,
1165 use C<command_close_pipe()>, which can throw the exception.
1166
1167 =cut
1168
1169 {
1170         package Git::Error::Command;
1171
1172         @Git::Error::Command::ISA = qw(Error);
1173
1174         sub new {
1175                 my $self = shift;
1176                 my $cmdline = '' . shift;
1177                 my $value = 0 + shift;
1178                 my $outputref = shift;
1179                 my(@args) = ();
1180
1181                 local $Error::Depth = $Error::Depth + 1;
1182
1183                 push(@args, '-cmdline', $cmdline);
1184                 push(@args, '-value', $value);
1185                 push(@args, '-outputref', $outputref);
1186
1187                 $self->SUPER::new(-text => 'command returned error', @args);
1188         }
1189
1190         sub stringify {
1191                 my $self = shift;
1192                 my $text = $self->SUPER::stringify;
1193                 $self->cmdline() . ': ' . $text . ': ' . $self->value() . "\n";
1194         }
1195
1196         sub cmdline {
1197                 my $self = shift;
1198                 $self->{'-cmdline'};
1199         }
1200
1201         sub cmd_output {
1202                 my $self = shift;
1203                 my $ref = $self->{'-outputref'};
1204                 defined $ref or undef;
1205                 if (ref $ref eq 'ARRAY') {
1206                         return @$ref;
1207                 } else { # SCALAR
1208                         return $$ref;
1209                 }
1210         }
1211 }
1212
1213 =over 4
1214
1215 =item git_cmd_try { CODE } ERRMSG
1216
1217 This magical statement will automatically catch any C<Git::Error::Command>
1218 exceptions thrown by C<CODE> and make your program die with C<ERRMSG>
1219 on its lips; the message will have %s substituted for the command line
1220 and %d for the exit status. This statement is useful mostly for producing
1221 more user-friendly error messages.
1222
1223 In case of no exception caught the statement returns C<CODE>'s return value.
1224
1225 Note that this is the only auto-exported function.
1226
1227 =cut
1228
1229 sub git_cmd_try(&$) {
1230         my ($code, $errmsg) = @_;
1231         my @result;
1232         my $err;
1233         my $array = wantarray;
1234         try {
1235                 if ($array) {
1236                         @result = &$code;
1237                 } else {
1238                         $result[0] = &$code;
1239                 }
1240         } catch Git::Error::Command with {
1241                 my $E = shift;
1242                 $err = $errmsg;
1243                 $err =~ s/\%s/$E->cmdline()/ge;
1244                 $err =~ s/\%d/$E->value()/ge;
1245                 # We can't croak here since Error.pm would mangle
1246                 # that to Error::Simple.
1247         };
1248         $err and croak $err;
1249         return $array ? @result : $result[0];
1250 }
1251
1252
1253 =back
1254
1255 =head1 COPYRIGHT
1256
1257 Copyright 2006 by Petr Baudis E<lt>pasky@suse.czE<gt>.
1258
1259 This module is free software; it may be used, copied, modified
1260 and distributed under the terms of the GNU General Public Licence,
1261 either version 2, or (at your option) any later version.
1262
1263 =cut
1264
1265
1266 # Take raw method argument list and return ($obj, @args) in case
1267 # the method was called upon an instance and (undef, @args) if
1268 # it was called directly.
1269 sub _maybe_self {
1270         UNIVERSAL::isa($_[0], 'Git') ? @_ : (undef, @_);
1271 }
1272
1273 # Check if the command id is something reasonable.
1274 sub _check_valid_cmd {
1275         my ($cmd) = @_;
1276         $cmd =~ /^[a-z0-9A-Z_-]+$/ or throw Error::Simple("bad command: $cmd");
1277 }
1278
1279 # Common backend for the pipe creators.
1280 sub _command_common_pipe {
1281         my $direction = shift;
1282         my ($self, @p) = _maybe_self(@_);
1283         my (%opts, $cmd, @args);
1284         if (ref $p[0]) {
1285                 ($cmd, @args) = @{shift @p};
1286                 %opts = ref $p[0] ? %{$p[0]} : @p;
1287         } else {
1288                 ($cmd, @args) = @p;
1289         }
1290         _check_valid_cmd($cmd);
1291
1292         my $fh;
1293         if ($^O eq 'MSWin32') {
1294                 # ActiveState Perl
1295                 #defined $opts{STDERR} and
1296                 #       warn 'ignoring STDERR option - running w/ ActiveState';
1297                 $direction eq '-|' or
1298                         die 'input pipe for ActiveState not implemented';
1299                 # the strange construction with *ACPIPE is just to
1300                 # explain the tie below that we want to bind to
1301                 # a handle class, not scalar. It is not known if
1302                 # it is something specific to ActiveState Perl or
1303                 # just a Perl quirk.
1304                 tie (*ACPIPE, 'Git::activestate_pipe', $cmd, @args);
1305                 $fh = *ACPIPE;
1306
1307         } else {
1308                 my $pid = open($fh, $direction);
1309                 if (not defined $pid) {
1310                         throw Error::Simple("open failed: $!");
1311                 } elsif ($pid == 0) {
1312                         if (defined $opts{STDERR}) {
1313                                 close STDERR;
1314                         }
1315                         if ($opts{STDERR}) {
1316                                 open (STDERR, '>&', $opts{STDERR})
1317                                         or die "dup failed: $!";
1318                         }
1319                         _cmd_exec($self, $cmd, @args);
1320                 }
1321         }
1322         return wantarray ? ($fh, join(' ', $cmd, @args)) : $fh;
1323 }
1324
1325 # When already in the subprocess, set up the appropriate state
1326 # for the given repository and execute the git command.
1327 sub _cmd_exec {
1328         my ($self, @args) = @_;
1329         _setup_git_cmd_env($self);
1330         _execv_git_cmd(@args);
1331         die qq[exec "@args" failed: $!];
1332 }
1333
1334 # set up the appropriate state for git command
1335 sub _setup_git_cmd_env {
1336         my $self = shift;
1337         if ($self) {
1338                 $self->repo_path() and $ENV{'GIT_DIR'} = $self->repo_path();
1339                 $self->repo_path() and $self->wc_path()
1340                         and $ENV{'GIT_WORK_TREE'} = $self->wc_path();
1341                 $self->wc_path() and chdir($self->wc_path());
1342                 $self->wc_subdir() and chdir($self->wc_subdir());
1343         }
1344 }
1345
1346 # Execute the given Git command ($_[0]) with arguments ($_[1..])
1347 # by searching for it at proper places.
1348 sub _execv_git_cmd { exec('git', @_); }
1349
1350 # Close pipe to a subprocess.
1351 sub _cmd_close {
1352         my ($fh, $ctx) = @_;
1353         if (not close $fh) {
1354                 if ($!) {
1355                         # It's just close, no point in fatalities
1356                         carp "error closing pipe: $!";
1357                 } elsif ($? >> 8) {
1358                         # The caller should pepper this.
1359                         throw Git::Error::Command($ctx, $? >> 8);
1360                 }
1361                 # else we might e.g. closed a live stream; the command
1362                 # dying of SIGPIPE would drive us here.
1363         }
1364 }
1365
1366
1367 sub DESTROY {
1368         my ($self) = @_;
1369         $self->_close_hash_and_insert_object();
1370         $self->_close_cat_blob();
1371 }
1372
1373
1374 # Pipe implementation for ActiveState Perl.
1375
1376 package Git::activestate_pipe;
1377 use strict;
1378
1379 sub TIEHANDLE {
1380         my ($class, @params) = @_;
1381         # FIXME: This is probably horrible idea and the thing will explode
1382         # at the moment you give it arguments that require some quoting,
1383         # but I have no ActiveState clue... --pasky
1384         # Let's just hope ActiveState Perl does at least the quoting
1385         # correctly.
1386         my @data = qx{git @params};
1387         bless { i => 0, data => \@data }, $class;
1388 }
1389
1390 sub READLINE {
1391         my $self = shift;
1392         if ($self->{i} >= scalar @{$self->{data}}) {
1393                 return undef;
1394         }
1395         my $i = $self->{i};
1396         if (wantarray) {
1397                 $self->{i} = $#{$self->{'data'}} + 1;
1398                 return splice(@{$self->{'data'}}, $i);
1399         }
1400         $self->{i} = $i + 1;
1401         return $self->{'data'}->[ $i ];
1402 }
1403
1404 sub CLOSE {
1405         my $self = shift;
1406         delete $self->{data};
1407         delete $self->{i};
1408 }
1409
1410 sub EOF {
1411         my $self = shift;
1412         return ($self->{i} >= scalar @{$self->{data}});
1413 }
1414
1415
1416 1; # Famous last words