v.0.1.1

Author: Ruslan Tolstov

lib/rswag_schema_export/capistrano.rb

@@ -1 +1 @@
-load File.expand_path("../tasks/rswag.rake", __FILE__)
+load File.expand_path("tasks/rswag.rake", __dir__)

lib/rswag_schema_export/schema_import.rb

@@ -2,14 +2,12 @@
 
 module RswagSchemaExport
   class Import
-    def run
+    def run(stage = "develop")
       abort("RSWAG_SCHEMA_PATH is not defined. Example: tmp/swagger/swagger.json") unless ENV["RSWAG_SCHEMA_PATH"]
       abort("RSWAG_ACCESS_KEY_ID is not defined") unless ENV["RSWAG_ACCESS_KEY_ID"]
       abort("RSWAG_SECRET_ACCESS_KEY is not defined") unless ENV["RSWAG_SECRET_ACCESS_KEY"]
       abort("RSWAG_REGION is not defined") unless ENV["RSWAG_REGION"]
       abort("RSWAG_BUCKET is not defined") unless ENV["RSWAG_BUCKET"]
-
-      stage = ENV["STAGE"] || "develop"
       app_name = ENV["APP_NAME"] || "app"
 
       begin
@@ -22,7 +20,8 @@ def run
         versions = bucket.objects(prefix: "schemas/#{app_name}/#{stage}_schemas/versions").collect(&:key)
 
         last_schema_key = versions.max
-        bucket.object(last_schema_key).copy_to("#{ENV['RSWAG_BUCKET']}/schemas/#{app_name}/#{stage}_schemas/schema.json")
+        bucket.object(last_schema_key)
+              .copy_to("#{ENV['RSWAG_BUCKET']}/schemas/#{app_name}/#{stage}_schemas/schema.json")
         # Download schema.json
         if block_given?
           bucket.object("schemas/#{app_name}/#{stage}_schemas/schema.json").download_file("schema.json")

lib/rswag_schema_export/tasks/rswag.rake

@@ -1,10 +1,11 @@
-require 'rswag_schema_export/schema_import'
+require "rswag_schema_export/schema_import"
 
 namespace :rswag do
   desc "Import latest schema.json to app"
   task :schema_import do
     on roles(:all) do
-      RswagSchemaExport::Import.new.run do
+      stage = ENV["STAGE"] || fetch(:stage, "develop")
+      RswagSchemaExport::Import.new.run(stage) do
         upload!("schema.json", "#{current_path}/#{ENV['RSWAG_SCHEMA_PATH']}")
       end
     end

lib/rswag_schema_export/version.rb

@@ -1,3 +1,3 @@
 module RswagSchemaExport
-  VERSION = "0.1.0".freeze
+  VERSION = "0.1.1".freeze
 end