rammer 1.1.0

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in rammer.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 manishaharidas
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,65 @@
+# Rammer
+
+Rammer is a framework dedicated to build high performance Async API servers on top of non-blocking (asynchronous) Ruby web server called Goliath. Rammer APIs are designed on top of REST-like API micro-framework Grape. Rammer is modular and integrates a powerful CLI called Viber to plug in and out its modules.
+
+## Installation
+
+Install it yourself as:
+
+    $ gem install rammer
+
+## Usage
+
+* rammer application_name
+* cd application_name
+* bundle install 	(To install all dependencies)
+* ruby server.rb -vs 	(To run the Goliath server)
+
+To include rammer modules into application use command viber:
+
+	$ viber module <COMMAND> <MODULE_NAME>
+
+COMMAND : 
+* -p or -plugin (Plugs in the specified module)
+* -u or -unplug (Unplugs the specified module)
+
+MODULE_NAME : 
+* authentication (Includes endpoint : sign_up)
+* authorization  (Includes endpoint : sign_in, sign_out)
+* oauth          (Includes endpoint : register_client, authorize, access_token, token, invalidate_token)
+
+Endpoints enabled with respective module:
+
+- Authentication
+* sign_up 			: "Enables user authentication using email, MD5 password and redirect url returning session token."
+
+- Authorization
+* sign_in 			: "Enables user login using credentials returning session token."
+* sign_out 			: "Enables user logout using credentials by invalidating the session token."
+
+- Oauth
+* register_client 	: "Registers ouath client using credentials returning client details like ID, Secret hash, authorization code."
+* authorize 		: "Enables functionality for activating authorization page access."
+* access_token 		: "Returns oauth access token for registered clients to authorized users."
+* token 			: "Returns bearer token to registered third party clients."
+* invalidate_token 	: "Invalidated the issued bearer token."
+
+## Contributors
+
+This list is open to all. You are all welcome :).
+
+* manishaharidas (Manisha Haridas) - 
+  * Github: https://github.com/manishaharidas
+
+## Comments/Requests
+
+If anyone has comments or questions please let me know (qbruby@qburst.com).
+If you have updates or patches or want to contribute I would love to see what you have or want to add.
+
+
+## Note on Patches/Pull Requests
+
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a future version unintentionally.
+* Send me a pull request. Bonus points for topic branches.

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/rammer

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+require 'rammer'
+
+generator = Rammer::Generator.new(*ARGV)
+generator.run

data/bin/viber

@@ -0,0 +1,73 @@
+#!/usr/bin/env ruby
+
+require 'rammer'
+
+def parse_options
+    case ARGV[1]
+    when "-p", "-plugin"
+        return true
+    when "-u", "-unplug"
+        return true
+    else
+        return false
+    end
+end
+
+def options_message
+STDOUT.puts <<-EOF
+Please include module name
+
+Usage:
+  viber module #{ARGV[1]}
+
+  options:
+    authentication 
+    authorization 
+    oauth 
+EOF
+end
+
+def execute(module_name,action)
+    target_dir = Dir.pwd.split('/',Dir.pwd.count('/')+1).last
+    name = target_dir.split.map(&:capitalize)*' '
+    case module_name
+    when "authentication"
+        module_class = "::#{name}::AuthenticationApis"
+    when "authorization"
+        module_class = "::#{name}::AuthorizationApis"
+    when "oauth"
+        module_class = "::#{name}::OauthApis"
+    end
+    options = { :target_dir => target_dir, :module_class => module_class, 
+                :module_name => module_name, :action => action}
+    module_generator = Rammer::ModuleGenerator.new(options)
+    module_generator.run
+end
+
+case ARGV[0]
+when "module"
+    if parse_options
+        case ARGV[2]
+        when "authentication"
+            execute("authentication",ARGV[1])
+        when "authorization"
+            execute("authorization",ARGV[1])
+        when "oauth"
+            execute("oauth",ARGV[1])
+        else
+            options_message
+            exit
+        end
+    else
+        STDOUT.puts <<-EOF
+Please provide command options
+
+Usage:
+  viber module 
+
+  options:
+    -p,-plugin      :Plugs in rammer specified modules. 
+    -u,-unplug      :Unplugs rammer specified modules.
+EOF
+    end    
+end

