rails_pwnerer 0.4 → 0.4.5

data/CHANGELOG

@@ -1,3 +1,5 @@
+v0.4.5. Implemented 'restore'.
+
 v0.4. Implemented 'backup'.
 
 v0.3.5. Implemented 'go live' and 'go panic' to take all application processes up / down.

data/README

@@ -9,13 +9,18 @@ welcome to test/update for other systems, and I'll be happy to merge patches,
 or add you to the RubyForge project. 
 
 
-Live without +rails_pwnerer+:
+Life without +rails_pwnerer+:
 
 Read a 20-page guide, and spend the better half of a day getting a box
 ready for production. You'll be frustrated in the process, but feel
 accomplished at the end. Assuming you don't do something wrong.
 
-Live with +rails_pwnerer+:
+Read a couple more 20-page guides, and learn to push newer versions
+of your application. You'll become lazy, and take shortcuts, like having 
+production passwords in the repository, insecure file permissions.
+I won't even mention backups. 
+
+Life with +rails_pwnerer+:
 
 1) Install rubygems:
 sudo apt-get install rubygems
@@ -32,5 +37,14 @@ sudo rpwn install svn+ssh://your.repository.host/path/to/your_application
 5) Reboot, or start the services right away:
 sudo rpwn go live
 
-6) Keep your application updated
+* Keep your application updated:
 sudo rpwn update your_application
+
+* Create a backup:
+sudo rpwn backup your_application
+
+* Restore from the latest backup:
+sudo rpwn restore your_application
+
+* Bring down all the applications (panic mode):
+sudo rpwn go down

data/lib/pwnage/app/cluster_config.rb

@@ -35,6 +35,9 @@ class RailsPwnage::App::ClusterConfig
   end
   
   def stop_mongrels(app_name)
+    # silently die if the app was completely busted
+    return unless File.exists? RailsPwnage::Config.path_to(app_name)
+    
     with_dir(RailsPwnage::Config.path_to(app_name)) do
       system 'mongrel_rails cluster::stop'
       
@@ -44,6 +47,9 @@ class RailsPwnage::App::ClusterConfig
   end
   
   def start_mongrels(app_name)
+    # silently die if the app was completely busted
+    return unless File.exists? RailsPwnage::Config.path_to(app_name)
+    
     with_dir(RailsPwnage::Config.path_to(app_name)) do
       system 'mongrel_rails cluster::start'
     end    
@@ -58,10 +64,21 @@ class RailsPwnage::App::ClusterConfig
     stop_mongrels(app_name)
     yield
   ensure
-    # TODO: reconfigure the mongrels if the configuration database changed  
+    configure_mongrels app_name
+    fix_permissions app_name
     start_mongrels(app_name)    
   end
   
+  def backup(app_name, action)
+    case action
+    when :checkpoint
+      # nothing to do here
+    when :rollback
+      configure_mongrels app_name
+      fix_permissions app_name      
+    end
+  end
+  
   def control_all(action)
     case action
     when :start
@@ -79,6 +96,10 @@ class RailsPwnage::App::ClusterConfig
     self.new.update(app_name, &update_proc)
   end
   
+  def self.backup(app_name, action)
+    self.new.backup(app_name, action)
+  end  
+  
   def self.control_all(action = :start)
     self.new.control_all(action) 
   end  

data/lib/pwnage/app/database.rb

@@ -5,25 +5,32 @@ require 'yaml'
 class RailsPwnage::App::Database
   include RailsPwnage::Base
 
-  # creates the mysql database for the application
-  def create_databse(app_name)
+  # creates/drops the mysql database for the application
+  def admin_database(app_name, action = :create)
     db_name = RailsPwnage::Config.app_dbname(app_name)
     db_user = RailsPwnage::Config.app_db_user(app_name)
     db_pass = RailsPwnage::Config.app_db_password(app_name)
     
     with_dir(RailsPwnage::Config.path_to(app_name)) do
-      # put together the creation script
+      # put together the admin script
+      case action
+      when :create
       sql_commands = <<ENDSQL
         CREATE DATABASE #{db_name};
         GRANT ALL ON #{db_name}.* TO '#{db_user}'@'localhost' IDENTIFIED BY '#{db_pass}' WITH GRANT OPTION;
 ENDSQL
+      when :drop
+      sql_commands = <<ENDSQL
+        DROP DATABASE #{db_name};
+ENDSQL
+      end
       
       # run it
