97 lines
2.6 KiB
Go
97 lines
2.6 KiB
Go
package gaypoints
|
|
|
|
import (
|
|
"database/sql"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"log"
|
|
"strconv"
|
|
)
|
|
|
|
var gpSelectUser *sql.Stmt
|
|
|
|
var gpSelectChat *sql.Stmt
|
|
|
|
var gpSet *sql.Stmt
|
|
|
|
func sqlGetAllGP(chatid int64) (gaypoints map[int64]int64) {
|
|
rows, err := gpSelectChat.Query(chatid)
|
|
if err != nil {
|
|
log.Panicf("Failed to query gaypoints: %v\n", err)
|
|
}
|
|
|
|
gaypoints = make(map[int64]int64)
|
|
var c int64
|
|
var b int64
|
|
for rows.Next() {
|
|
switch err := rows.Scan(&c, &b); err {
|
|
case sql.ErrNoRows:
|
|
gaypoints[c] = -1
|
|
case nil:
|
|
gaypoints[c] = b
|
|
default:
|
|
log.Panicf("Failed to query gaypoints: %v\n", err)
|
|
|
|
}
|
|
log.Println("[gaypoints.sql] Got gaypoints for " + strconv.FormatInt(c, 10) + " in chat " + strconv.FormatInt(chatid, 10) + ". Value is " + strconv.FormatInt(b, 10))
|
|
}
|
|
|
|
return gaypoints
|
|
|
|
}
|
|
|
|
func sqlGetGP(chatid int64, userid int64) (gaypoints int64) {
|
|
row := gpSelectUser.QueryRow(chatid, userid)
|
|
|
|
switch err := row.Scan(&gaypoints); err {
|
|
case sql.ErrNoRows:
|
|
gaypoints = -1
|
|
case nil:
|
|
break
|
|
default:
|
|
log.Panicf("Failed to query gaypoints: %v\n", err)
|
|
|
|
}
|
|
log.Println("[gaypoints.sql] Got gaypoints for " + strconv.FormatInt(userid, 10) + " in chat " + strconv.FormatInt(chatid, 10) + ". Value is " + strconv.FormatInt(gaypoints, 10))
|
|
return gaypoints
|
|
}
|
|
|
|
func sqlSetGP(chatid int64, userid int64, gaypoints int64) {
|
|
_, err := gpSet.Exec(chatid, userid, gaypoints)
|
|
if err != nil {
|
|
log.Panicf("Failed to set gaypoints: %v\n", err)
|
|
}
|
|
log.Println("[gaypoints.sql] Set gaypoints for " + strconv.FormatInt(userid, 10) + " in chat " + strconv.FormatInt(chatid, 10) + " to " + strconv.FormatInt(gaypoints, 10))
|
|
}
|
|
|
|
func InitDB() {
|
|
const dbPath string = "./bloater.db"
|
|
db, err := sql.Open("sqlite3", dbPath)
|
|
if err != nil {
|
|
log.Panicf("Failed to open sqlite database: %v\n", err)
|
|
}
|
|
|
|
_, err = db.Exec("CREATE TABLE IF NOT EXISTS gaypoints (chatid bigint, userid bigint,gaypoints bigint,UNIQUE(chatid,userid))")
|
|
if err != nil {
|
|
log.Panicf("Failed to create table: %v\n", err)
|
|
}
|
|
log.Println("[gaypoints.sql] Created sqlite table in database at " + dbPath)
|
|
|
|
gpSelectUser, err = db.Prepare("SELECT gaypoints FROM gaypoints WHERE chatid = ? AND userid = ?")
|
|
|
|
if err != nil {
|
|
log.Panicf("Failed to prepare sql select: %v\n", err)
|
|
}
|
|
|
|
gpSelectChat, err = db.Prepare("SELECT userid,gaypoints FROM gaypoints WHERE chatid = ? ORDER BY gaypoints DESC")
|
|
|
|
if err != nil {
|
|
log.Panicf("Failed to prepare sql select: %v\n", err)
|
|
}
|
|
|
|
gpSet, err = db.Prepare("INSERT OR REPLACE INTO gaypoints (chatid, userid, gaypoints) values (?,?,?)")
|
|
|
|
if err != nil {
|
|
log.Panicf("Failed to prepare sql insert: %v\n", err)
|
|
}
|
|
|
|
}
|