summaryrefslogtreecommitdiff
path: root/app/models/lfs_download_object.rb
blob: 3df6742fbc943be32a540c2b98b254aa29c2c4f5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# 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