bak-git-server.pl

   1 #!/usr/bin/perl
   2
   3 =head1 bak-git
   4
   5 Simple tracking of remote files in central git repository
   6 with only shell, netcat, rsync and ssh on client
   7
   8 Start server, install on remote-host or upgrade with:
   9
  10   ./bak-git-server.pl /path/to/backup 192.168.42.42
  11         [--install remote-host]
  12         [--upgrade]
  13
  14 You will want to add following to C<~/.ssh/config>
  15
  16   RemoteForward 9001 localhost:9001
  17
  18 bak command overview:
  19
  20   bak add /path
  21   bak commit [/path [message]]
  22   bak diff [host:][/path]
  23   bak status [/path]
  24   bak log [/path]
  25
  26   bak ch[anges]
  27   bak revert [host:]/path
  28
  29   bak - push all changed files to server
  30
  31 =cut
  32
  33 use warnings;
  34 use strict;
  35 use autodie;
  36 use IO::Socket::INET;
  37 use File::Path;
  38 use Getopt::Long;
  39
  40 my $upgrade = 0;
  41 my $install;
  42
  43 GetOptions(
  44         'upgrade!'  => \$upgrade,
  45         'install=s' => \$install,
  46 ) || die "$!\n";
  47
  48 my ( $dir, $server_ip ) = @ARGV;
  49 die "usage: $0 /backup/directory 127.0.0.1\n" unless $dir;
  50 $server_ip ||= '127.0.0.1';
  51
  52 my $shell_client = <<__SHELL_CLIENT__;
  53 #!/bin/sh
  54 echo \$USER/\$SUDO_USER `hostname -s` `pwd` \$* | nc $server_ip 9001
  55 __SHELL_CLIENT__
  56
  57 chdir $dir;
  58 system 'git init' unless -e '.git';
  59
  60 if ( $upgrade || $install ) {
  61         open(my $fh, '>', '/tmp/bak');
  62         print $fh $shell_client;
  63         close($fh);
  64         chmod 0755, '/tmp/bak';
  65
  66         my @hosts = grep { -d $_ } glob '*';
  67         @hosts = ( $install ) if $install;
  68
  69         foreach my $hostname ( @hosts ) {
  70                 warn "install on $hostname\n";
  71                 system 'ssh-copy-id', "root\@$hostname" if ! -d $hostname;
  72                 system "scp /tmp/bak root\@$hostname:/usr/local/bin/";
  73                 system "ssh root\@$hostname apt-get install -y rsync";
  74         }
  75 }
  76
  77 my $server = IO::Socket::INET->new(
  78         Proto     => 'tcp',
  79         LocalAddr => $server_ip,
  80         LocalPort => 9001,
  81         Listen    => SOMAXCONN,
  82         Reuse     => 1
  83 ) || die $!;
  84
  85
  86 warn "dir: $dir listen: $server_ip:9001\n"
  87         , $shell_client
  88 ;
  89
  90 sub pull_changes {
  91         my $hostname = shift;
  92         system "find $hostname -type f | sed 's,$hostname,,' > /tmp/$hostname.list";
  93         system "rsync -avv --files-from /tmp/$hostname.list root\@$hostname:/ $hostname/"
  94 }
  95
  96 while (my $client = $server->accept()) {
  97         my $line = <$client>;
  98         chomp($line);
  99         warn "<<< $line\n";
 100         my ($user,$hostname,$pwd,$command,$rel_path,$message) = split(/\s+/,$line,6);
 101
 102         my $on_host = $1 if $rel_path =~ s/^([^:]+):(.+)$/$2/ && -e $1;
 103         my $path = $rel_path =~ m{^/} ? $rel_path : "$pwd/$rel_path";
 104
 105         $message ||= '';
 106         warn "$hostname [$command] $path | $message\n";
 107         $message ||= "$hostname [$command] $path";
 108
 109         my $dir = $path;
 110         $dir =~ s{/[^/]+$}{};
 111
 112         my $backup_path = -e "$hostname/$path" ? "$hostname/$path" : $hostname;
 113
 114         sub git {
 115                 my $args = join(' ',@_);
 116                 warn "# git $args\n";
 117                 my $out = `git $args`;
 118                 warn "$out\n# [", length($out), " bytes]\n" if defined $out;
 119                 return $out;
 120         }
 121
 122         if ( ! $command ) {
 123                 pull_changes $hostname;
 124         } elsif ( $command eq 'add' ) {
 125                 mkpath "$hostname/$dir" unless -e "$hostname/$dir";
 126                 system 'rsync', '-avv', "root\@$hostname:$path", "$hostname/$path";
 127                 print $client git 'add', "$hostname/$path";
 128         } elsif ( $command eq 'commit' ) {
 129                 pull_changes $hostname;
 130                 $message =~ s/'/\\'/g;
 131                 $user =~ s/\/$//;
 132                 print $client git( "commit -m '$message' --author '$user <$hostname>' $backup_path" );
 133         } elsif ( $command =~ m{(diff|status|log|ch)} ) {
 134                 $command .= ' --stat' if $command eq 'log';
 135                 $command = 'log --patch-with-stat' if $command =~ m/^ch/;
 136                 pull_changes $hostname if $command eq 'diff';
 137                 if ( $on_host ) {
 138                         system 'rsync', '-avv', "root\@$on_host:$path", "$on_host/$path";
 139                         open(my $diff, '-|', "diff -uw $hostname$path $on_host$path");
 140                         while(<$diff>) {
 141                                 print $client $_;
 142                         }
 143                 } else {
 144                         # commands without path will show host-wide status/changes
 145                         print $client git($command, $rel_path ? $backup_path : $hostname);
 146                 }
 147         } elsif ( $command eq 'revert' ) {
 148                 if ( $on_host ) {
 149                         system 'rsync', '-avv', "$on_host/$path", "root\@$hostname:$path";
 150                 } else {
 151                         print $client git "checkout -- $hostname/$path";
 152                         system 'rsync', '-avv', "$hostname/$path", "root\@$hostname:$path";
 153                 }
 154         } else {
 155                 print $client "Unknown command: $command\n";
 156         }
 157
 158 }
 159