rails_pwnerer 0.3.5 → 0.4

data/CHANGELOG

@@ -1,3 +1,5 @@
+v0.4. Implemented 'backup'.
+
 v0.3.5. Implemented 'go live' and 'go panic' to take all application processes up / down.
 
 v0.3. Supports 'update' to update application from svn. Still no configuration outside of config/main.rb.

data/lib/pwnage/app/database.rb

@@ -63,6 +63,31 @@ ENDSQL
     migrate_database app_name
   end
   
+  # 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
+    when :rollback
+      # TODO: rollback
+    end
+  end
+  
   def control_all(action)
     case action
     when :start
@@ -80,7 +105,11 @@ ENDSQL
     self.new.update(app_name)
   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      
+  end  
 end

data/lib/pwnage/app/main.rb

@@ -15,6 +15,22 @@ module RailsPwnage::App
     end    
     NginxConfig.update(app_name)
   end
+
+  # performs a backup task (checkpoint / release
+  def self.backup(app_name, action = :checkpoint)
+    case action
+      when :checkpoint
+        Svn.backup app_name, action
+        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
+        end
+    end    
+  end
   
   # start or stop all apps
   def self.control_all(action = :start)

data/lib/pwnage/app/svn.rb

@@ -5,7 +5,7 @@ require 'fileutils'
 class RailsPwnage::App::Svn
   include RailsPwnage::Base
   
-  def checkout(svn_path)
+  def svn_checkout(svn_path)
     app_name = File.basename(svn_path)
 
     prod_apps = RailsPwnage::Config.path_to :prod_apps
@@ -17,18 +17,75 @@ class RailsPwnage::App::Svn
     return app_name
   end
   
-  def update(app_name)
+  def svn_update(app_name)
     with_dir(RailsPwnage::Config.path_to(app_name)) do
       system "svn update"
     end
   end
+  
+  def scaffold_backup(app_name)
+    pwnerer_user = RailsPwnage::Config.pwnerer_user
+    pwnerer_uid = uid_for_username(pwnerer_user)
+    pwnerer_gid = gid_for_username(pwnerer_user)
+
+    backup_dir = RailsPwnage::Config.path_to :backup, app_name
+    FileUtils.mkpath backup_dir unless File.exists? backup_dir
+    File.chown(pwnerer_uid, pwnerer_gid, backup_dir)
+    
+    with_dir(RailsPwnage::Config.path_to(:backup, app_name)) do
+      ['db', 'files', 'tmp'].each do |subdir|
+        Dir.mkdir subdir unless File.exists? subdir
+        File.chown pwnerer_uid, pwnerer_gid, subdir
+      end
+    end
+  end
+  
+  def backup(app_name, action)
+    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"
+      
+      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}"
+        end
+        File.chmod(400, dump_file)
+        File.chown(pwnerer_uid, pwnerer_gid, dump_file)
         
+        # clean up
+        FileUtils.rm_r cold_copy
+      end
+    when :rollback
+    end
+  end
+  
   def setup(svn_path)
-    app_name = checkout(svn_path)
+    app_name = svn_checkout(svn_path)
 
     # TODO: fetch configuration, feed to system 
 
-    return app_name    
+    scaffold_backup(app_name)
+
+    return app_name
+  end
+  
+  def update(app_name)
+    scaffold_backup(app_name)
+    
+    # TODO: maybe backup old version before issuing the svn update?
+    
+    svn_update(app_name)
   end
   
   def self.setup(svn_path)
@@ -39,5 +96,9 @@ class RailsPwnage::App::Svn
     self.new.update(app_name)
   end
   
+  def self.backup(app_name, action)
+    self.new.backup(app_name, action)
+  end
+  
   # TODO: uninstallation
 end

data/lib/pwnage/config/main.rb

@@ -1,6 +1,6 @@
 module RailsPwnage::Config
   # the path to something important (e.g. :prod_apps --> path to all production applications)
-  def self.path_to(what = :prod)
+  def self.path_to(what = :prod, app_name = nil)
     case what
     when :prod
       # the directory containing all the production data
