77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
func main() {
|
|
url := flag.String("url", "", "URL to check for broken links")
|
|
flag.Parse()
|
|
|
|
if *url == "" {
|
|
log.Fatal("Please provide a URL using the -url flag")
|
|
}
|
|
|
|
checker := NewLinkChecker()
|
|
brokenLinks, err := checker.CheckLinks(*url)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Create results file with current date
|
|
filename := fmt.Sprintf("result-%s.txt", time.Now().Format("2006-01-02"))
|
|
f, err := os.Create(filename)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
fmt.Fprintf(f, "\nTotal pages checked: %d\n\n", checker.pagesChecked)
|
|
|
|
if len(brokenLinks) == 0 {
|
|
fmt.Fprintf(f, "No issues found!\n")
|
|
fmt.Printf("Results written to %s\n", filename)
|
|
return
|
|
}
|
|
|
|
// First list redirects
|
|
var redirectCount int
|
|
for _, link := range brokenLinks {
|
|
if link.Redirect != nil {
|
|
if redirectCount == 0 {
|
|
fmt.Fprintf(f, "Redirects found:\n")
|
|
}
|
|
fmt.Fprintf(f, "- %s (Redirect %d -> %s)\n", link.URL, link.StatusCode, link.Redirect.ToURL)
|
|
redirectCount++
|
|
}
|
|
}
|
|
if redirectCount > 0 {
|
|
fmt.Fprintf(f, "\n")
|
|
}
|
|
|
|
// Then list broken links
|
|
var brokenCount int
|
|
for _, link := range brokenLinks {
|
|
if link.Redirect == nil {
|
|
if brokenCount == 0 {
|
|
fmt.Fprintf(f, "Broken links found:\n")
|
|
}
|
|
if link.Error != "" {
|
|
fmt.Fprintf(f, "- %s (Error: %s)\n", link.URL, link.Error)
|
|
} else {
|
|
fmt.Fprintf(f, "- %s (Status: %d)\n", link.URL, link.StatusCode)
|
|
}
|
|
brokenCount++
|
|
}
|
|
}
|
|
|
|
fmt.Fprintf(f, "\nTotal issues: %d (%d redirects, %d broken)\n",
|
|
len(brokenLinks), redirectCount, brokenCount)
|
|
|
|
fmt.Printf("Results written to %s\n", filename)
|
|
}
|