forked from mirrors/gotosocial
a156188b3e
* update dependencies, bump Go version to 1.19 * bump test image Go version * update golangci-lint * update gotosocial-drone-build * sign * linting, go fmt * update swagger docs * update swagger docs * whitespace * update contributing.md * fuckin whoopsie doopsie * linterino, linteroni * fix followrequest test not starting processor * fix other api/client tests not starting processor * fix remaining tests where processor not started * bump go-runners version * don't check last-webfingered-at, processor may have updated this * update swagger command * update bun to latest version * fix embed to work the same as before with new bun Signed-off-by: kim <grufwub@gmail.com> Co-authored-by: tsmethurst <tobi.smethurst@protonmail.com>
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
// Copyright 2018 Gin Core Team. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package render
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"strconv"
|
|
)
|
|
|
|
// Reader contains the IO reader and its length, and custom ContentType and other headers.
|
|
type Reader struct {
|
|
ContentType string
|
|
ContentLength int64
|
|
Reader io.Reader
|
|
Headers map[string]string
|
|
}
|
|
|
|
// Render (Reader) writes data with custom ContentType and headers.
|
|
func (r Reader) Render(w http.ResponseWriter) (err error) {
|
|
r.WriteContentType(w)
|
|
if r.ContentLength >= 0 {
|
|
if r.Headers == nil {
|
|
r.Headers = map[string]string{}
|
|
}
|
|
r.Headers["Content-Length"] = strconv.FormatInt(r.ContentLength, 10)
|
|
}
|
|
r.writeHeaders(w, r.Headers)
|
|
_, err = io.Copy(w, r.Reader)
|
|
return
|
|
}
|
|
|
|
// WriteContentType (Reader) writes custom ContentType.
|
|
func (r Reader) WriteContentType(w http.ResponseWriter) {
|
|
writeContentType(w, []string{r.ContentType})
|
|
}
|
|
|
|
// writeHeaders writes custom Header.
|
|
func (r Reader) writeHeaders(w http.ResponseWriter, headers map[string]string) {
|
|
header := w.Header()
|
|
for k, v := range headers {
|
|
if header.Get(k) == "" {
|
|
header.Set(k, v)
|
|
}
|
|
}
|
|
}
|