noveku 0.2 → 0.3

data/README.md

@@ -23,7 +23,7 @@ This gem exposes two executables :
 * `hrs` (heroku staging) : will execute the given commands for the `staging` remote
 * `hrp` (heroku production) : will execute the given commands for the `production` remote
 
-## Commands
+## Heroku commands
 
 Supported commands and their equivalent : 
 
@@ -32,12 +32,18 @@ Supported commands and their equivalent :
 * `migrate`: `heroku run rake db:migrate --remote ENV && heroku restart --remote ENV`
 * `tail`: `heroku logs --tail --remote ENV`
 
-When giving a command that is not specifically supported, it will be passed to `heroku` : `heroku ARGS --remote ENV".  
+When giving a command that is not specifically supported, it will be passed to `heroku` : `heroku ARGS --remote ENV`.  
 This makes several other commands available, such as `restart`, `releases`, ...
 
-## What's next
+## Advanced commands
 
-Commands that will be soon supported : commands for dumping mongolab database.
+* `mongodump`: Dumps the mongo database. Shortcut for both `mongolab_dump` and `mongohq_dump`: tries mongolab first, then mongohq.
+* `mongolab_dump`: Dumps the mongo database. Look in the config keys of `ENV` to find `MONGOLAB_URI`.
+* `mongohq_dump`: Dumps the mongo database. Look in the config keys of `ENV` to find `MONGOHQ_URL`.
+
+Those commands puts the dump in the `dump` dir, relatively to where you executed it.
+
+Since the restoration of the database does not involve any interaction with heroku, it is out of the scope of this gem at the moment. However, it is pretty easy to integrate the commands above and the restoration in a rake task.
 
 ## Contributions
 
@@ -48,5 +54,6 @@ Commands that will be soon supported : commands for dumping mongolab database.
 
 ## Changelog
 
-* `0.2`: Added rake command
-* `0.1`: First version. Available commands : `console`, `migrate`, `tail`
+* `0.3`: Added `mongodump`, `mongolab_dump`, `mongohq_dump`.
+* `0.2`: Added `rake` command.
+* `0.1`: First version. Available commands : `console`, `migrate`, `tail`.

data/lib/noveku/base.rb

@@ -1,7 +1,10 @@
 module Noveku
+  # Heroku config keys name
+  MONGOLAB_KEY = 'MONGOLAB_URI'
+  MONGOHQ_KEY = 'MONGOHQ_URL'
+
   # This exception will be raised if @environment is not present
-  class NotAValidEnvironment < StandardError
-  end
+  class NotAValidEnvironment < StandardError; end
 
   # Common functionnality
   class Base
@@ -42,6 +45,23 @@ module Noveku
       system "heroku logs --tail --remote #{@environment}"
     end
 
+    # Dump mongolab or mongohq db
+    def mongodump_cmd
+      uri = config_value_for(MONGOLAB_KEY) || config_value_for(MONGOHQ_KEY)
+
+      mongo_dump(uri)
+    end
+
+    # Dump mongolab db
+    def mongolab_dump_cmd
+      mongo_dump config_value_for(MONGOLAB_KEY)
+    end
+
+    # Dump mongohq db
+    def mongohq_dump_cmd
+      mongo_dump config_value_for(MONGOHQ_KEY)
+    end
+
     def method_missing(name, *args, &block)
       # If this is a command with no specific support, pass the raw arguments to `heroku` directly
       if name.to_s.end_with?('_cmd')
@@ -50,5 +70,39 @@ module Noveku
         super
       end
     end
+
+    private
+
+    # Returns config value for key
+    def config_value_for(key)
+      uri = `heroku config:get #{key} --remote #{@environment}`.strip
+      uri = nil if uri == ''
+      uri
+    end
+
+    # Returns a hash of data parsed from the mongo uri
+    def mongo_uri_parse(uri)
+      matches = uri.match /mongodb:\/\/([^:]*):([^@]*)@([^:]*):(\d*)\/(.*)/i
+
+      {
+        user:     matches[1],
+        password: matches[2],
+        host:     matches[3],
+        port:     matches[4],
+        database: matches[5]
+      }
+    end
+
+    # Get a dump of the database
+    def mongo_dump(uri)
+      if uri 
+        opts = mongo_uri_parse(uri)
+        puts "Dumping `#{opts[:database]}` from #{opts[:host]}..."
+        system "mongodump -h #{opts[:host]}:#{opts[:port]} -d #{opts[:database]} -u #{opts[:user]} -p #{opts[:password]}"
+        puts 'Done.'
+      else
+        puts 'No uri.'
+      end
+    end
   end
 end

data/lib/noveku/version.rb

@@ -1,3 +1,3 @@
 module Noveku
-  VERSION = '0.2'
+  VERSION = '0.3'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: noveku
 version: !ruby/object:Gem::Version
-  version: '0.2'
+  version: '0.3'
   prerelease: 
 platform: ruby
 authors: