git-annotate.perl

   1 #!/usr/bin/perl
   2 # Copyright 2006, Ryan Anderson <ryan@michonline.com>
   3 #
   4 # GPL v2 (See COPYING)
   5 #
   6 # This file is licensed under the GPL v2, or a later version
   7 # at the discretion of Linus Torvalds.
   8
   9 use warnings;
  10 use strict;
  11 use Getopt::Long;
  12 use POSIX qw(strftime gmtime);
  13 use File::Basename qw(basename dirname);
  14
  15 sub usage() {
  16         print STDERR "Usage: ${\basename $0} [-s] [-S revs-file] file [ revision ]
  17         -l, --long
  18                         Show long rev (Defaults off)
  19         -t, --time
  20                         Show raw timestamp (Defaults off)
  21         -r, --rename
  22                         Follow renames (Defaults on).
  23         -S, --rev-file revs-file
  24                         Use revs from revs-file instead of calling git-rev-list
  25         -h, --help
  26                         This message.
  27 ";
  28
  29         exit(1);
  30 }
  31
  32 our ($help, $longrev, $rename, $rawtime, $starting_rev, $rev_file) = (0, 0, 1);
  33
  34 my $rc = GetOptions(    "long|l" => \$longrev,
  35                         "time|t" => \$rawtime,
  36                         "help|h" => \$help,
  37                         "rename|r" => \$rename,
  38                         "rev-file|S=s" => \$rev_file);
  39 if (!$rc or $help or !@ARGV) {
  40         usage();
  41 }
  42
  43 my $filename = shift @ARGV;
  44 if (@ARGV) {
  45         $starting_rev = shift @ARGV;
  46 }
  47
  48 my @stack = (
  49         {
  50                 'rev' => defined $starting_rev ? $starting_rev : "HEAD",
  51                 'filename' => $filename,
  52         },
  53 );
  54
  55 our @filelines = ();
  56
  57 if (defined $starting_rev) {
  58         @filelines = git_cat_file($starting_rev, $filename);
  59 } else {
  60         open(F,"<",$filename)
  61                 or die "Failed to open filename: $!";
  62
  63         while(<F>) {
  64                 chomp;
  65                 push @filelines, $_;
  66         }
  67         close(F);
  68
  69 }
  70
  71 our %revs;
  72 our @revqueue;
  73 our $head;
  74
  75 my $revsprocessed = 0;
  76 while (my $bound = pop @stack) {
  77         my @revisions = git_rev_list($bound->{'rev'}, $bound->{'filename'});
  78         foreach my $revinst (@revisions) {
  79                 my ($rev, @parents) = @$revinst;
  80                 $head ||= $rev;
  81
  82                 if (!defined($rev)) {
  83                         $rev = "";
  84                 }
  85                 $revs{$rev}{'filename'} = $bound->{'filename'};
  86                 if (scalar @parents > 0) {
  87                         $revs{$rev}{'parents'} = \@parents;
  88                         next;
  89                 }
  90
  91                 if (!$rename) {
  92                         next;
  93                 }
  94
  95                 my $newbound = find_parent_renames($rev, $bound->{'filename'});
  96                 if ( exists $newbound->{'filename'} && $newbound->{'filename'} ne $bound->{'filename'}) {
  97                         push @stack, $newbound;
  98                         $revs{$rev}{'parents'} = [$newbound->{'rev'}];
  99                 }
 100         }
 101 }
 102 push @revqueue, $head;
 103 init_claim( defined $starting_rev ? $head : 'dirty');
 104 unless (defined $starting_rev) {
 105         my $diff = open_pipe("git","diff","HEAD", "--",$filename)
 106                 or die "Failed to call git diff to check for dirty state: $!";
 107
 108         _git_diff_parse($diff, [$head], "dirty", (
 109                                 'author' => gitvar_name("GIT_AUTHOR_IDENT"),
 110                                 'author_date' => sprintf("%s +0000",time()),
 111                                 )
 112                         );
 113         close($diff);
 114 }
 115 handle_rev();
 116
 117
 118 my $i = 0;
 119 foreach my $l (@filelines) {
 120         my ($output, $rev, $committer, $date);
 121         if (ref $l eq 'ARRAY') {
 122                 ($output, $rev, $committer, $date) = @$l;
 123                 if (!$longrev && length($rev) > 8) {
 124                         $rev = substr($rev,0,8);
 125                 }
 126         } else {
 127                 $output = $l;
 128                 ($rev, $committer, $date) = ('unknown', 'unknown', 'unknown');
 129         }
 130
 131         printf("%s\t(%10s\t%10s\t%d)%s\n", $rev, $committer,
 132                 format_date($date), ++$i, $output);
 133 }
 134
 135 sub init_claim {
 136         my ($rev) = @_;
 137         for (my $i = 0; $i < @filelines; $i++) {
 138                 $filelines[$i] = [ $filelines[$i], '', '', '', 1];
 139                         # line,
 140                         # rev,
 141                         # author,
 142                         # date,
 143                         # 1 <-- belongs to the original file.
 144         }
 145         $revs{$rev}{'lines'} = \@filelines;
 146 }
 147
 148
 149 sub handle_rev {
 150         my $revseen = 0;
 151         my %seen;
 152         while (my $rev = shift @revqueue) {
 153                 next if $seen{$rev}++;
 154
 155                 my %revinfo = git_commit_info($rev);
 156
 157                 if (exists $revs{$rev}{parents} &&
 158                     scalar @{$revs{$rev}{parents}} != 0) {
 159
 160                         git_diff_parse($revs{$rev}{'parents'}, $rev, %revinfo);
 161                         push @revqueue, @{$revs{$rev}{'parents'}};
 162
 163                 } else {
 164                         # We must be at the initial rev here, so claim everything that is left.
 165                         for (my $i = 0; $i < @{$revs{$rev}{lines}}; $i++) {
 166                                 if (ref ${$revs{$rev}{lines}}[$i] eq '' || ${$revs{$rev}{lines}}[$i][1] eq '') {
 167                                         claim_line($i, $rev, $revs{$rev}{lines}, %revinfo);
 168                                 }
 169                         }
 170                 }
 171         }
 172 }
 173
 174
 175 sub git_rev_list {
 176         my ($rev, $file) = @_;
 177
 178         my $revlist;
 179         if ($rev_file) {
 180                 open($revlist, '<' . $rev_file)
 181                     or die "Failed to open $rev_file : $!";
 182         } else {
 183                 $revlist = open_pipe("git-rev-list","--parents","--remove-empty",$rev,"--",$file)
 184                         or die "Failed to exec git-rev-list: $!";
 185         }
 186
 187         my @revs;
 188         while(my $line = <$revlist>) {
 189                 chomp $line;
 190                 my ($rev, @parents) = split /\s+/, $line;
 191                 push @revs, [ $rev, @parents ];
 192         }
 193         close($revlist);
 194
 195         printf("0 revs found for rev %s (%s)\n", $rev, $file) if (@revs == 0);
 196         return @revs;
 197 }
 198
 199 sub find_parent_renames {
 200         my ($rev, $file) = @_;
 201
 202         my $patch = open_pipe("git-diff-tree", "-M50", "-r","--name-status", "-z","$rev")
 203                 or die "Failed to exec git-diff: $!";
 204
 205         local $/ = "\0";
 206         my %bound;
 207         my $junk = <$patch>;
 208         while (my $change = <$patch>) {
 209                 chomp $change;
 210                 my $filename = <$patch>;
 211                 if (!defined $filename) {
 212                         next;
 213                 }
 214                 chomp $filename;
 215
 216                 if ($change =~ m/^[AMD]$/ ) {
 217                         next;
 218                 } elsif ($change =~ m/^R/ ) {
 219                         my $oldfilename = $filename;
 220                         $filename = <$patch>;
 221                         chomp $filename;
 222                         if ( $file eq $filename ) {
 223                                 my $parent = git_find_parent($rev, $oldfilename);
 224                                 @bound{'rev','filename'} = ($parent, $oldfilename);
 225                                 last;
 226                         }
 227                 }
 228         }
 229         close($patch);
 230
 231         return \%bound;
 232 }
 233
 234
 235 sub git_find_parent {
 236         my ($rev, $filename) = @_;
 237
 238         my $revparent = open_pipe("git-rev-list","--remove-empty", "--parents","--max-count=1","$rev","--",$filename)
 239                 or die "Failed to open git-rev-list to find a single parent: $!";
 240
 241         my $parentline = <$revparent>;
 242         chomp $parentline;
 243         my ($revfound,$parent) = split m/\s+/, $parentline;
 244
 245         close($revparent);
 246
 247         return $parent;
 248 }
 249
 250 sub git_find_all_parents {
 251         my ($rev) = @_;
 252
 253         my $revparent = open_pipe("git-rev-list","--remove-empty", "--parents","--max-count=1","$rev")
 254                 or die "Failed to open git-rev-list to find a single parent: $!";
 255
 256         my $parentline = <$revparent>;
 257         chomp $parentline;
 258         my ($origrev, @parents) = split m/\s+/, $parentline;
 259
 260         close($revparent);
 261
 262         return @parents;
 263 }
 264
 265 sub git_merge_base {
 266         my ($rev1, $rev2) = @_;
 267
 268         my $mb = open_pipe("git-merge-base", $rev1, $rev2)
 269                 or die "Failed to open git-merge-base: $!";
 270
 271         my $base = <$mb>;
 272         chomp $base;
 273
 274         close($mb);
 275
 276         return $base;
 277 }
 278
 279 # Construct a set of pseudo parents that are in the same order,
 280 # and the same quantity as the real parents,
 281 # but whose SHA1s are as similar to the logical parents
 282 # as possible.
 283 sub get_pseudo_parents {
 284         my ($all, $fake) = @_;
 285
 286         my @all = @$all;
 287         my @fake = @$fake;
 288
 289         my @pseudo;
 290
 291         my %fake = map {$_ => 1} @fake;
 292         my %seenfake;
 293
 294         my $fakeidx = 0;
 295         foreach my $p (@all) {
 296                 if (exists $fake{$p}) {
 297                         if ($fake[$fakeidx] ne $p) {
 298                                 die sprintf("parent mismatch: %s != %s\nall:%s\nfake:%s\n",
 299                                             $fake[$fakeidx], $p,
 300                                             join(", ", @all),
 301                                             join(", ", @fake),
 302                                            );
 303                         }
 304
 305                         push @pseudo, $p;
 306                         $fakeidx++;
 307                         $seenfake{$p}++;
 308
 309                 } else {
 310                         my $base = git_merge_base($fake[$fakeidx], $p);
 311                         if ($base ne $fake[$fakeidx]) {
 312                                 die sprintf("Result of merge-base doesn't match fake: %s,%s != %s\n",
 313                                        $fake[$fakeidx], $p, $base);
 314                         }
 315
 316                         # The details of how we parse the diffs
 317                         # mean that we cannot have a duplicate
 318                         # revision in the list, so if we've already
 319                         # seen the revision we would normally add, just use
 320                         # the actual revision.
 321                         if ($seenfake{$base}) {
 322                                 push @pseudo, $p;
 323                         } else {
 324                                 push @pseudo, $base;
 325                                 $seenfake{$base}++;
 326                         }
 327                 }
 328         }
 329
 330         return @pseudo;
 331 }
 332
 333
 334 # Get a diff between the current revision and a parent.
 335 # Record the commit information that results.
 336 sub git_diff_parse {
 337         my ($parents, $rev, %revinfo) = @_;
 338
 339         my @pseudo_parents;
 340         my @command = ("git-diff-tree");
 341         my $revision_spec;
 342
 343         if (scalar @$parents == 1) {
 344
 345                 $revision_spec = join("..", $parents->[0], $rev);
 346                 @pseudo_parents = @$parents;
 347         } else {
 348                 my @all_parents = git_find_all_parents($rev);
 349
 350                 if (@all_parents !=  @$parents) {
 351                         @pseudo_parents = get_pseudo_parents(\@all_parents, $parents);
 352                 } else {
 353                         @pseudo_parents = @$parents;
 354                 }
 355
 356                 $revision_spec = $rev;
 357                 push @command, "-c";
 358         }
 359
 360         my @filenames = ( $revs{$rev}{'filename'} );
 361
 362         foreach my $parent (@$parents) {
 363                 push @filenames, $revs{$parent}{'filename'};
 364         }
 365
 366         push @command, "-p", "-M", $revision_spec, "--", @filenames;
 367
 368
 369         my $diff = open_pipe( @command )
 370                 or die "Failed to call git-diff for annotation: $!";
 371
 372         _git_diff_parse($diff, \@pseudo_parents, $rev, %revinfo);
 373
 374         close($diff);
 375 }
 376
 377 sub _git_diff_parse {
 378         my ($diff, $parents, $rev, %revinfo) = @_;
 379
 380         my $ri = 0;
 381
 382         my $slines = $revs{$rev}{'lines'};
 383         my (%plines, %pi);
 384
 385         my $gotheader = 0;
 386         my ($remstart);
 387         my $parent_count = @$parents;
 388
 389         my $diff_header_regexp = "^@";
 390         $diff_header_regexp .= "@" x @$parents;
 391         $diff_header_regexp .= ' -\d+,\d+' x @$parents;
 392         $diff_header_regexp .= ' \+(\d+),\d+';
 393         $diff_header_regexp .= " " . ("@" x @$parents);
 394
 395         my %claim_regexps;
 396         my $allparentplus = '^' . '\\+' x @$parents . '(.*)$';
 397
 398         {
 399                 my $i = 0;
 400                 foreach my $parent (@$parents) {
 401
 402                         $pi{$parent} = 0;
 403                         my $r = '^' . '.' x @$parents . '(.*)$';
 404                         my $p = $r;
 405                         substr($p,$i+1, 1) = '\\+';
 406
 407                         my $m = $r;
 408                         substr($m,$i+1, 1) = '-';
 409
 410                         $claim_regexps{$parent}{plus} = $p;
 411                         $claim_regexps{$parent}{minus} = $m;
 412
 413                         $plines{$parent} = [];
 414
 415                         $i++;
 416                 }
 417         }
 418
 419         DIFF:
 420         while(<$diff>) {
 421                 chomp;
 422                 #printf("%d:%s:\n", $gotheader, $_);
 423                 if (m/$diff_header_regexp/) {
 424                         $remstart = $1 - 1;
 425                         # (0-based arrays)
 426
 427                         $gotheader = 1;
 428
 429                         foreach my $parent (@$parents) {
 430                                 for (my $i = $ri; $i < $remstart; $i++) {
 431                                         $plines{$parent}[$pi{$parent}++] = $slines->[$i];
 432                                 }
 433                         }
 434                         $ri = $remstart;
 435
 436                         next DIFF;
 437
 438                 } elsif (!$gotheader) {
 439                         # Skip over the leadin.
 440                         next DIFF;
 441                 }
 442
 443                 if (m/^\\/) {
 444                         ;
 445                         # Skip \No newline at end of file.
 446                         # But this can be internationalized, so only look
 447                         # for an initial \
 448
 449                 } else {
 450                         my %claims = ();
 451                         my $negclaim = 0;
 452                         my $allclaimed = 0;
 453                         my $line;
 454
 455                         if (m/$allparentplus/) {
 456                                 claim_line($ri, $rev, $slines, %revinfo);
 457                                 $allclaimed = 1;
 458
 459                         }
 460
 461                         PARENT:
 462                         foreach my $parent (keys %claim_regexps) {
 463                                 my $m = $claim_regexps{$parent}{minus};
 464                                 my $p = $claim_regexps{$parent}{plus};
 465
 466                                 if (m/$m/) {
 467                                         $line = $1;
 468                                         $plines{$parent}[$pi{$parent}++] = [ $line, '', '', '', 0 ];
 469                                         $negclaim++;
 470
 471                                 } elsif (m/$p/) {
 472                                         $line = $1;
 473                                         if (get_line($slines, $ri) eq $line) {
 474                                                 # Found a match, claim
 475                                                 $claims{$parent}++;
 476
 477                                         } else {
 478                                                 die sprintf("Sync error: %d\n|%s\n|%s\n%s => %s\n",
 479                                                                 $ri, $line,
 480                                                                 get_line($slines, $ri),
 481                                                                 $rev, $parent);
 482                                         }
 483                                 }
 484                         }
 485
 486                         if (%claims) {
 487                                 foreach my $parent (@$parents) {
 488                                         next if $claims{$parent} || $allclaimed;
 489                                         $plines{$parent}[$pi{$parent}++] = $slines->[$ri];
 490                                             #[ $line, '', '', '', 0 ];
 491                                 }
 492                                 $ri++;
 493
 494                         } elsif ($negclaim) {
 495                                 next DIFF;
 496
 497                         } else {
 498                                 if (substr($_,scalar @$parents) ne get_line($slines,$ri) ) {
 499                                         foreach my $parent (@$parents) {
 500                                                 printf("parent %s is on line %d\n", $parent, $pi{$parent});
 501                                         }
 502
 503                                         my @context;
 504                                         for (my $i = -2; $i < 2; $i++) {
 505                                                 push @context, get_line($slines, $ri + $i);
 506                                         }
 507                                         my $context = join("\n", @context);
 508
 509                                         my $justline = substr($_, scalar @$parents);
 510                                         die sprintf("Line %d, does not match:\n|%s|\n|%s|\n%s\n",
 511                                                     $ri,
 512                                                     $justline,
 513                                                     $context);
 514                                 }
 515                                 foreach my $parent (@$parents) {
 516                                         $plines{$parent}[$pi{$parent}++] = $slines->[$ri];
 517                                 }
 518                                 $ri++;
 519                         }
 520                 }
 521         }
 522
 523         for (my $i = $ri; $i < @{$slines} ; $i++) {
 524                 foreach my $parent (@$parents) {
 525                         push @{$plines{$parent}}, $slines->[$ri];
 526                 }
 527                 $ri++;
 528         }
 529
 530         foreach my $parent (@$parents) {
 531                 $revs{$parent}{lines} = $plines{$parent};
 532         }
 533
 534         return;
 535 }
 536
 537 sub get_line {
 538         my ($lines, $index) = @_;
 539
 540         return ref $lines->[$index] ne '' ? $lines->[$index][0] : $lines->[$index];
 541 }
 542
 543 sub git_cat_file {
 544         my ($rev, $filename) = @_;
 545         return () unless defined $rev && defined $filename;
 546
 547         my $blob = git_ls_tree($rev, $filename);
 548         die "Failed to find a blob for $filename in rev $rev\n" if !defined $blob;
 549
 550         my $catfile = open_pipe("git","cat-file", "blob", $blob)
 551                 or die "Failed to git-cat-file blob $blob (rev $rev, file $filename): " . $!;
 552
 553         my @lines;
 554         while(<$catfile>) {
 555                 chomp;
 556                 push @lines, $_;
 557         }
 558         close($catfile);
 559
 560         return @lines;
 561 }
 562
 563 sub git_ls_tree {
 564         my ($rev, $filename) = @_;
 565
 566         my $lstree = open_pipe("git","ls-tree",$rev,$filename)
 567                 or die "Failed to call git ls-tree: $!";
 568
 569         my ($mode, $type, $blob, $tfilename);
 570         while(<$lstree>) {
 571                 chomp;
 572                 ($mode, $type, $blob, $tfilename) = split(/\s+/, $_, 4);
 573                 last if ($tfilename eq $filename);
 574         }
 575         close($lstree);
 576
 577         return $blob if ($tfilename eq $filename);
 578         die "git-ls-tree failed to find blob for $filename";
 579
 580 }
 581
 582
 583
 584 sub claim_line {
 585         my ($floffset, $rev, $lines, %revinfo) = @_;
 586         my $oline = get_line($lines, $floffset);
 587         @{$lines->[$floffset]} = ( $oline, $rev,
 588                 $revinfo{'author'}, $revinfo{'author_date'} );
 589         #printf("Claiming line %d with rev %s: '%s'\n",
 590         #               $floffset, $rev, $oline) if 1;
 591 }
 592
 593 sub git_commit_info {
 594         my ($rev) = @_;
 595         my $commit = open_pipe("git-cat-file", "commit", $rev)
 596                 or die "Failed to call git-cat-file: $!";
 597
 598         my %info;
 599         while(<$commit>) {
 600                 chomp;
 601                 last if (length $_ == 0);
 602
 603                 if (m/^author (.*) <(.*)> (.*)$/) {
 604                         $info{'author'} = $1;
 605                         $info{'author_email'} = $2;
 606                         $info{'author_date'} = $3;
 607                 } elsif (m/^committer (.*) <(.*)> (.*)$/) {
 608                         $info{'committer'} = $1;
 609                         $info{'committer_email'} = $2;
 610                         $info{'committer_date'} = $3;
 611                 }
 612         }
 613         close($commit);
 614
 615         return %info;
 616 }
 617
 618 sub format_date {
 619         if ($rawtime) {
 620                 return $_[0];
 621         }
 622         my ($timestamp, $timezone) = split(' ', $_[0]);
 623         my $minutes = abs($timezone);
 624         $minutes = int($minutes / 100) * 60 + ($minutes % 100);
 625         if ($timezone < 0) {
 626             $minutes = -$minutes;
 627         }
 628         my $t = $timestamp + $minutes * 60;
 629         return strftime("%Y-%m-%d %H:%M:%S " . $timezone, gmtime($t));
 630 }
 631
 632 # Copied from git-send-email.perl - We need a Git.pm module..
 633 sub gitvar {
 634     my ($var) = @_;
 635     my $fh;
 636     my $pid = open($fh, '-|');
 637     die "$!" unless defined $pid;
 638     if (!$pid) {
 639         exec('git-var', $var) or die "$!";
 640     }
 641     my ($val) = <$fh>;
 642     close $fh or die "$!";
 643     chomp($val);
 644     return $val;
 645 }
 646
 647 sub gitvar_name {
 648     my ($name) = @_;
 649     my $val = gitvar($name);
 650     my @field = split(/\s+/, $val);
 651     return join(' ', @field[0...(@field-4)]);
 652 }
 653
 654 sub open_pipe {
 655         if ($^O eq '##INSERT_ACTIVESTATE_STRING_HERE##') {
 656                 return open_pipe_activestate(@_);
 657         } else {
 658                 return open_pipe_normal(@_);
 659         }
 660 }
 661
 662 sub open_pipe_activestate {
 663         tie *fh, "Git::ActiveStatePipe", @_;
 664         return *fh;
 665 }
 666
 667 sub open_pipe_normal {
 668         my (@execlist) = @_;
 669
 670         my $pid = open my $kid, "-|";
 671         defined $pid or die "Cannot fork: $!";
 672
 673         unless ($pid) {
 674                 exec @execlist;
 675                 die "Cannot exec @execlist: $!";
 676         }
 677
 678         return $kid;
 679 }
 680
 681 package Git::ActiveStatePipe;
 682 use strict;
 683
 684 sub TIEHANDLE {
 685         my ($class, @params) = @_;
 686         my $cmdline = join " ", @params;
 687         my  @data = qx{$cmdline};
 688         bless { i => 0, data => \@data }, $class;
 689 }
 690
 691 sub READLINE {
 692         my $self = shift;
 693         if ($self->{i} >= scalar @{$self->{data}}) {
 694                 return undef;
 695         }
 696         return $self->{'data'}->[ $self->{i}++ ];
 697 }
 698
 699 sub CLOSE {
 700         my $self = shift;
 701         delete $self->{data};
 702         delete $self->{i};
 703 }
 704
 705 sub EOF {
 706         my $self = shift;
 707         return ($self->{i} >= scalar @{$self->{data}});
 708 }