mongo-db-utils 0.0.9 → 0.0.9.2

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

@@ -1,68 +0,0 @@
-require 'aws/s3'
-
-module MongoDbUtils
-  module Commands
-
-    class MongoTools
-
-      # wrapper for monogdump shell command
-      def self.dump(host,port,db,output,username = "", password = "")
-
-        options = []
-        options << o("-h", "#{host}:#{port}")
-        options << o("-db", db)
-        options << o("-o", output)
-        options << o("-u", username)
-        options << o("-p", password)
-
-        cmd = "mongodump "
-
-        options.each do |o|
-          cmd << "#{o.key} #{o.value} " unless o.empty?
-        end
-        `#{cmd}`
-      end
-
-
-      # wrapper for mongorestore shell command
-      def self.restore(host,port,db,source_folder,username = "", password = "")
-
-        options = []
-        options << o("-h", "#{host}:#{port}")
-        options << o("-db", db)
-        options << o("-u", username)
-        options << o("-p", password)
-
-        cmd = "mongorestore "
-
-        options.each do |o|
-          cmd << "#{o.key} #{o.value} " unless o.empty?
-        end
-        #ensure that we drop everything before we restore.
-        cmd << "--drop "
-        cmd << "#{source_folder}"
-        `#{cmd}`
-      end
-
-      private 
-      def self.o(key,value)
-        Option.new(key,value)
-      end
-
-    end
-
-    class Option
-      attr_accessor :key, :value
-
-      def initialize(key,value)
-        @key = key
-        @value = value
-      end
-
-      def empty?
-        @value.nil? || @value.empty?
-      end
-    end
-
-  end
-end

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

@@ -1,161 +0,0 @@
-module MongoDbUtils
-
-  module Model
-    class Config
-      attr_reader :dbs, :buckets
-      attr_writer :writer
-      attr_accessor :backup_folder
-
-      def initialize
-        @dbs = []
-        @buckets = []
-      end
-      
-      def empty?
-        @dbs.nil? || @dbs.empty?
-      end
-
-      def has_buckets?
-        !@buckets.nil? && !@buckets.empty?
-      end
-
-
-      def flush
-        @dbs = []
-        @writer.flush
-      end
-
-      def remove_db(db)
-        @dbs = @dbs - [db]
-        @writer.save(self)
-      end
-
-      def add_db_from_uri(uri)
-        @dbs = [] if @dbs.nil?
-        db = Db.from_uri(uri)
-        unless db.nil? || already_contains(db)
-          @dbs << db
-          @dbs.sort!
-          @writer.save(self)
-        end
-        !db.nil?
-      end
-
-      def already_contains(db)
-        @dbs.each do |existing|
-          if( existing.to_s == db.to_s)
-            return true
-          end
-
-          if( existing.host == db.host && existing.name == db.name)
-            return true
-          end
-
-        end
-        return false
-      end
-
-      # because we are serializing the config - the bucket may be nil
-      # at this point
-      def add_bucket(bucket)
-        @buckets = [] if @buckets.nil?
-        unless bucket.nil? || already_contains_bucket?(bucket)
-          @buckets << bucket
-          @writer.save(self)
-        end
-      end
-
-      def already_contains_bucket?(bucket)
-        puts "@buckets: #{@buckets}"
-        @buckets.each do |b|
-          if( b.to_s == bucket.to_s )
-            return true
-          end
-        end
-        return false
-      end
-
-
-      def save
-        @writer.save(self)
-      end
-
-      def to_s
-        "Config"
-      end
-    end
-
-    class Bucket
-      attr_accessor :name, :access_key, :secret_key
-
-      def to_s
-        "#{name} | #{access_key} | #{secret_key}"
-      end
-
-      def <=> (other)
-        self.name <=> other.name
-      end
-        
-    end
-
-    # A Db stored in the config
-    class Db
-
-      URI_NO_USER = /mongodb:\/\/(.*):(.*)\/(.*$)/
-      URI_USER = /mongodb:\/\/(.*):(.*)@(.*):(.*)\/(.*$)/
-
-      attr_accessor :host, :username, :password, :port, :name
-
-      def initialize(name, host, port, username=nil, password=nil)
-        @host = host
-        @port = port
-        @name = name
-        @username = username
-        @password = password
-      end
-
-      def self.from_uri(uri)
-
-        user,pwd,host,port,db = nil
-
-        if( uri.match(URI_USER))
-          match, user, pwd, host, port, name = *uri.match(URI_USER)
-        elsif(uri.match(URI_NO_USER))
-          match, host, port, name = *uri.match(URI_NO_USER)
-          user = ""
-          pwd = ""
-        end
-
-        return nil if( host.nil? || port.nil? || name.nil? )
-
-        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 = ""
-        unless(@username.empty? || @password.empty? )
-          user_pass = "#{@username}:#{@password}@"
-        end
-        "mongodb://#{user_pass}#{@host}:#{@port}/#{@name}"
-      end
-
-      def to_s_simple
-        "#{@name} on #{@host}:#{@port} - (#{@username}:#{@password})"
-      end
-
-      def <=>(other)
-        self.to_s <=> other.to_s
-      end
-
-    end
-  end
-end

