dpickett-awesome_backup 0.5.1

data/VERSION.yml

@@ -0,0 +1,4 @@
+--- 
+:minor: 5
+:patch: 1
+:major: 0

data/lib/awesome_backup/recipes/backup.rb

@@ -0,0 +1,44 @@
+  # Perform backups
+  #
+  # To automatically transfer backups to your local machine, add:
+  #   after "backup:create", "backup:download"
+  #
+  if respond_to?(:_cset)
+    _cset(:backup_path) { "#{shared_path}/backups" }
+    _cset(:skip_backup_tables, ['sessions'])
+  else
+    set(:backup_path, "#{shared_path}/backups")
+    set(:skip_backup_tables, ['sessions'])
+  end
+
+namespace :backup do
+  def latest
+    capture("cd #{current_path}; rake -s db:backup:latest BACKUP_DIR=#{backup_path}").strip
+  end
+
+  desc "Create a backup on the server"
+  task :create, :roles => :db, :only => {:primary => true} do
+    rails_env = fetch(:rails_env, "production")
+    skip_tables = Array(skip_backup_tables).join(',')
+    run "cd #{current_path}; rake db:backup:create RAILS_ENV=#{rails_env} BACKUP_DIR=#{backup_path} SKIP_TABLES=#{skip_tables}"
+  end
+
+  desc "Retreive a backup from the server. Gets the latest by default, set :backup_version to specify which version to copy"
+  task :download, :roles => :db, :only => {:primary => true} do
+    version = fetch(:backup_version, latest)
+    run "tar -C #{backup_path} -czf #{backup_path}/#{version}.tar.gz #{version}"
+    `mkdir -p backups`
+    get "#{backup_path}/#{version}.tar.gz", "backups/#{version}.tar.gz"
+    run "rm #{backup_path}/#{version}.tar.gz"
+    `tar -C backups -zxf backups/#{version}.tar.gz`
+    `rm backups/#{version}.tar.gz`
+  end
+
+  desc "Creates a new remote backup and clones it to the local database"
+  task :mirror, :roles => :db, :only => {:primary => true} do
+    create
+    download
+    `rake db:backup:restore`
+  end
+
+end

data/lib/awesome_backup/recipes.rb

@@ -0,0 +1,3 @@
+Capistrano::Configuration.instance(:must_exist).load do
+  load File.join(File.dirname(__FILE__), "recipes", "backup.rb")  
+end

data/lib/awesome_backup/tasks/backup.rake

@@ -0,0 +1,29 @@
+namespace :db do
+  namespace :backup do
+  
+    def backup_directory
+      ENV['BACKUP_DIR'] ||= File.join(RAILS_ROOT, 'backups')
+    end
+  
+    # Set the VERSION to the latest if it's not set
+    task :latest do
+      last = Dir["#{backup_directory}/*/"].sort.last
+      puts ENV['VERSION'] ||= File.basename(last) if last
+    end
+  
+    # Setup environment variables for the schema and fixtures tasks
+    task :setup => :environment do
+      ENV['VERSION'] ||= Time.now.utc.strftime('%Y%m%d%H%M%S')
+      backup = File.join(backup_directory, ENV['VERSION'])
+      FileUtils.mkdir_p backup
+      ENV['FIXTURES_DIR'] = backup
+      ENV['SCHEMA'] = File.join(backup, 'schema.rb')
+    end
+  
+    desc 'Create a new backup of the database'
+    task :create => [:setup, 'db:fixtures:dump', 'db:schema:dump']
+  
+    desc 'Restore a backup of the database. Use VERSION to specify a version other than the latest.'
+    task :restore => [:latest, :setup, 'db:schema:load', 'db:fixtures:load']
+  end
+end

data/lib/awesome_backup/tasks/database.rake

@@ -0,0 +1,44 @@
+# These tasks replace the built-in Rails tasks for dumping and loading the schema,
+# allowing you to specify the FIXTURES_DIR to use for dumping and loading.
+Rake.application.instance_eval do
+  @tasks.delete "db:fixtures:load"
+  @tasks.delete "db:fixtures:dump"
+end
+
+
+namespace :db do  
+  namespace :fixtures do
+    desc "Load fixtures into the current environment's database.  Load specific fixtures using FIXTURES=x,y"
+    task :load => :environment do
+      require 'active_record/fixtures'
+      ActiveRecord::Base.establish_connection(RAILS_ENV.to_sym)
+      fixtures_dir = ENV['FIXTURES_DIR'] || File.join(RAILS_ROOT, 'test', 'fixtures')
+      (ENV['FIXTURES'] ? ENV['FIXTURES'].split(/,/) : Dir.glob(File.join(fixtures_dir, '*.{yml,csv}'))).each do |fixture_file|
+        Fixtures.create_fixtures(fixtures_dir, File.basename(fixture_file, '.*'))
+      end
+    end
+    
+    desc "Create YAML fixtures from data in the current environment's database. Dump specific tables using TABLES=x[,y,z]."
+    task :dump => :environment do
+      ActiveRecord::Base.establish_connection(RAILS_ENV.to_sym)
+      fixtures_dir = ENV['FIXTURES_DIR'] || 
+         File.join(RAILS_ROOT, 'test', 'fixtures')
+      skip_tables = ENV['SKIP_TABLES'] ? ENV['SKIP_TABLES'].split(/,/) : ["sessions"]
+      tables = ENV['TABLES'] ? ENV['TABLES'].split(/,/) : ActiveRecord::Base.connection.tables
+      sql = "SELECT * FROM %s LIMIT %d OFFSET %d"
+      limit = 500
+      (tables - skip_tables).each do |table_name| 
+        i = "000"
+        offset = 0
+        File.open("#{fixtures_dir}/#{table_name}.yml", 'w' ) do |file|
+          while !(data = ActiveRecord::Base.connection.select_all(sql % [table_name, limit, offset])).empty?
+            data.each do |record|
+              file.write({"#{table_name}_#{i.succ!}" => record}.to_yaml.sub(/^---.*\n/, ''))
+            end
+            offset += limit
+          end
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification 
+name: dpickett-awesome_backup
+version: !ruby/object:Gem::Version 
+  version: 0.5.1
+platform: ruby
+authors: 
+- Collective Ideas
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-01-22 00:00:00 -08:00
+default_executable: 
+dependencies: []
+
+description: ""
+email: dpickett@enlightsolutions.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- VERSION.yml
+- lib/awesome_backup
+- lib/awesome_backup/recipes
+- lib/awesome_backup/recipes/backup.rb
+- lib/awesome_backup/recipes.rb
+- lib/awesome_backup/tasks
+- lib/awesome_backup/tasks/backup.rake
+- lib/awesome_backup/tasks/database.rake
+- lib/awesome_backup.rb
+has_rdoc: false
+homepage: http://github.com/collectiveidea/awesome-backup/tree/master
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Provides Rake and Capistrano tasks for making database backups
+test_files: []
+