From b221d04845823808559e753df9472f48d5ce719f Mon Sep 17 00:00:00 2001
From: Daeho Ro <lamanus@outlook.kr>
Date: Tue, 17 Dec 2024 07:52:45 +0900
Subject: [PATCH] formula: add clap completion style

---
 Library/Homebrew/formula.rb | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/Library/Homebrew/formula.rb b/Library/Homebrew/formula.rb
index bede10e37813dc..47762c09ce6f1d 100644
--- a/Library/Homebrew/formula.rb
+++ b/Library/Homebrew/formula.rb
@@ -2105,6 +2105,15 @@ def extract_macho_slice_from(file, arch = Hardware::CPU.arch)
   #                                                     bin/"foo")
   # ```
   #
+  # Using predefined `shell_parameter_format :clap`.
+  #
+  # ```ruby
+  # generate_completions_from_executable(bin/"foo", shell_parameter_format: :clap, shells: [:zsh])
+  #
+  # # translates to
+  # (zsh_completion/"_foo").write Utils.safe_popen_read({ "SHELL" => "zsh", "COMPLETE" => "zsh" }, bin/"foo")
+  # ```
+  #
   # Using custom `shell_parameter_format`.
   #
   # ```ruby
@@ -2125,7 +2134,7 @@ def extract_macho_slice_from(file, arch = Hardware::CPU.arch)
   #   the shells to generate completion scripts for. Defaults to `[:bash, :zsh, :fish]`.
   # @param shell_parameter_format
   #   specify how `shells` should each be passed to the `executable`. Takes either a String representing a
-  #   prefix, or one of `[:flag, :arg, :none, :click]`. Defaults to plainly passing the shell.
+  #   prefix, or one of `[:flag, :arg, :none, :click, :clap]`. Defaults to plainly passing the shell.
   sig {
     params(
       commands: T.any(Pathname, String),
@@ -2158,6 +2167,9 @@ def generate_completions_from_executable(*commands,
         prog_name = File.basename(commands.first.to_s).upcase.tr("-", "_")
         popen_read_env["_#{prog_name}_COMPLETE"] = "#{shell}_source"
         nil
+      elsif shell_parameter_format == :clap
+        popen_read_env["COMPLETE"] = shell.to_s
+        nil
       else
         "#{shell_parameter_format}#{shell}"
       end