gosora/routes/poll.go

package routes

import (
	"bytes"
	"database/sql"
	"errors"
	"net/http"
	"strconv"

	c "git.tuxpa.in/a/gosora/common"
)

func PollVote(w http.ResponseWriter, r *http.Request, u *c.User, sPollID string) c.RouteError {
    pollID, err := strconv.Atoi(sPollID)
    if err != nil {
        return c.PreError("The provided PollID is not a valid number.", w, r)
    }
    poll, err := c.Polls.Get(pollID)
    if err == sql.ErrNoRows {
        return c.PreError("The poll you tried to vote for doesn't exist.", w, r)
    } else if err != nil {
        return c.InternalError(err, w, r)
    }

    var topic *c.Topic
    if poll.ParentTable == "replies" {
        reply, err := c.Rstore.Get(poll.ParentID)
        if err == sql.ErrNoRows {
            return c.PreError("The parent post doesn't exist.", w, r)
        } else if err != nil {
            return c.InternalError(err, w, r)
        }
        topic, err = c.Topics.Get(reply.ParentID)
    } else if poll.ParentTable == "topics" {
        topic, err = c.Topics.Get(poll.ParentID)
    } else {
        return c.InternalError(errors.New("Unknown parentTable for poll"), w, r)
    }

    if err == sql.ErrNoRows {
        return c.PreError("The parent topic doesn't exist.", w, r)
    } else if err != nil {
        return c.InternalError(err, w, r)
    }

    // TODO: Add hooks to make use of headerLite
    _, ferr := c.SimpleForumUserCheck(w, r, u, topic.ParentID)
    if ferr != nil {
        return ferr
    }
    if !u.Perms.ViewTopic {
        return c.NoPermissions(w, r, u)
    }

    optIndex, err := strconv.Atoi(r.PostFormValue("poll_option_input"))
    if err != nil {
        return c.LocalError("Malformed input", w, r, u)
    }
    err = poll.CastVote(optIndex, u.ID, u.GetIP())
    if err != nil {
        return c.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, u *c.User, sPollID string) c.RouteError {
    //log.Print("in PollResults")
    pollID, err := strconv.Atoi(sPollID)
    if err != nil {
        return c.PreError("The provided PollID is not a valid number.", w, r)
    }
    poll, err := c.Polls.Get(pollID)
    if err == sql.ErrNoRows {
        return c.PreError("The poll you tried to vote for doesn't exist.", w, r)
    } else if err != nil {
        return c.InternalError(err, w, r)
    }

    // TODO: Implement a version of this which doesn't rely so much on sequential order
    var ob bytes.Buffer
    ob.WriteRune('[')
    var i int
    e := poll.Resultsf(func(votes int) error {
        if i != 0 {
            ob.WriteRune(',')
        }
        ob.WriteString(strconv.Itoa(votes))
        i++
        return nil
    })
    if e != nil && e != sql.ErrNoRows {
        return c.InternalError(e, w, r)
    }
    ob.WriteRune(']')
    w.Write(ob.Bytes())

    return nil
}