feat: introduce --shortcode and --url as mutually exclusive flags

1 files changed, 23 insertions, 1 deletions

diff --git a/main.go b/main.go
index 3b40dad..9e35e72 100644
--- a/main.go
+++ b/main.go
@@ -30,8 +30,29 @@ func main() {
 			{
 				Name:  "get",
 				Usage: "Crawl the target",
+				MutuallyExclusiveFlags: []cli.MutuallyExclusiveFlags{
+					{
+						Required: true,
+						Flags: [][]cli.Flag{
+							{
+								&cli.StringFlag{
+									Name:  "shortcode",
+									Usage: "Specify target using a shortcode",
+								},
+							},
+							{
+								&cli.StringFlag{
+									Name:  "url",
+									Usage: "Specify target using a URL",
+								},
+							},
+						},
+					},
+				},
 				Action: func(ctx context.Context, cmd *cli.Command) error {
 					fmt.Println("get!")
+					fmt.Println("shortcode: ", cmd.String("shortcode"))
+					fmt.Println("urls: ", cmd.String("url"))
 					return nil
 				},
 			},
@@ -47,7 +68,8 @@ func main() {
 	}
 
 	if err := cmd.Run(context.Background(), os.Args); err != nil {
-		log.Fatal(err)
+		fmt.Printf("\n%v\n", err)
+		os.Exit(1)
 	}
 }