490 lines
		
	
	
		
			19 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			490 lines
		
	
	
		
			19 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # typed: false
 | |
| # frozen_string_literal: true
 | |
| 
 | |
| require "formula"
 | |
| require "cli/parser"
 | |
| require "utils/pypi"
 | |
| require "utils/tar"
 | |
| 
 | |
| module Homebrew
 | |
|   module_function
 | |
| 
 | |
|   def bump_formula_pr_args
 | |
|     Homebrew::CLI::Parser.new do
 | |
|       usage_banner <<~EOS
 | |
|         `bump-formula-pr` [<options>] [<formula>]
 | |
| 
 | |
|         Create a pull request to update <formula> with a new URL or a new tag.
 | |
| 
 | |
|         If a <URL> is specified, the <SHA-256> checksum of the new download should also
 | |
|         be specified. A best effort to determine the <SHA-256> and <formula> name will
 | |
|         be made if either or both values are not supplied by the user.
 | |
| 
 | |
|         If a <tag> is specified, the Git commit <revision> corresponding to that tag
 | |
|         should also be specified. A best effort to determine the <revision> will be made
 | |
|         if the value is not supplied by the user.
 | |
| 
 | |
|         If a <version> is specified, a best effort to determine the <URL> and <SHA-256> or
 | |
|         the <tag> and <revision> will be made if both values are not supplied by the user.
 | |
| 
 | |
|         *Note:* this command cannot be used to transition a formula from a
 | |
|         URL-and-SHA-256 style specification into a tag-and-revision style specification,
 | |
|         nor vice versa. It must use whichever style specification the formula already uses.
 | |
|       EOS
 | |
|       switch "-n", "--dry-run",
 | |
|              description: "Print what would be done rather than doing it."
 | |
|       switch "--write",
 | |
|              description: "Make the expected file modifications without taking any Git actions."
 | |
|       switch "--commit",
 | |
|              depends_on:  "--write",
 | |
|              description: "When passed with `--write`, generate a new commit after writing changes "\
 | |
|                           "to the formula file."
 | |
|       switch "--no-audit",
 | |
|              description: "Don't run `brew audit` before opening the PR."
 | |
|       switch "--strict",
 | |
|              description: "Run `brew audit --strict` before opening the PR."
 | |
|       switch "--online",
 | |
|              description: "Run `brew audit --online` before opening the PR."
 | |
|       switch "--no-browse",
 | |
|              description: "Print the pull request URL instead of opening in a browser."
 | |
|       switch "--no-fork",
 | |
|              description: "Don't try to fork the repository."
 | |
|       comma_array "--mirror",
 | |
|                   description: "Use the specified <URL> as a mirror URL. If <URL> is a comma-separated list "\
 | |
|                                "of URLs, multiple mirrors will be added."
 | |
|       flag   "--version=",
 | |
|              description: "Use the specified <version> to override the value parsed from the URL or tag. Note "\
 | |
|                           "that `--version=0` can be used to delete an existing version override from a "\
 | |
|                           "formula if it has become redundant."
 | |
|       flag   "--message=",
 | |
|              description: "Append <message> to the default pull request message."
 | |
|       flag   "--url=",
 | |
|              description: "Specify the <URL> for the new download. If a <URL> is specified, the <SHA-256> "\
 | |
|                           "checksum of the new download should also be specified."
 | |
|       flag   "--sha256=",
 | |
|              depends_on:  "--url=",
 | |
|              description: "Specify the <SHA-256> checksum of the new download."
 | |
|       flag   "--tag=",
 | |
|              description: "Specify the new git commit <tag> for the formula."
 | |
|       flag   "--revision=",
 | |
|              depends_on:  "--tag=",
 | |
|              description: "Specify the new git commit <revision> corresponding to the specified <tag>."
 | |
|       switch "-f", "--force",
 | |
|              description: "Ignore duplicate open PRs. Remove all mirrors if --mirror= was not specified."
 | |
| 
 | |
|       conflicts "--dry-run", "--write"
 | |
|       conflicts "--no-audit", "--strict"
 | |
|       conflicts "--no-audit", "--online"
 | |
