website/pkg/controller/forum/search.go

119 lines
2.7 KiB
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
}
})
}