diff --git a/main.go b/main.go index b318b64..759f4e3 100644 --- a/main.go +++ b/main.go @@ -1,7 +1,6 @@ package main import ( - "flag" "fmt" "log" "net/url" @@ -11,24 +10,22 @@ import ( ) func main() { - targetURL := flag.String("url", "", "URL to check for broken links") - flag.Parse() - - if *targetURL == "" { - log.Fatal("Please provide a URL using the -url flag") + if len(os.Args) < 2 { + log.Fatal("Please provide a URL as an argument") } + targetURL := os.Args[1] // Start time for the filename startTime := time.Now() checker := NewLinkChecker() - brokenLinks, err := checker.CheckLinks(*targetURL) + brokenLinks, err := checker.CheckLinks(targetURL) if err != nil { log.Fatal(err) } // Extract domain from URL - parsedURL, err := url.Parse(*targetURL) + parsedURL, err := url.Parse(targetURL) if err != nil { log.Fatal(err) }