mirror of
https://github.com/ordinary-dev/phoenix
synced 2024-09-20 03:40:29 +05:00
Ivan R.
5aa2cee5b1
Since I started simplifying, I decided to abandon ORM. I won’t say that this makes much sense, everything works more or less as before. Except that the size of the program has decreased slightly again, by about a megabyte.
63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
package pages
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
|
|
"github.com/ordinary-dev/phoenix/database"
|
|
"github.com/ordinary-dev/phoenix/jwttoken"
|
|
)
|
|
|
|
func ShowRegistrationForm(w http.ResponseWriter, _ *http.Request) {
|
|
userCount, err := database.CountAdmins()
|
|
if err != nil {
|
|
ShowError(w, http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
if userCount > 0 {
|
|
ShowError(w, http.StatusBadRequest, errors.New("at least 1 user already exists"))
|
|
return
|
|
}
|
|
|
|
Render("auth.html.tmpl", w, map[string]any{
|
|
"title": "Create an account",
|
|
"description": "To prevent other people from seeing your links, create an account.",
|
|
"button": "Create",
|
|
"formAction": "/registration",
|
|
})
|
|
}
|
|
|
|
func CreateUser(w http.ResponseWriter, r *http.Request) {
|
|
userCount, err := database.CountAdmins()
|
|
if err != nil {
|
|
ShowError(w, http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
if userCount > 0 {
|
|
ShowError(w, http.StatusBadRequest, errors.New("at least 1 user already exists"))
|
|
return
|
|
}
|
|
|
|
// Try to create a user.
|
|
username := r.FormValue("username")
|
|
password := r.FormValue("password")
|
|
_, err = database.CreateAdmin(username, password)
|
|
if err != nil {
|
|
ShowError(w, http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
// Generate access token.
|
|
token, err := jwttoken.GetJWTToken()
|
|
if err != nil {
|
|
ShowError(w, http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
http.SetCookie(w, jwttoken.TokenToCookie(token))
|
|
|
|
// Redirect to homepage.
|
|
http.Redirect(w, r, "/", http.StatusFound)
|
|
}
|