|       conflicts "--url", "--tag"
 | |
|       max_named 1
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def use_correct_linux_tap(formula, args:)
 | |
|     if OS.linux? && formula.tap.core_tap?
 | |
|       tap_full_name = formula.tap.full_name.gsub("linuxbrew", "homebrew")
 | |
|       homebrew_core_url = "https://github.com/#{tap_full_name}"
 | |
|       homebrew_core_remote = "homebrew"
 | |
|       homebrew_core_branch = "master"
 | |
|       origin_branch = "#{homebrew_core_remote}/#{homebrew_core_branch}"
 | |
|       previous_branch = Utils.popen_read("git -C \"#{formula.tap.path}\" symbolic-ref -q --short HEAD").chomp
 | |
|       previous_branch = "master" if previous_branch.empty?
 | |
|       formula_path = formula.path.to_s[%r{(Formula/.*)}, 1]
 | |
| 
 | |
|       if args.dry_run? || args.write?
 | |
|         ohai "git remote add #{homebrew_core_remote} #{homebrew_core_url}"
 | |
|         ohai "git fetch #{homebrew_core_remote} #{homebrew_core_branch}"
 | |
|         ohai "git cat-file -e #{origin_branch}:#{formula_path}"
 | |
|         ohai "git checkout #{origin_branch}"
 | |
|         return tap_full_name, origin_branch, previous_branch
 | |
|       else
 | |
|         formula.path.parent.cd do
 | |
|           unless Utils.popen_read("git remote -v").match?(%r{^homebrew.*Homebrew/homebrew-core.*$})
 | |
|             ohai "Adding #{homebrew_core_remote} remote"
 | |
|             safe_system "git", "remote", "add", homebrew_core_remote, homebrew_core_url
 | |
|           end
 | |
|           ohai "Fetching #{origin_branch}"
 | |
|           safe_system "git", "fetch", homebrew_core_remote, homebrew_core_branch
 | |
|           if quiet_system "git", "cat-file", "-e", "#{origin_branch}:#{formula_path}"
 | |
|             ohai "#{formula.full_name} exists in #{origin_branch}"
 | |
|             safe_system "git", "checkout", origin_branch
 | |
|             return tap_full_name, origin_branch, previous_branch
 | |
|           end
 | |
|         end
 | |
|       end
 | |
|     end
 | |
|     if formula.tap
 | |
|       origin_branch = Utils.popen_read("git", "-C", formula.tap.path.to_s, "symbolic-ref", "-q", "--short",
 | |
|                                        "refs/remotes/origin/HEAD").chomp.presence
 | |
|     end
 | |
|     origin_branch ||= "origin/master"
 | |
|     [formula.tap&.full_name, origin_branch, "-"]
 | |
|   end
 | |
| 
 | |
|   def bump_formula_pr
 | |
|     args = bump_formula_pr_args.parse
 | |
| 
 | |
|     # As this command is simplifying user-run commands then let's just use a
 | |
|     # user path, too.
 | |
|     ENV["PATH"] = ENV["HOMEBREW_PATH"]
 | |
| 
 | |
|     # Use the user's browser, too.
 | |
|     ENV["BROWSER"] = Homebrew::EnvConfig.browser
 | |
| 
 | |
|     formula = args.named.to_formulae.first
 | |
| 
 | |
|     new_url = args.url
 | |
|     formula ||= determine_formula_from_url(new_url) if new_url
 | |
|     raise FormulaUnspecifiedError unless formula
 | |
| 
 | |
|     odie "This formula is disabled!" if formula.disabled?
 | |
| 
 | |
|     tap_full_name, origin_branch, previous_branch = use_correct_linux_tap(formula, args: args)
 | |
|     check_open_pull_requests(formula, tap_full_name, args: args)
 | |
| 
 | |
|     new_version = args.version
 | |
|     check_closed_pull_requests(formula, tap_full_name, version: new_version, args: args) if new_version
 | |
| 
 | |
|     opoo "This formula has patches that may be resolved upstream." if formula.patchlist.present?
 | |
|     if formula.resources.any? { |resource| !resource.name.start_with?("homebrew-") }
 | |
