From 3639d23dccd33e0d3d4ad9f0902d8f29a9cd4b71 Mon Sep 17 00:00:00 2001
From: zeekrs <git@zeekrs.dev>
Date: Tue, 9 Jan 2024 22:12:50 +0800
Subject: [PATCH] feat: add completion for fish

---
 cmd/hostctl/actions/completion.go | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/cmd/hostctl/actions/completion.go b/cmd/hostctl/actions/completion.go
index da7804c..9d6a6fe 100644
--- a/cmd/hostctl/actions/completion.go
+++ b/cmd/hostctl/actions/completion.go
@@ -8,14 +8,15 @@ import (
 
 func newCompletionCmd(rootCmd *cobra.Command) *cobra.Command {
 	completionCmd := &cobra.Command{
-		Use:    "completion <bash|zsh>",
-		Short:  "Generate bash or zsh completion script",
+		Use:    "completion <bash|zsh|fish>",
+		Short:  "Generate bash zsh or fish completion script",
 		Hidden: true,
 	}
 
 	bashCompletionCmd := newBashCompletionCmd(rootCmd)
 	zshCompletionCmd := newZshCompletionCmd(rootCmd)
-	completionCmd.AddCommand(bashCompletionCmd, zshCompletionCmd)
+	fishCompletionCmd := newFishCompletionCmd(rootCmd)
+	completionCmd.AddCommand(bashCompletionCmd, zshCompletionCmd, fishCompletionCmd)
 
 	return completionCmd
 }
@@ -39,3 +40,13 @@ func newZshCompletionCmd(rootCmd *cobra.Command) *cobra.Command {
 		},
 	}
 }
+
+func newFishCompletionCmd(rootCmd *cobra.Command) *cobra.Command {
+	return &cobra.Command{
+		Use:   "fish",
+		Short: "Generate fish completion script",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return rootCmd.GenFishCompletion(os.Stdout, true)
+		},
+	}
+}