124 lines
2.6 KiB
Go
124 lines
2.6 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
defaultBinder = &DefaultBinder{}
|
|
)
|
|
|
|
type Context struct {
|
|
ctx context.Context
|
|
req *http.Request
|
|
res http.ResponseWriter
|
|
params map[string]string
|
|
statusCode int
|
|
}
|
|
|
|
func (ctx *Context) reset(req *http.Request, res http.ResponseWriter, ps map[string]string) {
|
|
ctx.statusCode = http.StatusOK
|
|
ctx.req, ctx.res, ctx.params = req, res, ps
|
|
}
|
|
|
|
func (ctx *Context) RealIp() string {
|
|
if ip := ctx.Request().Header.Get("X-Forwarded-For"); ip != "" {
|
|
i := strings.IndexAny(ip, ",")
|
|
if i > 0 {
|
|
return strings.TrimSpace(ip[:i])
|
|
}
|
|
return ip
|
|
}
|
|
if ip := ctx.Request().Header.Get("X-Real-IP"); ip != "" {
|
|
return ip
|
|
}
|
|
ra, _, _ := net.SplitHostPort(ctx.Request().RemoteAddr)
|
|
return ra
|
|
}
|
|
|
|
func (ctx *Context) Request() *http.Request {
|
|
return ctx.req
|
|
}
|
|
|
|
func (ctx *Context) Response() http.ResponseWriter {
|
|
return ctx.res
|
|
}
|
|
|
|
func (ctx *Context) Context() context.Context {
|
|
if ctx.Request().Context() != nil {
|
|
return ctx.Request().Context()
|
|
}
|
|
return ctx.ctx
|
|
}
|
|
|
|
func (ctx *Context) Bind(v any) (err error) {
|
|
return defaultBinder.Bind(v, ctx.Request())
|
|
}
|
|
|
|
func (ctx *Context) Query(k string) string {
|
|
return ctx.Request().FormValue(k)
|
|
}
|
|
|
|
func (ctx *Context) Param(k string) string {
|
|
var (
|
|
ok bool
|
|
v string
|
|
)
|
|
if v, ok = ctx.params[k]; ok {
|
|
return v
|
|
}
|
|
return ctx.Request().FormValue(k)
|
|
}
|
|
|
|
func (ctx *Context) send(res responsePayload) (err error) {
|
|
ctx.Response().Header().Set("Content-Type", "application/json")
|
|
encoder := json.NewEncoder(ctx.Response())
|
|
if strings.HasPrefix(ctx.Request().Header.Get("User-Agent"), "curl") {
|
|
encoder.SetIndent("", "\t")
|
|
}
|
|
return encoder.Encode(res)
|
|
}
|
|
|
|
func (ctx *Context) Success(v any) (err error) {
|
|
return ctx.send(responsePayload{Data: v})
|
|
}
|
|
|
|
func (ctx *Context) Status(code int) {
|
|
ctx.statusCode = code
|
|
}
|
|
|
|
func (ctx *Context) Error(code int, reason string) (err error) {
|
|
return ctx.send(responsePayload{Code: code, Reason: reason})
|
|
}
|
|
|
|
func (ctx *Context) Redirect(url string, code int) {
|
|
if code != http.StatusFound && code != http.StatusMovedPermanently {
|
|
code = http.StatusMovedPermanently
|
|
}
|
|
http.Redirect(ctx.Response(), ctx.Request(), url, code)
|
|
}
|
|
|
|
func (ctx *Context) SetCookie(cookie *http.Cookie) {
|
|
http.SetCookie(ctx.Response(), cookie)
|
|
}
|
|
|
|
func (ctx *Context) SendFile(filename string) (err error) {
|
|
var (
|
|
fi os.FileInfo
|
|
fp *os.File
|
|
)
|
|
if fi, err = os.Stat(filename); err == nil {
|
|
if fp, err = os.Open(filename); err == nil {
|
|
http.ServeContent(ctx.Response(), ctx.Request(), path.Base(filename), fi.ModTime(), fp)
|
|
err = fp.Close()
|
|
}
|
|
}
|
|
return
|
|
}
|