Contents

require 'heroku-api'
require 'rendezvous'
require 'paratrooper/local_api_key_extractor'

module Paratrooper
  class HerokuWrapper
    attr_reader :api_key, :app_name, :heroku_api, :key_extractor, :rendezvous

    def initialize(app_name, options = {})
      @app_name      = app_name
      @key_extractor = options[:key_extractor] || LocalApiKeyExtractor
      @api_key       = options[:api_key] || key_extractor.get_credentials
      @heroku_api    = options[:heroku_api] || Heroku::API.new(api_key: api_key)
      @rendezvous    = options[:rendezvous] || Rendezvous
    end

    def app_restart
      heroku_api.post_ps_restart(app_name)
    end

    def app_maintenance_off
      app_maintenance('0')
    end

    def app_maintenance_on
      app_maintenance('1')
    end

    def app_url
      app_domain_name
    end

    def run_migrations
      data = heroku_api.post_ps(app_name, 'rake db:migrate', attach: 'true').body
      rendezvous.start(:url => data['rendezvous_url'])
    end

    private
    def app_domain_name
      if custom_domain_response
        custom_domain_response['domain']
      else
        default_domain_name
      end
    end

    def app_maintenance(flag)
      heroku_api.post_app_maintenance(app_name, flag)
    end

    def default_domain_name
      heroku_api.get_app(app_name).body['domain_name']['domain']
    end

    def custom_domain_response
      @custom_domain_response ||= heroku_api.get_domains(app_name).body.last
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
paratrooper-1.3.2	lib/paratrooper/heroku_wrapper.rb
paratrooper-1.3.1	lib/paratrooper/heroku_wrapper.rb