X-Git-Url: http://git.rot13.org/?a=blobdiff_plain;f=bak-git-server.pl;h=1c8efc1303b5560ad340f3564bd7736bba45827b;hb=b3e5588a5f136f8c1c72f693327c39c8937ef9c1;hp=a826934d669a5832e14a60f94f19c2678663ac5d;hpb=8830aaf02a5f699fbad5c5381f1803b7417fabe3;p=bak-git.git

diff --git a/bak-git-server.pl b/bak-git-server.pl
index a826934..1c8efc1 100755
--- a/bak-git-server.pl
+++ b/bak-git-server.pl
@@ -26,8 +26,12 @@ bak command overview:
   bak ch[anges]
   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,7 @@ $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 `hostname` `pwd` \$* | nc $server_ip 9001
 __SHELL_CLIENT__
 
 chdir $dir;
@@ -90,6 +94,11 @@ warn "dir: $dir listen: $server_ip:9001\n"
 sub pull_changes {
 	my $hostname = shift;
 	system "find $hostname -type f | sed 's,$hostname,,' > /tmp/$hostname.list";
+	if ( @_ ) {
+		open(my $files, '>>', "/tmp/$hostname.list");
+		print $files "$_\n" foreach @_;
+		close($files);
+	}
 	system "rsync -avv --files-from /tmp/$hostname.list root\@$hostname:/ $hostname/"
 }
 
@@ -98,13 +107,25 @@ while (my $client = $server->accept()) {
 	chomp($line);
 	warn "<<< $line\n";
 	my ($user,$hostname,$pwd,$command,$rel_path,$message) = split(/\s+/,$line,6);
+	$hostname =~ s/\..+$//;
 
-	my $on_host = $1 if $rel_path =~ s/^([^:]+):(.+)$/$2/ && -e $1;
+	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{/[^/]+$}{};
@@ -123,8 +144,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 ) {
+			system '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;
@@ -133,8 +160,9 @@ 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';
+		pull_changes( $hostname ) if $command eq 'diff';
 		if ( $on_host ) {
+			system 'rsync', '-avv', "root\@$hostname:$path", "$hostname/$path";
 			system 'rsync', '-avv', "root\@$on_host:$path", "$on_host/$path";
 			open(my $diff, '-|', "diff -Nuw $hostname$path $on_host$path");
 			while(<$diff>) {
@@ -142,7 +170,11 @@ while (my $client = $server->accept()) {
 			}
 		} else {
 			# commands without path will show host-wide status/changes
-			print $client git($command, $rel_path ? $backup_path : $hostname);
+			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' ) {
 		if ( $on_host ) {
@@ -151,6 +183,13 @@ while (my $client = $server->accept()) {
 			print $client git "checkout -- $hostname/$path";
 			system '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);
 	} else {
 		print $client "Unknown command: $command\n";
 	}