Introduce a new EE::Concerns::LfsRequest module

Signed-off-by: Rémy Coutable <remy@rymai.me>

app/controllers/concerns/lfs_request.rb

@@ -11,6 +11,7 @@ module LfsRequest
   extend ActiveSupport::Concern
 
   included do
+    prepend EE::LfsRequest
     before_action :require_lfs_enabled!
     before_action :lfs_check_access!
   end
@@ -111,40 +112,4 @@ module LfsRequest
   def has_authentication_ability?(capability)
     (authentication_abilities || []).include?(capability)
   end
-
-  module EE
-    def lfs_forbidden!
-      raise NotImplementedError unless defined?(super)
-
-      if project.above_size_limit? || objects_exceed_repo_limit?
-        render_size_error
-      else
-        super
-      end
-    end
-
-    def render_size_error
-      render(
-        json: {
-          message: Gitlab::RepositorySizeError.new(project).push_error(@exceeded_limit),
-          documentation_url: help_url
-        },
-        content_type: "application/vnd.git-lfs+json",
-        status: 406
-      )
-    end
-
-    def objects_exceed_repo_limit?
-      return false unless project.size_limit_enabled?
-      return @limit_exceeded if defined?(@limit_exceeded)
-
-      lfs_push_size = objects.sum { |o| o[:size] }
-      size_with_lfs_push = project.repository_and_lfs_size + lfs_push_size
-
-      @exceeded_limit = size_with_lfs_push - project.actual_size_limit
-      @limit_exceeded = @exceeded_limit > 0
-    end
-  end
-
-  prepend EE
 end

ee/app/controllers/concerns/ee/lfs_request.rb

+module EE
+  module LfsRequest
+    extend ActiveSupport::Concern
+
+    def lfs_forbidden!
+      if project.above_size_limit? || objects_exceed_repo_limit?
+        render_size_error
+      else
+        super
+      end
+    end
+
+    def render_size_error
+      render(
+        json: {
+          message: ::Gitlab::RepositorySizeError.new(project).push_error(@exceeded_limit),
+          documentation_url: help_url
+        },
+        content_type: "application/vnd.git-lfs+json",
+        status: 406
+      )
+    end
+
+    def objects_exceed_repo_limit?
+      return false unless project.size_limit_enabled?
+      return @limit_exceeded if defined?(@limit_exceeded)
+
+      lfs_push_size = objects.sum { |o| o[:size] }
+      size_with_lfs_push = project.repository_and_lfs_size + lfs_push_size
+
+      @exceeded_limit = size_with_lfs_push - project.actual_size_limit
+      @limit_exceeded = @exceeded_limit > 0
+    end
+  end
+end