xevion/banner
1
0
Fork 0
mirror of https://github.com/Xevion/banner.git synced 2025-12-14 02:11:04 -06:00
Code Issues Packages Projects Releases Wiki Activity

Continue documenting helpers, improve contentLength log grab

Browse Source
This commit is contained in:
Xevion
2024-01-15 13:08:31 -06:00
parent 5ffdbb6139
commit 8425274b65
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
helpers.go
View File

@@ -15,6 +15,7 @@ import (
log "github.com/rs/zerolog/log" log "github.com/rs/zerolog/log"
) )
// BuildRequestWithBody builds a request with the given method, path, parameters, and body
func BuildRequestWithBody(method string, path string, params map[string]string, body io.Reader) *http.Request { func BuildRequestWithBody(method string, path string, params map[string]string, body io.Reader) *http.Request {
// Builds a URL for the given path and parameters // Builds a URL for the given path and parameters
url := baseURL + path url := baseURL + path
@@ -37,10 +38,12 @@ func BuildRequestWithBody(method string, path string, params map[string]string,
return request return request
} }
// BuildRequest builds a request with the given method, path, and parameters and an empty body
func BuildRequest(method string, path string, params map[string]string) *http.Request { func BuildRequest(method string, path string, params map[string]string) *http.Request {
return BuildRequestWithBody(method, path, params, nil) return BuildRequestWithBody(method, path, params, nil)
} }
// AddUserAgent adds a false but consistent user agent to the request
func AddUserAgent(req *http.Request) { 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") 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")
} }
@@ -100,6 +103,7 @@ func DiscordGoLogger(msgL, caller int, format string, a ...interface{}) {
} }
// Nonce returns a string made up of the current time in milliseconds, Unix epoch/UTC // Nonce returns a string made up of the current time in milliseconds, Unix epoch/UTC
// This is typically used as a query parameter to prevent request caching in the browser.
func Nonce() string { func Nonce() string {
return strconv.Itoa(int(time.Now().UnixMilli())) return strconv.Itoa(int(time.Now().UnixMilli()))
} }
@@ -111,7 +115,13 @@ func DoRequest(req *http.Request) (*http.Response, error) {
if err != nil { if err != nil {
log.Err(err).Str("method", req.Method).Msg("Request Failed") log.Err(err).Str("method", req.Method).Msg("Request Failed")
} else { } else {
log.Debug().Int("status", res.StatusCode).Int64("content-length", res.ContentLength).Strs("content-type", res.Header["Content-Type"]).Msg("Response") // Get the content length
contentLength := res.ContentLength
if contentLength == -1 {
contentLength, _ = strconv.ParseInt(res.Header.Get("Content-Length"), 10, 64)
}
log.Debug().Int("status", res.StatusCode).Int64("content-length", contentLength).Strs("content-type", res.Header["Content-Type"]).Msg("Response")
} }
return res, err return res, err
} }
@@ -202,3 +212,4 @@ func GetFirstEnv(key ...string) string {
} }
return "" return ""
} }