|       opoo "This formula has resources that may need to be updated."
 | |
|     end
 | |
| 
 | |
|     requested_spec = :stable
 | |
|     formula_spec = formula.stable
 | |
|     odie "#{formula}: no #{requested_spec} specification found!" unless formula_spec
 | |
| 
 | |
|     old_mirrors = formula_spec.mirrors
 | |
|     new_mirrors ||= args.mirror
 | |
|     new_mirror ||= determine_mirror(new_url)
 | |
|     new_mirrors ||= [new_mirror] unless new_mirror.nil?
 | |
| 
 | |
|     check_for_mirrors(formula, old_mirrors, new_mirrors, args: args) if new_url
 | |
| 
 | |
|     hash_type, old_hash = if (checksum = formula_spec.checksum)
 | |
|       [checksum.hash_type, checksum.hexdigest]
 | |
|     end
 | |
| 
 | |
|     new_hash = args[hash_type] if hash_type
 | |
|     new_tag = args.tag
 | |
|     new_revision = args.revision
 | |
|     old_url = formula_spec.url
 | |
|     old_tag = formula_spec.specs[:tag]
 | |
|     old_formula_version = formula_version(formula, requested_spec)
 | |
|     old_version = old_formula_version.to_s
 | |
|     forced_version = new_version.present?
 | |
|     new_url_hash = if new_url && new_hash
 | |
|       check_closed_pull_requests(formula, tap_full_name, url: new_url, args: args) unless new_version
 | |
|       true
 | |
|     elsif new_tag && new_revision
 | |
|       check_closed_pull_requests(formula, tap_full_name, url: old_url, tag: new_tag, args: args) unless new_version
 | |
|       false
 | |
|     elsif !hash_type
 | |
|       odie "#{formula}: no --tag= or --version= argument specified!" if !new_tag && !new_version
 | |
|       new_tag ||= old_tag.gsub(old_version, new_version)
 | |
|       if new_tag == old_tag
 | |
|         odie <<~EOS
 | |
|           You need to bump this formula manually since the new tag
 | |
|           and old tag are both #{new_tag}.
 | |
|         EOS
 | |
|       end
 | |
|       check_closed_pull_requests(formula, tap_full_name, url: old_url, tag: new_tag, args: args) unless new_version
 | |
|       resource_path, forced_version = fetch_resource(formula, new_version, old_url, tag: new_tag)
 | |
|       new_revision = Utils.popen_read("git -C \"#{resource_path}\" rev-parse -q --verify HEAD")
 | |
|       new_revision = new_revision.strip
 | |
|       false
 | |
|     elsif !new_url && !new_version
 | |
|       odie "#{formula}: no --url= or --version= argument specified!"
 | |
|     else
 | |
|       new_url ||= PyPI.update_pypi_url(old_url, new_version)
 | |
|       unless new_url
 | |
|         new_url = old_url.gsub(old_version, new_version)
 | |
|         if !new_mirrors && !old_mirrors.empty?
 | |
|           new_mirrors = old_mirrors.map do |old_mirror|
 | |
|             old_mirror.gsub(old_version, new_version)
 | |
|           end
 | |
|         end
 | |
|       end
 | |
|       if new_url == old_url
 | |
|         odie <<~EOS
 | |
|           You need to bump this formula manually since the new URL
 | |
|           and old URL are both:
 | |
|             #{new_url}
 | |
|         EOS
 | |
|       end
 | |
|       check_closed_pull_requests(formula, tap_full_name, url: new_url, args: args) unless new_version
 | |
|       resource_path, forced_version = fetch_resource(formula, new_version, new_url)
 | |
|       Utils::Tar.validate_file(resource_path)
 | |
|       new_hash = resource_path.sha256
 | |
|     end
 | |
| 
 | |
|     replacement_pairs = []
 | |
|     if requested_spec == :stable && formula.revision.nonzero?
 | |
