01a692ab5b
Added tests for the word filter store. Added qgen.NewAcc() to reduce the amount of boilerplate needed for creating an accumulator. Exposed the RecordError method on the accumulator. Added an Add method to PluginList and removed AddPlugin() in favour of that. More panel buttons on Nox should be styled now. Added the panel_update_button_text phrase for future use. More errors might be caught in the thumbnailer now. Removed ls from .travis.yml, it was there for debugging Code Climate.
113 lines
3.0 KiB
Go
113 lines
3.0 KiB
Go
package routes
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"log"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"../common"
|
|
"../query_gen/lib"
|
|
)
|
|
|
|
func PollVote(w http.ResponseWriter, r *http.Request, user common.User, sPollID string) common.RouteError {
|
|
pollID, err := strconv.Atoi(sPollID)
|
|
if err != nil {
|
|
return common.PreError("The provided PollID is not a valid number.", w, r)
|
|
}
|
|
|
|
poll, err := common.Polls.Get(pollID)
|
|
if err == sql.ErrNoRows {
|
|
return common.PreError("The poll you tried to vote for doesn't exist.", w, r)
|
|
} else if err != nil {
|
|
return common.InternalError(err, w, r)
|
|
}
|
|
|
|
var topic *common.Topic
|
|
if poll.ParentTable == "replies" {
|
|
reply, err := common.Rstore.Get(poll.ParentID)
|
|
if err == sql.ErrNoRows {
|
|
return common.PreError("The parent post doesn't exist.", w, r)
|
|
} else if err != nil {
|
|
return common.InternalError(err, w, r)
|
|
}
|
|
topic, err = common.Topics.Get(reply.ParentID)
|
|
} else if poll.ParentTable == "topics" {
|
|
topic, err = common.Topics.Get(poll.ParentID)
|
|
} else {
|
|
return common.InternalError(errors.New("Unknown parentTable for poll"), w, r)
|
|
}
|
|
|
|
if err == sql.ErrNoRows {
|
|
return common.PreError("The parent topic doesn't exist.", w, r)
|
|
} else if err != nil {
|
|
return common.InternalError(err, w, r)
|
|
}
|
|
|
|
// TODO: Add hooks to make use of headerLite
|
|
_, ferr := common.SimpleForumUserCheck(w, r, &user, topic.ParentID)
|
|
if ferr != nil {
|
|
return ferr
|
|
}
|
|
if !user.Perms.ViewTopic {
|
|
return common.NoPermissions(w, r, user)
|
|
}
|
|
|
|
optionIndex, err := strconv.Atoi(r.PostFormValue("poll_option_input"))
|
|
if err != nil {
|
|
return common.LocalError("Malformed input", w, r, user)
|
|
}
|
|
|
|
err = poll.CastVote(optionIndex, user.ID, user.LastIP)
|
|
if err != nil {
|
|
return common.InternalError(err, w, r)
|
|
}
|
|
|
|
http.Redirect(w, r, "/topic/"+strconv.Itoa(topic.ID), http.StatusSeeOther)
|
|
return nil
|
|
}
|
|
|
|
func PollResults(w http.ResponseWriter, r *http.Request, user common.User, sPollID string) common.RouteError {
|
|
log.Print("in PollResults")
|
|
pollID, err := strconv.Atoi(sPollID)
|
|
if err != nil {
|
|
return common.PreError("The provided PollID is not a valid number.", w, r)
|
|
}
|
|
|
|
poll, err := common.Polls.Get(pollID)
|
|
if err == sql.ErrNoRows {
|
|
return common.PreError("The poll you tried to vote for doesn't exist.", w, r)
|
|
} else if err != nil {
|
|
return common.InternalError(err, w, r)
|
|
}
|
|
|
|
// TODO: Abstract this
|
|
rows, err := qgen.NewAcc().Select("polls_options").Columns("votes").Where("pollID = ?").Orderby("option ASC").Query(poll.ID)
|
|
if err != nil {
|
|
return common.InternalError(err, w, r)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var optionList = ""
|
|
for rows.Next() {
|
|
var votes int
|
|
err := rows.Scan(&votes)
|
|
if err != nil {
|
|
return common.InternalError(err, w, r)
|
|
}
|
|
optionList += strconv.Itoa(votes) + ","
|
|
}
|
|
err = rows.Err()
|
|
if err != nil {
|
|
return common.InternalError(err, w, r)
|
|
}
|
|
|
|
// TODO: Implement a version of this which doesn't rely so much on sequential order
|
|
if len(optionList) > 0 {
|
|
optionList = optionList[:len(optionList)-1]
|
|
}
|
|
w.Write([]byte("[" + optionList + "]"))
|
|
return nil
|
|
}
|