7 "github.com/gin-gonic/gin"
13 JwtSecret string `toml:"jwt_secret"`
14 PasswordResetSecret string `toml:"password_reset_secret"`
15 FreeSMSUser string `toml:"free_sms_user"`
16 FreeSMSPass string `toml:"free_sms_pass"`
19 func SetConfig(config Config) {
22 JWT_SECRET = []byte(config.JwtSecret)
23 PASSWORD_RESET_SECRET = []byte(config.PasswordResetSecret)
32 func (e Error) Err() error {
40 func (e Error) Error() string {
48 func NewInternalError(err error) *Error {
49 return &Error{InternalError, "internal error", err}
52 func ToSnake(in string) string {
57 for i := 0; i < length; i++ {
58 if i > 0 && unicode.IsUpper(runes[i]) && ((i+1 < length && unicode.IsLower(runes[i+1])) || unicode.IsLower(runes[i-1])) {
59 out = append(out, '_')
61 out = append(out, unicode.ToLower(runes[i]))
67 type Response struct {
68 StatusCode Status `json:"-"`
69 ErrorCode ErrorCode `json:"-"`
71 Status string `json:"status"`
72 Code string `json:"code,omitempty"`
73 Response interface{} `json:"response,omitempty"`
74 Message string `json:"message,omitempty"`
77 func (r Response) populateStatus() Response {
78 r.Status = ToSnake(r.StatusCode.String())
81 r.Code = ToSnake(r.ErrorCode.String())
87 func ErrorResponse(code ErrorCode, message string) Response {
95 func SuccessResponse(data interface{}) Response {
102 func WriteJsonResponse(response Response, c *gin.Context) {
103 response = response.populateStatus()
105 c.JSON(StatusToHttpCode(response.StatusCode, response.ErrorCode), response)
108 func WriteBinary(contentType string, b []byte, c *gin.Context) {
109 c.Data(http.StatusOK, contentType, b)
112 type Middleware func(*gin.Context) *Error
114 func M(handler Middleware) gin.HandlerFunc {
115 return func(c *gin.Context) {
119 WriteJsonResponse(ErrorResponse(err.Code, err.UserMessage), c)
128 type Query interface {
129 ValidateParams() *Error
130 Run() (interface{}, *Error)
133 func RunQuery(query Query, c *gin.Context) {
134 if err := query.ValidateParams(); err != nil {
135 WriteJsonResponse(ErrorResponse(err.Code, err.UserMessage), c)
140 if out, err := query.Run(); err != nil {
141 WriteJsonResponse(ErrorResponse(err.Code, err.UserMessage), c)
144 WriteJsonResponse(SuccessResponse(out), c)