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
|
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)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue