1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
|
package main
import "github.com/mxk/go-sqlite/sqlite3"
import "errors"
import "strconv"
type Poll struct {
id int64
title string
userId int64
}
func addPoll(db *sqlite3.Conn, title string, options []string, nick string) (int64, error) {
user := getUserForName(db, nick)
if (user.id == 0) {
var err error
user, err = createUser(db, nick, false)
if (err != nil) {
return 0, err
}
}
args := sqlite3.NamedArgs{"$a": title, "$b": user.id}
sql := "INSERT INTO polls (title, user_id) VALUES ($a, $b)"
err := db.Exec(sql, args)
if (err != nil ) {
return 0, err
}
pollId := db.LastInsertId()
for _,option := range options {
addOption(db, option, pollId)
if (err != nil ) {
return 0, err
}
}
return pollId, nil
}
func addOption(db *sqlite3.Conn, text string, pollId int64) error {
args := sqlite3.NamedArgs{"$a": text, "$b": pollId}
sql := "INSERT INTO options (text, poll_id) VALUES ($a, $b)"
err := db.Exec(sql, args)
if (err != nil ) {
return err
}
return nil
}
func getPollFromTitle(db *sqlite3.Conn, title string) (Poll, error) {
args := sqlite3.NamedArgs{"$a": title}
sql := "SELECT * FROM polls WHERE title = $a"
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
poll := Poll{id:rowid, title:row["title"].(string), userId:row["user_id"].(int64)}
return poll, nil
}
//If we get here there are no matching polls, return an error
return Poll{id:0, title:"", userId:0}, errors.New("No poll found with title '" + title +"'")
}
func getPollFromId(db *sqlite3.Conn, id int) (Poll, error) {
args := sqlite3.NamedArgs{"$a": id}
sql := "SELECT * FROM polls WHERE id = $a"
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
poll := Poll{id:rowid, title:row["title"].(string), userId:row["user_id"].(int64)}
return poll, nil
}
//If we get here there are no matching polls, return an error
return Poll{id:0, title:"", userId:0}, errors.New("No poll found with ID " + strconv.Itoa(id))
}
func deletePoll(db *sqlite3.Conn, id int, nick string) error {
user := getUserForName(db, nick)
poll, err := getPollFromId(db, id)
if (err != nil) {
return err
}
if (user.isAdmin || poll.userId == user.id) {
if (poll.id == 0) {
return nil
} else {
sql := "DELETE FROM polls WHERE id = $a"
args := sqlite3.NamedArgs{"$a": id}
db.Exec(sql, args)
return nil
}
} else {
return errors.New("You do not have permission to delete this poll")
}
}
func hasUserVotedInPoll(db *sqlite3.Conn, poll Poll, user User) bool {
args := sqlite3.NamedArgs{"$a": poll.id, "$b": user.id}
sql := "SELECT count(*) FROM votes WHERE poll_id = $a AND user_id = $b"
s, err := db.Query(sql, args)
for ; err == nil ; err = s.Next() {
var count int64
s.Scan(&count) // Assigns 1st column to rowid, the rest to row
if (count == 0) {
return false
} else {
return true
}
}
//Not sure why we'd get here, but we probably don't want to record a vote if we did
return true
}
func getRecentPolls(db *sqlite3.Conn) ([]Poll, error) {
sql := "SELECT * FROM polls ORDER BY id DESC LIMIT 5"
s, err := db.Query(sql)
row := make(sqlite3.RowMap)
var polls = make([]Poll, 5)
i := 0
for ; err == nil ; err = s.Next() {
var rowid int64
s.Scan(&rowid, row) // Assigns 1st column to rowid, the rest to row
poll := Poll{id:rowid, title:row["title"].(string), userId:row["user_id"].(int64)}
polls[i] = poll
i++
}
//If we get here there are no matching polls, return an error
return polls, err
}
|
