rswag-api 2.1.1 → 2.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fdb4e3352a6d79128a80330791bfe3a82561c4e658c2e531316368098ddc2d26
-  data.tar.gz: b21ab1fd96389fb193f9fd487c214b2ef6f0a30f010e9a07c9a2ab4ebb6f2ae9
+  metadata.gz: 8f5a77aa4988d76dcd76d251288729514f07fc0e4cb28bb04a184d6be53001f6
+  data.tar.gz: f13c18e015741835531ed97014fffb3d0ba647044ce948ea1d334852f4a64f0b
 SHA512:
-  metadata.gz: 3dd345e95c82de2c375b71bbe4b443ee78f3648924ef12d84d44a9a1a144df3abbc40d35e57497eb58b94927890d997df2981aa4ced47ad75055b4e824196124
-  data.tar.gz: 3091da2ac5c5c54acdb23c4b643a27b6917a2685835a0599603078069e6eacf214705e992e535e005104b00bfd3fc14465670362c3a579ef33a7e7d6749f5723
+  metadata.gz: b3548dbef713f3e93890d7198c70fe69b9c201081d2797a266add689e0a7a47a8f7a7b85feb38ae01da73abf222138bd387e69beb9cb9c3ef0f24b3a2cc50676
+  data.tar.gz: 02a830a48706320dcdd8ffb3269793376ffb331353ef331a77c54e51a589d70c901f35b20fd3a95ef122e0b8e1d29410b6873be97d0f20f226d38ef58114bf5b

data/lib/rswag/api/middleware.rb

@@ -1,8 +1,10 @@
 require 'json'
+require 'yaml'
+require 'rack/mime'
 
 module Rswag
   module Api
-    class Middleware 
+    class Middleware
 
       def initialize(app, config)
         @app = app
@@ -14,24 +16,46 @@ module Rswag
         filename = "#{@config.resolve_swagger_root(env)}/#{path}"
 
         if env['REQUEST_METHOD'] == 'GET' && File.file?(filename)
-          swagger = load_json(filename)
+          swagger = parse_file(filename)
           @config.swagger_filter.call(swagger, env) unless @config.swagger_filter.nil?
+          mime = Rack::Mime.mime_type(::File.extname(path), 'text/plain')
+          body = unload_swagger(filename, swagger)
 
           return [
             '200',
-            { 'Content-Type' => 'application/json' },
-            [ JSON.dump(swagger) ]
+            { 'Content-Type' => mime },
+            [ body ]
           ]
         end
-          
+
         return @app.call(env)
       end
 
       private
 
+      def parse_file(filename)
+        if /\.ya?ml$/ === filename
+          load_yaml(filename)
+        else
+          load_json(filename)
+        end
+      end
+
+      def load_yaml(filename)
+        YAML.safe_load(File.read(filename))
+      end
+
       def load_json(filename)
         JSON.parse(File.read(filename))
       end
+
+      def unload_swagger(filename, swagger)
+        if /\.ya?ml$/ === filename
+          YAML.dump(swagger)
+        else
+          JSON.dump(swagger)
+        end
+      end
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rswag-api
 version: !ruby/object:Gem::Version
-  version: 2.1.1
+  version: 2.2.0
 platform: ruby
 authors:
 - Richie Morris
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-10-18 00:00:00.000000000 Z
+date: 2019-11-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: railties
@@ -67,7 +67,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.7.7
+rubygems_version: 2.7.8
 signing_key: 
 specification_version: 4
 summary: A Rails Engine that exposes Swagger files as JSON endpoints