From 4030ecb67d99ff35610b3b830dfebf9569201434 Mon Sep 17 00:00:00 2001 From: Alex Mayer Date: Thu, 20 Jun 2019 00:33:10 -0400 Subject: Check Args In Main Function Remove parseArgs function --- main.go | 18 ++++++------------ 1 file changed, 6 insertions(+), 12 deletions(-) diff --git a/main.go b/main.go index f214a09..32ddf6c 100644 --- a/main.go +++ b/main.go @@ -8,15 +8,6 @@ import ( "os" ) -func parseArgs() (string, error) { - // We can only accept one argument - if len(os.Args) != 2 { - return "", fmt.Errorf("Usage: gh_authkey_checker ") - } - - return os.Args[1], nil -} - func fetchKeys(username string) (string, error) { url := fmt.Sprintf("https://github.com/%s.keys", username) resp, err := http.Get(url) @@ -42,11 +33,14 @@ func fetchKeys(username string) (string, error) { } func main() { - username, err := parseArgs() - if err != nil { - log.Fatal(err) + // Ensure we have the correct number of arguments + if len(os.Args) != 2 { + fmt.Println("Usage: gh_authkey_checker ") + os.Exit(1) } + username := os.Args[1] + log.Printf("Fetching keys for user %s", username) keys, err := fetchKeys(username) if err != nil { -- cgit v1.2.3