mirror of
https://github.com/Xevion/banner.git
synced 2025-12-06 15:14:25 -06:00
143 lines
2.8 KiB
Go
143 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"math/rand"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
log "github.com/rs/zerolog/log"
|
|
)
|
|
|
|
func BuildRequestWithBody(method string, path string, params map[string]string, body io.Reader) *http.Request {
|
|
// Builds a URL for the given path and parameters
|
|
url := baseURL + path
|
|
|
|
if params != nil {
|
|
takenFirst := false
|
|
for key, value := range params {
|
|
paramChar := "&"
|
|
if !takenFirst {
|
|
paramChar = "?"
|
|
takenFirst = true
|
|
}
|
|
|
|
url += paramChar + key + "=" + value
|
|
}
|
|
}
|
|
|
|
request, _ := http.NewRequest(method, url, body)
|
|
AddUserAgent(request)
|
|
return request
|
|
}
|
|
|
|
func BuildRequest(method string, path string, params map[string]string) *http.Request {
|
|
return BuildRequestWithBody(method, path, params, nil)
|
|
}
|
|
|
|
func AddUserAgent(req *http.Request) {
|
|
req.Header.Add("User-Agent", "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Safari/537.36")
|
|
}
|
|
|
|
func ContainsContentType(header string, search string) bool {
|
|
// Split on commas, check if any of the types match
|
|
for _, content_type := range strings.Split(header, ";") {
|
|
if content_type == search {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
const letterBytes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
func RandomString(n int) string {
|
|
b := make([]byte, n)
|
|
for i := range b {
|
|
b[i] = letterBytes[rand.Intn(len(letterBytes))]
|
|
}
|
|
return string(b)
|
|
}
|
|
|
|
// Nonce returns a string made up of the current time in milliseconds, Unix epoch/UTC
|
|
func Nonce() string {
|
|
return strconv.Itoa(int(time.Now().UnixMilli()))
|
|
}
|
|
|
|
func doRequest(req *http.Request) (*http.Response, error) {
|
|
log.Debug().Str("method", req.Method).Msg("Request")
|
|
res, err := client.Do(req)
|
|
if err != nil {
|
|
log.Err(err).Str("method", req.Method).Msg("Request Failed")
|
|
} else {
|
|
log.Debug().Str("status", res.Status).Int64("content-length", res.ContentLength).Strs("content-type", res.Header["Content-Type"]).Msg("Response")
|
|
}
|
|
return res, err
|
|
}
|
|
|
|
func Plural(n int) string {
|
|
if n == 1 {
|
|
return ""
|
|
}
|
|
return "s"
|
|
}
|
|
|
|
func WeekdaysToString(days map[time.Weekday]bool) string {
|
|
str := ""
|
|
|
|
if days[time.Monday] {
|
|
str += "M"
|
|
}
|
|
|
|
if days[time.Tuesday] {
|
|
str += "Tu"
|
|
}
|
|
|
|
if days[time.Wednesday] {
|
|
str += "W"
|
|
}
|
|
|
|
if days[time.Thursday] {
|
|
str += "Th"
|
|
}
|
|
|
|
if days[time.Friday] {
|
|
str += "F"
|
|
}
|
|
|
|
if days[time.Saturday] {
|
|
str += "Sa"
|
|
}
|
|
|
|
if days[time.Sunday] {
|
|
str += "Su"
|
|
}
|
|
|
|
return str
|
|
}
|
|
|
|
type NaiveTime struct {
|
|
Hours uint
|
|
Minutes uint
|
|
}
|
|
|
|
func ParseNaiveTime(integer uint64) NaiveTime {
|
|
minutes := uint(integer % 100)
|
|
hours := uint(integer / 100)
|
|
|
|
return NaiveTime{Hours: hours, Minutes: minutes}
|
|
}
|
|
|
|
func (nt NaiveTime) String() string {
|
|
meridiem := "AM"
|
|
hour := nt.Hours
|
|
if nt.Hours >= 12 {
|
|
meridiem = "PM"
|
|
hour -= 12
|
|
}
|
|
return fmt.Sprintf("%d:%02d%s", hour, nt.Minutes, meridiem)
|
|
}
|