Blizzard
95 lines
1.8 KiB
Go
95 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
var db *gorm.DB
|
|
|
|
// Setting represents simple key-value store for app configs
|
|
type Setting struct {
|
|
Key string `gorm:"primaryKey"`
|
|
Value string
|
|
}
|
|
|
|
func initDB() error {
|
|
homeDir, err := os.UserHomeDir()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
configDir := filepath.Join(homeDir, ".betting-assistant")
|
|
if err := os.MkdirAll(configDir, 0755); err != nil {
|
|
return err
|
|
}
|
|
dbPath := filepath.Join(configDir, "data.db")
|
|
|
|
db, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Silent),
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Auto Migration
|
|
if err := db.AutoMigrate(&Game{}, &Setting{}); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Insert defaults if empty
|
|
var count int64
|
|
db.Model(&Game{}).Count(&count)
|
|
if count == 0 {
|
|
insertDefaults()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func insertDefaults() {
|
|
games := []Game{
|
|
{ID: uuid.New().String(), Name: "地鼠挖宝 (Moles)", Icon: "🐹", URL: "http://127.0.0.1:8080/api/v1/moles/start", Token: ""},
|
|
}
|
|
db.Create(&games)
|
|
|
|
db.Save(&Setting{Key: "active_game_id", Value: games[0].ID})
|
|
}
|
|
|
|
// -- DB Operations wrapped for app logic --
|
|
|
|
func getAllGamesDB() []Game {
|
|
var games []Game
|
|
db.Find(&games)
|
|
return games
|
|
}
|
|
|
|
func getActiveGameID() string {
|
|
var s Setting
|
|
db.First(&s, "key = ?", "active_game_id")
|
|
return s.Value
|
|
}
|
|
|
|
func setActiveGameID(id string) error {
|
|
return db.Save(&Setting{Key: "active_game_id", Value: id}).Error
|
|
}
|
|
|
|
func addGameDB(game *Game) error {
|
|
if game.ID == "" {
|
|
game.ID = uuid.New().String()
|
|
}
|
|
return db.Create(game).Error
|
|
}
|
|
|
|
func updateGameDB(game *Game) error {
|
|
return db.Save(game).Error
|
|
}
|
|
|
|
func deleteGameDB(id string) error {
|
|
return db.Delete(&Game{}, "id = ?", id).Error
|
|
}
|