gitlab-org--gitlab-foss/lib/gitlab/diff/position_collection.rb

# frozen_string_literal: true

module Gitlab
  module Diff
    class PositionCollection
      include Enumerable

      # collection - An array of Gitlab::Diff::Position
      def initialize(collection, diff_head_sha = nil)
        @collection = collection
        @diff_head_sha = diff_head_sha
      end

      def each(&block)
        filtered_positions.each(&block)
      end

      def concat(positions)
        tap { @collection.concat(positions) }
      end

      # Doing a lightweight filter in-memory given we're not prepared for querying
      # positions (https://gitlab.com/gitlab-org/gitlab/issues/33271).
      def unfoldable
        select do |position|
          position.unfoldable? && valid_head_sha?(position)
        end
      end

      private

      def filtered_positions
        @collection.select { |item| item.is_a?(Position) }
      end

      def valid_head_sha?(position)
        return true unless @diff_head_sha

        position.head_sha == @diff_head_sha
      end
    end
  end
end