NotAShelf/mdlinkt
1
0
Fork 0
mirror of https://github.com/NotAShelf/mdlinkt.git synced 2024-11-26 07:06:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

interepet invalid host errors as invalid links

Browse source
This commit is contained in:
raf 2023-12-24 21:58:26 +03:00
parent 768fc24f2b
commit 4ba90a1367
No known key found for this signature in database
GPG key ID: 02D1DD3FA08B6B29
1 changed files with 35 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
main.go
View file

@ -20,7 +20,7 @@ type LinkCheckResult struct {
}
func logWithColor(level string, msg string, args ...interface{}) {
timestamp := time.Now().Format("2006/01/02 15:04:05") // TODO: use time.RFC3339?
timestamp := time.Now().Format("2006/01/02 15:04:05")
colorFunc := color.New(color.FgWhite).SprintFunc()
switch level {
case "ERROR":
@ -49,9 +49,17 @@ func main() {
logWithColor("ERROR", "Failed to open file: %v", err)
os.Exit(1)
}
if file == nil {
logWithColor("ERROR", "File is nil")
os.Exit(1)
}
defer file.Close()
scanner := bufio.NewScanner(file)
if scanner == nil {
logWithColor("ERROR", "Scanner is nil")
os.Exit(1)
}
re := regexp.MustCompile(`\[(.*?)\]\((.*?)\)`)
results := []LinkCheckResult{}
@ -62,22 +70,28 @@ func main() {
for _, match := range matches {
link := strings.TrimSpace(match[2])
resp, err := http.Head(link)
if err != nil {
if err != nil || resp == nil {
logWithColor("ERROR", "Invalid link: %s", link)
continue
}
results = append(results, LinkCheckResult{
Link: link,
IsValid: false,
StatusCode: http.StatusBadRequest,
})
} else {
isValid := resp.StatusCode == http.StatusOK
result := LinkCheckResult{
Link: link,
IsValid: resp.StatusCode == http.StatusOK,
IsValid: isValid,
StatusCode: resp.StatusCode,
}
results = append(results, result)
if *verbose || (!*failedOnly && !result.IsValid) {
if *verbose || (!*failedOnly && !isValid) {
statusColor := color.GreenString
if !result.IsValid {
if !isValid {
statusColor = color.RedString
}
logWithColor("INFO", "%s: %s", link, statusColor("%d", result.StatusCode))
logWithColor("INFO", "%s: %s", link, statusColor("%d", resp.StatusCode))
}
}
}
}
@ -97,18 +111,18 @@ func main() {
invalidCount++
}
}
summaryColor := color.GreenString
if invalidCount > 0 {
summaryColor = color.RedString
}
logWithColor("INFO", summaryColor("Summary: %d valid links, %d invalid links"), validCount, invalidCount)
if *failedOnly {
for _, result := range results {
if !result.IsValid {
logWithColor("ERROR", "Failed link: %s", result.Link)
}
}
} else {
summaryColor := color.GreenString
if invalidCount > 0 {
summaryColor = color.RedString
}
logWithColor("INFO", summaryColor("Summary: %d valid links, %d invalid links"), validCount, invalidCount)
}
if invalidCount > 0 {