87 lines
1.5 KiB
Go
87 lines
1.5 KiB
Go
package store
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"strings"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
type Store struct {
|
|
Path string
|
|
Conn *sql.DB
|
|
|
|
userStore *UserStore
|
|
gameStore *GameStore
|
|
}
|
|
|
|
func addPathOptions(path string) string {
|
|
if !strings.HasPrefix(path, ":memory:") {
|
|
path = "file:" + path
|
|
}
|
|
return fmt.Sprintf("%s?_foreign_keys=true", path)
|
|
}
|
|
|
|
func NewStore(path string) (*Store, error) {
|
|
s := &Store{Path: path}
|
|
|
|
// ToDo: should do this at start time instead of create time?
|
|
conn, err := sql.Open("sqlite3", addPathOptions(path))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
s.Conn = conn
|
|
|
|
if err := s.EnsureSchema(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// init stores
|
|
s.userStore = NewUserStore(s)
|
|
s.gameStore = NewGameStore(s)
|
|
|
|
return s, nil
|
|
}
|
|
|
|
func (s *Store) Close() error {
|
|
return s.Conn.Close()
|
|
}
|
|
|
|
func (s *Store) User() *UserStore {
|
|
return s.userStore
|
|
}
|
|
|
|
func (s *Store) Game() *GameStore {
|
|
return s.gameStore
|
|
}
|
|
|
|
func (s *Store) EnsureSchema() error {
|
|
schema := `
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name VARCHAR(255) NOT NULL,
|
|
mail VARCHAR(255) UNIQUE NOT NULL,
|
|
username VARCHAR(255) UNIQUE NOT NULL,
|
|
password VARCHAR(255) NOT NULL,
|
|
admin BOOLEAN NOT NULL
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS game (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
user_id INTEGER NOT NULL,
|
|
name VARCHAR(255) NOT NULL
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS gamemember (
|
|
game_id INTEGER NOT NULL,
|
|
user_id INTEGER NOT NULL,
|
|
role VARCHAR(255) NOT NULL,
|
|
|
|
PRIMARY KEY (game_id, user_id)
|
|
);
|
|
`
|
|
|
|
_, err := s.Conn.Exec(schema)
|
|
return err
|
|
}
|