brew/Library/Homebrew/cmd/pyenv-sync.rb

# typed: true
# frozen_string_literal: true

require "cli/parser"
require "formula"

module Homebrew
  module_function

  sig { returns(CLI::Parser) }
  def pyenv_sync_args
    Homebrew::CLI::Parser.new do
      description <<~EOS
        Create symlinks for Homebrew's installed Python versions in `~/.pyenv/versions`.

        Note that older patch version symlinks will be created and linked to the minor
        version so e.g. Python 3.11.0 will also be symlinked to 3.11.3.
      EOS

      named_args :none
    end
  end

  sig { void }
  def pyenv_sync
    pyenv_root = Pathname(ENV.fetch("HOMEBREW_PYENV_ROOT", Pathname(Dir.home)/".pyenv"))

    # Don't run multiple times at once.
    pyenv_sync_running = pyenv_root/".pyenv_sync_running"
    return if pyenv_sync_running.exist?

    begin
      pyenv_versions = pyenv_root/"versions"
      pyenv_versions.mkpath
      FileUtils.touch pyenv_sync_running

      pyenv_sync_args.parse

      HOMEBREW_CELLAR.glob("python{,@*}")
                     .flat_map(&:children)
                     .each { |path| link_pyenv_versions(path, pyenv_versions) }

      pyenv_versions.children
                    .select(&:symlink?)
                    .reject(&:exist?)
                    .each { |path| FileUtils.rm_f path }
    ensure
      pyenv_sync_running.unlink if pyenv_sync_running.exist?
    end
  end

  sig { params(path: Pathname, pyenv_versions: Pathname).void }
  def link_pyenv_versions(path, pyenv_versions)
    pyenv_versions.mkpath

    version = Keg.new(path).version
    major_version = version.major.to_i
    minor_version = version.minor.to_i
    patch_version = version.patch.to_i

    (0..patch_version).each do |patch|
      link_path = pyenv_versions/"#{major_version}.#{minor_version}.#{patch}"

      FileUtils.rm_f link_path
      FileUtils.ln_sf path, link_path
    end
  end
end