data/lib/rammer/version.rb

@@ -0,0 +1,3 @@
+module Rammer
+  VERSION = "1.1.0"
+end

data/lib/rammer.rb

@@ -0,0 +1,287 @@
+require "rammer/version"
+require 'fileutils'
+$gem_file_name = "rammer-"+Rammer::VERSION
+module Rammer
+
+  	class Generator
+	  	attr_accessor :target_dir, :project_name, :module_name, :gem_path
+        BASE_DIR = ['app', 'app/apis', 'config', 'db', 'db/migrate', 'app/models']          
+    	TEMPLATE_FILES = ['Gemfile','Gemfile.lock','Procfile','Rakefile','server.rb', 'tree.rb']
+    	def initialize(dir_name)
+			self.project_name   = dir_name
+			if self.project_name.nil? || self.project_name.squeeze.strip == ""
+		        raise NoGitHubRepoNameGiven
+		    else
+		        path = File.split(self.project_name)
+
+		        if path.size > 1
+		          extracted_directory = File.join(path[0..-1])
+		          self.project_name = path.last
+		        end
+		    end
+		    self.target_dir  =  self.project_name
+		    path = `gem which rammer`
+	  		p self.gem_path = path.split($gem_file_name,2).first
+		end
+
+	    def run
+	      unless File.exists?(target_dir) || File.directory?(target_dir)
+	      	$stdout.puts "Creating goliath application under the directory #{target_dir}"
+	        FileUtils.mkdir target_dir
+	        create_base_dirs
+	        copy_files_to_target
+	        setup_api_module
+	        copy_files_to_dir 'application.rb','config'
+	        copy_files_to_dir 'database.yml','config'
+	        $stdout.puts "\e[33mRun `bundle install` to install missing gems.\e[0m"
+	      else
+	       	$stdout.puts "\e[1;31mError:\e[0m The directory #{target_dir} already exists, aborting. Maybe move it out of the way before continuing."
+	      end
+	    end
+
+	    private
+
+	    def create_base_dirs
+	    	BASE_DIR.each do |dir|
+	    		FileUtils.mkdir "#{target_dir}/#{dir}"
+	    		$stdout.puts "\e[1;32m \tcreate\e[0m\t#{dir}"
+	    	end
+	    	FileUtils.mkdir "#{target_dir}/app/apis/#{target_dir}"
+	    	$stdout.puts "\e[1;32m \tcreate\e[0m\tapp/apis/#{target_dir}"
+	    end
+
+	    def setup_api_module
+	    	self.module_name = target_dir.split('_').map(&:capitalize).join('')
+	    	create_api_module
+	    	config_server
+	    end
+
+	    def create_api_module
+	    	File.open("#{target_dir}/app/apis/#{target_dir}/base.rb", "w") do |f|
+	    		f.write('module ') 
+	    		f.puts(module_name)
+	    		f.write("\tclass Base < Grape::API\n\tend\nend")
+	    	end
+	    	$stdout.puts "\e[1;32m \tcreate\e[0m\tapp/apis/#{target_dir}/base.rb"
+	    end
+
+	    def config_server
+	    	file = File.open("#{target_dir}/server.rb", "r+")
+	    	file.each do |line|	   
+		        while line == "\tdef response(env)\n" do
+		            pos = file.pos
+		            rest = file.read
+		            file.seek pos
+		            file.write("\t\t::") 
+		            file.write(module_name)
+		            file.write("::Base.call(env)\n")
+		            file.write(rest)
+		            return
+		        end
+			end
+			$stdout.puts "\e[1;32m \tcreate\e[0m\tserver.rb"
+	    end
+
+	    def copy_files_to_target
+	    	TEMPLATE_FILES.each do |file|
+	    		source = File.join("#{gem_path}/#{$gem_file_name}/lib/template/",file)
+		    	FileUtils.cp(source,"#{target_dir}")
+		        $stdout.puts "\e[1;32m \tcreate\e[0m\t#{file}"
+		    end
+	    end
+
+	    def copy_files_to_dir(file,destination)
+	    	FileUtils.cp("#{gem_path}/#{$gem_file_name}/lib/template/#{file}","#{target_dir}/#{destination}")
+	        $stdout.puts "\e[1;32m \tcreate\e[0m\t#{destination}/#{file}"
+	    end
+  	end
+
+  	class ModuleGenerator
+	  	attr_accessor :target_dir, :module_class, :options, :module_name, :gem_path, :action
+	  	AUTH_MIGRATE = ['01_create_users.rb','02_create_sessions.rb']
+	    OAUTH_MIGRATE = ['03_create_owners.rb', '04_create_oauth2_authorizations.rb', '05_create_oauth2_clients.rb']
+	    AUTH_MODELS = ['user.rb', 'session.rb', 'oauth2_authorization.rb']
+	    OAUTH_MODELS = ['oauth2_client.rb', 'owner.rb']
+	  	def initialize(options)
+	  		self.options = options
+	  		self.target_dir = options[:target_dir]
+	  		self.module_class = options[:module_class]
+	  		self.module_name = options[:module_name]
+	  		self.action = options[:action]
+	  		path = `gem which rammer`
+	  		self.gem_path = path.split($gem_file_name,2).first
+	  	end
+
+	  	def run
+	  		case action
+	  		when "-p","-plugin"
+		  		flag = require_module_to_base
+		  		mount_module unless flag
+				copy_module
+				create_migrations
+				copy_model_files
+				add_gems
+				oauth_message if module_name == "oauth" && !flag
+			when "-u","-unplug"
+				unmount_module
+			end
+		end
+
+		def mount_module
+			file = File.open("#{Dir.pwd}/app/apis/#{target_dir}/base.rb", "r+")
+			file.each do |line|
+			    while line == "\tclass Base < Grape::API\n" do
+			        pos = file.pos
+			        rest = file.read
+			        file.seek pos
+			        file.write("\t\tmount ") 
+			        file.puts(module_class)
+			        file.write(rest)
+			        break
+			    end
+			end
+			$stdout.puts "\e[1;35m\tmounted\e[0m\t#{module_class}"
+		end
+
+		def require_module_to_base
+			file = File.open("#{Dir.pwd}/app/apis/#{target_dir}/base.rb", "r+")
+	  		file.each do |line|
+	  			while line == "require_relative './modules/#{module_name}_apis'\n" do
+	  				$stdout.puts "\e[33mModule already mounted.\e[0m"
+	  				return true
+	  			end
+	  		end
+			File.open("#{Dir.pwd}/app/apis/#{target_dir}/base.rb", "r+") do |f|	
+				pos = f.pos		
+				rest = f.read
+				f.seek pos
+				f.write("require_relative './modules/") 
+		        f.write(module_name)
+		        f.write("_apis'\n")
+		        f.write(rest)
+			end
+			return false
+		end
+
+		def copy_module		
+			src = "#{gem_path}/#{$gem_file_name}/lib/template/#{module_name}_apis.rb"
+			dest = "#{Dir.pwd}/app/apis/#{target_dir}/modules"
+			presence = File.exists?("#{dest}/#{module_name}_apis.rb")? true : false
+			FileUtils.mkdir dest unless File.exists?(dest)
+			FileUtils.cp(src,dest) unless presence
+			configure_module_files
+			$stdout.puts "\e[1;32m \tcreate\e[0m\tapp/apis/#{target_dir}/modules/#{module_name}_apis.rb" unless presence
+	  	end
+
+	  	def create_migrations
+	  		src = "#{gem_path}/#{$gem_file_name}/lib/template"
+	  		dest = "#{Dir.pwd}/db/migrate"
+	  		case module_name
+	  		when "authentication", "authorization"  	
+	  			common_migrations(src,dest)
+	  		when "oauth"
+	  			common_migrations(src,dest)
+	  			OAUTH_MIGRATE.each do |file|		
+	  				presence = File.exists?("#{dest}/#{file}")? true : false
+	  				unless presence
+		  				FileUtils.cp("#{src}/#{file}",dest)
+		  				$stdout.puts "\e[1;32m \tcreate\e[0m\tdb/migrate/#{file}"
+	  				end
+	  			end
+	  		end
+	  	end
+
+	  	def common_migrations(src,dest)
+	  		AUTH_MIGRATE.each do |file|		
+  				presence = File.exists?("#{dest}/#{file}")? true : false
+  				unless presence
+	  				FileUtils.cp("#{src}/#{file}",dest)
+	  				$stdout.puts "\e[1;32m \tcreate\e[0m\tdb/migrate/#{file}"
+  				end
+  			end
+	  	end
+
+	  	def common_models(src,dest)
+	  		AUTH_MODELS.each do |file|
+  				presence = File.exists?("#{dest}/#{file}")? true : false
+  				unless presence
+	  				FileUtils.cp("#{src}/#{file}",dest)
+	  				$stdout.puts "\e[1;32m \tcreate\e[0m\tapp/models/#{file}"
+  				end
+  			end
+	  	end
+
+	  	def copy_model_files
+	  		src = "#{gem_path}/#{$gem_file_name}/lib/template"
+	  		dest = "#{Dir.pwd}/app/models"
+	  		case module_name
+	  		when "authentication", "authorization"  	
+	  			common_models(src,dest)
+	  		when "oauth"
+	  			common_models(src,dest)
+	  			OAUTH_MODELS.each do |file|		
+	  				presence = File.exists?("#{dest}/#{file}")? true : false
+	  				unless presence
+		  				FileUtils.cp("#{src}/#{file}",dest)
+		  				$stdout.puts "\e[1;32m \tcreate\e[0m\tapp/models/#{file}"
+	  				end
+	  			end
+	  		end
+	  	end
+
+	  	def configure_module_files
+	  		file = File.read("#{Dir.pwd}/app/apis/#{target_dir}/modules/#{module_name}_apis.rb")
+	  		replace = file.gsub(/module GrapeGoliath/, "module #{target_dir.split('_').map(&:capitalize)*''}")
+	  		File.open("#{Dir.pwd}/app/apis/#{target_dir}/modules/#{module_name}_apis.rb", "w"){|f|
+	  			f.puts replace
+	  		}
+	  	end
+
+	  	def add_gems
+	  		file = File.open("#{Dir.pwd}/Gemfile", "r+")
+	  		file.each do |line|
+	  			while line == "gem 'oauth2'\n" do
+	  				return
+	  			end
+	  		end
+	  		File.open("#{Dir.pwd}/Gemfile", "a+") do |f|
+	  			f.write("gem 'multi_json'\ngem 'oauth2'\ngem 'songkick-oauth2-provider'\ngem 'ruby_regex'\ngem 'oauth'\n")
+	  		end
+  			$stdout.puts "\e[1;35m \tGemfile\e[0m\tgem 'multi_json'\n\t\tgem 'oauth2'
+\t\tgem 'songkick-oauth2-provider'\n\t\tgem 'ruby_regex'\n\t\tgem 'oauth'\n"
+			$stdout.puts "\e[1;32m \trun\e[0m\tbundle install"
+			system("bundle install")
+	  	end
+
+	  	def unmount_module
+	  		temp_file = "#{Dir.pwd}/app/apis/#{target_dir}/tmp.rb"
+	  		source = "#{Dir.pwd}/app/apis/#{target_dir}/base.rb"
+	  		delete_file = "#{Dir.pwd}/app/apis/#{target_dir}/modules/#{module_name}_apis.rb"
+	  		File.open(temp_file, "w") do |out_file|
+	  			File.foreach(source) do |line|
+	  				unless line == "require_relative './modules/#{module_name}_apis'\n"
+    					out_file.puts line unless line == "\t\tmount #{module_class}\n"
+    				end
+  				end
+  				FileUtils.mv(temp_file, source)
+  			end
+  			if File.exists?(delete_file)
+	  			FileUtils.rm(delete_file) 
+	  			$stdout.puts "\e[1;35m\tunmounted\e[0m\t#{module_class}" 
+	  			$stdout.puts "\e[1;31m\tdelete\e[0m\t\tapp/apis/#{target_dir}/modules/#{module_name}_apis.rb" 
+	  		else
+	  			$stdout.puts "\e[33mModule already unmounted.\e[0m"
+	  		end
+	  	end
+
+	  	def oauth_message
+$stdout.puts "\e[33m
+In app/apis/<APP_NAME>/modules/oauth_apis.rb
+	Specify redirection url to the respective authorization page into 'redirect_to_url'
+and uncomment the code to enable /oauth/authorize endpoint functionality.
+
+\e[0m"
+	  	end
+  	end
+end
+

data/lib/template/01_create_users.rb

@@ -0,0 +1,44 @@
+class CreateUsers < ActiveRecord::Migration
+  def change
+    create_table(:users) do |t|
+      t.string :email,              :null => false, :default => ""
+      t.string :encrypted_password, :null => false, :default => ""
+
+      ## Recoverable
+      t.string   :reset_password_token
+      t.datetime :reset_password_sent_at
+
+      ## Rememberable
+      t.datetime :remember_created_at
+
+      ## Trackable
+      t.integer  :sign_in_count, :default => 0, :null => false
+      t.datetime :current_sign_in_at
+      t.datetime :last_sign_in_at
+      t.string   :current_sign_in_ip
+      t.string   :last_sign_in_ip
+
+      ## Confirmable
+      # t.string   :confirmation_token
+      # t.datetime :confirmed_at
+      # t.datetime :confirmation_sent_at
+      # t.string   :unconfirmed_email # Only if using reconfirmable
+
+      ## Lockable
+      # t.integer  :failed_attempts, :default => 0, :null => false # Only if lock strategy is :failed_attempts
+      # t.string   :unlock_token # Only if unlock strategy is :email or :both
+      # t.datetime :locked_at
+
+      ## Token authenticatable
+      # t.string :authentication_token
+
+      t.timestamps
+    end
+
+    add_index  :users, :email,                :unique => true
+    add_index  :users, :reset_password_token, :unique => true 
+    # add_index  :users, :authentication_token, :unique => true
+    # add_index :operators, :confirmation_token,   :unique => true
+    # add_index :operators, :unlock_token,         :unique => true
+  end
+end

data/lib/template/02_create_sessions.rb

@@ -0,0 +1,9 @@
+class CreateSessions < ActiveRecord::Migration
+	def change
+    	create_table(:sessions) do |t|
+    		t.string :user_id
+    		t.string :session_token
+    	end
+    	add_index  :sessions, :session_token, :unique => true
+	end
+end

data/lib/template/03_create_owners.rb

@@ -0,0 +1,13 @@
+class CreateOwner < ActiveRecord::Migration
+	def self.up
+	    create_table :owners do |t|
+	      t.timestamps
+	      t.string     :username
+	     end
+		add_index :owners, [:username]
+ 	end
+
+ 	def self.down
+	    drop_table :owners
+  	end
+end

data/lib/template/04_create_oauth2_authorizations.rb

@@ -0,0 +1,23 @@
+class CreateOauth2Authorizations < ActiveRecord::Migration
+  def self.up
+    create_table :oauth2_authorizations do |t|
+      t.timestamps
+      t.string     :oauth2_resource_owner_type
+      t.integer    :oauth2_resource_owner_id
+      t.belongs_to :oauth2_client
+      t.string     :scope
+      t.string     :code,               :limit => 40
+      t.string     :access_token,  :limit => 40
+      t.string     :refresh_token, :limit => 40
+      t.datetime   :expires_at
+    end
+    add_index :oauth2_authorizations, [:oauth2_client_id, :code]
+    add_index :oauth2_authorizations, [:access_token]
+    add_index :oauth2_authorizations, [:oauth2_client_id, :access_token], :name => 'access_token_index'
+    add_index :oauth2_authorizations, [:oauth2_client_id, :refresh_token], :name => 'refresh_token_index'
+  end
+
+  def self.down
+    drop_table :oauth2_authorizations
+  end
+end

data/lib/template/05_create_oauth2_clients.rb

@@ -0,0 +1,18 @@
+class CreateOauth2Clients < ActiveRecord::Migration
+  def self.up
+    create_table :oauth2_clients do |t|
+      t.string     :name
+      t.string     :client_id
+      t.string     :client_secret_hash
+      t.string     :redirect_uri
+      t.string     :basic_code
+      t.timestamps
+    end
+    add_index :oauth2_clients, :client_id, :unique => true
+  end
+
+  def self.down
+    drop_table :oauth2_clients
+  end
+
+end

data/lib/template/Gemfile

@@ -0,0 +1,16 @@
+source "http://rubygems.org"
+
+
+gem 'pg'
+gem 'em-postgresql-adapter', :git => 'git://github.com/leftbee/em-postgresql-adapter.git'
+gem 'rack-fiber_pool',  :require => 'rack/fiber_pool'
+gem 'em-synchrony', :git => 'git://github.com/igrigorik/em-synchrony.git',
+                    :require => ['em-synchrony', 'em-synchrony/activerecord', 'em-synchrony/mysql2']
+                            
+gem 'grape'
+gem 'goliath'
+
+gem "activerecord", "~> 3.1.1"
+gem 'rack-fiber_pool',  :require => 'rack/fiber_pool'
+gem "mysql2"                    
+

data/lib/template/Procfile

@@ -0,0 +1 @@
+web: bundle exec ruby server.rb -sv -e prod -p $PORT

data/lib/template/Rakefile

@@ -0,0 +1,47 @@
+require "rubygems"
+require "bundler/setup"
+require 'em-synchrony/activerecord'
+require 'yaml'
+require 'erb'
+
+namespace :db do
+  desc "loads database configuration in for other tasks to run"
+  task :load_config do
+    ActiveRecord::Base.configurations = db_conf
+    
+    # drop and create need to be performed with a connection to the 'postgres' (system) database
+    ActiveRecord::Base.establish_connection db_conf["production"].merge('database' => 'postgres',
+                                                                       'schema_search_path' => 'public')
+  end
+  
+  desc "creates and migrates your database"
+  task :setup => :load_config do
+    Rake::Task["db:create"].invoke
+    Rake::Task["db:migrate"].invoke
+  end
+  
+  desc "migrate your database"
+  task :migrate do
+    ActiveRecord::Base.establish_connection db_conf["production"]
+
+    ActiveRecord::Migrator.migrate(
+      ActiveRecord::Migrator.migrations_paths, 
+      ENV["VERSION"] ? ENV["VERSION"].to_i : nil
+    )
+  end
+  
+  desc 'Drops the database'
+  task :drop => :load_config do
+    ActiveRecord::Base.connection.drop_database db_conf['production']['database']
+  end
+  
+  desc 'Creates the database'
+  task :create => :load_config do
+    ActiveRecord::Base.connection.create_database db_conf['production']['database']
+  end
+  
+end
+
+def db_conf
+  config = YAML.load(ERB.new(File.read('config/database.yml')).result)
+end

data/lib/template/application.rb

@@ -0,0 +1,21 @@
+require 'uri'
+require 'em-synchrony/activerecord'
+require 'yaml'
+require 'erb'
+
+# Sets up database configuration
+db = URI.parse(ENV['DATABASE_URL'] || 'http://localhost')
+if db.scheme == 'postgres' # Heroku environment
+  ActiveRecord::Base.establish_connection(
+    :adapter  => db.scheme == 'postgres' ? 'em_postgresql' : db.scheme,
+    :host     => db.host,
+    :username => db.user,
+    :password => db.password,
+    :database => db.path[1..-1],
+    :encoding => 'utf8'
+  )
+else # local environment
+  environment = ENV['DATABASE_URL'] ? 'production' : 'development'
+  db = YAML.load(ERB.new(File.read('config/database.yml')).result)[environment]
+  ActiveRecord::Base.establish_connection(db)
+end