39 lines
896 B
Go
39 lines
896 B
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/gin-contrib/sessions"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/jackc/pgx/v5/pgconn"
|
|
)
|
|
|
|
func handleDBError(err error) (int, string) {
|
|
pgErr, ok := err.(*pgconn.PgError)
|
|
if !ok {
|
|
return 500, err.Error()
|
|
}
|
|
statusStr := pgErr.Message[:3]
|
|
message := pgErr.Message[4:]
|
|
status, err := strconv.ParseInt(statusStr, 10, 0)
|
|
if err == nil {
|
|
return int(status), message
|
|
} else {
|
|
return 400, pgErr.Message
|
|
}
|
|
}
|
|
|
|
func MiddlewareAuth(c *gin.Context) {
|
|
session := sessions.Default(c)
|
|
user_id, ok := session.Get("user_id").(string)
|
|
if !ok && strings.HasPrefix(c.Request.URL.Path, "/api") {
|
|
c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": "Ты это, залогинься сначала что ли, а то чё как крыса"})
|
|
return
|
|
}
|
|
c.Set("authorized", ok)
|
|
c.Set("user_id", user_id)
|
|
c.Next()
|
|
}
|