From 9c9ebec3cd0666a1b0ec167cf3ef22d84989b9a8 Mon Sep 17 00:00:00 2001
From: iBug <git@ibugone.com>
Date: Sun, 14 Jul 2024 02:26:58 +0800
Subject: [PATCH] yukictl: Strip .yaml suffix from repo input

---
 pkg/yukictl/cmd/reload.go | 2 +-
 pkg/yukictl/cmd/sync.go   | 2 +-
 pkg/yukictl/cmd/util.go   | 9 +++++++++
 3 files changed, 11 insertions(+), 2 deletions(-)
 create mode 100644 pkg/yukictl/cmd/util.go

diff --git a/pkg/yukictl/cmd/reload.go b/pkg/yukictl/cmd/reload.go
index 4eaa389..6237527 100644
--- a/pkg/yukictl/cmd/reload.go
+++ b/pkg/yukictl/cmd/reload.go
@@ -42,7 +42,7 @@ func NewCmdReload(f factory.Factory) *cobra.Command {
 		Short: "Reload config of one or all repos",
 		RunE: func(cmd *cobra.Command, args []string) error {
 			if len(args) > 0 {
-				o.repo = args[0]
+				o.repo = stripSuffix(args[0])
 			}
 			return o.Run(f)
 		},
diff --git a/pkg/yukictl/cmd/sync.go b/pkg/yukictl/cmd/sync.go
index 2c91372..d5e22ec 100644
--- a/pkg/yukictl/cmd/sync.go
+++ b/pkg/yukictl/cmd/sync.go
@@ -43,7 +43,7 @@ func NewCmdSync(f factory.Factory) *cobra.Command {
 		Example: "  yukictl sync REPO",
 		Short:   "Sync local repository with remote",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			o.name = args[0]
+			o.name = stripSuffix(args[0])
 			return o.Run(f)
 		},
 	}
diff --git a/pkg/yukictl/cmd/util.go b/pkg/yukictl/cmd/util.go
new file mode 100644
index 0000000..58e6a60
--- /dev/null
+++ b/pkg/yukictl/cmd/util.go
@@ -0,0 +1,9 @@
+package cmd
+
+import "strings"
+
+const suffixYAML = ".yaml"
+
+func stripSuffix(s string) string {
+	return strings.TrimSuffix(s, suffixYAML)
+}