Register endpoint_auth_wrapper middleware only if not yet registered.

ruby-grape · Aug 22, 2016 · aa6a972 · aa6a972
1 parent 161c6e3
commit aa6a972
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/grape-swagger.rb b/lib/grape-swagger.rb
@@ -32,7 +32,11 @@ def add_swagger_documentation(options = {})
         options = { target_class: self }.merge(options)
         @target_class = options[:target_class]
 
-        use options[:endpoint_auth_wrapper] if !options[:endpoint_auth_wrapper].nil? && options[:endpoint_auth_wrapper].method_defined?(:before)
+        if !options[:endpoint_auth_wrapper].nil? &&
+           options[:endpoint_auth_wrapper].method_defined?(:before) &&
+           !middleware.flatten.include?(options[:endpoint_auth_wrapper])
+          use options[:endpoint_auth_wrapper]
+        end
 
         documentation_class.setup(options)
         mount(documentation_class)