Add db:remote:restore_latest task

Author: cmrd-senya

lib/capistrano-db-tasks/compressors/bzip2.rb

@@ -32,7 +32,7 @@ def decompress(from, to = nil)
                "-c --stdout > #{to}"
              end
 
-        "bunzip2 -f #{from} #{to}"
+        "bunzip2 -k -f #{from} #{to}"
       end
 
     end

lib/capistrano-db-tasks/database.rb

@@ -112,6 +112,7 @@ def initialize(cap_instance)
 
     def dump
       @cap.execute "cd #{@cap.current_path} && #{dump_cmd} | #{compressor.compress('-', output_file)}"
+      @cap.execute("ln -fs #{@cap.current_path}/#{output_file} #{@cap.current_path}/db/latest.sql.#{compressor.file_extension}")
       self
     end
 
@@ -212,6 +213,11 @@ def backup(instance)
       remote_db = Database::Remote.new(instance)
       remote_db.dump
     end
+
+    def restore_latest(instance)
+      remote_db = Database::Remote.new(instance)
+      remote_db.load("#{instance.current_path}/db/latest.sql.#{remote_db.compressor.file_extension}", true)
+    end
   end
 
 end

lib/capistrano-db-tasks/dbtasks.rb

@@ -37,6 +37,13 @@
         Database.backup(self)
       end
     end
+
+    desc 'Restores the latest database dump from the remote folder (pairs with db:remote:backup)'
+    task :restore_latest do
+      on roles(:db) do
+        Database.restore_latest(self)
+      end
+    end
   end
 
   namespace :local do