feat: Modify link checker to use first argument instead of -url flag

This commit is contained in:
Erik Winter (aider) 2025-02-01 12:15:09 +01:00
parent 1a73c9ab03
commit d68957a0c3
1 changed files with 5 additions and 8 deletions

13
main.go
View File

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