Merge pull request #13 from instacart/ISC-35647

Isc 35647 - add isc lookup for config

Gemfile

@@ -10,3 +10,4 @@ gem "activerecord-import"
 gem "combustion"
 gem "pg"
 gem "pg_query"
+gem "httparty"

lib/pghero.rb

@@ -24,6 +24,7 @@
 require "pghero/database"
 require "pghero/engine" if defined?(Rails)
 require "pghero/version"
+require "pghero/isc"
 
 module PgHero
   autoload :Connection, "pghero/connection"
@@ -135,6 +136,13 @@ def config
             "databases" => databases
           }
         end
+
+        @isc ||= PgHero::ISC.new
+        isc_pghero_config = @isc.pghero_config
+        Rails.logger.info("[PGHero] isc_pghero_config is loaded: #{not isc_pghero_config.nil?}")
+        Rails.logger.info("config is: #{isc_pghero_config}")
+        config.merge!(isc_pghero_config) unless isc_pghero_config.nil?
+
       end
     end
 

lib/pghero/isc.rb

@@ -0,0 +1,21 @@
+require 'httparty'
+
+def authorization
+  return "Bearer #{ENV['ISC_JWT_TOKEN']}" if ENV.has_key? "ISC_JWT_TOKEN"
+  "Token #{ENV['STORAGE_ISC_ACCESS_TOKEN']}"
+end
+
+module PgHero
+  class ISC
+    include HTTParty
+    base_uri "#{ENV['ISC_URL']}"
+    headers "Authorization" => authorization
+    raise_on [404, 400, 500, 503]
+
+    def pghero_config
+      self.class.get("/api/datastore_clusters/pghero_config/", {timeout: 10})
+    rescue
+      nil
+    end
+  end
+end