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, 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()
	})
}