jzitnik
6756b9f179
Build and Publish Package (amd64) / docker (push) Successful in 4m54s
This is vibecoded shit but i can't be bothered to do it myself. Nobody else besides me is gonna be using this, and this wont ever be extended or anything, so I just don't care.
112 lines
2.6 KiB
Go
112 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
type Database struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewDatabase(path string) (*Database, error) {
|
|
db, err := sql.Open("sqlite3", path+"?_journal_mode=WAL&_busy_timeout=5000")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
d := &Database{db: db}
|
|
if err = d.migrate(); err != nil {
|
|
db.Close()
|
|
return nil, err
|
|
}
|
|
return d, nil
|
|
}
|
|
|
|
func (d *Database) Close() error {
|
|
return d.db.Close()
|
|
}
|
|
|
|
func (d *Database) migrate() error {
|
|
for _, q := range []string{
|
|
`CREATE TABLE IF NOT EXISTS spaces (
|
|
user_mxid TEXT PRIMARY KEY,
|
|
space_room_id TEXT NOT NULL
|
|
)`,
|
|
`CREATE TABLE IF NOT EXISTS bridges (
|
|
ntfy_topic TEXT PRIMARY KEY,
|
|
matrix_room_id TEXT NOT NULL
|
|
)`,
|
|
} {
|
|
if _, err := d.db.Exec(q); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *Database) GetSpaceRoomID(userMXID string) (string, error) {
|
|
var roomID string
|
|
err := d.db.QueryRow("SELECT space_room_id FROM spaces WHERE user_mxid = ?", userMXID).Scan(&roomID)
|
|
if err == sql.ErrNoRows {
|
|
return "", nil
|
|
}
|
|
return roomID, err
|
|
}
|
|
|
|
func (d *Database) SetSpaceRoomID(userMXID, spaceRoomID string) error {
|
|
_, err := d.db.Exec(
|
|
"INSERT OR REPLACE INTO spaces (user_mxid, space_room_id) VALUES (?, ?)",
|
|
userMXID, spaceRoomID,
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (d *Database) GetAllBridges() (map[string]string, error) {
|
|
rows, err := d.db.Query("SELECT ntfy_topic, matrix_room_id FROM bridges")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
bridges := make(map[string]string)
|
|
for rows.Next() {
|
|
var topic, roomID string
|
|
if err := rows.Scan(&topic, &roomID); err != nil {
|
|
return nil, err
|
|
}
|
|
bridges[topic] = roomID
|
|
}
|
|
return bridges, rows.Err()
|
|
}
|
|
|
|
func (d *Database) GetBridgeByRoom(matrixRoomID string) (string, error) {
|
|
var topic string
|
|
err := d.db.QueryRow("SELECT ntfy_topic FROM bridges WHERE matrix_room_id = ?", matrixRoomID).Scan(&topic)
|
|
if err == sql.ErrNoRows {
|
|
return "", nil
|
|
}
|
|
return topic, err
|
|
}
|
|
|
|
func (d *Database) GetBridgeByTopic(ntfyTopic string) (string, error) {
|
|
var roomID string
|
|
err := d.db.QueryRow("SELECT matrix_room_id FROM bridges WHERE ntfy_topic = ?", ntfyTopic).Scan(&roomID)
|
|
if err == sql.ErrNoRows {
|
|
return "", nil
|
|
}
|
|
return roomID, err
|
|
}
|
|
|
|
func (d *Database) SetBridge(ntfyTopic, matrixRoomID string) error {
|
|
_, err := d.db.Exec(
|
|
"INSERT OR REPLACE INTO bridges (ntfy_topic, matrix_room_id) VALUES (?, ?)",
|
|
ntfyTopic, matrixRoomID,
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (d *Database) DeleteBridge(ntfyTopic string) error {
|
|
_, err := d.db.Exec("DELETE FROM bridges WHERE ntfy_topic = ?", ntfyTopic)
|
|
return err
|
|
}
|