gitlab-org--gitlab-foss/lib/gitlab/repository_size_checker.rb

# frozen_string_literal: true

module Gitlab
  # Centralized class for repository size related calculations.
  class RepositorySizeChecker
    attr_reader :limit

    # @param current_size_proc [Proc] returns repository size in bytes
    def initialize(current_size_proc:, limit:, enabled: true)
      @current_size_proc = current_size_proc
      @limit = limit
      @enabled = enabled && limit != 0
    end

    # @return [Integer] bytes
    def current_size
      @current_size ||= @current_size_proc.call
    end

    def enabled?
      @enabled
    end

    def above_size_limit?
      return false unless enabled?

      current_size > limit
    end

    # @param change_size [int] in bytes
    def changes_will_exceed_size_limit?(change_size)
      return false unless enabled?

      change_size > limit || exceeded_size(change_size) > 0
    end

    # @param change_size [int] in bytes
    def exceeded_size(change_size = 0)
      current_size + change_size - limit
    end

    def error_message
      @error_message_object ||= Gitlab::RepositorySizeErrorMessage.new(self)
    end
  end
end