yasarg 0.0.1 → 0.0.2

data/VERSION

@@ -1 +1 @@
-0.0.1
+0.0.2

data/lib/yasarg/connection.rb

@@ -0,0 +1,37 @@
+module Yasarg
+  class Connection
+    class << self
+      def establish
+        require File.expand_path('../../yasarg/rails_fake_app',  __FILE__)
+        if !ActiveRecord::Base.connected?
+          ActiveRecord::Base.establish_connection(configuration)
+        end
+      end
+
+      def configuration
+        configuration = {}
+        if ENV["DATABASE_URL"]# heroku
+          configuration = heroku_configuration
+        else
+          configuration = Yasarg::RailsFakeApp.config.database_configuration[Rails.env]
+        end
+
+        configuration
+      end
+
+      def heroku_configuration
+        require "uri"
+        db = URI.parse(ENV['DATABASE_URL'])
+        {
+          :adapter  => db.scheme == 'postgres' ? 'postgresql' : db.scheme,
+          :host     => db.host,
+          :port     => db.port,
+          :username => db.user,
+          :password => db.password,
+          :database => db.path[1..-1],
+          :encoding => 'utf8'
+        }
+      end
+    end
+  end
+end

data/lib/yasarg.rb

@@ -1,5 +1,6 @@
 require "rubygems"
 require "rails"
+require "active_record"
 
 lib = File.expand_path("../", __FILE__)
 $:.unshift lib unless $:.include?(lib)
@@ -8,3 +9,4 @@ APP_PATH = File.expand_path("#{lib}/yasarg/rails_fake_app",  __FILE__)
 
 require "yasarg/rails_fake_app"
 require "yasarg/tasks"
+require "yasarg/connection"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: yasarg
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -61,6 +61,7 @@ files:
 - VERSION
 - bin/yasarg
 - lib/yasarg.rb
+- lib/yasarg/connection.rb
 - lib/yasarg/rails_fake_app.rb
 - lib/yasarg/tasks.rb
 - yasarg.gemspec