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
|
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")
}
}
|
