92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
|
package comment
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"code.nonshy.com/nonshy/website/pkg/models"
|
||
|
|
"code.nonshy.com/nonshy/website/pkg/session"
|
||
|
|
"code.nonshy.com/nonshy/website/pkg/templates"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Subscription endpoint - to opt in or out of comment thread subscriptions.
|
||
|
|
func Subscription() http.HandlerFunc {
|
||
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
// Query params.
|
||
|
|
var (
|
||
|
|
tableName = r.FormValue("table_name")
|
||
|
|
tableID uint64
|
||
|
|
subscribe = r.FormValue("subscribe") == "true"
|
||
|
|
fromURL = r.FormValue("next") // what page to send back to
|
||
|
|
)
|
||
|
|
|
||
|
|
// Parse the table ID param.
|
||
|
|
if idStr := r.FormValue("table_id"); idStr == "" {
|
||
|
|
session.FlashError(w, r, "Comment table ID required.")
|
||
|
|
templates.Redirect(w, "/")
|
||
|
|
return
|
||
|
|
} else {
|
||
|
|
if idInt, err := strconv.Atoi(idStr); err != nil {
|
||
|
|
session.FlashError(w, r, "Comment table ID invalid.")
|
||
|
|
templates.Redirect(w, "/")
|
||
|
|
return
|
||
|
|
} else {
|
||
|
|
tableID = uint64(idInt)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Redirect URL must be relative.
|
||
|
|
if !strings.HasPrefix(fromURL, "/") {
|
||
|
|
// Maybe it's URL encoded?
|
||
|
|
fromURL, _ = url.QueryUnescape(fromURL)
|
||
|
|
if !strings.HasPrefix(fromURL, "/") {
|
||
|
|
fromURL = "/"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Validate everything else.
|
||
|
|
if _, ok := models.CommentableTables[tableName]; !ok {
|
||
|
|
session.FlashError(w, r, "You can not comment on that.")
|
||
|
|
templates.Redirect(w, "/")
|
||
|
|
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
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get their subscription.
|
||
|
|
sub, err := models.GetSubscription(currentUser, tableName, tableID)
|
||
|
|
if err != nil {
|
||
|
|
// If they want to subscribe, insert their row.
|
||
|
|
if subscribe {
|
||
|
|
if _, err := models.SubscribeTo(currentUser, tableName, tableID); err != nil {
|
||
|
|
session.FlashError(w, r, "Couldn't create subscription: %s", err)
|
||
|
|
} else {
|
||
|
|
session.Flash(w, r, "You will now be notified about comments on this page.")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// Toggle it.
|
||
|
|
sub.Subscribed = subscribe
|
||
|
|
if err := sub.Save(); err != nil {
|
||
|
|
session.FlashError(w, r, "Couldn't save your subscription settings: %s", err)
|
||
|
|
} else {
|
||
|
|
if subscribe {
|
||
|
|
session.Flash(w, r, "You will now be notified about comments on this page.")
|
||
|
|
} else {
|
||
|
|
session.Flash(w, r, "You will no longer be notified about new comments on this page.")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
templates.Redirect(w, fromURL)
|
||
|
|
})
|
||
|
|
}
|