From c138effe83fe68234d9f630b62b52c3fdfdaca6a Mon Sep 17 00:00:00 2001
From: Matthew Bloch <matthew@bytemark.co.uk>
Date: Mon, 6 Jan 2014 02:19:54 +0000
Subject: First commit of working code, needs a lot more writing and testing

---
 byteback-setup-client | 54 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 byteback-setup-client

(limited to 'byteback-setup-client')

diff --git a/byteback-setup-client b/byteback-setup-client
new file mode 100644
index 0000000..ddb6672
--- /dev/null
+++ b/byteback-setup-client
@@ -0,0 +1,54 @@
+#!/usr/bin/ruby
+#
+# Run on a client machine to set up backups for the first time
+
+require 'fileutils'
+require 'trollop'
+
+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
+
+error("Must be a remote path") unless colon
+if !@hostname
+	@hostname = `hostname -f`
+	print "No hostname set, using #{@hostname}"
+end
+
+error "This host already appears set up - you need to delete /etc/byteback if not" if
+  File.readable?("/etc/byteback/key")
+
+FileUtils.mkdir_p("/etc/byteback")
+
+error "Couldn't generate SSH key" unless 
+  system("ssh-keygen -q -t rsa -C \"byteback client key\" -N \"\" -f /etc/byteback/key")
+
+key_pub = File.read("/etc/byteback/key.pub")
+
+error "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") { |f| f.print @destination }
+
+print "Setup worked!  To take your first backup run: byteback-backup --verbose\n"
-- 
cgit v1.2.3