package main

import "github.com/mxk/go-sqlite/sqlite3"
import "errors"

type Option struct {
    id int64
    text string
    pollId int64
    numVotes int64
}

func getOptionFromText(db*sqlite3.Conn, text string, id int) (Option, error) {

    args := sqlite3.NamedArgs{"$a": text, "$b": id}
    sql := "SELECT * FROM options WHERE text = $a and poll_id = $b"
    s, err := db.Query(sql, args)
    row := make(sqlite3.RowMap)
    
    for ; err == nil ; err = s.Next() {
        var rowid int64
        s.Scan(&rowid, row)     // Assigns 1st column to rowid, the rest to row

        option := Option{id:rowid, text:row["text"].(string), pollId:row["poll_id"].(int64)}
        return option, nil
    }

    //If we get here there are no matching options
    return Option{id:0, text:"", pollId:0}, errors.New("No option found matching '" + text + "'")

}

func getVotesForOption(db*sqlite3.Conn, option Option) (Option, error) {
    args := sqlite3.NamedArgs{"$a": option.id, "$b": option.pollId}
    sql := "SELECT count(*) FROM votes WHERE option_id = $a AND poll_id = $b"
    s, err := db.Query(sql, args)
    
    for ; err == nil ; err = s.Next() {
        var votes int64
        s.Scan(&votes)     // Assigns 1st column to rowid, the rest to row

        option = Option{id:option.id, text:option.text, pollId:option.pollId, numVotes:votes}
        return option, nil
    }

    //If we get here there are no matching options
    return Option{id:0, text:"", pollId:0}, errors.New("No option found matching '" + option.text + "'")
}

func getOptionsForPoll(db*sqlite3.Conn, poll Poll) ([]Option, error) {
    args := sqlite3.NamedArgs{"$a": poll.id}
    sql := "SELECT count(*) FROM options WHERE poll_id = $a"
    s, err := db.Query(sql, args)
    
    var numOptions int64

    //Get the number of votes first so we can make the right size array
    for ; err == nil ; err = s.Next() {
        s.Scan(&numOptions)     // Assigns 1st column to rowid, the rest to row
    }

    var options = make([]Option, numOptions)

    sql = "SELECT * FROM options WHERE poll_id = $a"
    s, err = db.Query(sql, args)
    row := make(sqlite3.RowMap)
    id := 0

    for ; err == nil ; err = s.Next() {
        var rowid int64
        s.Scan(&rowid, row)     // Assigns 1st column to rowid, the rest to row

        option := Option{id:rowid, text:row["text"].(string), pollId:row["poll_id"].(int64)}
        options[id] = option
        id++
    }

    return options, err


    //If we get here there are no matching options
    // return Option{id:0, text:"", pollId:0}, errors.New("No option found matching '" + option.text + "'")
}