main.go

package main

import (
	"bufio"
	"fmt"
	"net/http"
	"os"

	"regexp"
	"strings"
)

// Declare the constants that contain the strings to the colors
const (
	colorRed    = "\033[31m"
	colorGreen  = "\033[32m"
	colorYellow = "\033[33m"
	colorReset  = "\033[0m"
)

func main() {
	filePath := getInput("Inform the filepath to the docs you want to check: ")
	file, err := openFile(filePath)
	if err != nil {
		fmt.Println("Error opening file:", err)
		return
	}
	defer file.Close()

	content, err := readFileContent(file)
	if err != nil {
		fmt.Println("Error reading file:", err)
		return
	}

	text := string(content)

	fmt.Println("\nScanning for typos powered by gpt-4")

	urlPatternHttps := `https?://[^\s()]+`
	generalUrlsPattnern := `\[(.*?)\]\((.*?)\)`
	buttonMatches := `href="([^"]+)"`

	matchesHttp := findMatches(text, urlPatternHttps)
	matches := findMatches(text, generalUrlsPattnern)
	matchesButton := findMatches(text, buttonMatches)

	for i := range matches {

		x := strings.Split(matches[i], "](")
		y := strings.Replace(x[1], ")", "", -1)

		if (strings.Contains(y, "en/") || strings.Contains(y, "pt/") || strings.Contains(y, "pt-br/")) && !strings.Contains(y, "www.azion.com") {
			y = formatURL(y)

		}
		matches[i] = y

	}

	for i := range matchesButton {
		x := strings.Replace(matchesButton[i], `href="`, "", -1)
		x = strings.ReplaceAll(x, "\"", "")

		if (strings.Contains(x, "en/") || strings.Contains(x, "pt/") || strings.Contains(x, "pt-br/")) && !strings.Contains(x, "www.azion.com") {
			x = formatURL(x)

		}
		matchesButton[i] = x

	}

	allURLs := append(matches, append(matchesHttp, matchesButton...)...)

	fmt.Println("\nTesting links")
	for _, link := range allURLs {
		statusCode, err := checkURL(link)
		if err != nil {
			fmt.Println(colorRed, "Link:", link, "- Error:", err.Error(), colorReset)
			continue
		}

		color := colorGreen
		if statusCode > 400 {
			color = colorRed
		} else if statusCode >= 300 && statusCode < 400 {
			color = colorYellow
		}

		fmt.Println(color, "Link:", link, "- Status Code:", statusCode, colorReset)
	}
}

// Get the input informed by the user
func getInput(prompt string) string {
	fmt.Print(prompt)
	scanner := bufio.NewScanner(os.Stdin)
	scanner.Scan()
	return scanner.Text()
}

// Open the file
func openFile(filePath string) (*os.File, error) {
	file, err := os.Open(filePath)
	if err != nil {
		return nil, err
	}
	return file, nil
}

// Read the content inside the file (.mdx)
func readFileContent(file *os.File) ([]byte, error) {
	stat, err := file.Stat()
	if err != nil {
		return nil, err
	}
	content := make([]byte, stat.Size())
	_, err = file.Read(content)
	if err != nil {
		return nil, err
	}
	return content, nil
}

// Find mathches by the pattern (regex) informed
func findMatches(text, pattern string) []string {
	re := regexp.MustCompile(pattern)
	return re.FindAllString(text, -1)
}

// Format the URL adding the host
func formatURL(url string) string {
	if !strings.HasSuffix(url, "/") {
		url += "/"
	}
	return "https://www.azion.com" + url
}

// Check if the link is working
func checkURL(link string) (int, error) {
	resp, err := http.Get(link)
	if err != nil {
		return 0, err
	}
	defer resp.Body.Close()
	return resp.StatusCode, nil
}