66 lines
1.2 KiB
Go
66 lines
1.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"portfolio/api/service/bcrypt"
|
|
"portfolio/api/service/jwt"
|
|
"portfolio/database/ent"
|
|
"portfolio/database/query"
|
|
"time"
|
|
)
|
|
|
|
func Login(w http.ResponseWriter, r *http.Request) {
|
|
var u *ent.User
|
|
|
|
isHtmx := r.Header.Get("HX-Request")
|
|
|
|
if isHtmx == "true" {
|
|
u = &ent.User{
|
|
Email: r.PostFormValue("email"),
|
|
Password: r.PostFormValue("password"),
|
|
}
|
|
} else {
|
|
err := json.NewDecoder(r.Body).Decode(&u)
|
|
if err != nil {
|
|
InternalServerErrorHandler(w, err)
|
|
}
|
|
}
|
|
|
|
User, err := query.GetLogin(context.Background(), u)
|
|
if err != nil {
|
|
UnprocessableEntityHandler(w, err)
|
|
return
|
|
}
|
|
|
|
if bcrypt.CheckPasswordHash(u.Password, User.Password) {
|
|
|
|
jwtToken := jwt.CreateUserJWT(User.Name, User.ID, string(User.Role))
|
|
|
|
if jwtToken != "" {
|
|
|
|
cookie := &http.Cookie{Name: "jwt",
|
|
Value: jwtToken,
|
|
HttpOnly: true,
|
|
Secure: true,
|
|
SameSite: http.SameSiteStrictMode,
|
|
Expires: time.Now().Add(24 * time.Hour),
|
|
}
|
|
|
|
http.SetCookie(w, cookie)
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
_, err = w.Write([]byte("login success"))
|
|
return
|
|
} else {
|
|
InternalServerErrorHandler(w, err)
|
|
return
|
|
}
|
|
|
|
} else {
|
|
UnauthorizedHandler(w)
|
|
return
|
|
}
|
|
}
|