brew/Library/Homebrew/cli/parser.rb

808 lines
28 KiB
Ruby
Raw Normal View History

2024-04-03 20:17:02 -07:00
# typed: strict
# frozen_string_literal: true
2024-03-18 15:50:50 -07:00
require "abstract_command"
2020-08-01 02:30:46 +02:00
require "env_config"
require "cask/config"
require "cli/args"
require "commands"
require "optparse"
2020-08-01 02:30:46 +02:00
require "utils/tty"
require "utils/formatter"
module Homebrew
module CLI
class Parser
# FIXME: Enable cop again when https://github.com/sorbet/sorbet/issues/3532 is fixed.
2024-04-03 20:17:02 -07:00
# rubocop:disable Style/MutableConstant
ArgType = T.type_alias { T.any(NilClass, Symbol, T::Array[String], T::Array[Symbol]) }
# rubocop:enable Style/MutableConstant
HIDDEN_DESC_PLACEHOLDER = "@@HIDDEN@@"
SYMBOL_TO_USAGE_MAPPING = T.let({
text_or_regex: "<text>|`/`<regex>`/`",
url: "<URL>",
}.freeze, T::Hash[Symbol, String])
private_constant :ArgType, :HIDDEN_DESC_PLACEHOLDER, :SYMBOL_TO_USAGE_MAPPING
2024-08-18 20:30:58 -07:00
sig { returns(Args::OptionsType) }
2024-04-03 20:17:02 -07:00
attr_reader :processed_options
2018-09-08 22:21:04 +05:30
2024-04-03 20:17:02 -07:00
sig { returns(T::Boolean) }
attr_reader :hide_from_man_page
sig { returns(ArgType) }
attr_reader :named_args_type
sig { params(cmd_path: Pathname).returns(T.nilable(CLI::Parser)) }
def self.from_cmd_path(cmd_path)
cmd_args_method_name = Commands.args_method_name(cmd_path)
2024-03-19 12:43:51 -07:00
cmd_name = cmd_args_method_name.to_s.delete_suffix("_args").tr("_", "-")
begin
2024-03-18 15:50:50 -07:00
if require?(cmd_path)
cmd = Homebrew::AbstractCommand.command(cmd_name)
if cmd
cmd.parser
else
2024-03-19 12:36:30 -07:00
# FIXME: remove once commands are all subclasses of `AbstractCommand`:
2024-03-18 15:50:50 -07:00
Homebrew.send(cmd_args_method_name)
end
end
rescue NoMethodError => e
raise if e.name.to_sym != cmd_args_method_name
nil
end
end
2024-04-03 20:17:02 -07:00
sig { returns(T::Array[[Symbol, String, { description: String }]]) }
def self.global_cask_options
[
[:flag, "--appdir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Applications " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:appdir]}`).",
}],
2023-03-26 08:10:40 +02:00
[:flag, "--keyboard-layoutdir=", {
description: "Target location for Keyboard Layouts " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:keyboard_layoutdir]}`).",
}],
[:flag, "--colorpickerdir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Color Pickers " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:colorpickerdir]}`).",
}],
[:flag, "--prefpanedir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Preference Panes " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:prefpanedir]}`).",
}],
[:flag, "--qlplugindir=", {
description: "Target location for Quick Look Plugins " \
2020-11-12 10:40:48 -05:00
"(default: `#{Cask::Config::DEFAULT_DIRS[:qlplugindir]}`).",
}],
[:flag, "--mdimporterdir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Spotlight Plugins " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:mdimporterdir]}`).",
}],
[:flag, "--dictionarydir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Dictionaries " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:dictionarydir]}`).",
}],
[:flag, "--fontdir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Fonts " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:fontdir]}`).",
}],
[:flag, "--servicedir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Services " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:servicedir]}`).",
}],
[:flag, "--input-methoddir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Input Methods " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:input_methoddir]}`).",
}],
[:flag, "--internet-plugindir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Internet Plugins " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:internet_plugindir]}`).",
}],
[:flag, "--audio-unit-plugindir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Audio Unit Plugins " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:audio_unit_plugindir]}`).",
}],
[:flag, "--vst-plugindir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for VST Plugins " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:vst_plugindir]}`).",
}],
[:flag, "--vst3-plugindir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for VST3 Plugins " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:vst3_plugindir]}`).",
}],
[:flag, "--screen-saverdir=", {
2020-11-12 10:40:48 -05:00
description: "Target location for Screen Savers " \
"(default: `#{Cask::Config::DEFAULT_DIRS[:screen_saverdir]}`).",
}],
[:comma_array, "--language", {
2020-11-12 10:40:48 -05:00
description: "Comma-separated list of language codes to prefer for cask installation. " \
"The first matching language is used, otherwise it reverts to the cask's " \
"default language. The default value is the language of your system.",
}],
]
end
2020-10-20 12:03:48 +02:00
sig { returns(T::Array[[String, String, String]]) }
def self.global_options
2020-07-30 18:40:10 +02:00
[
["-d", "--debug", "Display any debugging information."],
["-q", "--quiet", "Make some output more quiet."],
2020-07-30 18:40:10 +02:00
["-v", "--verbose", "Make some output more verbose."],
["-h", "--help", "Show this message."],
2020-07-30 18:40:10 +02:00
]
end
2024-04-03 20:17:02 -07:00
sig { params(option: String).returns(String) }
def self.option_to_name(option)
option.sub(/\A--?(\[no-\])?/, "").tr("-", "_").delete("=")
end
2024-03-18 16:11:28 -07:00
sig {
params(cmd: T.nilable(T.class_of(Homebrew::AbstractCommand)), block: T.nilable(T.proc.bind(Parser).void)).void
}
2024-03-18 15:50:50 -07:00
def initialize(cmd = nil, &block)
2024-04-03 20:17:02 -07:00
@parser = T.let(OptionParser.new, OptionParser)
@parser.summary_indent = " "
2020-08-01 02:30:46 +02:00
# Disable default handling of `--version` switch.
@parser.base.long.delete("version")
# Disable default handling of `--help` switch.
@parser.base.long.delete("help")
@args = T.let((cmd&.args_class || Args).new, Args)
2024-03-18 15:50:50 -07:00
if cmd
2024-04-03 20:17:02 -07:00
@command_name = T.let(cmd.command_name, String)
@is_dev_cmd = T.let(cmd.dev_cmd?, T::Boolean)
2024-03-18 15:50:50 -07:00
else
# FIXME: remove once commands are all subclasses of `AbstractCommand`:
# Filter out Sorbet runtime type checking method calls.
2024-04-03 20:17:02 -07:00
cmd_location = caller_locations.select do |location|
2024-03-18 15:50:50 -07:00
T.must(location.path).exclude?("/gems/sorbet-runtime-")
end.fetch(1)
2024-04-03 20:17:02 -07:00
@command_name = T.let(T.must(cmd_location.label).chomp("_args").tr("_", "-"), String)
@is_dev_cmd = T.let(T.must(cmd_location.absolute_path).start_with?(Commands::HOMEBREW_DEV_CMD_PATH),
T::Boolean)
# odeprecated(
# "`brew #{@command_name}'. This command needs to be refactored, as it is written in a style that",
# "inheritance from `Homebrew::AbstractCommand' ( see https://docs.brew.sh/External-Commands )",
# disable_for_developers: false,
# )
2024-04-03 20:17:02 -07:00
end
@constraints = T.let([], T::Array[[String, String]])
@conflicts = T.let([], T::Array[T::Array[String]])
@switch_sources = T.let({}, T::Hash[String, Symbol])
2024-08-18 20:30:58 -07:00
@processed_options = T.let([], Args::OptionsType)
2024-04-03 20:17:02 -07:00
@non_global_processed_options = T.let([], T::Array[[String, ArgType]])
@named_args_type = T.let(nil, T.nilable(ArgType))
@max_named_args = T.let(nil, T.nilable(Integer))
@min_named_args = T.let(nil, T.nilable(Integer))
@named_args_without_api = T.let(false, T::Boolean)
@description = T.let(nil, T.nilable(String))
@usage_banner = T.let(nil, T.nilable(String))
@hide_from_man_page = T.let(false, T::Boolean)
@formula_options = T.let(false, T::Boolean)
@cask_options = T.let(false, T::Boolean)
2020-07-30 18:40:10 +02:00
self.class.global_options.each do |short, long, desc|
2020-08-01 02:30:46 +02:00
switch short, long, description: desc, env: option_to_name(long), method: :on_tail
2020-07-30 18:40:10 +02:00
end
2020-11-16 22:18:56 +01:00
instance_eval(&block) if block
generate_banner
end
2024-04-03 20:17:02 -07:00
sig {
params(names: String, description: T.nilable(String), replacement: T.untyped, env: T.untyped,
depends_on: T.nilable(String), method: Symbol, hidden: T::Boolean, disable: T::Boolean).void
}
def switch(*names, description: nil, replacement: nil, env: nil, depends_on: nil,
2023-03-29 20:49:29 +02:00
method: :on, hidden: false, disable: false)
global_switch = names.first.is_a?(Symbol)
2020-07-30 18:40:10 +02:00
return if global_switch
2024-03-07 16:20:20 +00:00
description = option_description(description, *names, hidden:)
process_option(*names, description, type: :switch, hidden:) unless disable
if replacement || disable
description += " (#{disable ? "disabled" : "deprecated"}#{"; replaced by #{replacement}" if replacement})"
end
2020-08-01 02:30:46 +02:00
@parser.public_send(method, *names, *wrap_option_desc(description)) do |value|
# This odeprecated should stick around indefinitely.
2023-03-29 20:49:29 +02:00
odeprecated "the `#{names.first}` switch", replacement, disable: disable if !replacement.nil? || disable
2021-08-13 13:49:52 +01:00
value = true if names.none? { |name| name.start_with?("--[no-]") }
2020-08-14 20:03:15 +02:00
2024-03-07 16:20:20 +00:00
set_switch(*names, value:, from: :args)
end
names.each do |name|
2024-03-07 16:20:20 +00:00
set_constraints(name, depends_on:)
end
env_value = value_for_env(env)
2020-08-01 02:30:46 +02:00
set_switch(*names, value: env_value, from: :env) unless env_value.nil?
end
alias switch_option switch
2024-04-03 20:17:02 -07:00
sig { params(text: T.nilable(String)).returns(T.nilable(String)) }
2021-01-24 01:59:31 -05:00
def description(text = nil)
return @description if text.blank?
@description = text.chomp
end
2024-04-03 20:17:02 -07:00
sig { params(text: String).void }
2018-09-08 22:21:04 +05:30
def usage_banner(text)
@usage_banner, @description = text.chomp.split("\n\n", 2)
end
2024-04-03 20:17:02 -07:00
sig { returns(T.nilable(String)) }
def usage_banner_text = @parser.banner
2018-09-08 22:21:04 +05:30
2024-04-03 20:17:02 -07:00
sig { params(name: String, description: T.nilable(String), hidden: T::Boolean).void }
2021-06-08 22:02:32 -04:00
def comma_array(name, description: nil, hidden: false)
name = name.chomp "="
2024-03-07 16:20:20 +00:00
description = option_description(description, name, hidden:)
process_option(name, description, type: :comma_array, hidden:)
2018-09-08 22:21:04 +05:30
@parser.on(name, OptionParser::REQUIRED_ARGUMENT, Array, *wrap_option_desc(description)) do |list|
@args[option_to_name(name)] = list
end
end
2024-04-03 20:17:02 -07:00
sig {
2024-04-21 14:34:55 -07:00
params(names: String, description: T.nilable(String), replacement: T.any(Symbol, String, NilClass),
depends_on: T.nilable(String), hidden: T::Boolean).void
2024-04-03 20:17:02 -07:00
}
def flag(*names, description: nil, replacement: nil, depends_on: nil, hidden: false)
2021-01-18 11:03:23 -05:00
required, flag_type = if names.any? { |name| name.end_with? "=" }
[OptionParser::REQUIRED_ARGUMENT, :required_flag]
else
2021-01-18 11:03:23 -05:00
[OptionParser::OPTIONAL_ARGUMENT, :optional_flag]
end
names.map! { |name| name.chomp "=" }
2024-03-07 16:20:20 +00:00
description = option_description(description, *names, hidden:)
if replacement.nil?
2024-03-07 16:20:20 +00:00
process_option(*names, description, type: flag_type, hidden:)
else
description += " (disabled#{"; replaced by #{replacement}" if replacement.present?})"
end
@parser.on(*names, *wrap_option_desc(description), required) do |option_value|
# This odisabled should stick around indefinitely.
odisabled "the `#{names.first}` flag", replacement unless replacement.nil?
names.each do |name|
@args[option_to_name(name)] = option_value
end
end
names.each do |name|
2024-03-07 16:20:20 +00:00
set_constraints(name, depends_on:)
end
end
2024-04-03 20:17:02 -07:00
sig { params(options: String).returns(T::Array[T::Array[String]]) }
def conflicts(*options)
@conflicts << options.map { |option| option_to_name(option) }
end
2024-04-03 20:17:02 -07:00
sig { params(option: String).returns(String) }
2024-03-15 15:50:07 -07:00
def option_to_name(option) = self.class.option_to_name(option)
2024-04-03 20:17:02 -07:00
sig { params(name: String).returns(String) }
2018-06-01 14:19:00 +02:00
def name_to_option(name)
if name.length == 1
"-#{name}"
else
2018-11-07 23:44:34 +05:30
"--#{name.tr("_", "-")}"
2018-06-01 14:19:00 +02:00
end
end
2024-04-03 20:17:02 -07:00
sig { params(names: String).returns(T.nilable(String)) }
def option_to_description(*names)
names.map { |name| name.to_s.sub(/\A--?/, "").tr("-", " ") }.max
end
2024-04-03 20:17:02 -07:00
sig { params(description: T.nilable(String), names: String, hidden: T::Boolean).returns(String) }
2021-06-08 22:02:32 -04:00
def option_description(description, *names, hidden: false)
return HIDDEN_DESC_PLACEHOLDER if hidden
return description if description.present?
option_to_description(*names)
end
2024-04-03 20:17:02 -07:00
sig {
params(argv: T::Array[String], ignore_invalid_options: T::Boolean)
.returns([T::Array[String], T::Array[String]])
}
def parse_remaining(argv, ignore_invalid_options: false)
2020-07-30 18:40:10 +02:00
i = 0
remaining = []
argv, non_options = split_non_options(argv)
allow_commands = Array(@named_args_type).include?(:command)
2020-07-30 18:40:10 +02:00
while i < argv.count
begin
begin
arg = argv[i]
remaining << arg unless @parser.parse([arg]).empty?
rescue OptionParser::MissingArgument
raise if i + 1 >= argv.count
args = argv[i..(i + 1)]
@parser.parse(args)
i += 1
end
rescue OptionParser::InvalidOption
if ignore_invalid_options || (allow_commands && Commands.path(arg))
2020-07-30 18:40:10 +02:00
remaining << arg
else
$stderr.puts generate_help_text
raise
end
end
i += 1
end
[remaining, non_options]
end
2024-04-03 20:17:02 -07:00
sig { params(argv: T::Array[String], ignore_invalid_options: T::Boolean).returns(Args) }
def parse(argv = ARGV.freeze, ignore_invalid_options: false)
raise "Arguments were already parsed!" if @args_parsed
# If we accept formula options, but the command isn't scoped only
# to casks, parse once allowing invalid options so we can get the
# remaining list containing formula names.
if @formula_options && !only_casks?(argv)
remaining, non_options = parse_remaining(argv, ignore_invalid_options: true)
argv = [*remaining, "--", *non_options]
formulae(argv).each do |f|
next if f.options.empty?
f.options.each do |o|
name = o.flag
description = "`#{f.name}`: #{o.description}"
if name.end_with? "="
2024-03-07 16:20:20 +00:00
flag(name, description:)
else
2024-03-07 16:20:20 +00:00
switch name, description:
end
conflicts "--cask", name
end
end
end
2024-03-07 16:20:20 +00:00
remaining, non_options = parse_remaining(argv, ignore_invalid_options:)
2020-07-30 18:40:10 +02:00
named_args = if ignore_invalid_options
[]
else
remaining + non_options
end
2020-08-01 02:30:46 +02:00
unless ignore_invalid_options
unless @is_dev_cmd
set_default_options
validate_options
end
check_constraint_violations
2020-08-01 02:30:46 +02:00
check_named_args(named_args)
end
@args.freeze_named_args!(named_args, cask_options: @cask_options, without_api: @named_args_without_api)
@args.freeze_remaining_args!(non_options.empty? ? remaining : [*remaining, "--", non_options])
@args.freeze_processed_options!(@processed_options)
2021-03-18 14:46:48 +00:00
@args.freeze
2024-04-03 20:17:02 -07:00
@args_parsed = T.let(true, T.nilable(TrueClass))
2020-08-01 02:30:46 +02:00
if !ignore_invalid_options && @args.help?
puts generate_help_text
exit
2020-08-01 02:30:46 +02:00
end
2020-07-23 01:22:25 +02:00
@args
end
2024-04-03 20:17:02 -07:00
sig { void }
def set_default_options; end
2024-04-03 20:17:02 -07:00
sig { void }
2022-09-28 21:57:13 -07:00
def validate_options; end
2024-04-03 20:17:02 -07:00
sig { returns(String) }
def generate_help_text
2024-04-03 20:17:02 -07:00
Formatter.format_help_text(@parser.to_s, width: Formatter::COMMAND_DESC_WIDTH)
2021-06-08 22:02:32 -04:00
.gsub(/\n.*?@@HIDDEN@@.*?(?=\n)/, "")
2020-08-01 02:30:46 +02:00
.sub(/^/, "#{Tty.bold}Usage: brew#{Tty.reset} ")
.gsub(/`(.*?)`/m, "#{Tty.bold}\\1#{Tty.reset}")
.gsub(%r{<([^\s]+?://[^\s]+?)>}) { |url| Formatter.url(url) }
.gsub(/\*(.*?)\*|<(.*?)>/m) do |underlined|
underlined[1...-1].gsub(/^(\s*)(.*?)$/, "\\1#{Tty.underline}\\2#{Tty.reset}")
end
end
2024-04-03 20:17:02 -07:00
sig { void }
def cask_options
self.class.global_cask_options.each do |args|
2024-04-03 20:17:02 -07:00
options = T.cast(args.pop, T::Hash[Symbol, String])
send(*args, **options)
2024-04-03 20:17:02 -07:00
conflicts "--formula", args[1]
end
2021-03-18 14:46:48 +00:00
@cask_options = true
end
2020-10-20 12:03:48 +02:00
sig { void }
def formula_options
@formula_options = true
end
sig {
2021-01-10 14:26:40 -05:00
params(
2024-04-03 20:17:02 -07:00
type: ArgType,
number: T.nilable(Integer),
min: T.nilable(Integer),
max: T.nilable(Integer),
without_api: T::Boolean,
2021-01-10 14:26:40 -05:00
).void
}
def named_args(type = nil, number: nil, min: nil, max: nil, without_api: false)
2021-01-10 14:26:40 -05:00
if number.present? && (min.present? || max.present?)
raise ArgumentError, "Do not specify both `number` and `min` or `max`"
end
if type == :none && (number.present? || min.present? || max.present?)
raise ArgumentError, "Do not specify both `number`, `min` or `max` with `named_args :none`"
end
@named_args_type = type
if type == :none
@max_named_args = 0
2024-04-03 20:17:02 -07:00
elsif number
2021-01-10 14:26:40 -05:00
@min_named_args = @max_named_args = number
2024-04-03 20:17:02 -07:00
elsif min || max
2021-01-10 14:26:40 -05:00
@min_named_args = min
@max_named_args = max
end
@named_args_without_api = without_api
2021-01-10 14:26:40 -05:00
end
2020-10-20 12:03:48 +02:00
sig { void }
def hide_from_man_page!
@hide_from_man_page = true
end
private
2024-04-03 20:17:02 -07:00
sig { returns(String) }
def generate_usage_banner
command_names = ["`#{@command_name}`"]
aliases_to_skip = %w[instal uninstal]
command_names += Commands::HOMEBREW_INTERNAL_COMMAND_ALIASES.filter_map do |command_alias, command|
next if aliases_to_skip.include? command_alias
"`#{command_alias}`" if command == @command_name
end.sort
options = if @non_global_processed_options.empty?
""
elsif @non_global_processed_options.count > 2
" [<options>]"
else
2021-01-18 11:03:23 -05:00
required_argument_types = [:required_flag, :comma_array]
@non_global_processed_options.map do |option, type|
2023-02-10 23:15:40 -05:00
next " [`#{option}=`]" if required_argument_types.include? type
2023-02-10 23:15:40 -05:00
" [`#{option}`]"
end.join
end
named_args = ""
if @named_args_type.present? && @named_args_type != :none
arg_type = if @named_args_type.is_a? Array
types = @named_args_type.filter_map do |type|
next unless type.is_a? Symbol
next SYMBOL_TO_USAGE_MAPPING[type] if SYMBOL_TO_USAGE_MAPPING.key?(type)
"<#{type}>"
end
2021-03-01 13:43:47 +00:00
types << "<subcommand>" if @named_args_type.any?(String)
types.join("|")
elsif SYMBOL_TO_USAGE_MAPPING.key? @named_args_type
SYMBOL_TO_USAGE_MAPPING[@named_args_type]
else
"<#{@named_args_type}>"
end
named_args = if @min_named_args.blank? && @max_named_args == 1
" [#{arg_type}]"
elsif @min_named_args.blank?
" [#{arg_type} ...]"
elsif @min_named_args == 1 && @max_named_args == 1
" #{arg_type}"
elsif @min_named_args == 1
" #{arg_type} [...]"
else
" #{arg_type} ..."
end
end
"#{command_names.join(", ")}#{options}#{named_args}"
end
2024-04-03 20:17:02 -07:00
sig { returns(String) }
def generate_banner
@usage_banner ||= generate_usage_banner
@parser.banner = <<~BANNER
#{@usage_banner}
#{@description}
BANNER
end
2024-04-03 20:17:02 -07:00
sig { params(names: String, value: T.untyped, from: Symbol).void }
2020-08-01 02:30:46 +02:00
def set_switch(*names, value:, from:)
names.each do |name|
2019-02-15 11:01:03 -05:00
@switch_sources[option_to_name(name)] = from
2020-08-01 02:30:46 +02:00
@args["#{option_to_name(name)}?"] = value
end
end
2024-04-03 20:17:02 -07:00
sig { params(args: String).void }
def disable_switch(*args)
args.each do |name|
2021-03-18 14:46:48 +00:00
@args["#{option_to_name(name)}?"] = if name.start_with?("--[no-]")
nil
else
false
end
end
end
2024-04-03 20:17:02 -07:00
sig { params(name: String).returns(T::Boolean) }
def option_passed?(name)
2024-04-03 20:17:02 -07:00
!!(@args[name.to_sym] || @args[:"#{name}?"])
end
2024-04-03 20:17:02 -07:00
sig { params(desc: String).returns(T::Array[String]) }
def wrap_option_desc(desc)
2024-04-03 20:17:02 -07:00
Formatter.format_help_text(desc, width: Formatter::OPTION_DESC_WIDTH).split("\n")
end
2024-04-03 20:17:02 -07:00
sig { params(name: String, depends_on: T.nilable(String)).returns(T.nilable(T::Array[[String, String]])) }
def set_constraints(name, depends_on:)
return if depends_on.nil?
2018-09-17 02:45:00 +02:00
primary = option_to_name(depends_on)
secondary = option_to_name(name)
@constraints << [primary, secondary]
end
2024-04-03 20:17:02 -07:00
sig { void }
def check_constraints
@constraints.each do |primary, secondary|
primary_passed = option_passed?(primary)
secondary_passed = option_passed?(secondary)
next if !secondary_passed || (primary_passed && secondary_passed)
primary = name_to_option(primary)
secondary = name_to_option(secondary)
raise OptionConstraintError.new(primary, secondary, missing: true)
end
end
2024-04-03 20:17:02 -07:00
sig { void }
def check_conflicts
@conflicts.each do |mutually_exclusive_options_group|
violations = mutually_exclusive_options_group.select do |option|
option_passed? option
end
2018-06-01 14:19:00 +02:00
next if violations.count < 2
2018-09-17 02:45:00 +02:00
env_var_options = violations.select do |option|
2019-02-25 14:54:31 -05:00
@switch_sources[option_to_name(option)] == :env
end
select_cli_arg = violations.count - env_var_options.count == 1
raise OptionConflictError, violations.map { name_to_option(_1) } unless select_cli_arg
env_var_options.each { disable_switch(_1) }
end
end
2024-04-03 20:17:02 -07:00
sig { void }
def check_invalid_constraints
@conflicts.each do |mutually_exclusive_options_group|
@constraints.each do |p, s|
next unless Set[p, s].subset?(Set[*mutually_exclusive_options_group])
2018-09-17 02:45:00 +02:00
raise InvalidConstraintError.new(p, s)
end
end
end
2024-04-03 20:17:02 -07:00
sig { void }
def check_constraint_violations
check_invalid_constraints
check_conflicts
check_constraints
end
2018-09-08 22:21:04 +05:30
2024-04-03 20:17:02 -07:00
sig { params(args: T::Array[String]).void }
2020-07-30 18:40:10 +02:00
def check_named_args(args)
types = Array(@named_args_type).filter_map do |type|
next type if type.is_a? Symbol
:subcommand
end.uniq
exception = if @min_named_args && @max_named_args && @min_named_args == @max_named_args &&
args.size != @max_named_args
2024-03-07 16:20:20 +00:00
NumberOfNamedArgumentsError.new(@min_named_args, types:)
elsif @min_named_args && args.size < @min_named_args
2024-03-07 16:20:20 +00:00
MinNamedArgumentsError.new(@min_named_args, types:)
2020-10-20 12:03:48 +02:00
elsif @max_named_args && args.size > @max_named_args
2024-03-07 16:20:20 +00:00
MaxNamedArgumentsError.new(@max_named_args, types:)
end
2020-10-20 12:03:48 +02:00
raise exception if exception
end
2024-04-03 20:17:02 -07:00
sig { params(args: String, type: Symbol, hidden: T::Boolean).void }
def process_option(*args, type:, hidden: false)
2018-09-08 22:21:04 +05:30
option, = @parser.make_switch(args)
@processed_options.reject! { |existing| existing.second == option.long.first } if option.long.first.present?
2024-08-19 13:15:34 -07:00
@processed_options << [option.short.first, option.long.first, option.desc.first, hidden]
args.pop # last argument is the description
2021-03-18 14:46:48 +00:00
if type == :switch
disable_switch(*args)
else
args.each do |name|
@args[option_to_name(name)] = nil
end
end
2021-06-08 22:02:32 -04:00
return if hidden
return if self.class.global_options.include? [option.short.first, option.long.first, option.desc.first]
@non_global_processed_options << [option.long.first || option.short.first, type]
2018-09-08 22:21:04 +05:30
end
2024-04-03 20:17:02 -07:00
sig { params(argv: T::Array[String]).returns([T::Array[String], T::Array[String]]) }
def split_non_options(argv)
if (sep = argv.index("--"))
2020-07-30 18:40:10 +02:00
[argv.take(sep), argv.drop(sep + 1)]
else
[argv, []]
end
end
2024-04-03 20:17:02 -07:00
sig { params(argv: T::Array[String]).returns(T::Array[Formula]) }
2020-07-30 18:40:10 +02:00
def formulae(argv)
argv, non_options = split_non_options(argv)
2020-07-30 18:40:10 +02:00
named_args = argv.reject { |arg| arg.start_with?("-") } + non_options
2020-07-30 18:40:10 +02:00
spec = if argv.include?("--HEAD")
:head
else
:stable
end
# Only lowercase names, not paths, bottle filenames or URLs
named_args.filter_map do |arg|
next if arg.match?(HOMEBREW_CASK_TAP_CASK_REGEX)
begin
Formulary.factory(arg, spec, flags: argv.select { |a| a.start_with?("--") })
rescue FormulaUnavailableError, FormulaSpecificationError
nil
end
end.uniq(&:name)
end
2024-04-03 20:17:02 -07:00
sig { params(argv: T::Array[String]).returns(T::Boolean) }
def only_casks?(argv)
argv.include?("--casks") || argv.include?("--cask")
end
2024-04-03 20:17:02 -07:00
sig { params(env: T.any(NilClass, String, Symbol)).returns(T.untyped) }
def value_for_env(env)
return if env.blank?
method_name = :"#{env}?"
if Homebrew::EnvConfig.respond_to?(method_name)
Homebrew::EnvConfig.public_send(method_name)
else
ENV.fetch("HOMEBREW_#{env.upcase}", nil)
end
end
end
class OptionConstraintError < UsageError
2024-04-03 20:17:02 -07:00
sig { params(arg1: String, arg2: String, missing: T::Boolean).void }
def initialize(arg1, arg2, missing: false)
message = if missing
"`#{arg2}` cannot be passed without `#{arg1}`."
else
"`#{arg1}` and `#{arg2}` should be passed together."
end
super message
end
end
class OptionConflictError < UsageError
2024-04-03 20:17:02 -07:00
sig { params(args: T::Array[String]).void }
def initialize(args)
2024-04-03 20:17:02 -07:00
args_list = args.map { Formatter.option(_1) }.join(" and ")
super "Options #{args_list} are mutually exclusive."
end
end
class InvalidConstraintError < UsageError
2024-04-03 20:17:02 -07:00
sig { params(arg1: String, arg2: String).void }
def initialize(arg1, arg2)
super "`#{arg1}` and `#{arg2}` cannot be mutually exclusive and mutually dependent simultaneously."
end
end
class MaxNamedArgumentsError < UsageError
sig { params(maximum: Integer, types: T::Array[Symbol]).void }
def initialize(maximum, types: [])
super case maximum
when 0
"This command does not take named arguments."
else
types << :named if types.empty?
arg_types = types.map { |type| type.to_s.tr("_", " ") }
.to_sentence two_words_connector: " or ", last_word_connector: " or "
2023-02-27 20:49:02 -08:00
"This command does not take more than #{maximum} #{arg_types} #{Utils.pluralize("argument", maximum)}."
end
end
end
class MinNamedArgumentsError < UsageError
sig { params(minimum: Integer, types: T::Array[Symbol]).void }
def initialize(minimum, types: [])
types << :named if types.empty?
arg_types = types.map { |type| type.to_s.tr("_", " ") }
.to_sentence two_words_connector: " or ", last_word_connector: " or "
2023-02-27 20:49:02 -08:00
super "This command requires at least #{minimum} #{arg_types} #{Utils.pluralize("argument", minimum)}."
end
end
class NumberOfNamedArgumentsError < UsageError
sig { params(minimum: Integer, types: T::Array[Symbol]).void }
def initialize(minimum, types: [])
types << :named if types.empty?
arg_types = types.map { |type| type.to_s.tr("_", " ") }
.to_sentence two_words_connector: " or ", last_word_connector: " or "
2023-02-27 20:49:02 -08:00
super "This command requires exactly #{minimum} #{arg_types} #{Utils.pluralize("argument", minimum)}."
end
end
end
end
2022-09-28 21:57:13 -07:00
require "extend/os/parser"