62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
|
package common
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
|
||
|
|
qgen "github.com/Azareal/Gosora/query_gen"
|
||
|
|
)
|
||
|
|
|
||
|
|
var pollStmts PollStmts
|
||
|
|
|
||
|
|
type Poll struct {
|
||
|
|
ID int
|
||
|
|
ParentID int
|
||
|
|
ParentTable string
|
||
|
|
Type int // 0: Single choice, 1: Multiple choice, 2: Multiple choice w/ points
|
||
|
|
//AntiCheat bool // Apply various mitigations for cheating
|
||
|
|
// GroupPower map[gid]points // The number of points a group can spend in this poll, defaults to 1
|
||
|
|
|
||
|
|
Options map[int]string
|
||
|
|
Results map[int]int // map[optionIndex]points
|
||
|
|
QuickOptions []PollOption // TODO: Fix up the template transpiler so we don't need to use this hack anymore
|
||
|
|
VoteCount int
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Poll) CastVote(optionIndex int, uid int, ip string) error {
|
||
|
|
return Polls.CastVote(optionIndex, p.ID, uid, ip) // TODO: Move the query into a pollStmts rather than having it in the store
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Poll) Delete() error {
|
||
|
|
_, err := pollStmts.deletePollVotes.Exec(p.ID)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
_, err = pollStmts.deletePollOptions.Exec(p.ID)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
_, err = pollStmts.deletePoll.Exec(p.ID)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Poll) Copy() Poll {
|
||
|
|
return *p
|
||
|
|
}
|
||
|
|
|
||
|
|
type PollStmts struct {
|
||
|
|
deletePoll *sql.Stmt
|
||
|
|
deletePollOptions *sql.Stmt
|
||
|
|
deletePollVotes *sql.Stmt
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
DbInits.Add(func(acc *qgen.Accumulator) error {
|
||
|
|
pollStmts = PollStmts{
|
||
|
|
deletePoll: acc.Delete("polls").Where("pollID=?").Prepare(),
|
||
|
|
deletePollOptions: acc.Delete("polls_options").Where("pollID=?").Prepare(),
|
||
|
|
deletePollVotes: acc.Delete("polls_votes").Where("pollID=?").Prepare(),
|
||
|
|
}
|
||
|
|
return acc.FirstError()
|
||
|
|
})
|
||
|
|
}
|