feat: Modify link checker to use first argument instead of -url flag
This commit is contained in:
parent
1a73c9ab03
commit
d68957a0c3
13
main.go
13
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)
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue