72 lines
1.2 KiB
Go
72 lines
1.2 KiB
Go
|
package store
|
||
|
|
||
|
import (
|
||
|
"database/sql"
|
||
|
"fmt"
|
||
|
"strings"
|
||
|
|
||
|
_ "github.com/mattn/go-sqlite3"
|
||
|
)
|
||
|
|
||
|
type Store struct {
|
||
|
Path string
|
||
|
Conn *sql.DB
|
||
|
}
|
||
|
|
||
|
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}
|
||
|
|
||
|
conn, err := sql.Open("sqlite3", addPathOptions(path))
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
if err := s.EnsureSchema(); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
s.Conn = conn
|
||
|
|
||
|
return s, nil
|
||
|
}
|
||
|
|
||
|
func (s *Store) Close() error {
|
||
|
return s.Conn.Close()
|
||
|
}
|
||
|
|
||
|
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
|
||
|
);
|
||
|
|
||
|
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
|
||
|
}
|