73 lines
1.9 KiB
Go
73 lines
1.9 KiB
Go
|
package ginext
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"gogs.mikescher.com/BlackForestBytes/goext/exerr"
|
||
|
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type seekableResponse struct {
|
||
|
|
data io.ReadSeeker
|
||
|
|
contentType string
|
||
|
|
filename string
|
||
|
|
headers []headerval
|
||
|
|
cookies []cookieval
|
||
|
|
}
|
||
|
|
|
||
|
|
func (j seekableResponse) Write(g *gin.Context) {
|
||
|
|
g.Header("Content-Type", j.contentType) // if we don't set it here http.ServeContent does weird sniffing later...
|
||
|
|
|
||
|
|
for _, v := range j.headers {
|
||
|
|
g.Header(v.Key, v.Val)
|
||
|
|
}
|
||
|
|
for _, v := range j.cookies {
|
||
|
|
g.SetCookie(v.name, v.value, v.maxAge, v.path, v.domain, v.secure, v.httpOnly)
|
||
|
|
}
|
||
|
|
|
||
|
|
http.ServeContent(g.Writer, g.Request, j.filename, time.Unix(0, 0), j.data)
|
||
|
|
|
||
|
|
if clsr, ok := j.data.(io.ReadSeekCloser); ok {
|
||
|
|
err := clsr.Close()
|
||
|
|
if err != nil {
|
||
|
|
exerr.Wrap(err, "failed to close io.ReadSeerkClose in ginext.Seekable").Str("filename", j.filename).Print()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (j seekableResponse) WithHeader(k string, v string) HTTPResponse {
|
||
|
|
j.headers = append(j.headers, headerval{k, v})
|
||
|
|
return j
|
||
|
|
}
|
||
|
|
|
||
|
|
func (j seekableResponse) WithCookie(name string, value string, maxAge int, path string, domain string, secure bool, httpOnly bool) HTTPResponse {
|
||
|
|
j.cookies = append(j.cookies, cookieval{name, value, maxAge, path, domain, secure, httpOnly})
|
||
|
|
return j
|
||
|
|
}
|
||
|
|
|
||
|
|
func (j seekableResponse) IsSuccess() bool {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func (j seekableResponse) Statuscode() int {
|
||
|
|
return 200
|
||
|
|
}
|
||
|
|
|
||
|
|
func (j seekableResponse) BodyString(*gin.Context) *string {
|
||
|
|
return langext.Ptr("(seekable)")
|
||
|
|
}
|
||
|
|
|
||
|
|
func (j seekableResponse) ContentType() string {
|
||
|
|
return j.contentType
|
||
|
|
}
|
||
|
|
|
||
|
|
func (j seekableResponse) Headers() []string {
|
||
|
|
return langext.ArrMap(j.headers, func(v headerval) string { return v.Key + "=" + v.Val })
|
||
|
|
}
|
||
|
|
|
||
|
|
func Seekable(filename string, contentType string, data io.ReadSeeker) HTTPResponse {
|
||
|
|
return &seekableResponse{filename: filename, contentType: contentType, data: data}
|
||
|
|
}
|