make source and destination directory for diff

[bak-git.git] / bak-git-server.pl

diff --git a/bak-git-server.pl b/bak-git-server.pl
index ce4c099..5e656a0 100755 (executable)
--- a/bak-git-server.pl
+++ b/bak-git-server.pl
@@ -19,15 +19,19 @@ bak command overview:
 
   bak add /path
   bak commit [/path [message]]
-  bak diff [/path]
+  bak diff [host:][/path]
   bak status [/path]
   bak log [/path]
 
   bak ch[anges]
-  bak revert /path
+  bak revert [host:]/path
+
+  bak cat [host:]/path
 
   bak - push all changed files to server
 
+See L<http://blog.rot13.org/bak-git> for more information
+
 =cut
 
 use warnings;
@@ -51,7 +55,8 @@ $server_ip ||= '127.0.0.1';
 
 my $shell_client = <<__SHELL_CLIENT__;
 #!/bin/sh
-echo \$USER/\$SUDO_USER `hostname -s` `pwd` \$* | nc $server_ip 9001
+#echo \$USER/\$SUDO_USER $install `pwd` \$* | nc 127.0.0.1 9001
+echo \$USER/\$SUDO_USER `hostname` `pwd` \$* | nc $server_ip 9001
 __SHELL_CLIENT__
 
 chdir $dir;
@@ -87,23 +92,46 @@ warn "dir: $dir listen: $server_ip:9001\n"
        , $shell_client
 ;
 
+sub rsync {
+       warn "# rsync ",join(' ', @_), "\n";
+       system 'rsync', @_;
+}
+
 sub pull_changes {
        my $hostname = shift;
        system "find $hostname -type f | sed 's,$hostname,,' > /tmp/$hostname.list";
-       system "rsync -avv --files-from /tmp/$hostname.list root\@$hostname:/ $hostname/"
+       if ( @_ ) {
+               open(my $files, '>>', "/tmp/$hostname.list");
+               print $files "$_\n" foreach @_;
+               close($files);
+       }
+       rsync split / /, "-avv --files-from /tmp/$hostname.list root\@$hostname:/ $hostname/";
 }
 
 while (my $client = $server->accept()) {
        my $line = <$client>;
        chomp($line);
        warn "<<< $line\n";
-       my ($user,$hostname,$pwd,$command,$rel_path,$message) = split(/\s+/,$line,5);
-
+       my ($user,$hostname,$pwd,$command,$rel_path,$message) = split(/\s+/,$line,6);
+       $hostname =~ s/\..+$//;
+
+       my $on_host = '';
+       if ( $rel_path =~ s/^([^:]+):(.+)$/$2/ ) {
+               if ( -e $1 ) {
+                       $on_host = $1;
+               } else {
+                       print $client "host $1 doesn't exist in backup\n";
+                       next;
+               }
+       }
        my $path = $rel_path =~ m{^/} ? $rel_path : "$pwd/$rel_path";
 
-       $message ||= '';
-       warn "$hostname [$command] $path | $message\n";
-       $message ||= "$hostname [$command] $path";
+       warn "$hostname [$command] $on_host:$path | $message\n";
+
+       my $args_message = $message;
+
+       $message ||= "$path [$command]";
+       $message = "$hostname: $message";
 
        my $dir = $path;
        $dir =~ s{/[^/]+$}{};
@@ -122,8 +150,14 @@ while (my $client = $server->accept()) {
                pull_changes $hostname;
        } elsif ( $command eq 'add' ) {
                mkpath "$hostname/$dir" unless -e "$hostname/$dir";
-               system 'rsync', '-avv', "root\@$hostname:$path", "$hostname/$path";
-               print $client git 'add', "$hostname/$path";
+               while ( $path ) {
+                       rsync( '-avv', "root\@$hostname:$path", "$hostname/$path" );
+                       print $client git 'add', "$hostname/$path";
+
+                       $args_message =~ s/^(.+)\b// || last;
+                       $path = $1;
+                       warn "? $path";
+               }
        } elsif ( $command eq 'commit' ) {
                pull_changes $hostname;
                $message =~ s/'/\\'/g;
@@ -132,12 +166,39 @@ while (my $client = $server->accept()) {
        } elsif ( $command =~ m{(diff|status|log|ch)} ) {
                $command .= ' --stat' if $command eq 'log';
                $command = 'log --patch-with-stat' if $command =~ m/^ch/;
-               pull_changes $hostname if $command eq 'diff';
-               # commands without path will show host-wide status/changes
-               print $client git($command, $rel_path ? $backup_path : $hostname);
+               pull_changes( $hostname ) if $command eq 'diff';
+               if ( $on_host ) {
+                       mkpath $_ foreach grep { ! -e $_ } ( "$hostname/$dir", "$on_host/$dir" );
+                       rsync( '-avv', "root\@$hostname:$path", "$hostname/$path" );
+                       rsync( '-avv', "root\@$on_host:$path", "$on_host/$path" );
+                       open(my $diff, '-|', "diff -Nuw $hostname$path $on_host$path");
+                       while(<$diff>) {
+                               print $client $_;
+                       }
+               } else {
+                       # commands without path will show host-wide status/changes
+                       my $backup_path = $path ? "$hostname/$path" : "$hostname/";
+                       # hostname must end with / to prevent error from git:
+                       # ambiguous argument 'arh-hw': both revision and filename
+                       # to support branches named as hosts
+                       print $client git($command, $backup_path);
+               }
        } elsif ( $command eq 'revert' ) {
-               print $client git "checkout -- $hostname/$path";
-               system 'rsync', '-avv', "$hostname/$path", "root\@$hostname:$path";
+               if ( $on_host ) {
+                       rsync( '-avv', "$on_host/$path", "root\@$hostname:$path" );
+               } else {
+                       print $client git "checkout -- $hostname/$path";
+                       rsync( '-avv', "$hostname/$path", "root\@$hostname:$path" );
+               }
+       } elsif ( $command eq 'cat' ) {
+               my $file_path = ( $on_host ? $on_host : $hostname ) . "/$path";
+               open(my $file, '<', $file_path) || warn "ERROR $file_path: $!";
+               while(<$file>) {
+                       print $client $_;
+               }
+               close($file);
+       } elsif ( $command eq 'ls' ) {
+               print $client `ls $backup_path`;
        } else {
                print $client "Unknown command: $command\n";
        }