berty 0.0.1

data/bin/berty

@@ -0,0 +1,28 @@
+#!/usr/bin/env ruby
+$:.unshift File.join(File.dirname(__FILE__), '..', 'lib')
+require 'berty'
+
+servers = Berty.servers
+
+for server in servers
+  case ARGV.first
+  when 'setup'
+    Berty.logger.info "Started server setup..."
+    server.setup!
+    Berty.logger.info "Finished server setup..."
+    puts "Server '#{server.hostname}' setup..."
+  when 'info'
+    puts " * #{server.username}@#{server.hostname}"
+    for directory in server.directories
+      puts "   #{directory[:path]} -> #{directory[:destination]}"
+    end
+    if server.mysql_backup?
+      puts "   MySQL backup using password: #{server.mysql_password.gsub(/[a-zA-Z]/, '*')}"
+    end
+    puts unless server == servers.last
+  else
+    Berty.logger.info "Started backup..."
+    server.backup!
+    Berty.logger.info "Finished backup..."
+  end
+end

data/lib/berty/backup.rb

@@ -0,0 +1,12 @@
+module Berty
+  class BackupRun
+    
+    def run!
+      for server in Berty.servers
+        puts "server: #{server.hostname}"
+        server.backup!
+      end
+    end
+    
+  end
+end

data/lib/berty/server.rb

@@ -0,0 +1,94 @@
+module Berty
+  class Server
+    
+    attr_reader :hostname
+
+    def initialize(hostname, options = {})
+      @hostname = hostname
+      @options = options
+    end
+    
+    ## Return all directories which need to be backed up
+    def directories
+      (@options['directories'] || []).map do |dir|
+        path, label = dir.split(':')
+        label ||= path.split('/').last
+        {:path=> path, :label => label, :destination => File.join(backup_path, label)}
+      end
+    end
+    
+    ## Which username should be used when connecting to this server?
+    def username
+      @options['username']
+    end
+    
+    ## Return the full path where backups for this server should be stored on the local file system.
+    def backup_path
+      File.join(Berty.config['root'], hostname)
+    end
+    
+    ## Return the full path to where mysql backups are stored
+    def mysql_backup_path
+      File.join(backup_path, 'mysql')
+    end
+    
+    ## Execute a backup of this server (backing up all directories to the appropriate locations)
+    def backup!
+      run "mkdir -p #{backup_path}"
+      for dir in directories
+        destination = 
+        Berty.logger.info "  starting backup of: '#{dir[:path]}' to '#{dir[:destination]}'"
+        run "rsync -a #{username}@#{hostname}:#{dir[:path]}/ #{dir[:destination]}"
+        Berty.logger.info "  finished backup of: '#{dir[:path]}' to '#{dir[:destination]}'" 
+      end
+      backup_mysql!
+    end
+    
+    ## Setup the connection to this server by sshing to it and doing nothing. This gives the option to accept 
+    ## any host keys.
+    def setup!
+      key = File.read("#{ENV['HOME']}/.ssh/id_rsa.pub")
+      ssh "echo '#{key.chomp}' >> ~/.ssh/authorized_keys"
+    end
+    
+    ## Should this server backup a mysql database?
+    def mysql_backup?
+      @options['mysql_password'].is_a?(String)
+    end
+    
+    ## What password should be used for accessing the database?
+    def mysql_password
+      @options['mysql_password']
+    end
+    
+    ## Execute mysql backups?
+    def backup_mysql!
+      return unless mysql_backup?
+      run "mkdir -p #{mysql_backup_path}"
+      Berty.logger.info "  starting mysql backup on: #{hostname}"
+      ## create a dump on the remote system as '/tmp/bert-mysqldump'
+      ssh "mysqldump --all-databases -u root -p#{mysql_password} > /tmp/berty-mysqldump"
+      ## pull the file over to us
+      scp "/tmp/berty-mysqldump", File.join(mysql_backup_path, "#{Time.now.strftime("%Y-%m-%d-%H-%M")}.sql")
+      Berty.logger.info "  finishing mysql backup on: #{hostname}"
+    end
+    
+    private
+    
+    def run(command)
+      Berty.logger.info "executing: #{command}"
+      system(command)
+    end
+    
+    def ssh(command)
+      run "ssh #{username}@#{hostname} \"#{command}\""
+    end
+    
+    def scp(remote, local)
+      run "scp #{username}@#{hostname}:#{remote} #{local}"
+    end
+    
+    
+  end
+end
+ 

data/lib/berty.rb

@@ -0,0 +1,31 @@
+require 'logger'
+require 'yaml'
+
+require 'berty/backup'
+require 'berty/server'
+
+module Berty
+  
+  class Error < StandardError; end
+
+  class << self
+    def config_path
+      [File.expand_path('../../config/berty.example.conf', __FILE__), '/etc/berty.conf'].each do |file|
+        return file if File.exist?(file)
+      end
+      raise Error, "Configuration file not found. Please ensure it is defined in '/etc/berty.conf'"
+    end
+  
+    def config
+      @config ||= YAML.load_file(config_path)
+    end
+    
+    def logger
+      @logger ||= Logger.new(config['log_file'] || STDERR)
+    end
+    
+    def servers
+      @servers ||= config['servers'].map{|key,value| Berty::Server.new(key, value)  }
+    end
+  end
+end

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification 
+name: berty
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Adam Cooke
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-03-23 00:00:00 +00:00
+default_executable: 
+dependencies: []
+
+description: 
+email: adam@atechmedia.com
+executables: 
+- berty
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/berty
+- lib/berty/backup.rb
+- lib/berty/server.rb
+- lib/berty.rb
+has_rdoc: true
+homepage: http://www.atechmedia.com
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Berty the backup gem
+test_files: []
+