82 lines
2.5 KiB
Ruby
82 lines
2.5 KiB
Ruby
require 'heroku'
|
|
require 'heroku/client'
|
|
require 'locomotive/hosting/heroku/custom_domain'
|
|
require 'locomotive/hosting/heroku/first_installation'
|
|
|
|
module Locomotive
|
|
module Hosting
|
|
module Heroku
|
|
|
|
extend ActiveSupport::Concern
|
|
|
|
included do
|
|
class << self
|
|
attr_accessor :heroku_connection
|
|
attr_accessor :heroku_domains
|
|
end
|
|
end
|
|
|
|
module ClassMethods
|
|
|
|
def heroku?
|
|
self.config.hosting == :heroku ||
|
|
(self.config.hosting == :auto && ENV['HEROKU_SLUG'].present?)
|
|
end
|
|
|
|
def enable_heroku
|
|
self.config.domain = 'heroku.com' unless self.config.multi_sites?
|
|
|
|
self.config.heroku ||= {}
|
|
self.config.heroku[:name] = ENV['APP_NAME']
|
|
|
|
raise 'Heroku application name is mandatory' if self.config.heroku[:name].blank?
|
|
|
|
self.open_heroku_connection
|
|
|
|
self.enhance_site_model_with_heroku
|
|
|
|
self.apply_patches
|
|
|
|
# "cache" domains for better performance
|
|
self.heroku_domains = self.heroku_connection.list_domains(self.config.heroku[:name]).collect { |h| h[:domain] }
|
|
end
|
|
|
|
def open_heroku_connection
|
|
login = self.config.heroku[:login] || ENV['HEROKU_LOGIN']
|
|
password = self.config.heroku[:password] || ENV['HEROKU_PASSWORD']
|
|
|
|
self.heroku_connection = ::Heroku::Client.new(login, password)
|
|
end
|
|
|
|
def enhance_site_model_with_heroku
|
|
Site.send :include, Locomotive::Hosting::Heroku::CustomDomain
|
|
Site.send :include, Locomotive::Hosting::Heroku::FirstInstallation
|
|
end
|
|
|
|
def apply_patches
|
|
# for various reasons, Heroku can modify the behaviour of an application by changing the gem versions (json/pure for instance)
|
|
# so the purpose of this method is to correct those potential differences.
|
|
|
|
# http://blog.ethanvizitei.com/2010/11/json-pure-ruins-my-morning.html
|
|
Fixnum.class_eval { def to_json(options = nil); to_s; end }
|
|
end
|
|
|
|
# manage domains
|
|
|
|
def add_heroku_domain(name)
|
|
Locomotive.logger "[add heroku domain] #{name}"
|
|
self.heroku_connection.add_domain(self.config.heroku[:name], name)
|
|
self.heroku_domains << name
|
|
end
|
|
|
|
def remove_heroku_domain(name)
|
|
Locomotive.logger "[remove heroku domain] #{name}"
|
|
self.heroku_connection.remove_domain(self.config.heroku[:name], name)
|
|
self.heroku_domains.delete(name)
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
end |