gitlab-org--gitlab-foss/app/models/lfs_download_object.rb

# frozen_string_literal: true

class LfsDownloadObject
  include ActiveModel::Validations

  attr_accessor :oid, :size, :link, :headers
  delegate :sanitized_url, :credentials, to: :sanitized_uri

  validates :oid, format: { with: /\A\h{64}\z/ }
  validates :size, numericality: { greater_than_or_equal_to: 0 }
  validates :link, public_url: { protocols: %w(http https) }
  validate :headers_must_be_hash

  def initialize(oid:, size:, link:, headers: {})
    @oid = oid
    @size = size
    @link = link
    @headers = headers || {}
  end

  def sanitized_uri
    @sanitized_uri ||= Gitlab::UrlSanitizer.new(link)
  end

  def has_authorization_header?
    headers.keys.map(&:downcase).include?('authorization')
  end

  private

  def headers_must_be_hash
    errors.add(:base, "headers must be a Hash") unless headers.is_a?(Hash)
  end
end