Weetile/fastfact
Weetile/fastfact
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

curl support, no html in that case

Browse Source
This commit is contained in:
Weetile 2024-11-24 22:16:17 +00:00
parent 621314b7bc
commit 88ec940b9b
1 changed files with 33 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
main.go
View File

@ -17,7 +17,6 @@ func loadFacts() ([]string, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
// Split the file into lines and return as a slice // Split the file into lines and return as a slice
lines := string(data) lines := string(data)
return splitLines(lines), nil return splitLines(lines), nil
@ -32,7 +31,19 @@ func randomFactHandler(facts []string) http.HandlerFunc {
// Get a random index // Get a random index
index := rand.Intn(len(facts)) index := rand.Intn(len(facts))
// Write the random fact as an HTML response with CSS // Check if the request is from curl by looking at the User-Agent
userAgent := r.Header.Get("User-Agent")
isCurl := strings.HasPrefix(strings.ToLower(userAgent), "curl")
if isCurl {
// For curl requests, just return the fact as plain text
w.Header().Set("Content-Type", "text/plain")
fmt.Fprintln(w, facts[index])
return
}
// For browser requests, return the HTML version
w.Header().Set("Content-Type", "text/html")
fmt.Fprintf(w, ` fmt.Fprintf(w, `
<!DOCTYPE html> <!DOCTYPE html>
<html lang="en"> <html lang="en">
@ -49,7 +60,7 @@ func randomFactHandler(facts []string) http.HandlerFunc {
align-items: center; align-items: center;
height: 100vh; height: 100vh;
margin: 0; margin: 0;
font-size: 24px; /* Increase text size */ font-size: 24px;
font-weight: bold; font-weight: bold;
font-family: Arial, sans-serif; font-family: Arial, sans-serif;
} }
@ -78,3 +89,4 @@ func main() {
panic(err) panic(err)
} }
} }