chiark / gitweb /
wip tb-list
[topbloke.git] / Topbloke.pm
index f1c06154fc41dd1237249fc51c5eee107b68daba..f5c469fe48515c99abbb43696223133ad0769350 100644 (file)
@@ -5,6 +5,7 @@ use warnings;
 
 use POSIX;
 use IO::File;
+use IPC::Open2;
 
 package Topbloke;
 
@@ -14,12 +15,21 @@ BEGIN {
 
     $VERSION     = 1.00;
     @ISA         = qw(Exporter);
-    @EXPORT      = qw(parse_branch_spec current_tb_branch run_git_1line
-                      setup_config check_no_unwanted_metadata);
+    @EXPORT      = qw(debug
+                     run_git run_git_1line run_git_check_nooutput
+                     run_git_test_anyoutput
+                     git_config git_dir chdir_toplevel
+                     current_branch parse_patch_spec
+                     setup_config check_no_unwanted_metadata
+                     foreach_patch
+                     flagsfile_add_flag
+                     wf_start wf wf_abort wf_done wf_contents);
     %EXPORT_TAGS = ( );
     @EXPORT_OK   = qw();
 }
 
+our $git_command = 'git';
+
 sub debug ($) {
     my ($msg) = @_;
     print STDERR "DEBUG: $msg\n" or die $!;
@@ -43,18 +53,18 @@ sub run_git {
            die ref($ref)." @_ ?";
        }
     }
