From 4dfa91fd22d61cb476d70e0aa8e51f409d8f5783 Mon Sep 17 00:00:00 2001 From: Rui Chen Date: Thu, 30 Jan 2025 13:49:12 -0500 Subject: [PATCH] fix: use `flag.Args()` instead of `os.Args()` for leftover argument parsing Signed-off-by: Rui Chen --- main.go | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/main.go b/main.go index 777ffa0..29e9f72 100644 --- a/main.go +++ b/main.go @@ -33,14 +33,15 @@ func run() error { return nil } + args := flag.Args() var workdir string - switch { - case len(os.Args) > 2: - return fmt.Errorf("too many arguments, must be either 0 or 1") - case len(os.Args) == 2: - workdir = os.Args[1] - default: + switch len(args) { + case 0: workdir = "." + case 1: + workdir = args[0] + default: + return fmt.Errorf("too many arguments, must be either 0 or 1") } tf := terraform.NewRunner(workdir, terraformBin)