lib/heroku_config/aws_rotate.rb in heroku-config-0.3.0 vs lib/heroku_config/aws_rotate.rb in heroku-config-0.4.0

@@ -1,17 +1,18 @@
 module HerokuConfig
   class AwsRotate < Base
+    include AwsKeyNameMap
     class MaxKeysError < StandardError; end
 
     def initialize(options={})
       @options = options
       @app = options[:app]
     end
 
     def run
-      key_id = config.get("AWS_ACCESS_KEY_ID")
+      key_id = config.get(id_key_name)
       unless key_id
-        puts "WARN: No AWS_ACCESS_KEY_ID found for #{@app.color(:green)} app."
+        puts "WARN: No #{id_key_name} found for #{@app.color(:green)} app."
         if @options[:cli]
           puts "Exiting"
           exit 0
         else
           return