filbert 0.0.1 → 0.0.2

data/README.md

@@ -8,11 +8,22 @@ Filbert downloads a backup of a follower database for a given heroku application
 
 And reload your path (e.g. `rbenv rehash`). Then run
 
+    heroku login
+
+to get access to the app you are planning to backup.
 
 ## Usage
 
     filbert backup --app your-heroku-appname
 
+You can add it as a cron task if you want
+
+    crontab -e
+
+    # Then put this in there to run every 15 minutes.
+    # 'man 5 crontab' for more examples
+    */15 * * * * bash -lc "filbert backup --app heroku-app-name"
+
 ## Contributing
 
 1. Fork it

data/lib/filbert/db_config.rb

@@ -0,0 +1,24 @@
+module Filbert
+  class DbConfig
+    def initialize(config_path, env)
+      @config_path = config_path
+      @env = env || ENV['RAILS_ENV'] || 'development'
+    end
+
+    def username
+      config['username']
+    end
+
+    def database
+      config['database']
+    end
+
+    def password
+      config['password']
+    end
+
+    def config
+      @config ||= YAML.load_file(@config_path)[@env]
+    end
+  end
+end

data/lib/filbert/task.rb

@@ -1,9 +1,10 @@
+require 'filbert/db_config'
+
 module Filbert
   class Task < Thor
     include Thor::Actions
-    class_option :app, type: :string, required: true
 
-    method_options :real_emails => false
+    method_option :app, type: :string, required: true
     desc "backup", "capture and pull latest production snapshot and migrate local database"
     def backup
       say "Looking for the follower DB..."
@@ -19,7 +20,7 @@ module Filbert
       invoke :cleanup
     end
 
-    desc "cleanup", "remove backup files older than 24 hours"
+    desc "cleanup", "remove backup files older than 12 hours"
     def cleanup
       old_files.each do |file|
         say "Deleting old #{File.basename(file.path)}"
@@ -27,6 +28,27 @@ module Filbert
       end
     end
 
+    method_option :config, type: :string, default: "config/database.yml"
+    method_option :env, type: :string, default: "production"
+    desc "restore", "restore the latest db dump"
+    def restore
+      most_recent_file = ordered_dumps.last
+      db_config = DbConfig.new(options[:config], options[:env])
+
+      check_dump_ready(most_recent_file)
+      check_config_ready(db_config)
+
+      say "Restoring: #{db_config.database} <--- #{most_recent_file.path}"
+      kill_connections(db_config.database, db_config.username)
+      ENV['PGPASSWORD'] = db_config.password
+      run! "pg_restore -U #{db_config.username} -d #{db_config.database} -w #{most_recent_file.path}"
+
+    rescue Errno::ENOENT
+      say "Could not find config file #{options[:config]}. Please pass in --config with a path to database.yml"
+    ensure
+      ENV['PGPASSWORD'] = nil
+    end
+
     private
 
       def run!(cmd)
@@ -40,13 +62,17 @@ module Filbert
 
       def old_files
         hurdle = Time.now - 60*60*12
+        ordered_dumps.select{ |file|
+          file.mtime < hurdle
+        }
+      end
+
+      def ordered_dumps
         Dir.new(backups_dir).select{ |x|
           x.end_with? '.dump'
         }.map { |filename|
-          file = File.new(File.join(backups_dir, filename))
-        }.select{ |file|
-          file.mtime < hurdle
-        }
+          File.new(File.join(backups_dir, filename))
+        }.sort_by(&:mtime)
       end
 
       def file_path
@@ -56,5 +82,24 @@ module Filbert
       def backups_dir
         File.join(Dir.home, '.heroku_backups')
       end
+
+      def check_dump_ready(most_recent_file)
+        if most_recent_file.nil?
+          say "Didn't find any backup files in #{backups_dir}"
+          exit 0
+        end
+      end
+
+      def check_config_ready(db_config)
+        if db_config.config.nil?
+          say "Could not find config for \"#{options[:env]}\" in #{options[:config]}"
+          exit 0
+        end
+      end
+
+      def kill_connections(database, user)
+        sql = "SELECT pg_terminate_backend(procpid) FROM pg_stat_activity WHERE procpid <> pg_backend_pid();"
+        run "echo #{sql} |  psql -d #{database} -U #{user}"
+      end
   end
 end

data/lib/filbert/version.rb

@@ -1,3 +1,3 @@
 module Filbert
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: filbert
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-10-29 00:00:00.000000000 Z
+date: 2013-01-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: heroku
@@ -60,6 +60,7 @@ files:
 - bin/filbert
 - filbert.gemspec
 - lib/filbert.rb
+- lib/filbert/db_config.rb
 - lib/filbert/task.rb
 - lib/filbert/version.rb
 homepage: https://github.com/alphasights/filbert