2020-12-04 00:07:02 +01:00
|
|
|
# typed: false
|
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2020-12-09 03:12:21 +01:00
|
|
|
require "timeout"
|
2020-12-04 00:07:02 +01:00
|
|
|
require "cask/download"
|
|
|
|
require "cask/installer"
|
|
|
|
require "cask/cask_loader"
|
|
|
|
require "cli/parser"
|
|
|
|
require "tap"
|
2020-12-07 16:31:45 +01:00
|
|
|
require "unversioned_cask_checker"
|
2020-12-04 00:07:02 +01:00
|
|
|
|
|
|
|
module Homebrew
|
|
|
|
extend T::Sig
|
|
|
|
|
|
|
|
extend SystemCommand::Mixin
|
|
|
|
|
|
|
|
sig { returns(CLI::Parser) }
|
|
|
|
def self.bump_unversioned_casks_args
|
|
|
|
Homebrew::CLI::Parser.new do
|
2021-01-15 15:04:02 -05:00
|
|
|
description <<~EOS
|
2020-12-04 00:07:02 +01:00
|
|
|
Check all casks with unversioned URLs in a given <tap> for updates.
|
|
|
|
EOS
|
|
|
|
switch "-n", "--dry-run",
|
2020-12-07 16:31:45 +01:00
|
|
|
description: "Do everything except caching state and opening pull requests."
|
2020-11-27 11:41:08 -05:00
|
|
|
flag "--limit=",
|
|
|
|
description: "Maximum runtime in minutes."
|
2020-12-04 00:07:02 +01:00
|
|
|
flag "--state-file=",
|
2020-12-07 16:31:45 +01:00
|
|
|
description: "File for caching state."
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2021-01-10 14:26:40 -05:00
|
|
|
named_args [:cask, :tap], min: 1
|
2020-12-04 00:07:02 +01:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
sig { void }
|
|
|
|
def self.bump_unversioned_casks
|
|
|
|
args = bump_unversioned_casks_args.parse
|
|
|
|
|
|
|
|
state_file = if args.state_file.present?
|
|
|
|
Pathname(args.state_file).expand_path
|
|
|
|
else
|
|
|
|
HOMEBREW_CACHE/"bump_unversioned_casks.json"
|
|
|
|
end
|
|
|
|
state_file.dirname.mkpath
|
|
|
|
|
2020-12-05 21:30:02 +01:00
|
|
|
state = state_file.exist? ? JSON.parse(state_file.read) : {}
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2020-12-09 03:12:21 +01:00
|
|
|
casks = args.named.to_paths(only: :cask, recurse_tap: true).map { |path| Cask::CaskLoader.load(path) }
|
2020-12-08 01:03:39 +01:00
|
|
|
|
2020-12-09 03:12:21 +01:00
|
|
|
unversioned_casks = casks.select { |cask| cask.url&.unversioned? }
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2020-12-08 01:03:39 +01:00
|
|
|
ohai "Unversioned Casks: #{unversioned_casks.count} (#{state.size} cached)"
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2020-12-05 21:30:02 +01:00
|
|
|
checked, unchecked = unversioned_casks.partition { |c| state.key?(c.full_name) }
|
|
|
|
|
|
|
|
queue = Queue.new
|
|
|
|
|
2020-12-07 16:31:45 +01:00
|
|
|
# Start with random casks which have not been checked.
|
2020-12-05 21:30:02 +01:00
|
|
|
unchecked.shuffle.each do |c|
|
|
|
|
queue.enq c
|
|
|
|
end
|
2020-12-07 16:31:45 +01:00
|
|
|
|
|
|
|
# Continue with previously checked casks, ordered by when they were last checked.
|
2020-12-05 21:30:02 +01:00
|
|
|
checked.sort_by { |c| state.dig(c.full_name, "check_time") }.each do |c|
|
|
|
|
queue.enq c
|
|
|
|
end
|
|
|
|
|
|
|
|
limit = args.limit.presence&.to_i
|
|
|
|
end_time = Time.now + limit.minutes if limit
|
|
|
|
|
|
|
|
until queue.empty? || (end_time && end_time < Time.now)
|
|
|
|
cask = queue.deq
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2020-12-07 17:23:56 +01:00
|
|
|
key = cask.full_name
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2020-12-07 17:23:56 +01:00
|
|
|
new_state = bump_unversioned_cask(cask, state: state.fetch(key, {}), dry_run: args.dry_run?)
|
2020-12-07 16:31:45 +01:00
|
|
|
|
2020-12-07 17:23:56 +01:00
|
|
|
next unless new_state
|
2020-12-06 19:14:09 +01:00
|
|
|
|
2020-12-07 17:23:56 +01:00
|
|
|
state[key] = new_state
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2021-04-09 14:06:21 +01:00
|
|
|
state_file.atomic_write JSON.pretty_generate(state) unless args.dry_run?
|
2020-12-07 17:23:56 +01:00
|
|
|
end
|
|
|
|
end
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2021-01-17 22:45:55 -08:00
|
|
|
sig {
|
2020-12-07 17:23:56 +01:00
|
|
|
params(cask: Cask::Cask, state: T::Hash[String, T.untyped], dry_run: T.nilable(T::Boolean))
|
|
|
|
.returns(T.nilable(T::Hash[String, T.untyped]))
|
2021-01-17 22:45:55 -08:00
|
|
|
}
|
2020-12-07 17:23:56 +01:00
|
|
|
def self.bump_unversioned_cask(cask, state:, dry_run:)
|
|
|
|
ohai "Checking #{cask.full_name}"
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2020-12-07 17:23:56 +01:00
|
|
|
unversioned_cask_checker = UnversionedCaskChecker.new(cask)
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2021-01-07 13:49:05 -08:00
|
|
|
if !unversioned_cask_checker.single_app_cask? && !unversioned_cask_checker.single_pkg_cask?
|
2020-12-07 17:23:56 +01:00
|
|
|
opoo "Skipping, not a single-app or PKG cask."
|
|
|
|
return
|
|
|
|
end
|
|
|
|
|
|
|
|
last_check_time = state["check_time"]&.yield_self { |t| Time.parse(t) }
|
|
|
|
|
|
|
|
check_time = Time.now
|
|
|
|
if last_check_time && check_time < (last_check_time + 1.day)
|
|
|
|
opoo "Skipping, already checked within the last 24 hours."
|
|
|
|
return
|
|
|
|
end
|
|
|
|
|
|
|
|
last_sha256 = state["sha256"]
|
|
|
|
last_time = state["time"]&.yield_self { |t| Time.parse(t) }
|
|
|
|
last_file_size = state["file_size"]
|
|
|
|
|
|
|
|
download = Cask::Download.new(cask)
|
|
|
|
time, file_size = begin
|
|
|
|
download.time_file_size
|
|
|
|
rescue
|
|
|
|
[nil, nil]
|
|
|
|
end
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2020-12-07 17:23:56 +01:00
|
|
|
if last_time != time || last_file_size != file_size
|
2020-12-09 03:12:21 +01:00
|
|
|
sha256 = begin
|
|
|
|
Timeout.timeout(5.minutes) do
|
|
|
|
unversioned_cask_checker.installer.download.sha256
|
|
|
|
end
|
2020-12-07 17:23:56 +01:00
|
|
|
rescue => e
|
|
|
|
onoe e
|
|
|
|
end
|
2020-12-04 00:07:02 +01:00
|
|
|
|
2020-12-11 02:21:20 +01:00
|
|
|
if sha256.present? && last_sha256 != sha256
|
|
|
|
version = begin
|
|
|
|
Timeout.timeout(1.minute) do
|
|
|
|
unversioned_cask_checker.guess_cask_version
|
2020-12-07 17:23:56 +01:00
|
|
|
end
|
2020-12-11 02:21:20 +01:00
|
|
|
rescue Timeout::Error
|
|
|
|
onoe "Timed out guessing version for cask '#{cask}'."
|
|
|
|
end
|
2020-12-07 17:23:56 +01:00
|
|
|
|
2020-12-11 02:21:20 +01:00
|
|
|
if version
|
|
|
|
if cask.version == version
|
|
|
|
oh1 "Cask #{cask} is up-to-date at #{version}"
|
|
|
|
else
|
|
|
|
bump_cask_pr_args = [
|
|
|
|
"bump-cask-pr",
|
|
|
|
"--version", version.to_s,
|
|
|
|
"--sha256", ":no_check",
|
|
|
|
"--message", "Automatic update via `brew bump-unversioned-casks`.",
|
|
|
|
cask.sourcefile_path
|
|
|
|
]
|
|
|
|
|
|
|
|
if dry_run
|
|
|
|
bump_cask_pr_args << "--dry-run"
|
|
|
|
oh1 "Would bump #{cask} from #{cask.version} to #{version}"
|
|
|
|
else
|
|
|
|
oh1 "Bumping #{cask} from #{cask.version} to #{version}"
|
|
|
|
end
|
|
|
|
|
|
|
|
begin
|
|
|
|
system_command! HOMEBREW_BREW_FILE, args: bump_cask_pr_args
|
|
|
|
rescue ErrorDuringExecution => e
|
|
|
|
onoe e
|
|
|
|
end
|
2020-12-04 00:07:02 +01:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2020-12-06 20:04:53 +01:00
|
|
|
end
|
2020-12-07 17:23:56 +01:00
|
|
|
|
|
|
|
{
|
|
|
|
"sha256" => sha256,
|
|
|
|
"check_time" => check_time.iso8601,
|
|
|
|
"time" => time&.iso8601,
|
|
|
|
"file_size" => file_size,
|
|
|
|
}
|
2020-12-04 00:07:02 +01:00
|
|
|
end
|
|
|
|
end
|