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
|
package database
import (
"database/sql"
_ "github.com/mattn/go-sqlite3"
"log"
"time"
)
type KennelCat struct {
ID string `json:"id"`
UserID string `json:"user_id"`
Name string `json:"name"`
Link string `json:"link"`
Description string `json:"description"`
Spritesheet string `json:"spritesheet"`
CreatedAt time.Time `json:"created_at"`
}
type KennelState struct {
At time.Time `json:"at"`
EncodedState string `json:"state"`
}
func CountUserKennelCats(db *sql.DB, userID string) (int, error) {
log.Println("counting kennel cats for user", userID)
row := db.QueryRow("SELECT COUNT(*) FROM kennel_cat WHERE user_id = ?", userID)
var count int
err := row.Scan(&count)
if err != nil {
return 0, err
}
return count, nil
}
func GetUserKennelCats(db *sql.DB, userID string) ([]KennelCat, error) {
log.Println("getting kennel cats for user", userID)
rows, err := db.Query("SELECT * FROM kennel_cat WHERE user_id = ?", userID)
if err != nil {
return nil, err
}
defer rows.Close()
var cats []KennelCat
for rows.Next() {
var cat KennelCat
err := rows.Scan(&cat.ID, &cat.Name, &cat.UserID, &cat.Link, &cat.Description, &cat.Spritesheet, &cat.CreatedAt)
if err != nil {
return nil, err
}
cats = append(cats, cat)
}
return cats, nil
}
func SaveKennelCat(db *sql.DB, cat *KennelCat) (*KennelCat, error) {
log.Println("saving kennel cat", cat.ID)
if (cat.CreatedAt == time.Time{}) {
cat.CreatedAt = time.Now()
}
_, err := db.Exec("INSERT OR REPLACE INTO kennel_cat (id, user_id, name, link, description, spritesheet, created_at) VALUES (?, ?, ?, ?, ?, ?, ?)", cat.ID, cat.UserID, cat.Name, cat.Link, cat.Description, cat.Spritesheet, cat.CreatedAt)
if err != nil {
return nil, err
}
return cat, nil
}
func GetKennelCat(db *sql.DB, catID string) (*KennelCat, error) {
log.Println("getting kennel cat", catID)
row := db.QueryRow("SELECT * FROM kennel_cat WHERE id = ?", catID)
var cat KennelCat
err := row.Scan(&cat.ID, &cat.Name, &cat.UserID, &cat.Link, &cat.Description, &cat.Spritesheet, &cat.CreatedAt)
if err != nil {
return nil, err
}
return &cat, nil
}
func DeleteKennelCat(db *sql.DB, catID string) error {
log.Println("deleting kennel cat", catID)
_, err := db.Exec("DELETE FROM kennel_cat WHERE id = ?", catID)
if err != nil {
return err
}
return nil
}
func GetRandomKennelCat(dbConn *sql.DB) (*KennelCat, error) {
log.Println("getting random kennel cat")
row := dbConn.QueryRow("SELECT * FROM kennel_cat ORDER BY RANDOM() LIMIT 1")
var cat KennelCat
err := row.Scan(&cat.ID, &cat.Name, &cat.UserID, &cat.Link, &cat.Description, &cat.Spritesheet, &cat.CreatedAt)
if err != nil {
return nil, err
}
return &cat, nil
}
func GetNextKennelCat(dbConn *sql.DB, lastID string, next bool) (*KennelCat, error) {
log.Println("getting next kennel cat")
operation := ">"
sorted := "ASC"
if !next {
operation = "<"
sorted = "DESC"
}
row := dbConn.QueryRow("SELECT * FROM kennel_cat WHERE id "+operation+" ? ORDER BY id "+sorted+" LIMIT 1", lastID)
var cat KennelCat
err := row.Scan(&cat.ID, &cat.Name, &cat.UserID, &cat.Link, &cat.Description, &cat.Spritesheet, &cat.CreatedAt)
if err != nil {
if next {
// loop "back" to the first in the ring
row = dbConn.QueryRow("SELECT * FROM kennel_cat ORDER BY id ASC LIMIT 1")
} else {
// loop "forward" to the first in the ring
row = dbConn.QueryRow("SELECT * FROM kennel_cat ORDER BY id DESC LIMIT 1")
}
err = row.Scan(&cat.ID, &cat.Name, &cat.UserID, &cat.Link, &cat.Description, &cat.Spritesheet, &cat.CreatedAt)
}
if err != nil {
return nil, err
}
return &cat, nil
}
func GetKennel(dbConn *sql.DB) ([]KennelCat, error) {
log.Println("getting kennel")
rows, err := dbConn.Query("SELECT * FROM kennel_cat")
if err != nil {
return nil, err
}
defer rows.Close()
var cats []KennelCat
for rows.Next() {
var cat KennelCat
err := rows.Scan(&cat.ID, &cat.Name, &cat.UserID, &cat.Link, &cat.Description, &cat.Spritesheet, &cat.CreatedAt)
if err != nil {
return nil, err
}
cats = append(cats, cat)
}
return cats, nil
}
|
