require "stairs"

module Stairs
  module Steps
    class S3 < Stairs::Step
      title "S3"
      description "Setup AWS and S3 bucket access"

      def run
        env key_id_name, provide("AWS Access Key ID")
        env key_secret_name, provide("AWS Secret Access Key")
        env bucket_name, provide("S3 Bucket name")
      end

      private

      def key_id_name
        options[:key_id] || "AWS_ACCESS_KEY_ID"
      end

      def key_secret_name
        options[:key_secret] || "AWS_SECRET_ACCESS_KEY"
      end

      def bucket_name
        options[:bucket] || "AWS_S3_BUCKET"
      end
    end
  end
end

require "stairs"

module Stairs
  module Steps
    class S3 < Stairs::Step
      title "S3"
      description "Setup AWS and S3 bucket access"

      def run
        env "AWS_ACCESS_KEY_ID", provide("AWS Access Key ID")
        env "AWS_SECRET_ACCESS_KEY", provide("AWS Secret Access Key")
        env "AWS_S3_BUCKET", provide("S3 Bucket name")
      end
    end
  end
end

--- 
+++ 
@@ -7,9 +7,23 @@
       description "Setup AWS and S3 bucket access"
 
       def run
-        env "AWS_ACCESS_KEY_ID", provide("AWS Access Key ID")
-        env "AWS_SECRET_ACCESS_KEY", provide("AWS Secret Access Key")
-        env "AWS_S3_BUCKET", provide("S3 Bucket name")
+        env key_id_name, provide("AWS Access Key ID")
+        env key_secret_name, provide("AWS Secret Access Key")
+        env bucket_name, provide("S3 Bucket name")
+      end
+
+      private
+
+      def key_id_name
+        options[:key_id] || "AWS_ACCESS_KEY_ID"
+      end
+
+      def key_secret_name
+        options[:key_secret] || "AWS_SECRET_ACCESS_KEY"
+      end
+
+      def bucket_name
+        options[:bucket] || "AWS_S3_BUCKET"
       end
     end
   end
Add option to control name of env vars
