X-Git-Url: https://git.madduck.net/code/myrepos.git/blobdiff_plain/4c155e72e66ff5d1122b4c9f583514725308ad04..a80452ac1f76168ec147d8fabf742ddbb3d57326:/mr diff --git a/mr b/mr index 19c29aa..68772a0 100755 --- a/mr +++ b/mr @@ -22,17 +22,17 @@ B [options] log B [options] register [repository] -B [options] config section [parameter=[value] ...] +B [options] config section ["parameter=[value]" ...] B [options] action [params ...] =head1 DESCRIPTION -B is a Multiple Repository management tool. It -can checkout, update, or perform other actions on -a set of repositories as if they were one combined respository. It -supports any combination of subversion, git, cvs, and bzr repositories, -and support for other revision control systems can easily be added. +B is a Multiple Repository management tool. It can checkout, update, or +perform other actions on a set of repositories as if they were one combined +respository. It supports any combination of subversion, git, cvs, mecurial and +bzr repositories, and support for other revision control systems can easily be +added. B cds into and operates on all registered repositories at or below your working directory. Or, if you are in a subdirectory of a repository that @@ -86,16 +86,16 @@ List the repositories that mr will act on. =item register -Register an existing repository in the mrconfig file. By default, the +Register an existing repository in a mrconfig file. By default, the repository in the current directory is registered, or you can specify a directory to register. -By default it registers it to the ~/.mrconfig file. To make it write to a -different file, use the -c option. +The mrconfig file that is modified is chosen by either the -c option, or by +looking for the closest known one at or below the current directory. =item config -Adds, modifies, removes, or prints a value from the mrconfig file. The next +Adds, modifies, removes, or prints a value from a mrconfig file. The next parameter is the name of the section the value is in. To add or modify values, use one or more instances of "parameter=value". Use "parameter=" to remove a parameter. Use just "parameter" to get the value of a parameter. @@ -108,6 +108,13 @@ To show the command that mr uses to update the repository in src/foo: mr config src/foo update +To see the built-in library of shell functions contained in mr: + + mr config DEFAULT lib + +The ~/.mrconfig file is used by default. To use a different config file, +use the -c option. + =item help Displays this help. @@ -134,8 +141,7 @@ the current working directory. =item -c mrconfig -Use the specified mrconfig file, instead of looking for one in your home -directory. +Use the specified mrconfig file. The default is B<~/.mrconfig> =item -v @@ -151,6 +157,14 @@ about exactly which repositories failed and were skipped, if any. Just operate on the repository for the current directory, do not recurse into deeper repositories. +=item -j number + +Run the specified number of jobs in parallel. This can greatly speed up +operations such as updates. + +Note that in -j mode, all output of the jobs goes to standard output, even +output that would normally go to standard error. + =back =head1 FILES @@ -186,9 +200,9 @@ that contains the mrconfig file, but you can also choose to use absolute paths. Within a section, each parameter defines a shell command to run to handle a -given action. mr contains default handlers for the "update", "status", and -"commit" actions, so normally you only need to specify what to do for -"checkout". +given action. mr contains default handlers for "update", "status", +"commit", and other standard actions. Normally you only need to specify what +to do for "checkout". Note that these shell commands are run in a "set -e" shell environment, where any additional parameters you pass are available in @@ -197,9 +211,9 @@ directory, since the repository isn't checked out yet. All other commands are run inside the repository, though not necessarily at the top of it. The "MR_REPO" environment variable is set to the path to the top of the -repository, and "MR_CONFIG" is set to the .mrconfig file that defines the -repo being acted on, or, if the repo is not yet in a config file, the -.mrconfig file that mr thinks it should be added to. +repository. The "MR_CONFIG" environment variable is set to the .mrconfig file +that defines the repo being acted on, or, if the repo is not yet in a config +file, the .mrconfig file that should be modified to register the repo. A few parameters have special meanings: @@ -225,14 +239,6 @@ If the "chain" parameter is set and its command returns true, then B will try to load a .mrconfig file from the root of the repository. (You should avoid chaining from repositories with untrusted committers.) -=item deleted - -If the "deleted" parameter is set and its command returns true, then -B will treat the repository as deleted. It won't ever actually delete -the repository, but it will warn if it sees the repository's directory. -This is useful when one mrconfig file is shared amoung multiple machines, -to keep track of and remember to delete old repositories. - =item lib The "lib" parameter can specify some shell code that will be run before each @@ -257,6 +263,13 @@ use warnings; use strict; use Getopt::Long; use Cwd qw(getcwd abs_path); +use POSIX "WNOHANG"; +use constant { + OK => 0, + FAILED => 1, + SKIPPED => 2, + ABORT => 3, +}; $SIG{INT}=sub { print STDERR "mr: interrupted\n"; @@ -264,10 +277,12 @@ $SIG{INT}=sub { }; $ENV{MR_CONFIG}="$ENV{HOME}/.mrconfig"; +my $config_overridden=0; my $directory=getcwd(); my $verbose=0; my $stats=0; my $no_recurse=0; +my $jobs=1; my %config; my %configfiles; my %knownactions; @@ -276,10 +291,11 @@ my %alias; Getopt::Long::Configure("no_permute"); my $result=GetOptions( "d|directory=s" => sub { $directory=abs_path($_[1]) }, - "c|config=s" => \$ENV{MR_CONFIG}, + "c|config=s" => sub { $ENV{MR_CONFIG}=$_[1]; $config_overridden=1 }, "v|verbose" => \$verbose, "s|stats" => \$stats, "n|no-recurse" => \$no_recurse, + "j|jobs=i" => \$jobs, ); if (! $result || @ARGV < 1) { die("Usage: mr [-d directory] action [params ...]\n". @@ -293,17 +309,17 @@ if (! $result || @ARGV < 1) { if ($ENV{MR_CONFIG} !~ /^\//) { $ENV{MR_CONFIG}=getcwd()."/".$ENV{MR_CONFIG}; } +# Try to set MR_PATH to the path to the program. +eval { + use FindBin qw($Bin $Script); + $ENV{MR_PATH}=$Bin."/".$Script; +}; loadconfig(\*DATA); loadconfig($ENV{MR_CONFIG}); #use Data::Dumper; #print Dumper(\%config); -eval { - use FindBin qw($Bin $Script); - $ENV{MR_PATH}=$Bin."/".$Script; -}; - # alias expansion and command stemming my $action=shift @ARGV; if (exists $alias{$action}) { @@ -325,6 +341,7 @@ if (! exists $knownactions{$action}) { } } +# commands that do not operate on all repos if ($action eq 'help') { exec($config{''}{DEFAULT}{$action}) || die "exec: $!"; } @@ -354,6 +371,7 @@ elsif ($action eq 'config') { exists $config{$topdir}{$section}{$_}) { print $config{$topdir}{$section}{$_}."\n"; $found=1; + last if $section eq 'DEFAULT'; } } if (! $found) { @@ -365,6 +383,18 @@ elsif ($action eq 'config') { exit 0; } elsif ($action eq 'register') { + if (! $config_overridden) { + # Find the closest known mrconfig file to the current + # directory. + $directory.="/" unless $directory=~/\/$/; + foreach my $topdir (reverse sort keys %config) { + next unless length $topdir; + if ($directory=~/^\Q$topdir\E/) { + $ENV{MR_CONFIG}=$configfiles{$topdir}; + last; + } + } + } my $command="set -e; ".$config{''}{DEFAULT}{lib}."\n". "my_action(){ $config{''}{DEFAULT}{$action}\n }; my_action ". join(" ", map { s/\//\/\//g; s/"/\"/g; '"'.$_.'"' } @ARGV); @@ -405,10 +435,39 @@ if (! @repos) { $nochdir=1; } -my (@failed, @ok, @skipped); -foreach my $repo (@repos) { - action($action, @$repo); +# run the action on each repository and print stats +my (@ok, @failed, @skipped); +if ($jobs > 1) { + mrs(@repos); } +else { + foreach my $repo (@repos) { + record($repo, action($action, @$repo)); + } +} +if (! @ok && ! @failed && ! @skipped) { + die "mr $action: no repositories found to work on\n"; +} +print "mr $action: finished (".join("; ", + showstat($#ok+1, "ok", "ok"), + showstat($#failed+1, "failed", "failed"), + showstat($#skipped+1, "skipped", "skipped"), +).")\n"; +if ($stats) { + if (@skipped) { + print "mr $action: (skipped: ".join(" ", @skipped).")\n"; + } + if (@failed) { + print STDERR "mr $action: (failed: ".join(" ", @failed).")\n"; + } +} +if (@failed) { + exit 1; +} +elsif (! @ok && @skipped) { + exit 1; +} +exit 0; sub action { #{{{ my ($action, $dir, $topdir, $subdir) = @_; @@ -417,45 +476,17 @@ sub action { #{{{ my $lib=exists $config{$topdir}{$subdir}{lib} ? $config{$topdir}{$subdir}{lib}."\n" : ""; - if (exists $config{$topdir}{$subdir}{deleted}) { - my $test="set -e;".$lib.$config{$topdir}{$subdir}{deleted}; - print "mr $action: running deleted test >>$test<<\n" if $verbose; - my $ret=system($test); - if ($ret != 0) { - if (($? & 127) == 2) { - print STDERR "mr $action: interrupted\n"; - exit 2; - } - elsif ($? & 127) { - print STDERR "mr $action: deleted test received signal ".($? & 127)."\n"; - } - } - if ($ret >> 8 == 0) { - if (-d $dir) { - print STDERR "mr error: $dir should be deleted yet still exists\n\n"; - push @failed, $dir; - return; - } - else { - print "mr $action: $dir skipped (as deleted) per config file\n" if $verbose; - push @skipped, $dir; - return; - } - } - } - if ($action eq 'checkout') { if (-d $dir) { print "mr $action: $dir already exists, skipping checkout\n" if $verbose; - push @skipped, $dir; - return; + return SKIPPED; } $dir=~s/^(.*)\/[^\/]+\/?$/$1/; if (! -d $dir) { print "mr $action: creating parent directory $dir\n" if $verbose; - my $ret=system("mkdir", "-p", $dir); + system("mkdir", "-p", $dir); } } elsif ($action eq 'update') { @@ -474,27 +505,26 @@ sub action { #{{{ if ($ret != 0) { if (($? & 127) == 2) { print STDERR "mr $action: interrupted\n"; - exit 2; + return ABORT; } elsif ($? & 127) { print STDERR "mr $action: skip test received signal ".($? & 127)."\n"; - exit 1; + return ABORT; } } if ($ret >> 8 == 0) { print "mr $action: $dir skipped per config file\n" if $verbose; - push @skipped, $dir; - return; + return SKIPPED; } } if (! $nochdir && ! chdir($dir)) { print STDERR "mr $action: failed to chdir to $dir: $!\n"; - push @failed, $dir; + return FAILED; } elsif (! exists $config{$topdir}{$subdir}{$action}) { print STDERR "mr $action: no defined $action command for $topdir$subdir, skipping\n"; - push @skipped, $dir; + return SKIPPED; } else { if (! $nochdir) { @@ -508,34 +538,105 @@ sub action { #{{{ join(" ", map { s/\//\/\//g; s/"/\"/g; '"'.$_.'"' } @ARGV); print STDERR "mr $action: running >>$command<<\n" if $verbose; my $ret=system($command); + print "\n"; if ($ret != 0) { if (($? & 127) == 2) { print STDERR "mr $action: interrupted\n"; - exit 2; + return ABORT; } elsif ($? & 127) { print STDERR "mr $action: received signal ".($? & 127)."\n"; + return ABORT; } print STDERR "mr $action: failed ($ret)\n" if $verbose; - push @failed, $dir; if ($ret >> 8 != 0) { print STDERR "mr $action: command failed\n"; } elsif ($ret != 0) { print STDERR "mr $action: command died ($ret)\n"; } + return FAILED; } else { if ($action eq 'checkout' && ! -d $dir) { print STDERR "mr $action: $dir missing after checkout\n";; - push @failed, $dir; - return; + return FAILED; } - push @ok, $dir; + return OK; } + } +} #}}} - print "\n"; +# run actions on multiple repos, in parallel +sub mrs { #{{{ + $| = 1; + my @active; + my @fhs; + my @out; + my $running=0; + while (@fhs or @repos) { + while ($running < $jobs && @repos) { + $running++; + my $repo = shift @repos; + my $pid = open(my $fh, "-|"); + if (! $pid) { + open(STDERR, ">&STDOUT"); + exit action($action, @$repo); + } + push @active, $repo; + push @fhs, $fh; + push @out, ""; + } + my ($rin, $rout) = ('',''); + my $nfound; + foreach my $x (@fhs) { + next unless defined $x; + vec($rin, fileno($x), 1) = 1; + } + $nfound = select($rout=$rin, undef, undef, 1); + foreach my $i (0..$#fhs) { + my $fh = $fhs[$i]; + next unless defined $fh; + if (vec($rout, fileno($fh), 1) == 1) { + my $r = ''; + if (sysread($fh, $r, 1024) == 0) { + close($fh); + record($active[$i], $? >> 8); + $fhs[$i] = undef; + $running--; + print $out[$i]; + $out[$i]=''; + } + $out[$i] .= $r; + } + } + while (@fhs and !defined $fhs[0]) { + shift @active; + shift @fhs; + shift @out; + } + } +} #}}} + +sub record { #{{{ + my $dir=shift()->[0]; + my $ret=shift; + + if ($ret == OK) { + push @ok, $dir; + } + elsif ($ret == FAILED) { + push @failed, $dir; + } + elsif ($ret == SKIPPED) { + push @skipped, $dir; + } + elsif ($ret == ABORT) { + exit 1; + } + else { + die "unknown exit status $ret"; } } #}}} @@ -548,29 +649,6 @@ sub showstat { #{{{ } return; } #}}} -if (! @ok && ! @failed && ! @skipped) { - die "mr $action: no repositories found to work on\n"; -} -print "mr $action: finished (".join("; ", - showstat($#ok+1, "ok", "ok"), - showstat($#failed+1, "failed", "failed"), - showstat($#skipped+1, "skipped", "skipped"), -).")\n"; -if ($stats) { - if (@skipped) { - print "mr $action: (skipped: ".join(" ", @skipped).")\n"; - } - if (@failed) { - print STDERR "mr $action: (failed: ".join(" ", @failed).")\n"; - } -} -if (@failed) { - exit 1; -} -elsif (! @ok && @skipped) { - exit 1; -} -exit 0; my %loaded; sub loadconfig { #{{{ @@ -600,6 +678,10 @@ sub loadconfig { #{{{ $dir="."; } $dir=abs_path($dir)."/"; + + if (! exists $configfiles{$dir}) { + $configfiles{$dir}=$f; + } # copy in defaults from first parent my $parent=$dir; @@ -660,9 +742,6 @@ sub loadconfig { #{{{ else { $config{$dir}{$section}{$parameter}=$value; $knownactions{$parameter}=1; - if (! exists $configfiles{$dir}) { - $configfiles{$dir}=abs_path($f); - } if ($parameter eq 'chain' && length $dir && $section ne "DEFAULT" && -e $dir.$section."/.mrconfig") { @@ -805,7 +884,7 @@ lib = exit 1 } hours_since() { - for dir in .git .svn .bzr CVS; do + for dir in .git .svn .bzr CVS .hg; do if [ -e "$MR_REPO/$dir" ]; then flagfile="$MR_REPO/$dir/.mr_last$1" break @@ -822,11 +901,17 @@ update = if [ -d "$MR_REPO"/.svn ]; then svn update "$@" elif [ -d "$MR_REPO"/.git ]; then - git pull origin master "$@" + if [ -z "$@" ]; then + git pull -t origin master + else + git pull "$@" + fi elif [ -d "$MR_REPO"/.bzr ]; then bzr merge "$@" elif [ -d "$MR_REPO"/CVS ]; then cvs update "$@" + elif [ -d "$MR_REPO"/.hg ]; then + hg pull "$@" && hg update "$@" else error "unknown repo type" fi @@ -839,6 +924,8 @@ status = bzr status "$@" elif [ -d "$MR_REPO"/CVS ]; then cvs status "$@" + elif [ -d "$MR_REPO"/.hg ]; then + hg status "$@" else error "unknown repo type" fi @@ -851,6 +938,8 @@ commit = bzr commit "$@" && bzr push elif [ -d "$MR_REPO"/CVS ]; then cvs commit "$@" + elif [ -d "$MR_REPO"/.hg ]; then + hg commit -m "$@" && hg push else error "unknown repo type" fi @@ -863,6 +952,8 @@ diff = bzr diff "$@" elif [ -d "$MR_REPO"/CVS ]; then cvs diff "$@" + elif [ -d "$MR_REPO"/.hg ]; then + hg diff "$@" else error "unknown repo type" fi @@ -875,6 +966,8 @@ log = bzr log "$@" elif [ -d "$MR_REPO"/CVS ]; then cvs log "$@" + elif [ -d "$MR_REPO"/.hg ]; then + hg log "$@" else error "unknown repo type" fi @@ -913,6 +1006,11 @@ register = echo "Registering cvs repository $repo at root $root" mr -c "$MR_CONFIG" config "$(pwd)" \ checkout="cvs -d '$root' co -d $basedir $repo" + elif [ -d .hg ]; then + url=$(hg showconfig paths.default) + echo "Registering mercurial repo url: $url in $MR_CONFIG" + mr -c "$MR_CONFIG" config "$(pwd)" \ + checkout="hg clone $url $basedir" else error "unable to register this repo type" fi