chiark / gitweb /
fix a few bugs in THEORY
[topbloke.git] / Topbloke.pm
index b6de69022bd325306cdc6109394888c3507bebf6..5ea0ae73e178d5d1120e5f4f90279371e6621ad4 100644 (file)
@@ -1,13 +1,15 @@
 # -*- perl -*-
 
+package Topbloke;
+
 use strict;
 use warnings;
 
 use POSIX;
 use IO::File;
 use IPC::Open2;
-
-package Topbloke;
+use File::Path qw(make_path remove_tree);
+use File::Basename;
 
 BEGIN {
     use Exporter   ();
@@ -15,15 +17,17 @@ BEGIN {
 
     $VERSION     = 1.00;
     @ISA         = qw(Exporter);
-    @EXPORT      = qw(debug
+    @EXPORT      = qw(debug $tiprefs $baserefs %known_metadata
                      run_git run_git_1line run_git_check_nooutput
                      run_git_test_anyoutput git_get_object
-                     git_config git_dir chdir_toplevel
-                     current_branch parse_patch_spec parse_patch_name
-                     setup_config check_no_unwanted_metadata
+                     git_config git_dir chdir_toplevel enable_reflog
+                      check_no_metadata foreach_unknown_metadata
+                      check_clean_tree
+                     setup_config 
+                     current_branch parse_patch_name parse_patch_spec
                      patch_matches_spec
                      foreach_patch
-                     flagsfile_add_flag
+                     metafile_process depsfile_add_dep
                      wf_start wf wf_abort wf_done wf_contents
                      closeout);
     %EXPORT_TAGS = ( );
@@ -32,11 +36,15 @@ BEGIN {
 
 our $git_command = 'git';
 
+our %known_metadata;
+
 sub debug ($) {
     my ($msg) = @_;
     print STDERR "DEBUG: $msg\n" or die $!;
 }
 
+#----- general interaction with git -----
+
 sub run_git {
     # takes optional prefix arguments:
     #    coderef    hook to call for each line read,
@@ -95,8 +103,7 @@ 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))
+       $gro_pid = open2($gro_out, $gro_in, $git_command, qw(cat-file --batch))
            or die $!;
     }
     #debug("git_get_object $objname");
@@ -147,6 +154,8 @@ sub git_dir () {
     return $git_dir;
 }
 
+#----- specific interactions with git -----
+
 sub chdir_toplevel () {
     my $toplevel;
     run_git(sub { $toplevel = $_; }, 
@@ -155,6 +164,128 @@ sub chdir_toplevel () {
     chdir $toplevel or die "chdir toplevel $toplevel: $!\n";
 }
 
+sub enable_reflog ($) {
+    my ($branchref) = @_;
+    $branchref =~ m#^refs/# or die;
+    my $logsdir = git_dir().'/logs/';
+    my $dirname = $logsdir.dirname($branchref);
+    make_path($dirname) or die "$dirname $!";
+    open REFLOG, '>>', $logsdir.$branchref or die "$logsdir$branchref $!";
+    close REFLOG or die $!;
+}    
+
+sub check_no_metadata ($) {
+    # for checking foreign branches aren't contaminated
+    my ($gitbranch) = @_;
+    run_git_check_nooutput('foreign unexpectedly contains',
+                          qw(ls-tree --name-only),
+                          "$gitbranch:",
+                          qw(.topbloke));
+}
+
+sub foreach_unknown_metadata ($$) {
+    my ($ref, $code) = @_;
+    # Examines $ref.
+    # Executes $code for each tolerable unknown metadata found, with
+    # $_ being the (leaf) name of the metadata file
+    run_git(sub {
+       die unless s#^\.topbloke/##;
+       next if $known_metadata{$_};
+       m/-$/ or die "found unsupported metadata in $ref; you must upgrade\n";
+       $code->();
+           },
+           qw(ls-tree --name-only -r HEAD: .topbloke));
+}
+
+sub check_clean_tree ($) {
+    run_git_check_nooutput("operation requires working tree to be clean",
+                          qw(diff --name-only HEAD --));
+    run_git_check_nooutput("operation cannot proceed with staged changes",
+                          qw(diff --cached --name-only HEAD --));
+}
+
+$known_metadata{$_}=1 foreach qw(msg patch base deps deleted
+                                 +included +ends);
+
+#----- configuring a tree -----
+
+sub setup_config () {
+    my (@files) = (qw(lwildcard- msg patch base deps deleted
+                      +iwildcard- +included +ends));
+    my $version = 1;
+    my $drvname = sub {
+       my ($file) = @_;
+       $file =~ s/^\+//;
+       $file =~ s/\-$//;
+       return $file;
+    };
+    foreach my $iteration (qw(0 1)) {
+       foreach my $file (@files) {
+           my $cfgname = "merge.topbloke-".$drvname->($file);
+           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;
+           debug("setting merge driver $file");
+           run_git(qw(config), "$cfgname.name",
+                   "topbloke merge driver for $file");
+           run_git(qw(config), "$cfgname.driver",
+                   "topbloke-merge-driver --v$version".
+                   " $file %O %A %B %L");
+       }
+       my ($newattrsprefix, $newattrs, $attrsfile);
+
+       my $attrs = '';
+       my @needupdate;
+       foreach my $file (@files) {
+           my ($pat,$check) = ($file, $file);
+           if ($file =~ m/wildcard/) {
+               $pat = ($file =~ m/^\+/ ? '+' : '[^+]').'*';
+               $check =~ s/\w.*/xxxunknown/ or die;
+           }
+           $pat = ".topbloke/$pat";
+           $check = ".topbloke/$check";
+           my $want = "topbloke-".$drvname->($file);
+           $attrs .= "$pat\tmerge=$want\n";
+           my $current = run_git_1line(qw(check-attr merge), $check);
+           $current =~ s#^\Q$check\E: merge: ## or die "$file $current ?";
+           next if $current eq $want;
+           die "$file $current ?" unless 
+               $current eq 'unspecified' ||
+               $current =~ m/^topbloke-\wwildcard$/;
+           push @needupdate, "$file=$current";
+       }
+       if (@needupdate) {
+           $attrsfile = git_dir()."/info/attributes";
+           my $newattrsf = new IO::File "$attrsfile.tmp", 'w'
+                   or die "$attrsfile.tmp: $!";
+           die "@needupdate ?" if $iteration;
+           if (!open OA, '<', "$attrsfile") {
+               die "$attrsfile $!" unless $!==ENOENT;
+           } else {
+               while (<OA>) {
+                   next if m#^\.topbloke/#;
+                   print $newattrsf $_ or die $!;
+                   print $newattrsf "\n" or die $! unless chomp;
+               }
+               die $! if OA->error;
+               die $! unless close OA;
+           }
+           print $newattrsf $attrs or die $!;
+           close $newattrsf or die $!;
+           rename "$attrsfile.tmp", "$attrsfile" or die $!;
+       }
+    }
+}
+
+#----- branch and patch specs and parsed patch names -----
+
+our $tiprefs = 'refs/topbloke-tips';
+our $baserefs = 'refs/topbloke-bases';
+
 sub current_branch () {
     open R, git_dir().'/HEAD' or die "open HEAD $!";
     my $ref = <R>;  defined $ref or die $!;
@@ -168,7 +299,7 @@ sub current_branch () {
     }
     if ($ref =~ m#^refs/topbloke-(tip|base)s/([^/\@]*)\@([^/\@]*)/([^/]*)/#) {
        my $fullname = "$2\@$3/$4/$'";
-       return {
+       my $v = {
            Kind => $1,
            Email => $2,
            Domain => $3,
@@ -178,11 +309,12 @@ sub current_branch () {
            DepSpec => $fullname,
            Fullname => $fullname,
        };
+       return $v;
     } elsif ($ref =~ m#^refs/heads/#) {
        return {
            Kind => 'foreign',
            Ref => $ref,
-           DepSpec => "/$ref",
+           DepSpec => "$ref",
        };
     } else {
        return {
@@ -213,6 +345,7 @@ sub parse_patch_name ($) {
 sub parse_patch_spec ($) {
     my ($orig) = @_;
     local $_ = $orig;
+    warn 'FORMAT has new spec syntax nyi';
     my $spec = { }; # Email Domain DatePrefix DateNear Nick
     my $set = sub {
        my ($key,$val,$whats) = @_;
@@ -263,67 +396,6 @@ sub parse_patch_spec ($) {
     return $spec;
 }
 
-sub setup_config () {
-    my (@files) = (qw(msg deps included flags pflags));
-    my $version = 1;
-    foreach my $iteration (qw(0 1)) {
-       foreach my $file (@files) {
-           my $cfgname = "merge.topbloke-$file";
-           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;
-           debug("setting merge driver $file");
-           run_git(qw(config), "$cfgname.name",
-                   "topbloke merge driver for $file");
-           run_git(qw(config), "$cfgname.driver",
-                   "topbloke-merge-driver --v$version".
-                   " $file %O %A %B %L");
-       }
-       my ($newattrs, $attrsfile);
-       foreach my $file (@files) {
-           my $path = ".topbloke/$file";
-           my $current = run_git_1line(qw(check-attr merge), $path);
-           $current =~ s#^\Q$path\E: merge: ## or die "$file $current ?";
-           my $want = "topbloke-$file";
-           next if $current eq $want;
-           die "$file $current ?" unless $current eq 'unspecified';
-           die "$file $current ?" if $iteration;
-           if (!$newattrs) {
-               $attrsfile = git_dir()."/info/attributes";
-               $newattrs = new IO::File "$attrsfile.tmp", 'w'
-                   or die "$attrsfile.tmp: $!";
-               if (!open OA, '<', "$attrsfile") {
-                   die "$attrsfile $!" unless $!==&ENOENT;
-               } else {
-                   while (<OA>) {
-                       print $newattrs $_ or die $!;
-                       print "\n" or die $! unless chomp;
-                   }
-                   die $! if OA->error;
-                   die $! unless close OA;
-               }
-           }
-           print $newattrs "$path\tmerge=$want\n" or die $!;
-       }
-       last if !$newattrs;
-       close $newattrs or die $!;
-       rename "$attrsfile.tmp", "$attrsfile" or die $!;
-    }
-}
-
-sub check_no_unwanted_metadata ($) {
-    # for checking foreign branches aren't contaminated
-    my ($gitbranch) = @_;
-    run_git_check_nooutput('foreign unexpectedly contains',
-                          qw(ls-tree --name-only),
-                          "$gitbranch:",
-                          qw(.topbloke));
-}
-
 sub patch_matches_spec ($$) {
     my ($parsedname, $spec) = @_;
     foreach my $k (qw(Email Domain Nick)) {
@@ -338,76 +410,126 @@ sub patch_matches_spec ($$) {
     debug("patch_matches_spec  match"), return 1;
 }
 
+#----- reading topbloke metadata -----
+
 sub foreach_patch ($$$$) {
     my ($spec, $deleted_ok, $want, $body) = @_;
-    # runs $body->($patch, $parsedname, \%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
+    # runs $body->($patch, $parsedname, \%meta)
+    # where $meta{<metadata filename>} is, for <metadata filename> in @$want:
+    #              undefined if metadata file doesn't exist
+    #              defined with contents of file
     # and $parsedname is only valid if $spec is not undef
     #  (say $spec { }  if you want the name parsed but no restrictions)
+    # entries in want may also be "<metadata filename>_"
+    #  which means "strip trailing newlines" (result key in %meta is the same)
+    # <metadata filename> may instead be "B_<metadata filename>"
+    #  which means to look in the corresponding base branch
+    my @want = @$want;
+    my $atfront = sub {
+       my ($thing) = @_;
+       @want = ($thing, grep { $_ ne $thing } @want);
+    };
+    $atfront->(' patch');
+    $atfront->('deleted') unless $deleted_ok;
     run_git(sub {
        debug("foreach_patch considering $_");
        m/ / or die "$_ ?";
        my $objname = $`;
-       my @out;
+       my %meta;
+       my $parsedname;
        my $patch = substr($',19); #');
-       $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
+       my $wantix = 0;
+       foreach my $wantent (@want) {
+           my $file = $wantent;
+           my $stripnl = ($file =~ s/_$//);
+           my $key = $file;
+           my $inbase = ($file =~ s/^B_//);
+
+           if ($file eq ' patch') {
                if ($spec) {
-                   my $have = parse_patch_name($patch);
-                   debug("foreach_patch  mismatch"), return
-                       unless patch_matches_spec($have, $spec);
-                   unshift @out, $have;
-               } else {
-                   unshift @out, undef;
+                   $parsedname = parse_patch_name($patch);
+                   if (!patch_matches_spec($parsedname, $spec)) {
+                       debug("foreach_patch  mismatch");
+                       return;
+                   }
                }
-           }
-
-           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};
+           my $objkey = (!$inbase ? "$objname" : 
+                         "$baserefs/$patch").":.topbloke/$file";
+           my ($got, $data) = git_get_object($objkey);
+           if ($got eq 'missing') {
+               $meta{$key} = undef;
+           } elsif ($got eq 'blob') {
+               $meta{$key} = $data;
+               if ($file eq 'deleted' && !$deleted_ok) {
+                   debug("foreach_patch  Deleted");
+                   return;
+               }
+           } else {
+               warn "patch $patch object $objkey has unexpected type $got!\n";
+               return;
            }
-
-           push @out, \%data;
        }
-       debug("foreach_patch  YES @out"), return
-       $body->($patch, @out);
+       debug("foreach_patch  YES $patch");
+       $body->($patch, $parsedname, \%meta);
            },
            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");
+#----- updating topbloke metadata -----
+
+sub metafile_process ($$$$$) {
+    my ($metafile, $startcode, $linecode, $endcode, $enoentcode) = @_;
+    # runs $startcode->($outwf) at start
+    # runs $linecode->($outwf) for each old line, with $_ the chomped line
+    #   may modify $_, which will be written to $outf
+    # at end runs $endcode->($outwf);
+    # runs $enoentcode->($outwf) instead of ever calling $linecode
+    #  if the existing file does not exist;
+    #  if it's false dies instead
+    # any of these may return false, in which case we quit immediately
+    # any of these except enoentcode may be undef to mean "noop"
+    # if they all return true, we install the new file
+    my $wf = wf_start(".topbloke/$metafile");
+    my $call = sub {
+       return 1 unless $_->[0];
+       return 1 if $_->[0]($wf);
+       wf_abort($wf);
+       close FI;
+       return 0;
+    };
+    return unless $call->($startcode);
+    if (!open FI, '<', ".topbloke/$metafile") {
+       die "$metafile $!" unless $!==ENOENT;
+       die "$metafile $!" unless $enoentcode;
+       return unless $call->($enoentcode);
+    } else {
+       while (<FI>) {
+           chomp or die;
+           return unless $call->($linecode);
+           wf($wf, "$_\n");
+       }
+       FI->error and die $!;
+       close FI or die $!;
     }
-    FI->error and die $!;
-    close FI or die $!;
-    wf($wf, "$flag\n");
+    return unless $call->($endcode);
     wf_done($wf);
 }
+    
+
+sub depsfile_add_dep ($$) {
+    my ($depsfile, $depspec) = @_;
+    metafile_process($depsfile, undef, sub {
+       die "dep $depspec already set in $depsfile ?!" if $_ eq $depspec;
+    }, sub {
+       wf($_->[0], "$depspec\n");
+    }, undef);
+}
+
+#----- general utilities -----
 
 sub wf_start ($) {
     my ($path) = @_;