package main import ( "bytes" "errors" "fmt" "net/http" "strconv" "gitea.local.lab/Lbenedar/snippetbox/internal/models" "gitea.local.lab/Lbenedar/snippetbox/internal/validator" "github.com/julienschmidt/httprouter" ) type snippetCreateForm struct { Title string `form:"title"` Content string `form:"content"` Expires int `form:"expires"` validator.Validator `form:"-"` } type userSignupForm struct { Name string `form:"name"` Email string `form:"email"` Password string `form:"password"` validator.Validator `form:"-"` } type userLoginForm struct { Email string `form:"email"` Password string `form:"password"` validator.Validator `form:"-"` } func (app *application) render(w http.ResponseWriter, status int, page string, data *templateData) { ts, ok := app.templateCache[page] if !ok { err := fmt.Errorf("the template %s does not exist", page) app.serverError(w, err) return } buf := new(bytes.Buffer) err := ts.ExecuteTemplate(buf, "base", data) if err != nil { app.serverError(w, err) return } w.WriteHeader(status) buf.WriteTo(w) } func (app *application) home(w http.ResponseWriter, r *http.Request) { if r.URL.Path != "/" { app.notFound(w) return } snippets, err := app.snippets.Latest() if err != nil { app.serverError(w, err) return } data := app.newTemplateData(r) data.Snippets = snippets app.render(w, http.StatusOK, "home.tmpl", data) } func (app *application) snippetView(w http.ResponseWriter, r *http.Request) { params := httprouter.ParamsFromContext(r.Context()) id, err := strconv.Atoi(params.ByName("id")) if err != nil || id < 1 { app.notFound(w) return } snippet, err := app.snippets.Get(id) if err != nil { if errors.Is(err, models.ErrNoRecord) { app.notFound(w) } else { app.serverError(w, err) } return } data := app.newTemplateData(r) data.Snippet = snippet app.render(w, http.StatusOK, "view.tmpl", data) } func (app *application) snippetCreate(w http.ResponseWriter, r *http.Request) { data := app.newTemplateData(r) data.Form = snippetCreateForm{ Expires: 365, } app.render(w, http.StatusOK, "create.tmpl", data) } func (app *application) snippetCreatePost(w http.ResponseWriter, r *http.Request) { var form snippetCreateForm err := app.decodePostForm(r, &form) if err != nil { app.clientError(w, http.StatusBadRequest) return } form.CheckField(validator.NotBlank(form.Title), "title", "This field cannot be blank") form.CheckField(validator.MaxChars(form.Title, 100), "title", "This field cannot be more than 100 character long") form.CheckField(validator.NotBlank(form.Content), "content", "This field cannot be blank") form.CheckField(validator.PermittedValue(form.Expires, 1, 7, 365), "expires", "This field must equal 1, 7 and 365") if !form.Valid() { data := app.newTemplateData(r) data.Form = form app.render(w, http.StatusUnprocessableEntity, "create.tmpl", data) return } id, err := app.snippets.Insert(form.Title, form.Content, form.Expires) if err != nil { app.serverError(w, err) return } app.sessionManager.Put(r.Context(), "flash", "Snippet successfully created!") http.Redirect(w, r, fmt.Sprintf("/snippet/view/%d", id), http.StatusSeeOther) } func (app *application) userSignupPost(w http.ResponseWriter, r *http.Request) { var form userSignupForm err := app.decodePostForm(r, &form) if err != nil { app.clientError(w, http.StatusBadRequest) return } form.CheckField(validator.NotBlank(form.Name), "name", "This field cannot be blank") form.CheckField(validator.NotBlank(form.Email), "email", "This field cannot be blank") form.CheckField(validator.Matches(form.Email, validator.EmailRX), "email", "This field must be a valid email address") form.CheckField(validator.NotBlank(form.Password), "password", "This field cannot be blank") form.CheckField(validator.MinChars(form.Password, 8), "password", "This field must be at least 8 character long") if !form.Valid() { data := app.newTemplateData(r) data.Form = form app.render(w, http.StatusUnprocessableEntity, "signup.tmpl", data) return } err = app.users.Insert(form.Name, form.Email, form.Password) if err != nil { if errors.Is(err, models.ErrDuplicateEmail) { form.AddFieldError("email", "Email address already in use") data := app.newTemplateData(r) data.Form = form app.render(w, http.StatusUnprocessableEntity, "signup.tmpl", data) } else { app.serverError(w, err) } return } app.sessionManager.Put(r.Context(), "flash", "Your singup was successful. Please log in.") http.Redirect(w, r, "/user/login", http.StatusSeeOther) } func (app *application) userLoginPost(w http.ResponseWriter, r *http.Request) { var form userLoginForm err := app.decodePostForm(r, &form) if err != nil { app.clientError(w, http.StatusBadRequest) return } form.CheckField(validator.NotBlank(form.Email), "email", "This field cannot be blank") form.CheckField(validator.Matches(form.Email, validator.EmailRX), "email", "This field must be a valid email address") form.CheckField(validator.NotBlank(form.Password), "password", "This field cannot be blank") if !form.Valid() { data := app.newTemplateData(r) data.Form = form app.render(w, http.StatusUnprocessableEntity, "login.tmpl", data) return } id, err := app.users.Authenticate(form.Email, form.Password) if err != nil { if errors.Is(err, models.ErrInvalidCredentials) { form.AddNonFieldError("Email or password is incorrect") data := app.newTemplateData(r) data.Form = form app.render(w, http.StatusUnprocessableEntity, "login.tmpl", data) } else { app.serverError(w, err) } return } err = app.sessionManager.RenewToken(r.Context()) if err != nil { app.serverError(w, err) return } app.sessionManager.Put(r.Context(), "authenticatedUserID", id) http.Redirect(w, r, "/snippet/create", http.StatusSeeOther) } func (app *application) userLogoutPost(w http.ResponseWriter, r *http.Request) { err := app.sessionManager.RenewToken(r.Context()) if err != nil { app.serverError(w, err) return } app.sessionManager.Remove(r.Context(), "authenticatedUserID") app.sessionManager.Put(r.Context(), "flash", "You've been logged out succefully!") http.Redirect(w, r, "/", http.StatusSeeOther) } func (app *application) userSignup(w http.ResponseWriter, r *http.Request) { data := app.newTemplateData(r) data.Form = userSignupForm{} app.render(w, http.StatusOK, "signup.tmpl", data) } func (app *application) userLogin(w http.ResponseWriter, r *http.Request) { data := app.newTemplateData(r) data.Form = userLoginForm{} app.render(w, http.StatusOK, "login.tmpl", data) } func ping(w http.ResponseWriter, r *http.Request) { w.Write([]byte("OK")) }