123 lines
3.3 KiB
Go
123 lines
3.3 KiB
Go
package handlers
|
|
|
|
import (
|
|
"log"
|
|
|
|
"git.qowevisa.me/Qowevisa/fin-check-api/consts"
|
|
"git.qowevisa.me/Qowevisa/fin-check-api/db"
|
|
"git.qowevisa.me/Qowevisa/fin-check-api/tokens"
|
|
"git.qowevisa.me/Qowevisa/fin-check-api/types"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// @Summary Register an user
|
|
// @Description Creates user in database as db.User
|
|
// @Tags user
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param user body types.User true "User info"
|
|
// @Success 200 {object} types.Account
|
|
// @Failure 400 {object} types.ErrorResponse
|
|
// @Failure 500 {object} types.ErrorResponse
|
|
// @Router /user/register [post]
|
|
func UserRegister(c *gin.Context) {
|
|
var user types.User
|
|
if err := c.ShouldBindJSON(&user); err != nil {
|
|
c.JSON(400, types.ErrorResponse{Message: "Invalid request"})
|
|
return
|
|
}
|
|
|
|
dbUser := &db.User{
|
|
Username: user.Username,
|
|
Password: user.Password,
|
|
}
|
|
dbc := db.Connect()
|
|
if err := dbc.Create(dbUser).Error; err != nil {
|
|
c.JSON(500, types.ErrorResponse{Message: err.Error()})
|
|
return
|
|
}
|
|
var token1 *tokens.Token
|
|
if token, err := tokens.AddToken(dbUser.ID); err != nil {
|
|
c.JSON(500, types.ErrorResponse{Message: err.Error()})
|
|
return
|
|
} else {
|
|
if token == nil {
|
|
log.Printf("DAFUQ: 002\n")
|
|
c.JSON(500, types.ErrorResponse{Message: "DAFUQ002"})
|
|
return
|
|
}
|
|
token1 = token
|
|
}
|
|
err := tokens.CreateSessionFromToken(token1.Val, dbUser.ID)
|
|
if err != nil {
|
|
log.Printf("tokens.CreateSessionFromToken: %v\n", err)
|
|
c.JSON(500, types.ErrorResponse{Message: "ERROR: 1000"})
|
|
}
|
|
c.SetCookie(consts.COOKIE_SESSION, token1.Val, 3600, "/", "localhost", false, true)
|
|
acc := types.Account{
|
|
ID: dbUser.ID,
|
|
Token: token1.Val,
|
|
Username: dbUser.Username,
|
|
}
|
|
c.JSON(200, acc)
|
|
}
|
|
|
|
// @Summary Login for user
|
|
// @Description Checks user in database as db.User and gives token
|
|
// @Tags user
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param user body types.User true "User info"
|
|
// @Success 200 {object} types.Account
|
|
// @Failure 400 {object} types.ErrorResponse
|
|
// @Failure 500 {object} types.ErrorResponse
|
|
// @Router /user/login [post]
|
|
func UserLogin(c *gin.Context) {
|
|
var user types.User
|
|
if err := c.ShouldBindJSON(&user); err != nil {
|
|
c.JSON(400, types.ErrorResponse{Message: "Invalid request"})
|
|
return
|
|
}
|
|
|
|
dbUser := db.User{
|
|
Username: user.Username,
|
|
Password: user.Password,
|
|
}
|
|
foundUser := db.User{}
|
|
dbc := db.Connect()
|
|
if err := dbc.Find(&foundUser, dbUser).Error; err != nil {
|
|
c.JSON(500, types.ErrorResponse{Message: err.Error()})
|
|
return
|
|
}
|
|
if foundUser.ID == 0 {
|
|
c.JSON(400, types.ErrorResponse{Message: "Credentials are incorrect"})
|
|
return
|
|
}
|
|
var token1 *tokens.Token
|
|
if token, err := tokens.AddToken(foundUser.ID); err != nil {
|
|
c.JSON(500, types.ErrorResponse{Message: err.Error()})
|
|
return
|
|
} else {
|
|
if token == nil {
|
|
log.Printf("DAFUQ: 002\n")
|
|
c.JSON(500, types.ErrorResponse{Message: "DAFUQ002"})
|
|
return
|
|
}
|
|
token1 = token
|
|
}
|
|
err := tokens.CreateSessionFromToken(token1.Val, foundUser.ID)
|
|
if err != nil {
|
|
log.Printf("tokens.CreateSessionFromToken: %v\n", err)
|
|
c.JSON(500, types.ErrorResponse{Message: "ERROR: 1000"})
|
|
}
|
|
c.SetCookie(consts.COOKIE_SESSION, token1.Val, 3600, "/", "localhost", false, true)
|
|
acc := types.Account{
|
|
ID: foundUser.ID,
|
|
Token: token1.Val,
|
|
Username: dbUser.Username,
|
|
}
|
|
c.JSON(200, acc)
|
|
}
|
|
|
|
func isSessionTokenForUserInvalid(userID uint) {}
|