Simplify robots.txt parsing logic

This commit is contained in:
2024-10-23 14:06:55 +03:00
parent 8d9ea6cdec
commit c49a69728a

View File

@@ -1,46 +1,31 @@
package gemini
import (
"bufio"
"fmt"
"strings"
)
// ParseRobotsTxt takes robots.txt content and a host, returns list of full URLs that shouldn't be visited
// Takes robots.txt content and a host, and
// returns a list of full URLs that shouldn't
// be visited.
// TODO Also take into account the user agent?
// Check gemini://geminiprotocol.net/docs/companion/robots.gmi
func ParseRobotsTxt(content string, host string) []string {
scanner := bufio.NewScanner(strings.NewReader(content))
var disallowedPaths []string
// Skip everything until we find "User-agent: *" line
for scanner.Scan() {
line := strings.TrimSpace(scanner.Text())
if strings.ToLower(line) == "user-agent: *" {
break
}
}
// Now collect all Disallow paths
for scanner.Scan() {
line := strings.TrimSpace(scanner.Text())
// Stop if we hit another User-agent section
if strings.HasPrefix(strings.ToLower(line), "user-agent:") {
break
}
// Parse Disallow lines
if strings.HasPrefix(strings.ToLower(line), "disallow:") {
for _, line := range strings.Split(content, "\n") {
line = strings.TrimSpace(line)
line = strings.ToLower(line)
if strings.HasPrefix(line, "disallow:") {
parts := strings.SplitN(line, ":", 2)
if len(parts) == 2 {
path := strings.TrimSpace(parts[1])
if path != "" {
// Construct full Gemini URL
disallowedPaths = append(disallowedPaths,
disallowedPaths = append(disallowedPaths,
fmt.Sprintf("gemini://%s%s", host, path))
}
}
}
}
return disallowedPaths
}