brew/Library/Homebrew/livecheck/strategy/header_match.rb

# typed: false
# frozen_string_literal: true

require_relative "page_match"

module Homebrew
  module Livecheck
    module Strategy
      # The {HeaderMatch} strategy follows all URL redirections and scans
      # the resulting headers for matching text using the provided regex.
      #
      # @api private
      class HeaderMatch
        extend T::Sig

        NICE_NAME = "Match HTTP Headers"

        # We set the priority to zero since this cannot
        # be detected automatically.
        PRIORITY = 0

        # Whether the strategy can be applied to the provided URL.
        # The strategy will technically match any HTTP URL but is
        # only usable with a `livecheck` block containing a regex
        # or block.
        sig { params(url: String).returns(T::Boolean) }
        def self.match?(url)
          url.match?(%r{^https?://})
        end

        # Checks the final URL for new versions after following all redirections,
        # using the provided regex for matching.
        sig { params(url: String, regex: T.nilable(Regexp)).returns(T::Hash[Symbol, T.untyped]) }
        def self.find_versions(url, regex, &block)
          match_data = { matches: {}, regex: regex, url: url }

          headers = Strategy.page_headers(url)

          # Merge the headers from all responses into one hash
          merged_headers = {}
          headers.each { |resp_headers| merged_headers.merge!(resp_headers) }

          if block
            match = block.call(merged_headers)
          else
            match = nil

            if (filename = merged_headers["content-disposition"])
              if regex
                match ||= filename[regex, 1]
              else
                v = Version.parse(filename, detected_from_url: true)
                match ||= v.to_s unless v.null?
              end
            end

            if (location = merged_headers["location"])
              if regex
                match ||= location[regex, 1]
              else
                v = Version.parse(location, detected_from_url: true)
                match ||= v.to_s unless v.null?
              end
            end
          end

          match_data[:matches][match] = Version.new(match) if match

          match_data
        end
      end
    end
  end
end