handy 0.0.1 → 0.0.2

data/README.md

@@ -0,0 +1,11 @@
+# handy provides follwing tools
+
+##`rake handy:db:restore file=xyz.sql.gz##
+
+restores the data and structure from file 
+
+ 
+* cap production handy:db:pull
+
+
+Copyright (c) 2010 Neeraj Singh. See LICENSE for details.

data/lib/handy/railtie.rb

@@ -1,13 +1,11 @@
-module Handy
-  class Engine < Rails::Engine
+class Engine < Rails::Engine
+
+  initializer :after_initialize do
+  end
 
-    initializer "handy.setup" do
-    end
 
-    rake_tasks do
-      desc "lab99" do
-        puts "You're in my_gem"
-      end
-    end
+  rake_tasks do
+    load 'handy/tasks.rb'
   end
+
 end

data/lib/handy/tasks.rb

@@ -0,0 +1,111 @@
+class Util
+  attr_accessor :username, :password, :database
+  def initialize(*args)
+    @username, @password, @database = *args
+  end
+
+  def self.retrieve_db_info(database_yml_file, env)
+    config = YAML.load_file(database_yml_file)
+    [ config[env]['database'], config[env]['user'], config[env]['password'] ]
+  end
+
+  def mysql_command
+    password.blank? ? "mysql -u #{user} #{database}" : "mysql -u #{user}  -p'#{password}' #{database}"
+  end
+
+  def self.execute_cmd(cmd)
+    puts cmd
+    system cmd
+  end
+end
+
+namespace :handy do
+  namespace :db do
+
+    desc "Load schema and data from a local sql file."
+    task :restore => :environment do
+      puts "Usage: rake handy:db:restore file=xxxxxxxxx.sql[.gz]"
+      file_name = ENV['file']
+      raise "file was not supplied. Check Usage." unless file_name
+      restore_file = File.join(Rails.root, 'tmp', file_name)
+      raise "file was not found" unless File.exists?(restore_file)
+
+      db_info = Util.retrieve_db_info("#{Rails.root}/config/database.yml", Rails.env)
+      database, user, password = db_info
+      util = Util.new(db_info)
+
+      if restore_file =~ /\.gz/
+        puts "decompressing backup"
+        result = system("gzip -d #{restore_file}" )
+        raise("backup decompression failed. msg: #{$?}" ) unless result
+      end
+
+      cmd = "#{util.mysql_command} < #{restore_file.gsub('.gz','')}"
+
+      Util.execute_cmd(cmd)
+      puts "database has been restored"
+    end
+
+    desc "bakup database to a file"
+    task :backup => [:environment] do
+      desc <<-DESC
+      This task backups the database to a file. It will keep a maximum of 10 backed up copies.
+      The files are backed up at shared directory on remote server.
+      This task should be executed before any deploy to production.
+      Files are backed at Rails.root/../../shared/db_backups on the remote server
+      DESC
+
+      timestamp = ENV['timestamp'] || Time.zone.now.strftime("%Y-%m-%d-%H-%M-%S")
+      file_name = "#{timestamp}.sql"
+      backup_dir = File.join (Rails.root, 'tmp')
+
+      backup_file = File.join(backup_dir, "#{file_name}.gz")
+
+      FileUtils.mkdir_p(backup_dir) unless File.exists?(backup_dir) && File.directory?(backup_dir)
+
+      database, user, password = Util.retrieve_db_info("#{Rails.root}/config/database.yml", Rails.env)
+      cmd = "mysqldump --opt --skip-add-locks -u#{user} -p#{password} #{database} >> #{file_name}"
+      Util.execute_cmd(cmd)
+
+      #-c --stdout write on standard output, keep original files unchanged
+      #-q  quite
+      #-9 best compression
+      sh "gzip -q9 #{file_name}"
+      sh "mv #{file_name}.gz  #{backup_file}"
+      puts "Backup done at #{File.expand_path(backup_file)}"
+
+    end
+
+
+    desc "Copy production database to staging database. Or copy data from any database to any other datbase."
+    task :db2db => :environment  do
+      puts "Usage: handy:db:db2db from_env=production to_env=staging"
+      from_env = ENV['from_env'] || 'production'
+      to_env = ENV['to_env'] || 'staging'
+      file_name = "#{Rails.root}/tmp/#{from_env}.data"
+      config_file =  "#{Rails.root}/config/database.yml"
+
+      db_config = YAML.load_file(config_file)
+
+      from_user = db_config[from_env]['username']
+      from_password = db_config[from_env]['password']
+      from_database =  db_config[from_env]['database']
+      from_params = "-Q --add-drop-table -O add-locks=FALSE -O lock-tables=FALSE"
+
+      cmd = "mysqldump -u #{from_user} -p#{from_password} #{from_params} #{from_database} > #{file_name} "
+      puts cmd
+      system cmd
+
+      to_username = db_config[to_env]['username']
+      to_password = db_config[to_env]['password']
+      to_database =  db_config[to_env]['database']
+
+      cmd = "mysql -u #{to_username} -p#{to_password} #{to_database} < #{file_name}"
+      puts cmd
+      system cmd
+
+      puts "#{to_env} database has been restored with #{from_env} database"
+    end
+
+  end
+end

data/lib/handy/version.rb

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

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: handy
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 27
   prerelease: false
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - Neeraj Singh
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-10-01 00:00:00 -04:00
+date: 2010-10-07 00:00:00 -04:00
 default_executable: 
 dependencies: []
 
@@ -27,13 +27,14 @@ extensions: []
 
 extra_rdoc_files: 
 - LICENSE
-- README.rdoc
+- README.md
 files: 
 - LICENSE
-- README.rdoc
+- README.md
 - Rakefile
 - lib/handy.rb
 - lib/handy/railtie.rb
+- lib/handy/tasks.rb
 - lib/handy/version.rb
 - test/helper.rb
 - test/test_handy.rb

data/README.rdoc

@@ -1,17 +0,0 @@
-= handy
-
-Description goes here.
-
-== Note on Patches/Pull Requests
- 
-* Fork the project.
-* Make your feature addition or bug fix.
-* Add tests for it. This is important so I don't break it in a
-  future version unintentionally.
-* Commit, do not mess with rakefile, version, or history.
-  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
-* Send me a pull request. Bonus points for topic branches.
-
-== Copyright
-
-Copyright (c) 2010 Neeraj Singh. See LICENSE for details.