36 lines
844 B
Ruby
36 lines
844 B
Ruby
# 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
|