Merge pull request #2031 from CarterLi/master

fastfetch: initial commit

completers/fastfetch_completer/cmd/root.go

@@ -0,0 +1,146 @@
+package cmd
+
+import (
+	"encoding/json"
+	"os/exec"
+	"strings"
+
+	"github.com/rsteube/carapace"
+	"github.com/spf13/cobra"
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "fastfetch",
+	Short: "A neofetch-like tool for fetching system information and displaying them in a pretty way",
+	Long:  "https://github.com/fastfetch-cli/fastfetch",
+	Run:   func(cmd *cobra.Command, args []string) {},
+}
+
+func Execute() error {
+	return rootCmd.Execute()
+}
+
+type FlagData struct {
+	Short  string
+	Long   string
+	Desc   string
+	Remark string
+	Pseudo bool
+	Arg    *struct {
+		Type     string
+		Optional bool
+		Default  interface{}
+		Enum     map[string]string
+	}
+}
+
+func init() {
+	carapace.Gen(rootCmd).PreRun(func(cmd *cobra.Command, args []string) {
+		carapace.Gen(rootCmd).Standalone()
+
+		var groups map[string][]FlagData
+		if out, err := exec.Command("fastfetch", "--help-raw").Output(); err == nil {
+			json.Unmarshal(out, &groups)
+			if len(groups) == 0 {
+				return
+			}
+		} else {
+			return
+		}
+
+		actionBools := carapace.ActionValues("true", "false")
+		actionColors := carapace.ActionValues("black", "red", "green", "yellow", "blue", "magenta", "cyan", "white", "default")
+		actionMap := carapace.ActionMap{}
+
+		for _, flags := range groups {
+			for _, flag := range flags {
+				if flag.Pseudo {
+					continue
+				}
+
+				if flag.Arg != nil {
+					switch flag.Arg.Type {
+					case "bool":
+						rootCmd.Flags().BoolP(flag.Long, flag.Short, false, flag.Desc)
+						actionMap[flag.Long] = actionBools
+
+					case "color":
+						rootCmd.Flags().StringP(flag.Long, flag.Short, "", flag.Desc)
+						actionMap[flag.Long] = actionColors
+
+					case "command":
+						rootCmd.Flags().StringP(flag.Long, flag.Short, "", flag.Desc)
+						actionMap[flag.Long] = carapace.ActionExecCommand("fastfetch", "--list-modules", "autocompletion")(func(output []byte) carapace.Action {
+							options := []string{"color", "format"}
+							for _, flags := range groups {
+								for _, flag := range flags {
+									options = append(options, flag.Long)
+								}
+							}
+							for _, line := range strings.Split(strings.TrimRight(string(output), "\n"), "\n") {
+								x, _, _ := strings.Cut(line, ":")
+								options = append(options, x)
+							}
+
+							return carapace.ActionValues(options...)
+						})
+
+					case "config":
+						rootCmd.Flags().StringP(flag.Long, flag.Short, "", flag.Desc)
+						actionMap[flag.Long] = carapace.ActionExecCommand("fastfetch", "--list-presets", "autocompletion")(func(output []byte) carapace.Action {
+							presets := strings.Split(strings.TrimRight(string(output), "\n"), "\n")
+							return carapace.ActionValues(presets...)
+						})
+
+					case "enum":
+						rootCmd.Flags().StringP(flag.Long, flag.Short, "", flag.Desc)
+						var valuesDescribed []string
+						for key, desc := range flag.Arg.Enum {
+							valuesDescribed = append(valuesDescribed, key, desc)
+						}
+						actionMap[flag.Long] = carapace.ActionValuesDescribed(valuesDescribed...)
+
+					case "logo":
+						rootCmd.Flags().StringP(flag.Long, flag.Short, "", flag.Desc)
+						actionMap[flag.Long] = carapace.ActionExecCommand("fastfetch", "--list-logos", "autocompletion")(func(output []byte) carapace.Action {
+							res := []string{"none", "Disable logo", "small", "Show small logo if supported"}
+							for _, logo := range strings.Split(strings.TrimRight(string(output), "\n"), "\n") {
+								res = append(res, logo, "Builtin logo")
+							}
+							return carapace.ActionValuesDescribed(res...)
+						})
+
+					case "num":
+						rootCmd.Flags().IntP(flag.Long, flag.Short, 0, flag.Desc)
+
+					case "path":
+						rootCmd.Flags().StringP(flag.Long, flag.Short, "", flag.Desc)
+						actionMap[flag.Long] = carapace.ActionFiles()
+
+					case "structure":
+						rootCmd.Flags().StringP(flag.Long, flag.Short, "", flag.Desc)
+						actionMap[flag.Long] = carapace.ActionExecCommand("fastfetch", "--list-modules", "autocompletion")(func(output []byte) carapace.Action {
+							var texts []string
+							for _, line := range strings.Split(strings.TrimRight(string(output), "\n"), "\n") {
+								name, desc, _ := strings.Cut(line, ":")
+								texts = append(texts, name, desc)
+							}
+							return carapace.ActionValuesDescribed(texts...).UniqueList(":")
+						})
+
+					default:
+						rootCmd.Flags().StringP(flag.Long, flag.Short, "", flag.Desc)
+					}
+				} else {
+					rootCmd.Flags().BoolP(flag.Long, flag.Short, false, flag.Desc)
+				}
+			}
+		}
+
+		carapace.Gen(rootCmd).FlagCompletion(actionMap)
+
+		carapace.Gen(rootCmd).PositionalCompletion(
+			carapace.ActionDirectories(),
+		)
+	})
+}

completers/fastfetch_completer/main.go

@@ -0,0 +1,7 @@
+package main
+
+import "github.com/rsteube/carapace-bin/completers/fastfetch_completer/cmd"
+
+func main() {
+	cmd.Execute()
+}