package web import ( "html/template" "net/http" "git.snrd.de/Spaenny/goddit" "github.com/alexedwards/scs/v2" "github.com/go-chi/chi" "github.com/google/uuid" "github.com/gorilla/csrf" ) type ThreadHandler struct { store goddit.Store sessions *scs.SessionManager } func (h *ThreadHandler) List() http.HandlerFunc { type data struct { SessionData Threads []goddit.Thread } tmpl := template.Must(template.ParseFiles("templates/layout.html", "templates/threads.html")) return func(w http.ResponseWriter, r *http.Request) { tt, err := h.store.Threads() if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } tmpl.Execute(w, data{ SessionData: GetSessionData(h.sessions, r.Context()), Threads: tt, }) } } func (h *ThreadHandler) Create() http.HandlerFunc { type data struct { SessionData CSRF template.HTML } tmpl := template.Must(template.ParseFiles("templates/layout.html", "templates/thread_create.html")) return func(w http.ResponseWriter, r *http.Request) { tmpl.Execute(w, data{ SessionData: GetSessionData(h.sessions, r.Context()), CSRF: csrf.TemplateField(r), }) } } func (h *ThreadHandler) Show() http.HandlerFunc { type data struct { SessionData CSRF template.HTML Thread goddit.Thread Posts []goddit.Post } tmpl := template.Must(template.ParseFiles("templates/layout.html", "templates/thread.html")) return func(w http.ResponseWriter, r *http.Request) { idStr := chi.URLParam(r, "id") id, err := uuid.Parse(idStr) if err != nil { http.NotFound(w, r) return } t, err := h.store.Thread(id) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } pp, err := h.store.PostsByThread(t.ID) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } tmpl.Execute(w, data{ SessionData: GetSessionData(h.sessions, r.Context()), CSRF: csrf.TemplateField(r), Thread: t, Posts: pp, }) } } func (h *ThreadHandler) Store() http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { form := CreateThreadForm{ Title: r.FormValue("title"), Description: r.FormValue("description"), } if !form.Validate() { h.sessions.Put(r.Context(), "form", form) http.Redirect(w, r, r.Referer(), http.StatusFound) return } if err := h.store.CreateThread(&goddit.Thread{ ID: uuid.New(), Title: form.Title, Description: form.Description, }); err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } h.sessions.Put(r.Context(), "flash", "Your new thread has been created.") http.Redirect(w, r, "/threads", http.StatusFound) } } func (h *ThreadHandler) Delete() http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { idStr := chi.URLParam(r, "id") id, err := uuid.Parse(idStr) if err != nil { http.Error(w, err.Error(), http.StatusBadRequest) return } if err := h.store.DeleteThread(id); err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } h.sessions.Put(r.Context(), "flash", "The thread has been deleted.") http.Redirect(w, r, "/threads", http.StatusFound) } }