Add requst param parser regestry

Author: numbata

lib/grape-swagger.rb

@@ -10,12 +10,17 @@
 
 require 'grape-swagger/doc_methods'
 require 'grape-swagger/model_parsers'
+require 'grape-swagger/request_param_parser_registry'
 
 module GrapeSwagger
   class << self
     def model_parsers
       @model_parsers ||= GrapeSwagger::ModelParsers.new
     end
+
+    def request_param_parsers
+      @request_param_parsers ||= GrapeSwagger::RequestParamParserRegistry.new
+    end
   end
   autoload :Rake, 'grape-swagger/rake/oapi_tasks'
 

lib/grape-swagger/endpoint.rb

@@ -8,12 +8,6 @@
 
 module Grape
   class Endpoint # rubocop:disable Metrics/ClassLength
-    REQUEST_PARAM_PARSERS = [
-      GrapeSwagger::RequestParamParsers::Headers,
-      GrapeSwagger::RequestParamParsers::Route,
-      GrapeSwagger::RequestParamParsers::Body
-    ].freeze
-
     def content_types_for(target_class)
       content_types = (target_class.content_types || {}).values
 
@@ -389,7 +383,7 @@ def build_file_response(memo)
     end
 
     def build_request_params(route, settings)
-      REQUEST_PARAM_PARSERS.each_with_object({}) do |parser_klass, accum|
+      GrapeSwagger.request_param_parsers.each_with_object({}) do |parser_klass, accum|
         params = parser_klass.parse(
           route,
           accum,

lib/grape-swagger/request_param_parser_registry.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require_relative 'request_param_parsers/headers'
+require_relative 'request_param_parsers/route'
+require_relative 'request_param_parsers/body'
+
+module GrapeSwagger
+  class RequestParamParserRegistry
+    DEFAULT_PARSERS = [
+      GrapeSwagger::RequestParamParsers::Headers,
+      GrapeSwagger::RequestParamParsers::Route,
+      GrapeSwagger::RequestParamParsers::Body
+    ].freeze
+
+    include Enumerable
+
+    def initialize
+      @parsers = DEFAULT_PARSERS.dup
+    end
+
+    def register(klass)
+      @parsers << klass
+    end
+
+    def insert_before(before_klass, klass)
+      insert_at = @parsers.index(before_klass)
+      insert_at = @parsers.length - 1 if insert_at.nil?
+      @parsers.insert(insert_at, klass)
+    end
+
+    def insert_after(after_klass, klass)
+      insert_at = @parsers.index(after_klass)
+      insert_at = @parsers.length - 1 if insert_at.nil?
+      @parsers.insert(insert_at + 1, klass)
+    end
+
+    def each
+      @parsers.each do |klass|
+        yield klass
+      end
+    end
+  end
+end