-    open GIT, "-|", 'git', @_ or die $!;
+    open GIT, "-|", $git_command, @_ or die $!;
     if ($linecallr) {
        while (<GIT>) {
-           chomp or die "$_ ?";
+           chomp or die "$git_command @_ gave $_ ?";
            $linecallr->();
        }
        GIT->eof or die $!;
     }
     if (!close GIT) {
-       die "git @_ $!" if $!;
+       die "$git_command @_ $!" if $!;
        die unless $?;
-       die "git @_ ($?)" unless $estatusr;
+       die "$git_command @_ ($?)" unless $estatusr;
        $$estatusr = $?;
     } else {
        $$estatusr = 0 if $estatusr;
@@ -79,6 +89,31 @@ sub run_git_test_anyoutput {
     return $any;
 }
 
+sub git_get_object ($) {
+    my ($objname) = @_;
+    our ($gro_pid, $gro_out, $gro_in);
+    if (!$gro_pid) {
+       $gro_pid = IPC::Open2::open2($gro_out, $gro_in,
+                                    $git_command, qw(cat-file --batch))
+           or die $!;
+    }
+    $SIG{'PIPE'} = 'IGN';
+    print $gro_in $objname,"\n" or die $!;
+    $gro_in->flush or die "$objname $!";
+    $SIG{'PIPE'} = 'DFL';
+    my $l = <$gro_out>;
+    chomp $l or die "$objname $l ?";
+    if ($l =~ m/ missing$/) {
+       return 'missing';
+    } elsif (my ($type,$bytes) = $l =~ m/^\S+ (\w+) (\d+)$/) {
+       my $data;
+       read $gro_out, $data, $bytes == $bytes or die "$objname $!";
+       return ($type, $data);
+    } else {
+       die "$objname $l";
+    }
+}
+
 sub git_config ($$) {
     my ($cfgvar, $default) = @_;
     my ($l, $estatus);
@@ -103,7 +138,15 @@ sub git_dir () {
     return $git_dir;
 }
 
-sub current_tb_branch () {
+sub chdir_toplevel () {
+    my $toplevel;
+    run_git(sub { $toplevel = $_; }, 
+           qw(rev-parse --show-toplevel));
+    die "not in working tree?\n" unless defined $toplevel;
+    chdir $toplevel or die "chdir toplevel $toplevel: $!\n";
+}
+
+sub current_branch () {
     open R, git_dir().'/HEAD' or die "open HEAD $!";
     my $ref = <R>;  defined $ref or die $!;
     close R;
@@ -138,13 +181,27 @@ sub current_tb_branch () {
     }
 }
 
-sub parse_branch_spec ($) {
+sub parse_patch_name ($) {
+    my ($patch) = @_;
+    my ($eaddr, $date, $nick) = split /\//, $patch, 3;
+    defined $nick && length $nick or die "$patch ?";
+    my ($email, $domain) = $eaddr =~ m/^(.*)\@([^\@]+)$/
+       or die "$patch eaddr ?";
+    return {
+       Email => $email,
+       Domain => $domain,
+       Date => $date,
+       Nick => $nick,
+    };
+}
+
+sub parse_patch_spec ($) {
     my ($orig) = @_;
     local $_ = $orig;
     my $spec = { }; # Email Domain DatePrefix DateNear Nick
     my $set = sub {
        my ($key,$val,$whats) = @_;
-       die "multiple $whats in branch spec\n" if exists $spec->{$key};
+       die "multiple $whats in patch spec\n" if exists $spec->{$key};
        $spec->{$key} = $val;
     };
     my $rel_levels;
@@ -176,17 +233,18 @@ sub parse_branch_spec ($) {
        }
     }
     if (defined $rel_levels) {
-       my $branch = current_tb_branch();
+       my $branch = current_branch();
        if (!defined $branch->{Nick}) {
-           die "relative branch spec \`$orig',".
-               " but current branch not a topbloke branch\n";
+           die "relative patch spec \`$orig',".
+               " but current branch not a topbloke patch\n";
        }
        my ($ceaddr,$cdate,@l) = split /\//, $branch->{Nick};
        @l >= $rel_levels or
-           die "relative branch spec \`$orig' has too many ../s\n";
+           die "relative patch spec \`$orig' has too many ../s\n";
        $_ = (join '/', @l[0..$#l-$rel_levels]).'/'.$_;
+    } elsif (length) {
+       $spec->{Nick} = $_;
     }
-    $spec->{Nick} = $_;
     return $spec;
 }
 
@@ -196,9 +254,10 @@ sub setup_config () {
     foreach my $iteration (qw(0 1)) {
        foreach my $file (@files) {
            my $cfgname = "merge.topbloke-$file";
-           my $current_estatus;
-           my $current = run_git_1line(\$current_estatus,
-                                       qw(config), "$cfgname.driver");
+           my ($current, $current_estatus);
+           run_git(\$current_estatus,
+                   sub { $current = $_; },
+                   qw(config), "$cfgname.driver");
            $current = "## failed $current_estatus" if $current_estatus;
            next if $current =~ m/^topbloke-merge-driver --v$version /o;
            die "$file $current ?" if $iteration;
@@ -250,4 +309,111 @@ sub check_no_unwanted_metadata ($) {
                           qw(.topbloke));
 }
 
+sub foreach_patch ($$$$) {
+    my ($spec, $deleted_ok, $want, $body) = @_;
+    # runs $body->($fullname, \%flags, \%deps, \%pflags, \%included)
+    #                        $Want->[0]   1        2         3
+    # where $deps->{$fullname} etc. are 1 for true or nonexistent for false
+    #  and if $want->[$item] is not true, the corresponding item may be undef
+    run_git(sub {
+       debug("foreach_patch considering $_");
+       m/ / or die "$_ ?";
+       my $objname = $`;
+       my @out;
+       my $patch = substr($',19); #');
+       push @out, $patch;
+       $want->[0] ||= !$deleted_ok;
+       foreach my $file (qw(flags deps pflags included)) {
+
+           if ($file eq 'deps') {
+               # do this check after checking for deleted patches,
+               # so we don't parse deleted patches' names
+               # right, check the spec next
+               if ($spec) {
+                   my $have = parse_patch_name($patch);
+                   foreach my $k (qw(Email Domain Nick)) {
+                       debug("foreach_patch  mismatch $k"), return
+                           if defined $spec->{$k} &&
+                              $have->{$k} ne $spec->{$k};
+                   }
+                   debug("foreach_patch  mismatch DatePrefix"), return
+                       if defined $spec->{DatePrefix} &&
+                          substr($have->{Date}, 0, length $spec->{DatePrefix})
+                              ne $spec->{DatePrefix};
+               }
+           }
+
+           if (!shift @$want) {
+               push @out, undef;
+               next;
+           }
+
+           my ($got, $data) = git_get_object("$objname:.topbloke/$file");
+           die "$patch $file ?" unless defined $data;
+           my %data;
+           $data{$_}=1 foreach split /\n/, $data;
+
+           if ($file eq 'flags') {
+               debug("foreach_patch  Deleted"), return
+                   if !$deleted_ok && $data{Deleted};
+           }
+
+           push @out, \%data;
+       }
+       debug("foreach_patch  YES @out"), return
+       $body->(@out);
+           },
+           qw(for-each-ref --format), '%(objectname) %(refname)',
+               qw(refs/topbloke-tips));
+}
+
+sub flagsfile_add_flag ($$) {
+    # works on "deps" too
+    my ($flagsfile, $flag) = @_;
+    my $wf = wf_start(".topbloke/$flagsfile");
+    open FI, '<', ".topbloke/$flagsfile" or die $!;
+    while (<FI>) {
+       chomp or die;
+       die "flag $flag already set in $flagsfile ?!" if $_ eq $flag;
+       wf($wf, "$_\n");
+    }
+    FI->error and die $!;
+    close FI or die $!;
+    wf($wf, "$flag\n");
+    wf_done($wf);
+}
+
+sub wf_start ($) {
+    my ($path) = @_;
+    my $fh = new IO::File "$path.tmp", '>' or die "create $path.tmp: $!\n";
+    return [ $fh, $path ];
+}
+
+sub wf ($$) {
+    my ($wf, $data) = @_;
+    my ($fh, $path) = @$wf;
+    print $fh $data or die "write $path.tmp: $!\n";
+}
+
+sub wf_abort ($) {
+    my ($wf) = @_;
+    my ($fh, $path) = @$wf;
+    close $fh;
+    unlink "$path.tmp" or die "remove $path.tmp: $!\n";
+}
+
+sub wf_done ($) {
+    my ($wf) = @_;
+    my ($fh, $path) = @$wf;
+    close $fh or die "finish writing $path.tmp: $!\n";
+    rename "$path.tmp", $path or die "install new $path: $!\n";
+}
+
+sub wf_contents ($$) {
+    my ($path,$contents) = @_;
+    my $wf = wf_start($path);
+    wf($wf, $contents);
+    wf_done($wf);
+}
+
 1;