RenewCMS/api/controllers/pages/register.go

package pages

import (
	"RenewCMS/api"
	"RenewCMS/api/controllers/auth"
	"net/http"
	"os"

	"github.com/google/uuid"
)

type RegisterPageError struct {
	Email    bool `json:"email"`
	Password bool `json:"password"`
	Username bool `json:"username"`
}

type RegisterPage struct {
	PageError *RegisterPageError `json:"error"`
	Username  string             `json:"username"`
	Email     string             `json:"email"`
}

var EmptyRegisterPage = &RegisterPage{
	PageError: &RegisterPageError{
		Email:    false,
		Password: false,
		Username: false,
	},
	Username: "",
	Email:    "",
}

func GetRegisterPageHandler(registerPage *RegisterPage) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if !auth.IsLoggedIn(r) && auth.SomeUsersVerified() {
			http.Redirect(w, r, "/login?failure=There is already a verified account, please login.", http.StatusSeeOther)
		}
		if (auth.IsLoggedIn(r) && auth.IsVerified(r)) || auth.SomeUsersVerified() {
			http.Redirect(w, r, "/home", http.StatusSeeOther)
			return
		}
		if r.Method == http.MethodPost {
			PostRegisterPage(w, r)
			return
		}
		bs, err := api.Container.GetPageUseCase.GetPage("register", map[string]any{
			"PageError": registerPage.PageError,
			"Username":  registerPage.Username,
			"Email":     registerPage.Email,
			"Head":      headTmpl,
		})
		if err != nil {
			_, _ = w.Write([]byte(err.Error()))
		}
		_, _ = w.Write(bs)
	}
}

func PostRegisterPage(w http.ResponseWriter, r *http.Request) {
	_ = r.ParseForm()

	credentials := auth.RegisterCredentials{
		Username: r.FormValue("username"),
		Password: r.FormValue("password"),
		Email:    r.FormValue("email"),
	}
	err := api.Validate.Struct(credentials)
	if err != nil {
		r.Method = http.MethodGet
		GetRegisterPageHandler(&RegisterPage{
			PageError: &RegisterPageError{
				Email:    true,
				Password: true,
				Username: true,
			},
			Username: r.FormValue("username"),
			Email:    r.FormValue("email"),
		})(w, r)
		return
	}

	verificationCode := uuid.NewString()
	createdUser, err := auth.GetNewUser(credentials, verificationCode)
	if err != nil {
		r.Method = http.MethodGet
		GetRegisterPageHandler(&RegisterPage{
			PageError: &RegisterPageError{
				Email:    true,
				Password: false,
				Username: false,
			},
			Username: r.FormValue("username"),
			Email:    r.FormValue("email"),
		})(w, r)
		return
	}

	_ = api.Container.SendMailUseCase.SendMail(createdUser.Email, "mailValidation", map[string]string{
		"Host":             os.Getenv("HOST"),
		"VerificationCode": verificationCode,
	})

	_ = auth.SetJwtCookie(&w, createdUser.ID)

	http.Redirect(w, r, "/register/pending", http.StatusSeeOther)
}