chomp path to fix add/commit without message
[bak-git.git] / 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 [/path]
23   bak status [/path]
24   bak log [/path]
25
26   bak ch[anges]
27   bak revert /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,5);
101
102         my $path = $rel_path =~ m{^/} ? $rel_path : "$pwd/$rel_path";
103
104         $message ||= '';
105         warn "$hostname [$command] $path | $message\n";
106         $message ||= "$hostname [$command] $path";
107
108         my $dir = $path;
109         $dir =~ s{/[^/]+$}{};
110
111         my $backup_path = -e "$hostname/$path" ? "$hostname/$path" : $hostname;
112
113         sub git {
114                 my $args = join(' ',@_);
115                 warn "# git $args\n";
116                 my $out = `git $args`;
117                 warn "$out\n# [", length($out), " bytes]\n" if defined $out;
118                 return $out;
119         }
120
121         if ( ! $command ) {
122                 pull_changes $hostname;
123         } elsif ( $command eq 'add' ) {
124                 mkpath "$hostname/$dir" unless -e "$hostname/$dir";
125                 system 'rsync', '-avv', "root\@$hostname:$path", "$hostname/$path";
126                 print $client git 'add', "$hostname/$path";
127         } elsif ( $command eq 'commit' ) {
128                 pull_changes $hostname;
129                 $message =~ s/'/\\'/g;
130                 $user =~ s/\/$//;
131                 print $client git( "commit -m '$message' --author '$user <$hostname>' $backup_path" );
132         } elsif ( $command =~ m{(diff|status|log|ch)} ) {
133                 $command .= ' --stat' if $command eq 'log';
134                 $command = 'log --patch-with-stat' if $command =~ m/^ch/;
135                 pull_changes $hostname if $command eq 'diff';
136                 # commands without path will show host-wide status/changes
137                 print $client git($command, $rel_path ? $backup_path : $hostname);
138         } elsif ( $command eq 'revert' ) {
139                 print $client git "checkout -- $hostname/$path";
140                 system 'rsync', '-avv', "$hostname/$path", "root\@$hostname:$path";
141         } else {
142                 print $client "Unknown command: $command\n";
143         }
144
145 }
146