105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"log"
|
|
|
|
sn "github.com/ekzyis/snappy"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
var (
|
|
db = getDb()
|
|
)
|
|
|
|
func getDb() *sql.DB {
|
|
var (
|
|
db *sql.DB
|
|
err error
|
|
)
|
|
if db, err = sql.Open("sqlite3", "chessbot.sqlite3?_foreign_keys=on"); err != nil {
|
|
log.Fatal(err)
|
|
} else {
|
|
if err = migrate(db); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
return db
|
|
}
|
|
|
|
func migrate(db *sql.DB) error {
|
|
_, err := db.Exec(`
|
|
CREATE TABLE IF NOT EXISTS items (
|
|
id INTEGER PRIMARY KEY,
|
|
user_id INTEGER NOT NULL,
|
|
text TEXT NOT NULL,
|
|
parent_id INTEGER REFERENCES items(id),
|
|
created_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
updated_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
`)
|
|
|
|
return err
|
|
}
|
|
|
|
func ItemHasReply(parentId int, userId int) (bool, error) {
|
|
var (
|
|
count int
|
|
err error
|
|
)
|
|
|
|
if err = db.QueryRow(`SELECT COUNT(1) FROM items WHERE parent_id = ? AND user_id = ?`, parentId, userId).Scan(&count); err != nil {
|
|
return true, err
|
|
}
|
|
|
|
if count > 0 {
|
|
return true, nil
|
|
}
|
|
|
|
// check if parent already exists, this means we ignored it
|
|
if err = db.QueryRow(`SELECT COUNT(1) FROM items WHERE id = ?`, parentId).Scan(&count); err != nil {
|
|
return true, err
|
|
}
|
|
|
|
return count > 0, nil
|
|
}
|
|
|
|
func InsertItem(item *sn.Item) error {
|
|
if _, err := db.Exec(``+
|
|
`INSERT INTO items(id, user_id, text, parent_id) VALUES (?, ?, ?, NULLIF(?, 0)) `+
|
|
`ON CONFLICT DO UPDATE SET text = EXCLUDED.text, updated_at = CURRENT_TIMESTAMP`,
|
|
item.Id, item.User.Id, item.Text, item.ParentId); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func GetThread(id int) ([]sn.Item, error) {
|
|
var (
|
|
items []sn.Item
|
|
err error
|
|
)
|
|
|
|
var item sn.Item
|
|
item.ParentId = id
|
|
|
|
for item.ParentId > 0 {
|
|
// TODO: can't select created_at because sqlite3 doesn't support timestamps natively
|
|
// see https://github.com/mattn/go-sqlite3/issues/142
|
|
if err = db.QueryRow(
|
|
`SELECT id, user_id, text, COALESCE(parent_id, 0) FROM items WHERE id = ?`, item.ParentId).
|
|
Scan(&item.Id, &item.User.Id, &item.Text, &item.ParentId); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, errors.New("item not found in db")
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
items = append([]sn.Item{item}, items...)
|
|
}
|
|
|
|
return items, nil
|
|
}
|