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

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