|       replacement_pairs << [
 | |
|         /^  revision \d+\n(\n(  head "))?/m,
 | |
|         "\\2",
 | |
|       ]
 | |
|     end
 | |
| 
 | |
|     replacement_pairs += formula_spec.mirrors.map do |mirror|
 | |
|       [
 | |
|         / +mirror "#{Regexp.escape(mirror)}"\n/m,
 | |
|         "",
 | |
|       ]
 | |
|     end
 | |
| 
 | |
|     replacement_pairs += if new_url_hash
 | |
|       [
 | |
|         [
 | |
|           /#{Regexp.escape(formula_spec.url)}/,
 | |
|           new_url,
 | |
|         ],
 | |
|         [
 | |
|           old_hash,
 | |
|           new_hash,
 | |
|         ],
 | |
|       ]
 | |
|     else
 | |
|       [
 | |
|         [
 | |
|           formula_spec.specs[:tag],
 | |
|           new_tag,
 | |
|         ],
 | |
|         [
 | |
|           formula_spec.specs[:revision],
 | |
|           new_revision,
 | |
|         ],
 | |
|       ]
 | |
|     end
 | |
| 
 | |
|     old_contents = File.read(formula.path) unless args.dry_run?
 | |
| 
 | |
|     if new_mirrors
 | |
|       replacement_pairs << [
 | |
|         /^( +)(url "#{Regexp.escape(new_url)}"\n)/m,
 | |
|         "\\1\\2\\1mirror \"#{new_mirrors.join("\"\n\\1mirror \"")}\"\n",
 | |
|       ]
 | |
|     end
 | |
| 
 | |
|     # When bumping a linux-only formula, one needs to also delete the
 | |
|     # sha256 linux bottle line if it exists. That's because of running
 | |
|     # test-bot with --keep-old option in linuxbrew-core.
 | |
|     formula_contents = formula.path.read
 | |
|     if formula_contents.include?("depends_on :linux") && formula_contents.include?("=> :x86_64_linux")
 | |
|       replacement_pairs << [
 | |
|         /^    sha256 ".+" => :x86_64_linux\n/m,
 | |
|         "\\2",
 | |
|       ]
 | |
|     end
 | |
| 
 | |
|     if forced_version && new_version != "0"
 | |
|       replacement_pairs << if File.read(formula.path).include?("version \"#{old_formula_version}\"")
 | |
|         [
 | |
|           old_formula_version.to_s,
 | |
|           new_version,
 | |
|         ]
 | |
|       elsif new_mirrors
 | |
|         [
 | |
|           /^( +)(mirror "#{Regexp.escape(new_mirrors.last)}"\n)/m,
 | |
|           "\\1\\2\\1version \"#{new_version}\"\n",
 | |
|         ]
 | |
|       elsif new_url
 | |
|         [
 | |
|           /^( +)(url "#{Regexp.escape(new_url)}"\n)/m,
 | |
|           "\\1\\2\\1version \"#{new_version}\"\n",
 | |
|         ]
 | |
|       elsif new_revision
 | |
|         [
 | |
|           /^( {2})( +)(:revision => "#{new_revision}"\n)/m,
 | |
|           "\\1\\2\\3\\1version \"#{new_version}\"\n",
 | |
|         ]
 | |
|       end
 | |
|     elsif forced_version && new_version == "0"
 | |
|       replacement_pairs << [
 | |
|         /^  version "[\w.\-+]+"\n/m,
 | |
|         "",
 | |
|       ]
 | |
|     end
 | |
|     new_contents = Utils::Inreplace.inreplace_pairs(formula.path,
 | |
|                                                     replacement_pairs.uniq.compact,
 | |
|                                                     read_only_run: args.dry_run?,
 | |
|                                                     silent:        args.quiet?)
 | |
| 
 | |
|     new_formula_version = formula_version(formula, requested_spec, new_contents)
 | |
| 
 | |
|     if new_formula_version < old_formula_version
 | |
|       formula.path.atomic_write(old_contents) unless args.dry_run?
 | |
|       odie <<~EOS
 | |
|         You need to bump this formula manually since changing the
 | |
|         version from #{old_formula_version} to #{new_formula_version} would be a downgrade.
 | |
|       EOS
 | |
|     elsif new_formula_version == old_formula_version
 | |
|       formula.path.atomic_write(old_contents) unless args.dry_run?
 | |
|       odie <<~EOS
 | |
