low 0.0.3 → 0.0.4

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    low (0.0.2)
+    low (0.0.4)
       bson_ext
       mongo
       rack
@@ -9,12 +9,12 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
-    bson (1.7.0)
-    bson_ext (1.7.0)
-      bson (~> 1.7.0)
+    bson (1.6.4)
+    bson_ext (1.6.4)
+      bson (~> 1.6.4)
     diff-lcs (1.1.3)
-    mongo (1.7.0)
-      bson (~> 1.7.0)
+    mongo (1.6.4)
+      bson (~> 1.6.4)
     rack (1.4.1)
     rack-test (0.6.1)
       rack (>= 1.0)

data/lib/low/mongo/util.rb

@@ -0,0 +1,57 @@
+require 'uri'
+require 'mongo'
+
+module Low
+  module Mongo
+    # The `Util` module provides some handy static Mongo helper methods.
+    module Util
+      # Given a `string`, return all the mongdb URIs contained therein.
+      def self.extract_mongodb_uris(string)
+        URI.extract(string, 'mongodb')
+      end
+
+      def self.heroku_remote_mongo
+        # If a mongodb URI can be extracted from heroku config,
+        if uri = extract_mongodb_uris(`heroku config`).first
+
+          # build a Mongo::Remote with it.
+          Mongo::Remote.new(uri)
+        end
+      end
+
+      def self.sync_from_remote(local_database_or_mongo, remote)
+        # If a Mongo::Local is specified,
+        local = local_database_or_mongo.is_a?(Mongo::Local) ?
+
+          # use it,
+          local_database_or_mongo :
+
+          # otherwise, assume that it is a database name and build one.
+          Mongo::Local.new(local_database_or_mongo)
+
+        # Extract host and port from the remote URI,
+        match = remote.uri.match(URI.regexp('mongodb'))
+        remote_host = match[4]
+        remote_port = match[5]
+
+        # and copy the database.
+        local.connection.copy_database(
+          remote.database,
+          local.database,
+          "#{remote_host}:#{remote_port}",
+          remote.username,
+          remote.password
+        )
+      end
+
+      def self.sync_heroku(local_database_or_mongo)
+        # If there is a remote Heroku Mongo,
+        if remote = heroku_remote_mongo
+
+          # sync it to the specified database or Mongo.
+          sync_from_remote(local_database_or_mongo, remote)
+        end
+      end
+    end
+  end
+end

data/lib/low/mongo.rb

@@ -4,6 +4,8 @@ module Low
   # The `Mongo` module defines an interface for a Mongo helper
   # class. It also includes two basic classes, `Local` and `Remote`.
   module Mongo
+    autoload :Util, 'low/mongo/util'
+
     # Simple access to a Mongo::Collection instance.
     def [](collection)
       db[collection]
@@ -33,6 +35,12 @@ module Low
     def database
     end
 
+    def username
+    end
+
+    def password
+    end
+
     # Force a new connection the next time one is needed
     def reset_connection!
       @grid = nil
@@ -44,13 +52,15 @@ module Low
     class Local
       include Mongo
 
-      attr_reader :host, :database
+      attr_reader :host, :database, :username, :password
 
       # Specify the database upon initialization and
       # assume that the host is localhost (unless told otherwise).
-      def initialize(database, host = nil)
+      def initialize(database, opts = {})
         @database = database
-        @host = host || 'localhost'
+        @host = opts[:host] || 'localhost'
+        @username = opts[:username]
+        @password = opts[:password]
       end
     end
 
@@ -70,9 +80,23 @@ module Low
         @connection ||= ::Mongo::Connection.from_uri(uri)
       end
 
-      # The database can be extracted from the URI.
+      # The database can be extracted from the URI,
       def database
-        uri.match(/.*\/(.*)$/)[1]
+        @uri.match(/.*\/(.*)$/)[1]
+      end
+
+      # as can the username,
+      def username
+        if match = @uri.match(/^.*:\/\/(.*?):.*/)
+          match[1]
+        end
+      end
+
+      # and password.
+      def password
+        if match = @uri.match(/^.*:\/\/.*?:(.*?)@.*/)
+          match[1]
+        end
       end
     end
   end

data/lib/low/version.rb

@@ -1,3 +1,3 @@
 module Low
-  VERSION = '0.0.3'
+  VERSION = '0.0.4'
 end

data/spec/low/mongo/util_spec.rb

