// Package httpx contains the stable JSON response contract shared by HTTP // modules. It intentionally keeps the existing Gin adapter so callers can // migrate without changing their handler flow. package httpx import ( "net/http" "strings" "time" "github.com/gin-gonic/gin" ) const ( CodeSuccess = 0 CodeError = 7 ) type Response struct { Code int `json:"code"` Data any `json:"data"` Msg string `json:"msg"` } type PageResult struct { List any `json:"list"` Total int64 `json:"total"` Page int `json:"page"` PageSize int `json:"pageSize"` } func Write(c *gin.Context, code int, data any, message string) { if code == CodeError { message = sanitizeFailureMessage(message) } c.JSON(http.StatusOK, Response{Code: code, Data: data, Msg: message}) } func OK(c *gin.Context) { Write(c, CodeSuccess, gin.H{}, "操作成功") } func OKWithData(c *gin.Context, data any) { Write(c, CodeSuccess, data, "成功") } func Fail(c *gin.Context, message string) { Write(c, CodeError, gin.H{}, message) } func sanitizeFailureMessage(message string) string { message = strings.TrimSpace(message) lower := strings.ToLower(message) for _, marker := range []string{"gorm", "sql:", "redis", "mysql", "mongo", "dial ", "connection", "provider", "sdk", "json:", "serialize", "timeout", "http 4", "http 5", "tls:"} { if strings.Contains(lower, marker) { return "操作失败" } } if strings.Contains(message, "失败:") { return strings.TrimSpace(strings.SplitN(message, "失败:", 2)[0]) + "失败" } for _, marker := range []string{"参数", "不能为空", "无效", "不存在", "权限", "未登录", "验证码", "密码", "重复", "不支持", "超出", "分片", "上传", "文件", "名称", "令牌", "token"} { if strings.Contains(message, marker) { return message } } return message } func NoAuth(c *gin.Context, message string) { c.AbortWithStatusJSON(http.StatusUnauthorized, Response{Code: CodeError, Data: nil, Msg: message}) } func requestUsesHTTPS(request *http.Request) bool { if request != nil && request.TLS != nil { return true } if request == nil { return false } forwarded := strings.SplitN(request.Header.Get("X-Forwarded-Proto"), ",", 2)[0] return strings.EqualFold(strings.TrimSpace(forwarded), "https") } // SetCookie applies the transport-level cookie policy without knowing the // application's cookie names or authentication semantics. func SetCookie(c *gin.Context, name, value string, maxAge int) { if c == nil { return } cookie := &http.Cookie{Name: name, Value: value, Path: "/", MaxAge: maxAge, Secure: requestUsesHTTPS(c.Request), HttpOnly: true, SameSite: http.SameSiteStrictMode} if maxAge < 0 { cookie.Expires = time.Unix(1, 0) } http.SetCookie(c.Writer, cookie) }