mongo-db-utils 0.0.2 → 0.0.3

data/README.md

@@ -1,17 +1,21 @@
 # MongoDbUtils
 
+## Warning - not safe for production use - undergoing development!
+
 A little gem that simplifies backing up and copying your mongo dbs.
 
+![Sample](https://github.com/edeustace/mongo-db-utils/raw/master/images/sample.png)
+
 It saves your database urls so any task is just a few clicks.
 
 * backup a database
-* copy a database from one server to another (whilst backing up the target db if it exits)
+* copy a database from one server to another (whilst backing up the target db if it exists)
 
 
 
 ## Installation
 
-You need to have mongodump on your path.
+You need to have *mongodump* and *mongorestore* on your path.
 
     gem install 'mongo-db-utils'
 
@@ -33,10 +37,3 @@ When it does backups it stores them in ````~/.mongo-db-utils/backups/````. The n
     rake build
     gem install pkg/mongo-db-utils.gem
 
-## Contributing
-
-1. Fork it
-2. Create your feature branch (`git checkout -b my-new-feature`)
-3. Commit your changes (`git commit -am 'Added some feature'`)
-4. Push to the branch (`git push origin my-new-feature`)
-5. Create new Pull Request

data/lib/mongo-db-utils/cmd/mongotools.rb

@@ -35,6 +35,28 @@ module MongoDbUtils
         `#{cmd}`
 
       end
+
+
+      def self.restore(host,port,db,source_folder,username = "", password = "")
+
+        options = []
+        options << Option.new("-h", "#{host}:#{port}")
+        options << Option.new("-db", db)
+        options << Option.new("-u", username)
+        options << Option.new("-p", password)
+
+        cmd = "mongorestore "
+
+        options.each do |o|
+          cmd << "#{o.key} #{o.value} " unless o.empty?
+        end
+        cmd << "#{source_folder}"
+        puts "cmd:"
+        puts cmd
+        `#{cmd}`
+
+      end
+
     end
   end
 end

data/lib/mongo-db-utils/cmd.rb

@@ -5,58 +5,114 @@ require 'mongo/connection'
 module MongoDbUtils
   class Cmd
 
-    def self.backup(db, path)
-
+    def self.backup(db, folder, final_path = nil, tar_it = true)
+      puts "--"
+      puts ">> Backing up: #{db}, #{folder}, #{final_path}"
       unless( db_exists?(db) )
         return false
       end
 
       t = Time.new
       timestamp = t.strftime("%Y.%m.%d__%H.%M")
-      out_path = "#{path}/#{db.host}_#{db.port}/#{db.name}/#{timestamp}"
+
+      if( final_path.nil? )
+        out_path = "#{folder}/#{db.host}_#{db.port}/#{db.name}/#{timestamp}"
+      else
+        puts "final path not nil out_path: #{out_path}"
+      end
+
       full_path = File.expand_path(out_path)
 
+      puts ">> final backup path: #{full_path}"
+
       FileUtils.mkdir_p(full_path)
-      MongoDbUtils::Commands::MongoTools.dump(db.host,db.port,db.name,full_path,db.username,db.password)
+      MongoDbUtils::Commands::MongoTools.dump(
+        db.host,
+        db.port,
+        db.name,
+        full_path,
+        db.username,
+      db.password)
+
+      if( tar_it )
+        Dir.chdir(full_path)
+        `tar cvf #{db.name}.tar #{db.name}`
+        `rm -fr #{full_path}/#{db.name}`
+      end
 
-      `tar cvf #{full_path}/#{db.name}.tar.gz #{full_path}/#{db.name}`
-      `rm -fr #{full_path}/#{db.name}`
       true
     end
 
+    # With remote dbs you can't do a copy_database if you're not an admin.
+    # so using restore instead
+    #  connection = Mongo::Connection.from_uri(destination.to_s)
+    # mongo_db = connection[destination.name]
+    # if( destination.authentication_required? )
+    #  login_result = mongo_db.authenticate(destination.username, destination.password)
+    # end
+    # host = "#{source.host}:#{source.port}"
+    # connection.copy_database(source.name, destination.name, host, source.username, source.password)
+    #
+    #
     def self.copy(path, source, destination, halt_on_no_backup = true)
 
       backup_made = self.backup(destination, path)
 
-
       if( !backup_made && halt_on_no_backup)
         puts "aborting - no backup was made"
         return
       end
 
-      remove_db(destination)
-      puts "copying... please wait..."
-      connection = Mongo::Connection.from_uri(destination.to_s)
-      source_server = "#{source.host}:#{source.port}"
-      connection.copy_database(source.name,destination.name,source_server,source.username, source.password)
-      connection.close
+      tmp_path = "~/.mongo-db-utils/tmp"
+
+      FileUtils.mkdir_p(tmp_path)
+
+      puts "backup to: #{tmp_path}/#{source.name}"
+      backup(source,tmp_path, source.name, false)
+
+      MongoDbUtils::Commands::MongoTools.restore(
+        destination.host,
+        destination.port,
+        destination.name,
+        "#{tmp_path}/#{source.name}/#{source.name}",
+        destination.username,
+      destination.password)
+
+      `rm -fr #{tmp_path}`
     end
 
 
     private
     def self.db_exists?(db)
+
+      puts "DB exists? #{db.to_s}"
+
       connection = Mongo::Connection.from_uri(db.to_s)
-      contains_db = connection.database_names.include?(db.name)
+      mongo_db = connection[db.name]
+
+      exists = !mongo_db.nil?
+
+      puts "mongo_db: #{mongo_db}"
+      if( db.authentication_required? && exists )
+        login_result = mongo_db.authenticate(db.username, db.password)
+        exists = !login_result.nil?
+      else
+      end
       connection.close
-      contains_db
+      exists
     end
 
     def self.remove_db(db)
-      connection = Mongo::Connection.from_uri(db.to_s)
-      if( connection.database_names.include?(db.name))
+      if( db_exists?(db))
+        connection = Mongo::Connection.from_uri(db.to_s)
+        if( db.authentication_required? )
+          mongo_db = connection[db.name]
+          mongo_db.authenticate(db.username, db.password)
+        end
+
         connection.drop_database(db.name)
+        connection.close
       end
-      connection.close
     end
   end
 end

data/lib/mongo-db-utils/console.rb

@@ -10,6 +10,7 @@ module MongoDbUtils
 Mongo Db Utils - Version: #{MongoDbUtils::VERSION}
 ===================================
 eos
+   
 
     def initialize(config, cmd)
       @config = config
@@ -18,6 +19,7 @@ eos
 
     def run
       say(HEADER)
+      say(MongoDbUtils::READY_FOR_USE)
       main_menu
     end
 
@@ -77,7 +79,8 @@ eos
     def add_config
       entry = Hash.new
       entry[:mongo_uri] = ask("Mongo uri (eg: 'mongodb://user:pass@locahost:27017/db')")
-      successful = @config.add_db_from_uri(entry[:mongo_uri])
+      new_uri = entry[:mongo_uri].gsub(" ", "")
+      successful = @config.add_db_from_uri(new_uri)
 
       if successful
         say("added server")
@@ -99,7 +102,7 @@ eos
       choose do |menu|
         prep_menu(menu)
         @config.dbs.each do |db|
-          menu.choice "#{db.to_s_simple}" do backup(db) end
+          menu.choice "#{db.to_s}" do backup(db) end
         end
         menu.choice "back" do main_menu end
       end
@@ -114,7 +117,7 @@ eos
       choose do |menu|
         prep_menu(menu)
         @config.dbs.sort.each do |db|
-          menu.choice "#{db.to_s_simple}" do 
+          menu.choice "#{db.to_s}" do 
             copy_plan[:source] = db 
           end
         
@@ -130,7 +133,7 @@ eos
       choose do |menu|
         prep_menu(menu)
         @config.dbs.sort.each do |db|
-          menu.choice "#{db.to_s_simple}" do 
+          menu.choice "#{db.to_s}" do 
             copy_plan[:destination] = db
           end unless db == copy_plan[:source]
         end
@@ -141,7 +144,7 @@ eos
 
     def show_copy_plan(plan)
       say("Copy: (we'll backup the destination before we copy)")
-      say("#{plan[:source].to_s_simple} --> #{plan[:destination].to_s_simple}")
+      say("#{plan[:source].to_s} --> #{plan[:destination].to_s}")
 
       choose do |menu|
         prep_menu(menu)

data/lib/mongo-db-utils/models.rb

@@ -73,6 +73,14 @@ module MongoDbUtils
         Db.new(name,host,port,user,pwd)
       end
 
+      def authentication_required?
+        has?(self.username) && has?(self.password)
+      end
+
+      def has?(s)
+        !s.nil? && !s.empty?
+      end
+
 
       def to_s
         user_pass = ""

data/lib/mongo-db-utils/version.rb

@@ -1,3 +1,4 @@
 module MongoDbUtils
-      VERSION = "0.0.2"
+  VERSION = "0.0.3"
+  READY_FOR_USE = "(Alpha version - not for production use!!)"
 end

data/spec/mongo_db_utils_spec.rb

@@ -13,6 +13,8 @@ describe MongoDbUtils::Model do
     db.name.should eql("ed-backup")
     db.username.should eql("")
 
+    db.authentication_required?.should eql(false)
+
   end
 
   it "should parse mongo uris" do
@@ -25,6 +27,7 @@ describe MongoDbUtils::Model do
     db.name.should eql("ed-backup")
     db.username.should eql("ed")
     db.password.should eql("password")
+    db.authentication_required?.should eql(true)
 
   end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongo-db-utils
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-07-22 00:00:00.000000000 Z
+date: 2012-07-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -156,6 +156,7 @@ files:
 - bin/mongo-db-utils
 - features/mongo-db-utils.feature
 - features/support/setup.rb
+- images/sample.png
 - install_rdebug.sh
 - lib/mongo-db-utils/cli.rb
 - lib/mongo-db-utils/cmd.rb