Fix heroku client bits

to reintroduce the methods used by bin/heroku-wait-deploy-scale

Author: meatballhat

.rubocop_todo.yml

@@ -1,6 +1,6 @@
 # This configuration was generated by
 # `rubocop --auto-gen-config`
-# on 2018-11-30 09:10:04 -0500 using RuboCop version 0.60.0.
+# on 2018-12-07 00:06:34 -0500 using RuboCop version 0.60.0.
 # The point is for the user to remove these configuration records
 # one by one as the offenses are removed from the code base.
 # Note that changes in the inspected code, or installation of new
@@ -16,7 +16,7 @@ Metrics/AbcSize:
 Metrics/BlockLength:
   Max: 30
 
-# Offense count: 1
+# Offense count: 2
 # Configuration parameters: CountComments.
 Metrics/ClassLength:
   Max: 138
@@ -25,7 +25,7 @@ Metrics/ClassLength:
 Metrics/CyclomaticComplexity:
   Max: 7
 
-# Offense count: 27
+# Offense count: 30
 # Configuration parameters: CountComments, ExcludedMethods.
 Metrics/MethodLength:
   Max: 81
@@ -55,7 +55,7 @@ Style/RegexpLiteral:
   Exclude:
     - 'bin/travis-worker-verify-config'
 
-# Offense count: 7
+# Offense count: 9
 # Configuration parameters: AllowHeredoc, AllowURI, URISchemes, IgnoreCopDirectives, IgnoredPatterns.
 # URISchemes: http, https
 Metrics/LineLength:

lib/heroku_client.rb

@@ -15,6 +15,74 @@ def initialize(api_key: ENV.fetch('HEROKU_API_KEY', ''),
   private :api_key
   private :api_host
 
+  def wait(heroku_app, timeout: 120)
+    c = 0
+
+    loop do
+      return true if get("/apps/#{heroku_app}")
+      return false if c >= timeout
+
+      sleep 10
+      c += 10
+    end
+
+    true
+  end
+
+  def deploy(repo_slug, heroku_app, version)
+    warn "\nDeploying #{repo_slug} #{version} to #{heroku_app}"
+
+    response = post(
+      JSON.generate(
+        'source_blob' => {
+          'url' => "https://github.com/#{repo_slug}/archive/#{version}.tar.gz",
+          'version' => version
+        }
+      ),
+      "/apps/#{heroku_app}/builds"
+    )
+
+    raise 'Could not request a deployment' if response.nil?
+
+    warn "\nStreaming deploy output"
+
+    raise 'Could not stream deployment output' unless stream(URI(response.fetch('output_stream_url')))
+  end
+
+  def scale(heroku_app, ps_scales)
+    Array(ps_scales).each do |ps_scale|
+      formation = parse_formation(ps_scale)
+      response = patch(
+        JSON.generate(
+          'quantity' => formation.fetch(:qty),
+          'size' => formation.fetch(:size)
+        ),
+        "/apps/#{heroku_app}/formation/#{formation.fetch(:type)}"
+      )
+      raise "Could not scale #{heroku_app} #{ps_scales.inspect}" if response.nil?
+
+      warn "---> scaled #{heroku_app} #{ps_scale}:"
+      warn JSON.pretty_generate(response)
+    end
+  end
+
+  def parse_formation(ps_scale)
+    ret = {
+      type: '',
+      qty: 0,
+      size: ''
+    }
+
+    parts = ps_scale.split('=', 2)
+    ret[:type] = parts.fetch(0)
+
+    subparts = parts.fetch(1).split(':', 2)
+    ret[:qty] = Integer(subparts.fetch(0))
+    ret[:size] = subparts.fetch(1).strip
+
+    ret
+  end
+
   def get(path)
     request(Net::HTTP::Get.new(path))
   end