website/pkg/controller/forum/add_edit.go

package forum

import (
	"net/http"
	"strconv"
	"strings"

	"code.nonshy.com/nonshy/website/pkg/config"
	"code.nonshy.com/nonshy/website/pkg/models"
	"code.nonshy.com/nonshy/website/pkg/session"
	"code.nonshy.com/nonshy/website/pkg/templates"
)

// AddEdit page.
func AddEdit() http.HandlerFunc {
	tmpl := templates.Must("forum/add_edit.html")
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// Are we editing a forum or creating a new one?
		var editID uint64
		if editStr := r.FormValue("id"); editStr != "" {
			if i, err := strconv.Atoi(editStr); err == nil {
				editID = uint64(i)
			} else {
				session.FlashError(w, r, "Edit parameter: id was not an integer")
				templates.Redirect(w, "/forum/admin")
				return
			}
		}

		// Get the current user.
		currentUser, err := session.CurrentUser(r)
		if err != nil {
			session.FlashError(w, r, "Couldn't get current user: %s", err)
			templates.Redirect(w, "/")
			return
		}

		// If editing, look up the existing forum.
		var forum *models.Forum
		if editID > 0 {
			if found, err := models.GetForum(editID); err != nil {
				session.FlashError(w, r, "Couldn't get forum: %s", err)
				templates.Redirect(w, "/forum/admin")
				return
			} else {
				// Do we have permission?
				if found.OwnerID != currentUser.ID && !currentUser.IsAdmin {
					templates.ForbiddenPage(w, r)
					return
				}

				forum = found
			}
		}

		// Saving?
		if r.Method == http.MethodPost {
			var (
				title          = strings.TrimSpace(r.PostFormValue("title"))
				fragment       = strings.TrimSpace(strings.ToLower(r.PostFormValue("fragment")))
				description    = strings.TrimSpace(r.PostFormValue("description"))
				category       = strings.TrimSpace(r.PostFormValue("category"))
				isExplicit     = r.PostFormValue("explicit") == "true"
				isPrivileged   = r.PostFormValue("privileged") == "true"
				isPermitPhotos = r.PostFormValue("permit_photos") == "true"
			)

			// Sanity check admin-only settings.
			if !currentUser.IsAdmin {
				isPrivileged = false
				isPermitPhotos = false
			}

			// Were we editing an existing forum?
			if forum != nil {
				forum.Title = title
				forum.Description = description
				forum.Category = category
				forum.Explicit = isExplicit
				forum.Privileged = isPrivileged
				forum.PermitPhotos = isPermitPhotos

				// Save it.
				if err := forum.Save(); err == nil {
					session.Flash(w, r, "Forum has been updated!")
					templates.Redirect(w, "/forum/admin")
					return
				} else {
					session.FlashError(w, r, "Error saving the forum: %s", err)
				}
			} else {
				// Validate the fragment. Front-end enforces the pattern so this
				// is just a sanity check.
				if m := FragmentRegexp.FindStringSubmatch(fragment); m == nil {
					session.FlashError(w, r, "The fragment format is invalid.")
					templates.Redirect(w, "/forum/admin")
					return
				}

				// Ensure the fragment is unique.
				if _, err := models.ForumByFragment(fragment); err == nil {
					session.FlashError(w, r, "The forum fragment is already in use.")
				} else {
					// Create the forum.
					forum = &models.Forum{
						Owner:        *currentUser,
						Category:     category,
						Fragment:     fragment,
						Title:        title,
						Description:  description,
						Explicit:     isExplicit,
						Privileged:   isPrivileged,
						PermitPhotos: isPermitPhotos,
					}

					if err := models.CreateForum(forum); err == nil {
						session.Flash(w, r, "The forum has been created!")
						templates.Redirect(w, "/forum/admin")
						return
					} else {
						session.FlashError(w, r, "Error creating the forum: %s", err)
					}
				}
			}
		}

		_ = editID

		var vars = map[string]interface{}{
			"EditID":     editID,
			"EditForum":  forum,
			"Categories": config.ForumCategories,
		}
		if err := tmpl.Execute(w, r, vars); err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
	})
}