blob: 2c00e6a2095cb639b3997157315fedd0a63fb91e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
#!/usr/bin/ruby
#
# Run on a client machine to set up backups for the first time
#
$LOAD_PATH.unshift("/usr/lib/byteback")
require 'fileutils'
require 'trollop'
require 'byteback/util'
require 'byteback/log'
include Byteback::Util
include Byteback::Log
def error(message)
STDERR.print "*** #{message}\n"
exit 1
end
def verbose(message)
print "#{message}\n"
end
opts = Trollop::options do
opt :hostname, "Set host name for backups",
:type => :string
opt :destination, "Backup destination (i.e. user@host:/path)",
:type => :string
end
@destination = opts[:destination]
@hostname = opts[:hostname]
_dummy, @destination_user, @destination_host, colon, @destination_path =
/^(.*)?(?:@)([^:]+)(:)(.*)?$/.match(@destination).to_a
@destination_user ||= 'byteback'
@destination_path ||= ''
@destination_host ||= @destination
if !@hostname
@hostname = `hostname -f`.chomp
warn "No hostname set, using #{@hostname}\n"
end
FileUtils.mkdir_p("/etc/byteback")
if File.readable?("/etc/byteback/key")
warn "Skipping key generation, delete /etc/byteback/key if that's wrong"
else
error "Couldn't generate SSH key" unless
system <<-KEYGEN
ssh-keygen -q -t rsa -C "byteback client key" \
-N "" -f /etc/byteback/key
KEYGEN
end
key_pub = File.read("/etc/byteback/key.pub").chomp
error "Remote setup didn't work" unless
system("ssh -i /etc/byteback/key -l #{@destination_user} #{@destination_host} byteback-setup-client-receive #{@hostname} #{key_pub}")
File.open("/etc/byteback/destination", "w") do |f|
f.print "#{@destination_user}@#{@destination_host}:#{@destination_path}"
end
print "Setup worked! To take your first backup run: byteback-backup --verbose\n"
|
