website/pkg/controller/forum/search.go

package forum

import (
	"net/http"

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

// Search the forums.
func Search() http.HandlerFunc {
	tmpl := templates.Must("forum/search.html")

	// Whitelist for ordering options.
	var sortWhitelist = []string{
		"created_at desc",
		"created_at asc",
	}

	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		var (
			searchTerm = r.FormValue("q")
			byUsername = r.FormValue("username")
			postType   = r.FormValue("type")
			sort       = r.FormValue("sort")
			sortOK     bool
		)

		// Sort options.
		for _, v := range sortWhitelist {
			if sort == v {
				sortOK = true
				break
			}
		}
		if !sortOK {
			sort = sortWhitelist[0]
		}

		// All comments, or threads only?
		if postType != "threads" {
			postType = "all"
		}

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

		// Filters: find the user ID.
		var filterUserID uint64
		if byUsername != "" {
			user, err := models.FindUser(byUsername)
			if err != nil {
				session.FlashError(w, r, "Couldn't search posts by that username: no such username found.")
				templates.Redirect(w, r.URL.Path)
				return
			}
			filterUserID = user.ID
		}

		// Parse their search term.
		var (
			search  = models.ParseSearchString(searchTerm)
			filters = models.ForumSearchFilters{
				UserID:      filterUserID,
				ThreadsOnly: postType == "threads",
			}
			pager = &models.Pagination{
				Page:    1,
				PerPage: config.PageSizeThreadList,
				Sort:    sort,
			}
		)
		pager.ParsePage(r)

		posts, err := models.SearchForum(currentUser, search, filters, pager)
		if err != nil {
			session.FlashError(w, r, "Couldn't search the forums: %s", err)
			templates.Redirect(w, "/")
			return
		}

		// Map the originating threads to each comment.
		threadMap, err := models.MapForumCommentThreads(posts)
		if err != nil {
			log.Error("Couldn't map forum threads to comments: %s", err)
		}

		// Get any photo attachments for these comments.
		photos, err := models.MapCommentPhotos(posts)
		if err != nil {
			log.Error("Couldn't MapCommentPhotos: %s", err)
		}

		var vars = map[string]interface{}{
			"Pager":     pager,
			"Comments":  posts,
			"ThreadMap": threadMap,
			"PhotoMap":  photos,

			"SearchTerm": searchTerm,
			"ByUsername": byUsername,
			"Type":       postType,
			"Sort":       sort,
		}
		if err := tmpl.Execute(w, r, vars); err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
	})
}