From 49c5a4393a679bff908799b96b1df5a276cb1ca1 Mon Sep 17 00:00:00 2001
From: Jonathan Chang <jchang641@gmail.com>
Date: Mon, 30 Mar 2020 19:35:54 +1100
Subject: [PATCH] bintray: add Bintray API functions

---
 Library/Homebrew/bintray.rb | 125 ++++++++++++++++++++++++++++++++++++
 1 file changed, 125 insertions(+)
 create mode 100644 Library/Homebrew/bintray.rb

diff --git a/Library/Homebrew/bintray.rb b/Library/Homebrew/bintray.rb
new file mode 100644
index 0000000000..990b3ed3da
--- /dev/null
+++ b/Library/Homebrew/bintray.rb
@@ -0,0 +1,125 @@
+# frozen_string_literal: true
+
+require "utils/curl"
+require "json"
+
+class Bintray
+  API_URL = "https://api.bintray.com"
+
+  class Error < RuntimeError
+  end
+
+  def inspect
+    "#<Bintray: user=#{@bintray_user} org=#{@bintray_org} key=***>"
+  end
+
+  def initialize(user: nil, key: nil, org: nil, clear: true)
+    @bintray_user = user || ENV["HOMEBREW_BINTRAY_USER"]
+    @bintray_key = key || ENV["HOMEBREW_BINTRAY_KEY"]
+
+    if !@bintray_user || !@bintray_key
+      raise Error, "Missing HOMEBREW_BINTRAY_USER or HOMEBREW_BINTRAY_KEY variables!" unless Homebrew.args.dry_run?
+    end
+
+    @bintray_org = org || "homebrew"
+    ENV["HOMEBREW_FORCE_HOMEBREW_ON_LINUX"] = "1" if @bintray_org == "homebrew" && !OS.mac?
+
+    ENV.clear_sensitive_environment! if clear
+  end
+
+  def open_api(url, *extra_curl_args, auth: true)
+    args = extra_curl_args
+    args += ["--user", "#{@bintray_user}:#{@bintray_key}"] if auth
+    curl(*args, url,
+         show_output: Homebrew.args.verbose?,
+         secrets:     @bintray_key)
+  end
+
+  def upload(local_file, repo:, package:, version:, remote_file:, sha256: nil)
+    url = "#{API_URL}/content/#{@bintray_org}/#{repo}/#{package}/#{version}/#{remote_file}"
+    args = ["--upload-file", local_file]
+    args += ["--header", "X-Checksum-Sha2: #{sha256}"] unless sha256.blank?
+    open_api url, *args
+  end
+
+  def publish(repo:, package:, version:)
+    url = "#{API_URL}/content/#{@bintray_org}/#{repo}/#{package}/#{version}/publish"
+    open_api url, "--request", "POST"
+  end
+
+  def official_org?(org: @bintray_org)
+    %w[homebrew linuxbrew].include? org
+  end
+
+  def create_package(repo:, package:, **extra_data_args)
+    url = "#{API_URL}/packages/#{@bintray_org}/#{repo}/#{package}"
+    data = { name: package, public_download_numbers: true }
+    data[:public_stats] = official_org?
+    data.merge! extra_data_args
+    open_api url, "--request", "POST", "--data", data.to_json
+  end
+
+  def package_exists?(repo:, package:)
+    url = "#{API_URL}/packages/#{@bintray_org}/#{repo}/#{package}"
+    open_api url, "--output", "/dev/null", auth: false
+  end
+
+  def file_published?(repo:, remote_file:)
+    url = "https://dl.bintray.com/#{@bintray_org}/#{repo}/#{remote_file}"
+    begin
+      curl "--silent", "--head", "--output", "/dev/null", url
+    rescue ErrorDuringExecution => e
+      stderr = e.output.select { |type,| type == :stderr }
+                .map { |_, line| line }
+                .join
+      raise if e.status.exitstatus != 22 && !stderr.include?("404 Not Found")
+
+      false
+    else
+      true
+    end
+  end
+
+  def upload_bottle_json(json_files, publish_package: false)
+    bottles_hash = json_files.reduce({}) do |hash, json_file|
+      hash.deep_merge(JSON.parse(IO.read(json_file)))
+    end
+
+    formula_packaged = {}
+
+    bottles_hash.each do |formula_name, bottle_hash|
+      version = bottle_hash["formula"]["pkg_version"]
+      bintray_package = bottle_hash["bintray"]["package"]
+      bintray_repo = bottle_hash["bintray"]["repository"]
+
+      bottle_hash["bottle"]["tags"].each do |_tag, tag_hash|
+        filename = tag_hash["filename"]
+        sha256 = tag_hash["sha256"]
+
+        if file_published? repo: bintray_repo, remote_file: filename
+          raise Error, <<~EOS
+            #{filename} is already published.
+            Please remove it manually from:
+              https://bintray.com/#{@bintray_org}/#{bintray_repo}/#{bintray_package}/view#files
+            Or run:
+              curl -X DELETE -u $HOMEBREW_BINTRAY_USER:$HOMEBREW_BINTRAY_KEY \\
+              https://api.bintray.com/content/#{@bintray_org}/#{bintray_repo}/#{filename}
+          EOS
+        end
+
+        if !formula_packaged[formula_name] && !package_exists?(repo: bintray_repo, package: bintray_package)
+          create_package repo: bintray_repo, package: bintray_package
+          formula_packaged[formula_name] = true
+        end
+
+        upload(tag_hash["local_filename"],
+               repo:        bintray_repo,
+               package:     bintray_package,
+               version:     version,
+               remote_file: filename,
+               sha256:      sha256)
+      end
+      publish repo: bintray_repo, package: bintray_package, version: version if publish_package
+    end
+  end
+end