|         You need to bump this formula manually since the new version
 | |
|         and old version are both #{new_formula_version}.
 | |
|       EOS
 | |
|     end
 | |
| 
 | |
|     alias_rename = alias_update_pair(formula, new_formula_version)
 | |
|     if alias_rename.present?
 | |
|       ohai "renaming alias #{alias_rename.first} to #{alias_rename.last}"
 | |
|       alias_rename.map! { |a| formula.tap.alias_dir/a }
 | |
|     end
 | |
| 
 | |
|     unless args.dry_run?
 | |
|       resources_checked = PyPI.update_python_resources! formula, new_formula_version,
 | |
|                                                         silent: args.quiet?, ignore_non_pypi_packages: true
 | |
|     end
 | |
| 
 | |
|     run_audit(formula, alias_rename, old_contents, args: args)
 | |
| 
 | |
|     pr_message = "Created with `brew bump-formula-pr`."
 | |
|     if resources_checked.nil? && formula.resources.any? { |resource| !resource.name.start_with?("homebrew-") }
 | |
|       pr_message += <<~EOS
 | |
| 
 | |
| 
 | |
|         `resource` blocks may require updates.
 | |
|       EOS
 | |
|     end
 | |
| 
 | |
|     pr_info = {
 | |
|       sourcefile_path:  formula.path,
 | |
|       old_contents:     old_contents,
 | |
|       additional_files: alias_rename,
 | |
|       origin_branch:    origin_branch,
 | |
|       branch_name:      "bump-#{formula.name}-#{new_formula_version}",
 | |
|       commit_message:   "#{formula.name} #{new_formula_version}",
 | |
|       previous_branch:  previous_branch,
 | |
|       tap:              formula.tap,
 | |
|       tap_full_name:    tap_full_name,
 | |
|       pr_message:       pr_message,
 | |
|     }
 | |
|     GitHub.create_bump_pr(pr_info, args: args)
 | |
|   end
 | |
| 
 | |
|   def determine_formula_from_url(url)
 | |
|     # Split the new URL on / and find any formulae that have the same URL
 | |
|     # except for the last component, but don't try to match any more than the
 | |
|     # first five components since sometimes the last component isn't the only
 | |
|     # one to change.
 | |
|     url_split = url.split("/")
 | |
|     maximum_url_components_to_match = 5
 | |
|     components_to_match = [url_split.count - 1, maximum_url_components_to_match].min
 | |
|     base_url = url_split.first(components_to_match).join("/")
 | |
|     base_url = /#{Regexp.escape(base_url)}/
 | |
|     guesses = []
 | |
|     Formula.each do |f|
 | |
|       guesses << f if f.stable&.url && f.stable.url.match(base_url)
 | |
|     end
 | |
|     return guesses.shift if guesses.count == 1
 | |
|     return if guesses.count <= 1
 | |
| 
 | |
|     odie "Couldn't guess formula for sure; could be one of these:\n#{guesses.map(&:name).join(", ")}"
 | |
|   end
 | |
| 
 | |
|   def determine_mirror(url)
 | |
|     case url
 | |
|     when %r{.*ftp.gnu.org/gnu.*}
 | |
|       url.sub "ftp.gnu.org/gnu", "ftpmirror.gnu.org"
 | |