data/spec/config-loader-spec.rb

@@ -1,21 +0,0 @@
-require 'mongo-db-utils/config-loader'
-
-describe MongoDbUtils do
-
-    it "should create a config if one doesn't exist" do
-    
-      tmp_file = ".tmp_path/config.yml"
-
-      FileUtils.rm_rf(".tmp_path")
-
-      File.exist?(tmp_file).should eql(false)
-
-      config = MongoDbUtils::ConfigLoader.load(".tmp_path/config.yml")
-      
-      File.exist?(tmp_file).should eql(true)
-      
-      FileUtils.rm_rf(".tmp_path")
-
-    end
-
-end

data/spec/mongo_db_utils_spec.rb

@@ -1,65 +0,0 @@
-require 'mongo-db-utils/models'
-
-
-describe MongoDbUtils::Model do
-
-  it "should parse mongo uris" do
-
-    uri = "mongodb://localhost:27017/ed-backup"
-    db = MongoDbUtils::Model::Db.from_uri(uri)
-    db.to_s.should eql(uri)
-    db.host.should eql("localhost")
-    db.port.should eql("27017")
-    db.name.should eql("ed-backup")
-    db.username.should eql("")
-
-    db.authentication_required?.should eql(false)
-
-  end
-
-  it "should parse mongo uris" do
-
-    uri = "mongodb://ed:password@localhost:27017/ed-backup"
-    db = MongoDbUtils::Model::Db.from_uri(uri)
-    db.to_s.should eql(uri)
-    db.host.should eql("localhost")
-    db.port.should eql("27017")
-    db.name.should eql("ed-backup")
-    db.username.should eql("ed")
-    db.password.should eql("password")
-    db.authentication_required?.should eql(true)
-
-  end
-
-  it "should return nil if its a bad uri" do
-
-    uri = ""
-    db = MongoDbUtils::Model::Db.from_uri(uri)
-    db.should be(nil)
-  end
-
-  class MockWriter
-    def save(config)
-    end
-
-    def flush
-    end
-  end
-
-
-  it "config should add dbs only if they are different" do
-    config = MongoDbUtils::Model::Config.new
-    config.writer = MockWriter.new
-
-    config.add_db_from_uri("mongodb://blah:3333@server:123/blah")
-    config.dbs.length.should eql(1)
-    config.add_db_from_uri("mongodb://blah:3333@server:123/blah")
-    config.dbs.length.should eql(1)
-    # still the same server + name
-    config.add_db_from_uri("mongodb://server:123/blah")
-    config.dbs.length.should eql(1)
-     config.add_db_from_uri("mongodb://server2:123/blah")
-    config.dbs.length.should eql(2)
-  end
-
-end