1 files changed, 54 insertions, 0 deletions
diff --git a/lib/gitlab/checks/file_size_check/hook_environment_aware_any_oversized_blobs.rb b/lib/gitlab/checks/file_size_check/hook_environment_aware_any_oversized_blobs.rb
new file mode 100644
index 00000000000..952def83658
--- /dev/null
+++ b/lib/gitlab/checks/file_size_check/hook_environment_aware_any_oversized_blobs.rb
@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Checks
+    module FileSizeCheck
+      class HookEnvironmentAwareAnyOversizedBlobs
+        def initialize(project:, changes:, file_size_limit_megabytes:)
+          @project = project
+          @repository = project.repository
+          @changes = changes
+          @file_size_limit_megabytes = file_size_limit_megabytes
+        end
+
+        def find(timeout: nil)
+          if ignore_alternate_directories?
+            blobs = repository.list_all_blobs(bytes_limit: 0, dynamic_timeout: timeout,
+              ignore_alternate_object_directories: true).to_a
+
+            blobs.select! do |blob|
+              ::Gitlab::Utils.bytes_to_megabytes(blob.size) > file_size_limit_megabytes
+            end
+            filter_existing(blobs)
+          else
+            any_oversize_blobs.find(timeout: timeout)
+          end
+        end
+
+        private
+
+        attr_reader :project, :repository, :changes, :file_size_limit_megabytes
+
+        def filter_existing(blobs)
+          gitaly_repo = repository.gitaly_repository.dup.tap { |repo| repo.git_object_directory = "" }
+
+          map_blob_id_to_existence = repository.gitaly_commit_client.object_existence_map(blobs.map(&:id),
+            gitaly_repo: gitaly_repo)
+
+          blobs.reject { |blob| map_blob_id_to_existence[blob.id].present? }
+        end
+
+        def ignore_alternate_directories?
+          git_env = ::Gitlab::Git::HookEnv.all(repository.gl_repository)
+
+          git_env['GIT_OBJECT_DIRECTORY_RELATIVE'].present?
+        end
+
+        def any_oversize_blobs
+          AnyOversizedBlobs.new(project: project, changes: changes,
+            file_size_limit_megabytes: file_size_limit_megabytes)
+        end
+      end
+    end
+  end
+end