-      File.open('tmp/create_db.sql', 'w') { |f| f.write sql_commands }
-      system "mysql -uroot < tmp/create_db.sql"
+      File.open('tmp/admin_db.sql', 'w') { |f| f.write sql_commands }
+      system "mysql -uroot < tmp/admin_db.sql"
       
       # cleanup
-      File.delete('tmp/create_db.sql')
+      File.delete('tmp/admin_db.sql')
     end    
   end
   
@@ -44,16 +51,49 @@ ENDSQL
     File.chown(uid_for_username(pwnerer_user), gid_for_username(pwnerer_user), config_file)
   end
   
-  # migrates the database so it's ready
+  # migrates the database to the latest schema version
   def migrate_database(app_name)
     with_dir(RailsPwnage::Config.path_to(app_name)) do
       # now migrate the database
       system "rake db:migrate RAILS_ENV=production"
     end
   end
+  
+  # creates a database dump in the backup area
+  def dump_database(app_name)
+    db_name = RailsPwnage::Config.app_dbname(app_name)
+    db_user = RailsPwnage::Config.app_db_user(app_name)
+    db_pass = RailsPwnage::Config.app_db_password(app_name)    
+
+    pwnerer_user = RailsPwnage::Config.pwnerer_user
+    pwnerer_uid = uid_for_username(pwnerer_user)
+    pwnerer_gid = gid_for_username(pwnerer_user)
+    
+    timestamp = Time.now.strftime '%Y%m%d%H%M%S'
+    dump_file = "db/#{app_name}_#{timestamp}.sql"
+    with_dir(RailsPwnage::Config.path_to(:backup, app_name)) do
+      system "mysqldump --add-drop-database --add-drop-table --extended-insert --single-transaction -u#{db_user} -p#{db_pass} #{db_name} > #{dump_file}"
+      # lockdown the file
+      File.chmod(0400, dump_file)
+      File.chown(pwnerer_uid, pwnerer_gid, dump_file)
+    end    
+  end
+  
+  # loads the latest database dump from the backup area
+  def load_database(app_name)
+    db_name = RailsPwnage::Config.app_dbname(app_name)
+    db_user = RailsPwnage::Config.app_db_user(app_name)
+    db_pass = RailsPwnage::Config.app_db_password(app_name)    
+    
+    with_dir(RailsPwnage::Config.path_to(:backup, app_name)) do
+      # find the latest dump and load it in
+      dump_file = Dir.glob("db/#{app_name}_*").max
+      system "mysql -u#{db_user} -p#{db_pass} #{db_name} < #{dump_file}"
+    end
+  end
 
   def setup(app_name)
-    create_databse app_name
+    admin_database app_name, :create
     configure_rails app_name
     migrate_database app_name
   end
@@ -65,26 +105,16 @@ ENDSQL
   
   # backs up or restores the database
   def backup(app_name, action)
-    db_name = RailsPwnage::Config.app_dbname(app_name)
-    db_user = RailsPwnage::Config.app_db_user(app_name)
-    db_pass = RailsPwnage::Config.app_db_password(app_name)    
-
-    pwnerer_user = RailsPwnage::Config.pwnerer_user
-    pwnerer_uid = uid_for_username(pwnerer_user)
-    pwnerer_gid = gid_for_username(pwnerer_user)
     
     case action
     when :checkpoint
-      timestamp = Time.now.strftime '%Y%m%d%H%M%S'
-      dump_file = "db/#{app_name}_#{timestamp}.sql"
-      with_dir(RailsPwnage::Config.path_to(:backup, app_name)) do
-        system "mysqldump --add-drop-database --add-drop-table --extended-insert --single-transaction -u#{db_user} -p#{db_pass} #{db_name} > #{dump_file}"
-        # lockdown the file
-        File.chmod(0400, dump_file)
-        File.chown(pwnerer_uid, pwnerer_gid, dump_file)
-      end
+      dump_database app_name
     when :rollback
-      # TODO: rollback
+      admin_database app_name, :drop
+      admin_database app_name, :create
+      load_database app_name
+      configure_rails app_name      
+      migrate_database app_name
     end
   end
   

data/lib/pwnage/app/main.rb

@@ -20,14 +20,16 @@ module RailsPwnage::App
   def self.backup(app_name, action = :checkpoint)
     case action
       when :checkpoint
+        ClusterConfig.backup app_name, action
         Svn.backup app_name, action
-        ClusterConfig.update(app_name) do
+        ClusterConfig.update(app_name) do          
           Database.backup app_name, action
         end
       when :rollback
         ClusterConfig.update(app_name) do
           Svn.backup app_name, action
           Database.backup app_name, action
+          ClusterConfig.backup app_name, action
         end
     end    
   end

data/lib/pwnage/app/svn.rb

@@ -40,33 +40,63 @@ class RailsPwnage::App::Svn
     end
   end
   
-  def backup(app_name, action)
+  # dump the application files to the backup area
+  def dump_files(app_name)
     pwnerer_user = RailsPwnage::Config.pwnerer_user
     pwnerer_uid = uid_for_username(pwnerer_user)
     pwnerer_gid = gid_for_username(pwnerer_user)
     
-    case action
-    when :checkpoint
-      timestamp = Time.now.strftime '%Y%m%d%H%M%S'
-      dump_file = "files/#{app_name}_#{timestamp}.tar.gz"
+    timestamp = Time.now.strftime '%Y%m%d%H%M%S'
+    dump_file = "files/#{app_name}_#{timestamp}.tar.gz"
+    
+    with_dir(RailsPwnage::Config.path_to(:backup, app_name)) do
+      # create a cold copy of the application files
+      cold_copy = "tmp#{File::SEPARATOR}#{app_name}"
+      FileUtils.rm_r cold_copy if File.exists? cold_copy
+      FileUtils.cp_r RailsPwnage::Config.path_to(app_name), 'tmp'
       
-      with_dir(RailsPwnage::Config.path_to(:backup, app_name)) do
-        # create a cold copy of the application files
-        cold_copy = "tmp#{File::SEPARATOR}#{app_name}"
-        FileUtils.rm_r cold_copy if File.exists? cold_copy
-        FileUtils.cp_r RailsPwnage::Config.path_to(app_name), 'tmp'
-        
-        # pack and protect the cold copy
-        with_dir 'tmp' do
-          system "tar -czf ../#{dump_file} #{app_name}"
+      # remove the temp garbage in the cold copy
+      fs = File::SEPARATOR
+      ["cache#{fs}*", "pids#{fs}*.pid", "sessions#{fs}*"].each do |pattern|
+        Dir.glob("tmp#{fs}#{app_name}#{pattern}") do |entry|
+          next if entry =~ /\.svn$/
+          FileUtils.rm_r entry
         end
-        File.chmod(400, dump_file)
-        File.chown(pwnerer_uid, pwnerer_gid, dump_file)
-        
-        # clean up
-        FileUtils.rm_r cold_copy
       end
+      
+      # pack and protect the cold copy
+      with_dir 'tmp' do
+        system "tar -czf ../#{dump_file} #{app_name}"
+      end
+      File.chmod(400, dump_file)
+      File.chown(pwnerer_uid, pwnerer_gid, dump_file)
+      
+      # clean up
+      FileUtils.rm_r cold_copy
+    end    
+  end
+  
+  # loads the latest file dump from the backup area
+  def load_files(app_name)
+    dump_file = Dir.glob(RailsPwnage::Config.path_to(:backup, app_name) + File::SEPARATOR + "files/#{app_name}_*").max
+    with_dir(RailsPwnage::Config.path_to(:prod_apps)) do
+      # find the latest dump and load it in            
+      system "tar -xzf #{dump_file}"
+    end    
+  end
+  
+  # remove the application files
+  def drop_files(app_name)    
+    FileUtils.rm_r RailsPwnage::Config.path_to(app_name) if File.exists? RailsPwnage::Config.path_to(app_name)
+  end
+  
+  def backup(app_name, action)
+    case action
+    when :checkpoint
+      dump_files app_name
     when :rollback
+      drop_files app_name
+      load_files app_name
     end
   end
   

data/rails_pwnerer.gemspec

@@ -1,10 +1,10 @@
 
-# Gem::Specification for Rails_pwnerer-0.4
+# Gem::Specification for Rails_pwnerer-0.4.5
 # Originally generated by Echoe
 
 Gem::Specification.new do |s|
   s.name = %q{rails_pwnerer}
-  s.version = "0.4"
+  s.version = "0.4.5"
 
   s.specification_version = 2 if s.respond_to? :specification_version=
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: rails_pwnerer
 version: !ruby/object:Gem::Version 
-  version: "0.4"
+  version: 0.4.5
 platform: ruby
 authors: 
 - Victor Costan