feat: add custom download strategy

Author: gabolaev

Casks/ds.rb

@@ -1,8 +1,11 @@
+require_relative "../custom_download_strategy"
+
 cask "ds" do
   version "0.13.0"
   sha256 "d7d543ea64aaf15e8bfed812aab0a220c36777d42f6203f604e32d318f8ee99f"
 
-  url "https://github.com/docker/dash-releases/releases/download/v#{version}/ds-darwin-arm64.tar.gz"
+  url "https://github.com/docker/dash-releases/releases/download/v#{version}/ds-darwin-arm64.tar.gz",
+      using: GitHubPrivateRepositoryReleaseDownloadStrategy
   name "Docker Dash CLI"
   desc "Docker Dash CLI"
   homepage "https://github.com/docker/dash-releases"

Casks/ds@0.13.0.rb

@@ -1,8 +1,11 @@
+require_relative "../custom_download_strategy"
+
 cask "ds@0.13.0" do
   version "0.13.0"
   sha256 "d7d543ea64aaf15e8bfed812aab0a220c36777d42f6203f604e32d318f8ee99f"
 
-  url "https://github.com/docker/dash-releases/releases/download/v#{version}/ds-darwin-arm64.tar.gz"
+  url "https://github.com/docker/dash-releases/releases/download/v#{version}/ds-darwin-arm64.tar.gz",
+      using: GitHubPrivateRepositoryReleaseDownloadStrategy
   name "Docker Dash CLI"
   desc "Docker Dash CLI"
   homepage "https://github.com/docker/dash-releases"

custom_download_strategy.rb

@@ -0,0 +1,81 @@
+require "download_strategy"
+require "shellwords"
+require "utils/curl"
+
+class GitHubPrivateRepositoryReleaseDownloadStrategy < AbstractFileDownloadStrategy
+  def initialize(url, name, version, **meta)
+    super
+    @github_token = ENV["HOMEBREW_GITHUB_API_TOKEN"]
+    parse_url_pattern
+  end
+
+  def resolved_basename
+    super.gsub(' ', '+')
+  end
+
+  def parse_url_pattern
+    url_pattern =
+      %r{https://github.com/([^/]+)/([^/]+)/releases/download/([^/]+)/(\S+)}
+    unless @url =~ url_pattern
+      raise CurlDownloadStrategyError, "Invalid url pattern for GitHub Release."
+    end
+
+    _, @owner, @repo, @tag, @filename = *@url.match(url_pattern)
+  end
+
+  def fetch(timeout: nil)
+    if cached_location.exist?
+      ohai "Already downloaded: #{cached_location}"
+    else
+      if @github_token != nil
+        ohai "PAT found... using it"
+        asset_url =
+          "https://api.github.com/repos/#{@owner}/#{@repo}/releases/assets/#{asset_id}"
+
+        ::Utils::Curl.curl_download asset_url,
+                                    "--header",
+                                    "Authorization: Bearer #{@github_token}",
+                                    "--header",
+                                    "X-GitHub-Api-Version: 2022-11-28",
+                                    "--header",
+                                    "Accept: application/octet-stream",
+                                    to: temporary_path
+      else
+        ohai "Downloading to: #{Shellwords.escape(temporary_path.to_s)}"
+        cmd =
+          "gh --repo=#{@owner}/#{@repo} release download #{@tag} --pattern=#{@filename} --output=#{Shellwords.escape(temporary_path.to_s)} --clobber"
+        puts `export PATH=$PATH:/opt/homebrew/bin:/usr/local/bin:~/.local/bin && #{cmd}`
+        if not $?.success?
+          raise CurlDownloadStrategyError, "You may need to run 'gh auth login' or 'gh auth refresh' to login to GitHub."
+        end
+      end
+
+      ignore_interrupts { temporary_path.rename(cached_location) }
+    end
+  end
+
+  def clear_cache
+    super
+    rm_rf(temporary_path)
+  end
+
+  private
+
+  def asset_id
+    @asset_id ||= resolve_asset_id
+  end
+
+  def resolve_asset_id
+    release_metadata = fetch_release_metadata
+    assets = release_metadata["assets"].select { |a| a["name"] == @filename }
+    raise CurlDownloadStrategyError, "Asset file not found." if assets.empty?
+
+    assets.first["id"]
+  end
+
+  def fetch_release_metadata
+    release_url =
+      "https://api.github.com/repos/#{@owner}/#{@repo}/releases/tags/#{@tag}"
+    GitHub::API.open_rest(release_url)
+  end
+end