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) }