61 lines
1.9 KiB
Ruby
61 lines
1.9 KiB
Ruby
# typed: true
|
|
# frozen_string_literal: true
|
|
|
|
require "download_strategy"
|
|
require "utils/github"
|
|
|
|
module GitHub
|
|
# Download an artifact from GitHub Actions and unpack it into the current working directory.
|
|
#
|
|
# @param url [String] URL to download from
|
|
# @param artifact_id [String] a value that uniquely identifies the downloaded artifact
|
|
sig { params(url: String, artifact_id: String).void }
|
|
def self.download_artifact(url, artifact_id)
|
|
raise API::MissingAuthenticationError if API.credentials == :none
|
|
|
|
# We use a download strategy here to leverage the Homebrew cache
|
|
# to avoid repeated downloads of (possibly large) bottles.
|
|
token = API.credentials
|
|
downloader = GitHubArtifactDownloadStrategy.new(url, artifact_id, token:)
|
|
downloader.fetch
|
|
downloader.stage
|
|
end
|
|
end
|
|
|
|
# Strategy for downloading an artifact from GitHub Actions.
|
|
class GitHubArtifactDownloadStrategy < AbstractFileDownloadStrategy
|
|
def initialize(url, artifact_id, token:)
|
|
super(url, "artifact", artifact_id)
|
|
@cache = HOMEBREW_CACHE/"gh-actions-artifact"
|
|
@token = token
|
|
end
|
|
|
|
def fetch(timeout: nil)
|
|
ohai "Downloading #{url}"
|
|
if cached_location.exist?
|
|
puts "Already downloaded: #{cached_location}"
|
|
else
|
|
begin
|
|
Utils::Curl.curl("--location", "--create-dirs", "--output", temporary_path, url,
|
|
"--header", "Authorization: token #{@token}",
|
|
secrets: [@token],
|
|
timeout:)
|
|
rescue ErrorDuringExecution
|
|
raise CurlDownloadStrategyError, url
|
|
end
|
|
cached_location.dirname.mkpath
|
|
temporary_path.rename(cached_location)
|
|
end
|
|
|
|
symlink_location.dirname.mkpath
|
|
FileUtils.ln_s cached_location.relative_path_from(symlink_location.dirname), symlink_location, force: true
|
|
end
|
|
|
|
private
|
|
|
|
sig { returns(String) }
|
|
def resolved_basename
|
|
"artifact.zip"
|
|
end
|
|
end
|