heroku-mongo-backup 0.1.0 → 0.1.1

data/lib/heroku_mongo_backup.rb

@@ -1,4 +1,4 @@
-# coding: UTF-8
+# encoding: UTF-8
 
 require 'mongo'
 require 'bson'
@@ -9,7 +9,9 @@ require 'yaml'
 require 'rubygems'
 require 's3'
 
-module Heroku::Mongo
+module HerokuMongoBackup
+  require 'heroku_mongo_backup/railtie' if defined?(Rails)
+
   class Backup
     def chdir
       Dir.chdir("tmp")
@@ -19,30 +21,30 @@ module Heroku::Mongo
       end
       Dir.chdir("dump")
     end
-  
+
     def store
       backup = {}
-    
+  
       @db.collections.each do |col|
         backup['system.indexes.db.name'] = col.db.name if col.name == "system.indexes"
-      
+    
         records = []
-      
+    
         col.find().each do |record|
           records << record
         end
 
         backup[col.name] = records
       end
-    
+  
       marshal_dump = Marshal.dump(backup)
-    
+  
       file = File.new(@file_name, 'w')
       file = Zlib::GzipWriter.new(file)
       file.write marshal_dump
       file.close
     end
-  
+
     def load
       file = Zlib::GzipReader.open(@file_name)
       obj = Marshal.load file.read
@@ -50,15 +52,15 @@ module Heroku::Mongo
 
       obj.each do |col_name, records|
         next if col_name =~ /^system\./
-      
+    
         @db.drop_collection(col_name)
         dest_col = @db.create_collection(col_name)
-      
+    
         records.each do |record|
           dest_col.insert record
         end
       end
-    
+  
       # Load indexes here
       col_name = "system.indexes"
       dest_index_col = @db.collection(col_name)
@@ -69,65 +71,64 @@ module Heroku::Mongo
         end
       end
     end
-  
+
     def connect
       uri = URI.parse(@url)
       connection = ::Mongo::Connection.new(uri.host, uri.port)
       @db = connection.db(uri.path.gsub(/^\//, ''))
       @db.authenticate(uri.user, uri.password) if uri.user
     end
-  
+
     def s3_connect
-      bucket            = YAML.load_file("config/assets.yml")['s3_bucket']
-      access_key_id     = ENV['S3_KEY']
-      secret_access_key = ENV['S3_SECRET']
+      bucket            = ENV['S3_BUCKET']
+      access_key_id     = ENV['S3_KEY_ID']
+      secret_access_key = ENV['S3_SECRET_KEY']
 
       service = S3::Service.new(:access_key_id => access_key_id,
                                 :secret_access_key => secret_access_key)
       @bucket = service.buckets.find(bucket)
     end
-  
+
     def s3_upload
       object = @bucket.objects.build("backups/#{@file_name}")
       object.content = open(@file_name)
       object.save
     end
-  
+
     def s3_download
       open(@file_name, 'w') do |file|
         object = @bucket.objects.find("backups/#{@file_name}")
         file.write object.content
       end
     end
-  
+
     def initialize
-      # Backup settings
       @file_name = Time.now.strftime("%Y-%m-%d_%H-%M-%S.gz")
-    
-      # MongoDB config
-      local_db_name = YAML.load_file("config/mongoid.yml")['development']['database']
-    
-      development_uri = "mongodb://localhost:27017/#{local_db_name}"
-      production_uri = YAML.load_file("config/mongoid.yml")['production']['uri']
-    
-      #@url = development_uri
-      @url = production_uri
-    
+  
+      if ENV[:RAILS_ENV] == 'production'
+        uri = YAML.load_file("config/mongoid.yml")['production']['uri']
+      else
+        config = YAML.load_file("config/mongoid.yml")['development']
+        uri = "mongodb://#{config['host']}:#{config['port']}/#{config['database']}"
+      end
+  
+      @url = uri
+  
       puts "Using databased: #{@url}"
-    
+  
       self.connect
       self.s3_connect
     end
-  
+
     def backup
       self.chdir    
       self.store
       self.s3_upload
     end
-  
+
     def restore file_name
       @file_name = file_name
-    
+  
       self.chdir
       self.s3_download
       self.load

data/lib/heroku_mongo_backup/railtie.rb

@@ -0,0 +1,12 @@
+# encoding: UTF-8
+
+require 'heroku_mongo_backup'
+require 'rails'
+
+module HerokuMongoBackup
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      load "tasks/heroku_mongo_backup.rake"
+    end
+  end
+end

data/lib/tasks/heroku_mongo_backup.rake

@@ -1,15 +1,17 @@
-# coding: UTF-8
+# encoding: UTF-8
 
 namespace :mongo do
-  desc "Mongo backup and restore commands.\n\tBackup: rake mongo:backup\n\tRestore: rake mongo:restore FILE=<backup-file.gz>"
-  
-  task :backup => [:environment] do
-    Heroku::Mongo::Backup.new.backup
+  desc "Backup prodution database and store it on S3.\n
+        Example of usage: rake mongo:backup"
+  task :backup => :environment do
+    HerokuMongoBackup::Backup.new.backup
   end
   
-  task :restore => [:environment] do
+  desc "Restore command gets backup file from S3 server and pushes data to production db.\n
+        Example of usage: rake mongo:restore FILE=<backup-file.gz>"
+  task :restore => :environment do
     if ENV['FILE']
-      Heroku::Mongo::Backup.new.restore ENV['FILE']
+      HerokuMongoBackup::Backup.new.restore ENV['FILE']
     else
       puts "Please provide backup file to restore from. Thanks!"
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: heroku-mongo-backup
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2011-09-13 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70318711197760 !ruby/object:Gem::Requirement
+  requirement: &70358971982280 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70318711197760
+  version_requirements: *70358971982280
 - !ruby/object:Gem::Dependency
   name: mocha
-  requirement: &70318711197280 !ruby/object:Gem::Requirement
+  requirement: &70358971981540 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,13 +32,14 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70318711197280
+  version_requirements: *70358971981540
 description: Rake task for backing up mongo database on heroku and push it to S3.
 email: mail@alexkravets.com
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/heroku_mongo_backup/railtie.rb
 - lib/heroku_mongo_backup.rb
 - lib/tasks/heroku_mongo_backup.rake
 homepage: https://github.com/alexkravets/heroku-mongo-backup