mirror of
https://github.com/NotAShelf/mdlinkt.git
synced 2024-11-22 13:20:46 +00:00
null check for http.Head response
This commit is contained in:
parent
e5e16be391
commit
50298ea412
1 changed files with 14 additions and 15 deletions
29
main.go
29
main.go
|
@ -62,23 +62,22 @@ func main() {
|
|||
for _, match := range matches {
|
||||
link := strings.TrimSpace(match[2])
|
||||
resp, err := http.Head(link)
|
||||
isInvalid := err != nil || resp.StatusCode == http.StatusNotFound || strings.Contains(err.Error(), "no such host")
|
||||
if isInvalid {
|
||||
if err != nil {
|
||||
logWithColor("ERROR", "Invalid link: %s", link)
|
||||
} else {
|
||||
result := LinkCheckResult{
|
||||
Link: link,
|
||||
IsValid: resp.StatusCode == http.StatusOK,
|
||||
StatusCode: resp.StatusCode,
|
||||
}
|
||||
results = append(results, result)
|
||||
if *verbose || (!*failedOnly && !result.IsValid) {
|
||||
statusColor := color.GreenString
|
||||
if !result.IsValid {
|
||||
statusColor = color.RedString
|
||||
}
|
||||
logWithColor("INFO", "%s: %s", link, statusColor("%d", result.StatusCode))
|
||||
continue
|
||||
}
|
||||
result := LinkCheckResult{
|
||||
Link: link,
|
||||
IsValid: resp.StatusCode == http.StatusOK,
|
||||
StatusCode: resp.StatusCode,
|
||||
}
|
||||
results = append(results, result)
|
||||
if *verbose || (!*failedOnly && !result.IsValid) {
|
||||
statusColor := color.GreenString
|
||||
if !result.IsValid {
|
||||
statusColor = color.RedString
|
||||
}
|
||||
logWithColor("INFO", "%s: %s", link, statusColor("%d", result.StatusCode))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue