Captain ALM
012d464128
All checks were successful
continuous-integration/drone/push Build is passing
34 lines
773 B
Go
34 lines
773 B
Go
package web
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
)
|
|
|
|
func writeResponseHeaderCanWriteBody(method string, rw http.ResponseWriter, statusCode int, message string) bool {
|
|
hasBody := method != http.MethodHead && method != http.MethodOptions
|
|
if hasBody && message != "" {
|
|
rw.Header().Set("Content-Type", "text/plain; charset=utf-8")
|
|
rw.Header().Set("X-Content-Type-Options", "nosniff")
|
|
rw.Header().Set("Content-Length", strconv.Itoa(len(message)+2))
|
|
}
|
|
rw.WriteHeader(statusCode)
|
|
if hasBody {
|
|
if message != "" {
|
|
_, _ = rw.Write([]byte(message + "\r\n"))
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
type BufferedWriter struct {
|
|
Data []byte
|
|
}
|
|
|
|
func (c *BufferedWriter) Write(p []byte) (n int, err error) {
|
|
c.Data = append(c.Data, p...)
|
|
return len(p), nil
|
|
}
|