104 lines
2.6 KiB
Go
104 lines
2.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"axolotl/service"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var jsonFlag bool
|
|
var cfg service.Config
|
|
var rootCmd = &cobra.Command{Use: "ax", Short: "The axolotl issue tracker"}
|
|
|
|
func Execute() {
|
|
var err error
|
|
cfg, err = service.LoadConfigFile()
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, "failed to load config:", err)
|
|
os.Exit(1)
|
|
}
|
|
registerAliasCommands()
|
|
rootCmd.SetArgs(transformArgs(os.Args[1:]))
|
|
if err := rootCmd.Execute(); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.PersistentFlags().BoolVar(&jsonFlag, "json", false, "")
|
|
}
|
|
|
|
func addPropertyFlags(cmd *cobra.Command) {
|
|
cmd.Flags().String("type", "", "node type")
|
|
cmd.Flags().String("status", "", "node status")
|
|
cmd.Flags().String("prio", "", "node priority")
|
|
cmd.Flags().String("namespace", "", "node namespace")
|
|
}
|
|
|
|
func registerAliasCommands() {
|
|
rootCmd.AddGroup(&cobra.Group{ID: "aliases", Title: "Aliases:"})
|
|
aliases, _ := cfg.ListAliases()
|
|
for _, a := range aliases {
|
|
rootCmd.AddCommand(&cobra.Command{
|
|
Use: a.Name,
|
|
Short: a.Description,
|
|
GroupID: "aliases",
|
|
DisableFlagParsing: true,
|
|
Run: func(ccmd *cobra.Command, args []string) {
|
|
acmd := a.Command
|
|
acmd = strings.ReplaceAll(acmd, "$me", cfg.GetUser())
|
|
parts := strings.Fields(acmd)
|
|
var expanded []string
|
|
for _, part := range parts {
|
|
if part == "$@" {
|
|
expanded = append(expanded, args...)
|
|
continue
|
|
}
|
|
hasCatchAll := strings.Contains(part, "$@")
|
|
replaced := part
|
|
if hasCatchAll {
|
|
replaced = strings.ReplaceAll(replaced, "$@", strings.Join(args, " "))
|
|
}
|
|
for i := len(args) - 1; i >= 0; i-- {
|
|
placeholder := fmt.Sprintf("$%d", i+1)
|
|
replaced = strings.ReplaceAll(replaced, placeholder, args[i])
|
|
}
|
|
expanded = append(expanded, replaced)
|
|
}
|
|
rootCmd.SetArgs(transformArgs(expanded))
|
|
if err := rootCmd.Execute(); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
},
|
|
})
|
|
}
|
|
}
|
|
|
|
func transformArgs(args []string) []string {
|
|
aliases := map[string]string{
|
|
"--status": "_status",
|
|
"--prio": "_prio",
|
|
"--type": "_type",
|
|
"--namespace": "_namespace",
|
|
}
|
|
result := []string{}
|
|
|
|
for i := 0; i < len(args); i++ {
|
|
if idx := strings.Index(args[i], "="); idx != -1 {
|
|
if prop, ok := aliases[args[i][:idx]]; ok {
|
|
result = append(result, "--tag", prop+"::"+args[i][idx+1:])
|
|
continue
|
|
}
|
|
}
|
|
if prop, ok := aliases[args[i]]; ok && i+1 < len(args) {
|
|
result, i = append(result, "--tag", prop+"::"+args[i+1]), i+1
|
|
continue
|
|
}
|
|
result = append(result, args[i])
|
|
}
|
|
return result
|
|
}
|