brew/Library/Homebrew/dev-cmd/bump-unversioned-casks.rb

# typed: false
# frozen_string_literal: true

require "cask/download"
require "cask/installer"
require "cask/cask_loader"
require "cli/parser"
require "tap"

module Homebrew
  extend T::Sig

  extend SystemCommand::Mixin

  sig { returns(CLI::Parser) }
  def self.bump_unversioned_casks_args
    Homebrew::CLI::Parser.new do
      usage_banner <<~EOS
        `bump-unversioned-casks` [<options>] [<tap>]

        Check all casks with unversioned URLs in a given <tap> for updates.
      EOS
      switch "-n", "--dry-run",
             description: "List what would be done, but do not actually do anything."
      flag  "--limit=",
            description: "Maximum number of casks to update."
      flag   "--state-file=",
             description: "File for keeping track of state."

      named 1
    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

    tap = Tap.fetch(args.named.first)

    old_state = state_file.exist? ? JSON.parse(state_file.read) : {}

    new_state = {}

    cask_files = tap.cask_files
    unversioned_cask_files = cask_files.select do |cask_file|
      url = cask_file.each_line do |line|
        url = line[/\s*url\s+"([^"]+)"\s*/, 1]
        break url if url
      end

      url.present? && url.exclude?('#{')
    end.sort

    unversioned_casks = unversioned_cask_files.map { |path| Cask::CaskLoader.load(path) }

    ohai "Static Casks:"
    puts "Total:      #{unversioned_casks.count}"
    puts "Single-App: #{unversioned_casks.count { |c| single_app_cask?(c) }}"
    puts "Single-Pkg: #{unversioned_casks.count { |c| single_pkg_cask?(c) }}"

    limit = args.limit.presence&.to_i || unversioned_casks.count

    unversioned_casks.shuffle.each do |cask|
      ohai "Checking #{cask.full_name}"

      unless single_app_cask?(cask)
        opoo "Skipping, not a single-app cask."
        next
      end

      download = Cask::Download.new(cask)
      time, file_size = begin
        download.time_file_size
      rescue
        opoo "Skipping, cannot get time and file size."
        next
      end

      odebug "Time: #{time.inspect}"
      odebug "Size: #{file_size.inspect}"

      last_state = old_state.fetch(cask.full_name, {})
      last_check_time = last_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."
        next
      end

      last_sha256 = last_state["sha256"]
      last_time = last_state["time"]&.yield_self { |t| Time.parse(t) }
      last_file_size = last_state["file_size"]

      next if last_time == time && last_file_size == file_size

      installer = Cask::Installer.new(cask, verify_download_integrity: false)

      begin
        cached_download = installer.download
      rescue => e
        onoe e
        next
      end

      sha256 = cached_download.sha256

      if last_sha256 != sha256 && (version = guess_cask_version(cask, installer))
        odebug "Version: #{version.inspect}"

        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 args.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
            Homebrew.failed = true
          end
        end
      end

      unless args.dry_run?
        new_state[cask.full_name] = {
          "sha256"     => sha256,
          "check_time" => check_time.iso8601,
          "time"       => time&.iso8601,
          "file_size"  => file_size,
        }
      end

      break if (limit -= 1).zero?
    end

    state_file.atomic_write JSON.pretty_generate(old_state.merge(new_state))
  end

  sig { params(cask: Cask::Cask, installer: Cask::Installer).returns(T.nilable(String)) }
  def self.guess_cask_version(cask, installer)
    apps = cask.artifacts.select { |a| a.is_a?(Cask::Artifact::App) }
    return if apps.count != 1

    Dir.mktmpdir do |dir|
      dir = Pathname(dir)

      installer.extract_primary_container(to: dir)

      plists = apps.flat_map do |app|
        Pathname.glob(dir/"**"/app.source.basename/"Contents"/"Info.plist")
      end
      next if plists.empty?

      plist = plists.first

      system_command! "plutil", args: ["-convert", "xml1", plist]
      plist = Plist.parse_xml(plist.read)

      short_version = plist["CFBundleShortVersionString"]
      version = plist["CFBundleVersion"]

      return "#{short_version},#{version}" if cask.version.include?(",")

      return cask.version.to_s if [short_version, version].include?(cask.version.to_s)

      return short_version if short_version&.match(/\A\d+(\.\d+)+\Z/)
      return version if version&.match(/\A\d+(\.\d+)+\Z/)

      short_version || version
    end
  end

  def self.single_app_cask?(cask)
    cask.artifacts.count { |a| a.is_a?(Cask::Artifact::App) } == 1
  end

  def self.single_pkg_cask?(cask)
    cask.artifacts.count { |a| a.is_a?(Cask::Artifact::Pkg) } == 1
  end
end