|     when %r{.*download.savannah.gnu.org/*}
 | |
|       url.sub "download.savannah.gnu.org", "download-mirror.savannah.gnu.org"
 | |
|     when %r{.*www.apache.org/dyn/closer.lua\?path=.*}
 | |
|       url.sub "www.apache.org/dyn/closer.lua?path=", "archive.apache.org/dist/"
 | |
|     when %r{.*mirrors.ocf.berkeley.edu/debian.*}
 | |
|       url.sub "mirrors.ocf.berkeley.edu/debian", "mirrorservice.org/sites/ftp.debian.org/debian"
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def check_for_mirrors(formula, old_mirrors, new_mirrors, args:)
 | |
|     return if new_mirrors || old_mirrors.empty?
 | |
| 
 | |
|     if args.force?
 | |
|       opoo "#{formula}: Removing all mirrors because a --mirror= argument was not specified."
 | |
|     else
 | |
|       odie <<~EOS
 | |
|         #{formula}: a --mirror= argument for updating the mirror URL(s) was not specified.
 | |
|         Use --force to remove all mirrors.
 | |
|       EOS
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def fetch_resource(formula, new_version, url, **specs)
 | |
|     resource = Resource.new
 | |
|     resource.url(url, specs)
 | |
|     resource.owner = Resource.new(formula.name)
 | |
|     forced_version = new_version && new_version != resource.version
 | |
|     resource.version = new_version if forced_version
 | |
|     odie "No --version= argument specified!" unless resource.version
 | |
|     [resource.fetch, forced_version]
 | |
|   end
 | |
| 
 | |
|   def formula_version(formula, spec, contents = nil)
 | |
|     name = formula.name
 | |
|     path = formula.path
 | |
|     if contents
 | |
|       Formulary.from_contents(name, path, contents, spec).version
 | |
|     else
 | |
|       Formulary::FormulaLoader.new(name, path).get_formula(spec).version
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def check_open_pull_requests(formula, tap_full_name, args:)
 | |
|     GitHub.check_for_duplicate_pull_requests(formula.name, tap_full_name, state: "open", args: args)
 | |
|   end
 | |
| 
 | |
|   def check_closed_pull_requests(formula, tap_full_name, args:, version: nil, url: nil, tag: nil)
 | |
|     unless version
 | |
|       specs = {}
 | |
|       specs[:tag] = tag if tag
 | |
|       version = Version.detect(url, **specs)
 | |
|     end
 | |
|     # if we haven't already found open requests, try for an exact match across closed requests
 | |
|     GitHub.check_for_duplicate_pull_requests("#{formula.name} #{version}", tap_full_name, state: "closed", args: args)
 | |
|   end
 | |
| 
 | |
|   def alias_update_pair(formula, new_formula_version)
 | |
|     versioned_alias = formula.aliases.grep(/^.*@\d+(\.\d+)?$/).first
 | |
|     return if versioned_alias.nil?
 | |
| 
 | |
|     name, old_alias_version = versioned_alias.split("@")
 | |
|     new_alias_regex = (old_alias_version.split(".").length == 1) ? /^\d+/ : /^\d+\.\d+/
 | |
|     new_alias_version, = *new_formula_version.to_s.match(new_alias_regex)
 | |
|     return if Version.create(new_alias_version) <= Version.create(old_alias_version)
 | |
| 
 | |
|     [versioned_alias, "#{name}@#{new_alias_version}"]
 | |
|   end
 | |
| 
 | |
|   def run_audit(formula, alias_rename, old_contents, args:)
 | |
|     audit_args = []
 | |
|     audit_args << "--strict" if args.strict?
 | |
|     audit_args << "--online" if args.online?
 | |
|     if args.dry_run?
 | |
|       if args.no_audit?
 | |
|         ohai "Skipping `brew audit`"
 | |
|       elsif audit_args.present?
 | |
|         ohai "brew audit #{audit_args.join(" ")} #{formula.path.basename}"
 | |
|       else
 | |
|         ohai "brew audit #{formula.path.basename}"
 | |
|       end
 | |
|       return
 | |
|     end
 | |
|     FileUtils.mv alias_rename.first, alias_rename.last if alias_rename.present?
 | |
|     failed_audit = false
 | |
|     if args.no_audit?
 | |
|       ohai "Skipping `brew audit`"
 | |
|     elsif audit_args.present?
 | |
|       system HOMEBREW_BREW_FILE, "audit", *audit_args, formula.path
 | |
|       failed_audit = !$CHILD_STATUS.success?
 | |
|     else
 | |
|       system HOMEBREW_BREW_FILE, "audit", formula.path
 | |
|       failed_audit = !$CHILD_STATUS.success?
 | |
|     end
 | |
|     return unless failed_audit
 | |
| 
 | |
|     formula.path.atomic_write(old_contents)
 | |
|     FileUtils.mv alias_rename.last, alias_rename.first if alias_rename.present?
 | |
|     odie "`brew audit` failed!"
 | |
|   end
 | |
| end
 | 
