package main import ( "errors" "fmt" "net/http" "strconv" "strings" "unicode/utf8" "github.com/julienschmidt/httprouter" "snippetbox.chaosfem.tw/internal/models" ) // home ... func (app *application) home(w http.ResponseWriter, r *http.Request) { snippets, err := app.snippets.Latest() if err != nil { app.serverError(w, r, err) return } data := app.newTemplateData(r) data.Snippets = snippets app.render(w, r, http.StatusOK, "home.tmpl", data) } // snippetView ... 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 || id < 1 { if errors.Is(err, models.ErrNoRecord) { app.notFound(w) } else { app.serverError(w, r, err) } return } data := app.newTemplateData(r) data.Snippet = snippet app.render(w, r, http.StatusOK, "view.tmpl", data) } // (app *application) snippetCreate ... func (app *application) snippetCreate(w http.ResponseWriter, r *http.Request) { data := app.newTemplateData(r) data.Form = snippetCreateForm{ Expires: 365, } app.render(w, r, http.StatusOK, "create.tmpl", data) } type snippetCreateForm struct { Title string Content string Expires int FieldErrors map[string]string } // snippetCreatePost ... func (app *application) snippetCreatePost(w http.ResponseWriter, r *http.Request) { err := r.ParseForm() if err != nil { app.clientError(w, http.StatusBadRequest) return } expires, err := strconv.Atoi(r.PostForm.Get("expires")) if err != nil { app.clientError(w, http.StatusBadRequest) return } form := snippetCreateForm{ Title: r.PostForm.Get("title"), Content: r.PostForm.Get("content"), Expires: expires, FieldErrors: map[string]string{}, } if strings.TrimSpace(form.Title) == "" { form.FieldErrors["title"] = "This field cannot be blank" } else if utf8.RuneCountInString(form.Title) > 100 { form.FieldErrors["title"] = "This field cannot contain more than 100 characters" } if strings.TrimSpace(form.Content) == "" { form.FieldErrors["content"] = "This field cannot be blank" } if form.Expires != 1 && form.Expires != 7 && form.Expires != 365 { form.FieldErrors["expires"] = "This field must equal 1, 7 or 365" } if len(form.FieldErrors) > 0 { data := app.newTemplateData(r) data.Form = form app.render(w, r, http.StatusUnprocessableEntity, "create.tmpl", data) return } id, err := app.snippets.Insert(form.Title, form.Content, form.Expires) if err != nil { app.serverError(w, r, err) return } http.Redirect(w, r, fmt.Sprintf("/snippet/view/%d", id), http.StatusSeeOther) }