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

@@ -1,17 +1,24 @@
 module HerokuConfig
   class CLI < Command
     class_option :verbose, type: :boolean
     class_option :noop, type: :boolean
 
+    key_name_options = Proc.new do
+      option :id_key_name, default: "AWS_ACCESS_KEY_ID", desc: "Heroku config variable name for AWS_ACCESS_KEY_ID"
+      option :secret_key_name, default: "AWS_SECRET_ACCESS_KEY", desc: "Heroku config variable name for AWS_SECRET_ACCESS_KEY"
+    end
+
     desc "aws-rotate APP", "Rotates AWS key for app"
     long_desc Help.text(:aws_rotate)
+    key_name_options.call
     def aws_rotate(app)
       AwsRotate.new(options.merge(app: app, cli: true)).run
     end
 
     desc "aws-rotate-all FILE", "Rotates AWS key for list of apps"
     long_desc Help.text(:aws_rotate_all)
+    key_name_options.call
     def aws_rotate_all(file)
       AwsRotateAll.new(options.merge(file: file)).run
     end
 
     desc "completion *PARAMS", "Prints words for auto-completion."