@@ -0,0 +1,33 @@
+require File.dirname(__FILE__) + '/../../spec_helper'
+
+describe Low::Mongo::Util do
+  describe '.extract_mongodb_uris' do
+    it 'should return an array of mongodb URIs in the specified string' do
+      string = <<-BLOCK
+        GEM_PATH:     vendor/bundle/ruby/1.9.1
+        LANG:         en_US.UTF-8
+        MONGOLAB_URI: mongodb://heroku_app1:secret1@mongolab.com:29827/heroku_app1
+        MONGOHQ_URI:  mongodb://heroku_app2:secret2@mongohq.com:29827/heroku_app2
+        RANDOM_URI:   http://www.gilt.com 
+        PATH:         bin:vendor/bundle/ruby/1.9.1/bin:/usr/local/bin:/usr/bin:/bin
+        RACK_ENV:     production
+      BLOCK
+
+      uris = Low::Mongo::Util.extract_mongodb_uris(string)
+      uris.length.should == 2
+      uris.should include('mongodb://heroku_app1:secret1@mongolab.com:29827/heroku_app1')
+      uris.should include('mongodb://heroku_app2:secret2@mongohq.com:29827/heroku_app2')
+    end
+  end
+
+  describe '.sync_from_remote' do
+    it 'should delegate to #copy_database on the local connection' do
+      local = Low::Mongo::Local.new('low_local')
+      remote = Low::Mongo::Remote.new('mongodb://khy:secret@mongo.com:1234/low_remote')
+      connection = mock(:connection)
+      local.should_receive(:connection).and_return(connection)
+      connection.should_receive(:copy_database).with('low_remote', 'low_local', 'mongo.com:1234', 'khy', 'secret')
+      Low::Mongo::Util.sync_from_remote(local, remote)
+    end
+  end
+end

data/spec/low/mongo_spec.rb

@@ -59,11 +59,35 @@ describe Low::Mongo::Local do
       mongo.host.should == 'localhost'
     end
 
-    it 'should return the value specified at initialization' do
-      mongo = Low::Mongo::Local.new('low_test', '0.0.0.0')
+    it 'should return the value specified by the named parameter at initialization' do
+      mongo = Low::Mongo::Local.new('low_test', host: '0.0.0.0')
       mongo.host.should == '0.0.0.0'
     end
   end
+
+  describe '#username' do
+    it 'should return nil if none is specified at initialization' do
+      mongo = Low::Mongo::Local.new('low_test')
+      mongo.username.should be_nil
+    end
+
+    it 'should return the value specified by the named parameter at initialization' do
+      mongo = Low::Mongo::Local.new('low_test', username: 'khy')
+      mongo.username.should == 'khy'
+    end
+  end
+
+  describe '#password' do
+    it 'should return nil if none is specified at initialization' do
+      mongo = Low::Mongo::Local.new('low_test')
+      mongo.username.should be_nil
+    end
+
+    it 'should return the value specified by the named parameter at initialization' do
+      mongo = Low::Mongo::Local.new('low_test', password: 'secret')
+      mongo.password.should == 'secret'
+    end
+  end
 end
 
 describe Low::Mongo::Remote do
@@ -81,6 +105,30 @@ describe Low::Mongo::Remote do
     end
   end
 
+  describe '#username' do
+    it 'should return nil if the URI does not specify a username' do
+      mongo = Low::Mongo::Remote.new('mongodb://mongo.com/low_remote')
+      mongo.username.should be_nil
+    end
+
+    it 'should return the appropriate value extracted from the URI' do
+      mongo = Low::Mongo::Remote.new('mongodb://khy:secret@mongo.com/low_remote')
+      mongo.username.should == 'khy'
+    end
+  end
+
+  describe '#password' do
+    it 'should return nil if the URI does not specify a password' do
+      mongo = Low::Mongo::Remote.new('mongodb://mongo.com/low_remote')
+      mongo.password.should be_nil
+    end
+
+    it 'should return the appropriate value extracted from the URI' do
+      mongo = Low::Mongo::Remote.new('mongodb://khy:secret@mongo.com/low_remote')
+      mongo.password.should == 'secret'
+    end
+  end
+
   describe '#connection' do
     it 'should pass uri to the ::Mongo::Connection.from_uri method' do
       connection = double('mongo connection')

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: low
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-29 00:00:00.000000000 Z
+date: 2012-09-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rack
@@ -106,9 +106,11 @@ files:
 - Rakefile
 - lib/low.rb
 - lib/low/mongo.rb
+- lib/low/mongo/util.rb
 - lib/low/subdomain_map.rb
 - lib/low/version.rb
 - low.gemspec
+- spec/low/mongo/util_spec.rb
 - spec/low/mongo_spec.rb
 - spec/low/subdomain_map_spec.rb
 - spec/spec_helper.rb
@@ -137,6 +139,7 @@ signing_key:
 specification_version: 3
 summary: A low-level utility library.
 test_files:
+- spec/low/mongo/util_spec.rb
 - spec/low/mongo_spec.rb
 - spec/low/subdomain_map_spec.rb
 - spec/spec_helper.rb