Contents

require 'heroku-api'
require 'netrc'
require 'json'

require_relative 'helper'

module HerokuRailsSaas
  class HerokuClient
    HEROKU_API_HOST = "api.heroku.com"

    def initialize
      @user, @api_token = netrc[HEROKU_API_HOST]
      @heroku = Heroku::API.new(:api_key => @api_token)
    end

    attr_accessor :user, :heroku, :api_token

  private
    # Redirects method calls to the Heroku::API client, parse the JSON and returns the body.
    def method_missing(method_name, *args, &block)
      begin
        response = @heroku.__send__(method_name.to_sym, *args, &block)
        response.body
      rescue Heroku::API::Errors::ErrorWithResponse => error
        message = error.response.status == 404 ? "#{Helper.yellow(args[0])} does not exists" : JSON.parse(error.response.body)["error"]
        status = error.response.headers["Status"]

        raise <<-OUTPUT
              #{Helper.red(error.class)}:
                Status: #{status}
                Message: #{message}
              OUTPUT
      end
    end

    def netrc # :nodoc:
      @netrc ||= begin
        File.exists?(netrc_path) ? Netrc.read(netrc_path) : raise(StandardError)
      rescue => error
        raise ".netrc missing or no entry found. Try `heroku auth:login`"
      end
    end

    def netrc_path # :nodoc:
      default = Netrc.default_path
      encrypted = default + ".gpg"
      if File.exists?(encrypted)
        encrypted
      else
        default
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
heroku-rails-saas-1.0.3	lib/heroku-rails-saas/heroku_client.rb
heroku-rails-saas-1.0.2	lib/heroku-rails-saas/heroku_client.rb