From d38cfc8036b03f6d1a3d16a7b8c7062bbe26d73d Mon Sep 17 00:00:00 2001 From: John O'Keefe Date: Sat, 9 Dec 2023 11:51:31 -0500 Subject: [PATCH] allowed file to be entered as argument --- main.go | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/main.go b/main.go index c4a8950..8e91f11 100644 --- a/main.go +++ b/main.go @@ -21,10 +21,18 @@ func main() { Error(err) // Get name of file to be processed. - pterm.Printf(pterm.Cyan("What is the name of the file in this directory?: ")) - reader := bufio.NewReader(os.Stdin) - input, err := reader.ReadString('\n') - Error(err) + args := os.Args[1:] + + var input string + + if len(args) > 0 { + input = args[0] + } else { + pterm.Printf(pterm.Cyan("What is the name of the file in this directory?: ")) + reader := bufio.NewReader(os.Stdin) + input, err = reader.ReadString('\n') + Error(err) + } // Create upcs array and push all upc strings into it input = strings.TrimSpace(input)