brew/Library/Homebrew/cmd/pyenv-sync.rb
John Britton 03b455f460
cmd/pyenv-sync: add new command.
Similar to `rbenv-sync` and `nodenv-sync`, but for use with `pyenv`.

Python has separate formulae for minor Python versions, as such this
will symlink all patch versions to the latest minor version.
2023-06-29 09:37:56 -04:00

69 lines
1.8 KiB
Ruby

# 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
dot_pyenv = Pathname(Dir.home)/".pyenv"
# Don't run multiple times at once.
pyenv_sync_running = dot_pyenv/".pyenv_sync_running"
return if pyenv_sync_running.exist?
begin
pyenv_versions = dot_pyenv/"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