From 56ed245db098491bead6825ffe140cbdb39e7f77 Mon Sep 17 00:00:00 2001
From: Crimer <crimer90@gmail.com>
Date: Thu, 7 Oct 2010 14:29:17 +0000
Subject: [PATCH] MongoMapper now not overrides database name in MONGOHQ_URL
 configuration if provided. It also works on Heroku.

---
 .gitignore                    | 19 +++++++++----------
 config/initializers/_mongo.rb | 11 ++++-------
 2 files changed, 13 insertions(+), 17 deletions(-)

diff --git a/.gitignore b/.gitignore
index 1fbfd2e04..571cd38ae 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,22 +1,21 @@
 .bundle
+.DS_Store
 .idea
+.rvmrc
 *.swap
-*.swp
 *.swo
+*.swp
 *~
-.bundle
+bin/*
+config/app_config.yml
+config/fb_config.yml
+config/initializers/secret_token.rb
 db/*.sqlite3
 log/*
-tmp/**/*
+nbproject
 gpg/diaspora-development/*.gpg
 gpg/diaspora-production/*.gpg
 gpg/*/random_seed
 public/uploads/*
-.rvmrc
-.DS_Store
-config/app_config.yml
-bin/*
-nbproject
-config/initializers/secret_token.rb
 public/source.tar
-config/fb_config.yml
+tmp/**/*
diff --git a/config/initializers/_mongo.rb b/config/initializers/_mongo.rb
index 8948a0c59..7d14315a7 100644
--- a/config/initializers/_mongo.rb
+++ b/config/initializers/_mongo.rb
@@ -2,13 +2,10 @@
 #   licensed under the Affero General Public License version 3.  See
 #   the COPYRIGHT file.
 
-if ENV['MONGOHQ_URL']
-  MongoMapper.config = {RAILS_ENV => {'uri' => ENV['MONGOHQ_URL']}}
-else
-  MongoMapper.connection = Mongo::Connection.new(APP_CONFIG['mongo_host'], APP_CONFIG['mongo_port'])
-end
+ENV['MONGODB_URL'] = ENV['MONGOHQ_URL'] || URI::Generic.build(:scheme => 'mongodb', :host => APP_CONFIG['mongo_host'], :port => APP_CONFIG['mongo_port'], :path => "/diaspora-#{Rails.env}").to_s
 
-MongoMapper.database = "diaspora-#{Rails.env}"
+MongoMapper.config = {RAILS_ENV => {'uri' => ENV['MONGODB_URL']}}
+MongoMapper.connect RAILS_ENV
 
 if defined?(PhusionPassenger)
    PhusionPassenger.on_event(:starting_worker_process) do |forked|
@@ -16,4 +13,4 @@ if defined?(PhusionPassenger)
    end
 end
 
-Magent.connection  = Mongo::Connection.new(APP_CONFIG['mongo_host'], APP_CONFIG['mongo_port'])
+Magent.connection = MongoMapper.connection