@@ -11,12 +11,17 @@ module RailsPwnage::Config
     when :prod_config
       # the directory containing the config files
       '/prod/config'
+    when :prod_backups
+      '/prod/backups'
     when :mongrel_configs
       # the directory containing the mongrel_cluster config files
       '/etc/mongrel_cluster'
     when :nginx_configs
       # the directory containing the nginx config files
       '/etc/nginx/sites-enabled'
+    when :backup
+      # the backup directory for an application
+      '/prod/backups/' + app_name
     when String
       # an application
       self.path_to(:prod_apps) + File::SEPARATOR + what

data/lib/pwnage/executor.rb

@@ -4,38 +4,48 @@ class RailsPwnage::Executor
   # standalone runner
   def run(args)
     case args[0]        
-      when 'scaffold'
-        case args[1]
-        when 'dirs'
-          Dirs.go
-        when 'gems'
-          Gems.go
-        when 'packages'
-          Packages.go
-        when 'mongrels'
-          HookMongrels.go
-        else
-          Packages.go
-          Gems.go
-          Dirs.go
-          HookMongrels.go
-        end
-        
-      when 'install'
-        svn_path = args[1]
-        RailsPwnage::App.install svn_path
-        
-      when 'update'
-        app_name = args[1]
-        RailsPwnage::App.update app_name
-        
-      when 'go'
-        case args[1]
-        when 'live'
-          RailsPwnage::App.control_all :start
-        when 'panic'
-          RailsPwnage::App.control_all :stop
-        end
+    when 'scaffold', 'den00b'
+      case args[1]
+      when 'dirs'
+        Dirs.go
+      when 'gems'
+        Gems.go
+      when 'packages'
+        Packages.go
+      when 'mongrels'
+        HookMongrels.go
+      else
+        Packages.go
+        Gems.go
+        Dirs.go
+        HookMongrels.go
+      end
+      
+    when 'install', 'micro'
+      svn_path = args[1]
+      RailsPwnage::App.install svn_path
+      
+    when 'update', 'ubermicro'
+      app_name = args[1]
+      RailsPwnage::App.update app_name
+      
+    when 'go'
+      case args[1]
+      when 'live', 'pwn'
+        RailsPwnage::App.control_all :start
+      when 'down', 'panic'
+        RailsPwnage::App.control_all :stop
+      else
+        print "Unrecognized go command #{args[1]}\n"
+      end
+      
+    when 'backup', 'checkpoint', 'save'
+      RailsPwnage::App.backup args[1], :checkpoint
+    when 'restore', 'reload', 'recover', 'rollback'
+      RailsPwnage::App.backup args[1], :rollback
+      
+    else
+      print "Unrecognized command #{args[0]}\n"
     end
   end
   

data/lib/pwnage/scaffolds/dirs.rb

@@ -9,9 +9,9 @@ class RailsPwnage::Scaffolds::Dirs
   def run
     pwnerer_uid = uid_for_username(RailsPwnage::Config.pwnerer_user)
     with_dir('/') do
-      [:prod_apps, :prod_config].map { |k| RailsPwnage::Config.path_to k }.each do |path|
+      [:prod_apps, :prod_config, :prod_backups].map { |k| RailsPwnage::Config.path_to k }.each do |path|
         FileUtils.mkpath path
-        File.chown(pwnerer_uid, nil, path)        
+        File.chown(pwnerer_uid, nil, path)
       end
     end
   end

data/rails_pwnerer.gemspec

@@ -1,10 +1,10 @@
 
-# Gem::Specification for Rails_pwnerer-0.3.5
+# Gem::Specification for Rails_pwnerer-0.4
 # Originally generated by Echoe
 
 Gem::Specification.new do |s|
   s.name = %q{rails_pwnerer}
-  s.version = "0.3.5"
+  s.version = "0.4"
 
   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.3.5
+  version: "0.4"
 platform: ruby
 authors: 
 - Victor Costan