| package common |
|
|
| import ( |
| "context" |
| "fmt" |
| "html" |
| "net/http" |
| "strings" |
|
|
| "github.com/OpenListTeam/OpenList/v4/cmd/flags" |
| "github.com/OpenListTeam/OpenList/v4/internal/conf" |
| "github.com/gin-gonic/gin" |
| log "github.com/sirupsen/logrus" |
| ) |
|
|
| func hidePrivacy(msg string) string { |
| for _, r := range conf.PrivacyReg { |
| msg = r.ReplaceAllStringFunc(msg, func(s string) string { |
| return strings.Repeat("*", len(s)) |
| }) |
| } |
| return msg |
| } |
|
|
| |
| |
| func ErrorResp(c *gin.Context, err error, code int, l ...bool) { |
| ErrorWithDataResp(c, err, code, nil, l...) |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| } |
|
|
| |
| |
| |
| func ErrorPage(c *gin.Context, err error, code int, l ...bool) { |
|
|
| if len(l) > 0 && l[0] { |
| if flags.Debug || flags.Dev { |
| log.Errorf("%+v", err) |
| } else { |
| log.Errorf("%v", err) |
| } |
| } |
|
|
| codes := fmt.Sprintf("%d %s", code, http.StatusText(code)) |
|
|
| html := fmt.Sprintf(`<!DOCTYPE html> |
| <html lang="en"> |
| <head> |
| <meta charset="utf-8" /> |
| <meta name="viewport" content="width=device-width, initial-scale=1" /> |
| <meta name="color-scheme" content="dark light" /> |
| <meta name="robots" content="noindex" /> |
| <title>%s</title> |
| </head> |
| <body> |
| <h1>%s</h1> |
| <hr> |
| <p>%s</p> |
| </body> |
| </html>`, |
| codes, codes, html.EscapeString(hidePrivacy(err.Error()))) |
| c.Data(code, "text/html; charset=utf-8", []byte(html)) |
| c.Abort() |
| } |
|
|
| func ErrorWithDataResp(c *gin.Context, err error, code int, data interface{}, l ...bool) { |
| if len(l) > 0 && l[0] { |
| if flags.Debug || flags.Dev { |
| log.Errorf("%+v", err) |
| } else { |
| log.Errorf("%v", err) |
| } |
| } |
| c.JSON(200, Resp[interface{}]{ |
| Code: code, |
| Message: hidePrivacy(err.Error()), |
| Data: data, |
| }) |
| c.Abort() |
| } |
|
|
| func ErrorStrResp(c *gin.Context, str string, code int, l ...bool) { |
| if len(l) != 0 && l[0] { |
| log.Error(str) |
| } |
| c.JSON(200, Resp[interface{}]{ |
| Code: code, |
| Message: hidePrivacy(str), |
| Data: nil, |
| }) |
| c.Abort() |
| } |
|
|
| func SuccessResp(c *gin.Context, data ...interface{}) { |
| SuccessWithMsgResp(c, "success", data...) |
| } |
|
|
| func SuccessWithMsgResp(c *gin.Context, msg string, data ...interface{}) { |
| var respData interface{} |
| if len(data) > 0 { |
| respData = data[0] |
| } |
|
|
| c.JSON(200, Resp[interface{}]{ |
| Code: 200, |
| Message: msg, |
| Data: respData, |
| }) |
| } |
|
|
| func Pluralize(count int, singular, plural string) string { |
| if count == 1 { |
| return singular |
| } |
| return plural |
| } |
|
|
| func GinWithValue(c *gin.Context, keyAndValue ...any) { |
| c.Request = c.Request.WithContext( |
| ContentWithValue(c.Request.Context(), keyAndValue...), |
| ) |
| } |
|
|
| func ContentWithValue(ctx context.Context, keyAndValue ...any) context.Context { |
| if len(keyAndValue) < 1 || len(keyAndValue)%2 != 0 { |
| panic("keyAndValue must be an even number of arguments (key, value, ...)") |
| } |
| for len(keyAndValue) > 0 { |
| ctx = context.WithValue(ctx, keyAndValue[0], keyAndValue[1]) |
| keyAndValue = keyAndValue[2:] |
| } |
| return ctx |
| } |
|
|
