lib/apipie/generator/swagger/method_description/response_service.rb in apipie-rails-1.3.0 vs lib/apipie/generator/swagger/method_description/response_service.rb in apipie-rails-1.4.0

@@ -37,11 +37,12 @@
           schema: Apipie::Generator::Swagger::MethodDescription::ResponseSchemaService.new(
             response,
             allow_null: false,
             http_method: @http_method,
             controller_method: @method_description
-          ).to_swagger
+          ).to_swagger,
+          headers: response_headers(response.headers)
         }.compact
       end
   end
 
   # @return [Hash]
@@ -52,7 +53,19 @@
       Apipie::Generator::Swagger::Warning::NO_RETURN_CODES_SPECIFIED_CODE,
       @method_description.ruby_name
     ).warn_through_writer
 
     { 200 => { description: 'ok' } }
+  end
+
+  # @param [Array<Hash>] headers
+  #
+  # https://swagger.io/specification/v2/#header-object
+  def response_headers(headers)
+    headers.each_with_object({}) do |header, result|
+      result[header[:name].to_s] = {
+        description: header[:description],
+        type: header[:validator]
+      }
+    end
   end
 end