Gacha pool overhaul and DB backup wizard

server/cmd/wizard-restore/main.go

@@ -0,0 +1,135 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"sort"
+	"strings"
+
+	"charm.land/huh/v2"
+	"charm.land/lipgloss/v2"
+)
+
+const (
+	gameDBPath   = "db/game.db"
+	backupDir    = "db/backups"
+	backupSuffix = ".bak"
+)
+
+const banner = `
+  _                        _____
+ | |  _  _ _ _  __ _ _ _  |_   _|___ __ _ _ _
+ | |_| || | ' \/ _` + "`" + ` | '_|   | |/ -_)/ _` + "`" + ` | '_|
+ |____\_,_|_||_\__,_|_|     |_|\___|\__,_|_|
+
+ ╭──────────────────────────────╮
+ │   RESTORE                    │
+ ╰──────────────────────────────╯
+`
+
+func main() {
+	lipgloss.EnableLegacyWindowsANSI(os.Stdout)
+	lipgloss.EnableLegacyWindowsANSI(os.Stderr)
+	fmt.Print(banner)
+
+	chosen, ok := pickBackup()
+	if !ok {
+		return
+	}
+	if !confirmOverwrite(chosen) {
+		fmt.Println("  cancelled — nothing changed")
+		return
+	}
+	if err := doRestore(chosen); err != nil {
+		fmt.Fprintf(os.Stderr, "  restore failed: %v\n", err)
+		os.Exit(1)
+	}
+	fmt.Printf("  restored %s from %s\n", gameDBPath, chosen)
+}
+
+func pickBackup() (string, bool) {
+	entries, err := os.ReadDir(backupDir)
+	if err != nil {
+		fmt.Fprintln(os.Stderr, "  no backups found in", backupDir)
+		return "", false
+	}
+	var backups []string
+	for _, e := range entries {
+		if !e.IsDir() && strings.HasPrefix(e.Name(), "game.db.") && strings.HasSuffix(e.Name(), backupSuffix) {
+			backups = append(backups, e.Name())
+		}
+	}
+	if len(backups) == 0 {
+		fmt.Fprintln(os.Stderr, "  no backups found in", backupDir)
+		return "", false
+	}
+	sort.Slice(backups, func(i, j int) bool { return backups[i] > backups[j] })
+
+	options := make([]huh.Option[string], 0, len(backups)+1)
+	for _, name := range backups {
+		options = append(options, huh.NewOption(name, name))
+	}
+	options = append(options, huh.NewOption("Cancel", ""))
+
+	var chosen string
+	if err := huh.NewSelect[string]().
+		Title("Pick a backup to restore").
+		Description("db/game.db will be replaced by the chosen file.").
+		Options(options...).
+		Value(&chosen).
+		Run(); err != nil || chosen == "" {
+		return "", false
+	}
+	return chosen, true
+}
+
+func confirmOverwrite(chosen string) bool {
+	confirm := false
+	if err := huh.NewConfirm().
+		Title("Overwrite db/game.db?").
+		Description(fmt.Sprintf(
+			"This will REPLACE db/game.db with %s.\n"+
+				"Any progress since that backup will be lost.\n"+
+				"(A fresh backup will be taken on the next ./wizard launch.)",
+			chosen)).
+		Affirmative("Yes, restore").
+		Negative("Cancel").
+		Value(&confirm).
+		Run(); err != nil {
+		return false
+	}
+	return confirm
+}
+
+func doRestore(chosen string) error {
+	src := filepath.Join(backupDir, chosen)
+	if _, err := os.Stat(src); errors.Is(err, os.ErrNotExist) {
+		return fmt.Errorf("%s no longer exists", src)
+	}
+	if err := copyFile(src, gameDBPath); err != nil {
+		return err
+	}
+	_ = os.Remove(gameDBPath + "-wal")
+	_ = os.Remove(gameDBPath + "-shm")
+	return nil
+}
+
+func copyFile(src, dst string) error {
+	in, err := os.Open(src)
+	if err != nil {
+		return err
+	}
+	defer in.Close()
+	out, err := os.Create(dst)
+	if err != nil {
+		return err
+	}
+	defer out.Close()
+	if _, err := io.Copy(out, in); err != nil {
+		return err
+	}
+	return out.Sync()
+}

server/cmd/wizard/backup.go

@@ -0,0 +1,75 @@
+package main
+
+import (
+	"database/sql"
+	"errors"
+	"fmt"
+	"os"
+	"path/filepath"
+	"sort"
+	"strings"
+	"time"
+
+	"charm.land/huh/v2/spinner"
+
+	_ "modernc.org/sqlite"
+)
+
+const (
+	gameDBPath    = "db/game.db"
+	backupDir     = "db/backups"
+	backupSuffix  = ".bak"
+	backupRetainN = 10
+)
+
+func backupGameDB() {
+	if _, err := os.Stat(gameDBPath); errors.Is(err, os.ErrNotExist) {
+		return
+	}
+
+	_ = spinner.New().Title("  Backing up db/game.db...").Action(func() {
+		if err := os.MkdirAll(backupDir, 0o755); err != nil {
+			fmt.Fprintf(os.Stderr, "  failed to create %s: %v\n", backupDir, err)
+			return
+		}
+
+		ts := time.Now().UTC().Format("20060102T150405Z")
+		dest := filepath.Join(backupDir, fmt.Sprintf("game.db.%s%s", ts, backupSuffix))
+
+		db, err := sql.Open("sqlite", gameDBPath+"?_pragma=busy_timeout(5000)")
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "  failed to open %s: %v\n", gameDBPath, err)
+			return
+		}
+		defer db.Close()
+
+		escaped := strings.ReplaceAll(dest, "'", "''")
+		if _, err := db.Exec(fmt.Sprintf("VACUUM INTO '%s'", escaped)); err != nil {
+			fmt.Fprintf(os.Stderr, "  VACUUM INTO failed: %v\n", err)
+			_ = os.Remove(dest)
+			return
+		}
+
+		pruneOldBackups()
+	}).Run()
+}
+
+func pruneOldBackups() {
+	entries, err := os.ReadDir(backupDir)
+	if err != nil {
+		return
+	}
+	var backups []os.DirEntry
+	for _, e := range entries {
+		if !e.IsDir() && strings.HasPrefix(e.Name(), "game.db.") && strings.HasSuffix(e.Name(), backupSuffix) {
+			backups = append(backups, e)
+		}
+	}
+	if len(backups) <= backupRetainN {
+		return
+	}
+	sort.Slice(backups, func(i, j int) bool { return backups[i].Name() < backups[j].Name() })
+	for _, old := range backups[:len(backups)-backupRetainN] {
+		_ = os.Remove(filepath.Join(backupDir, old.Name()))
+	}
+}

server/cmd/wizard/main.go

@@ -79,6 +79,7 @@ func main() {
 		validateTools()
 		validateProtocIncludes()
 		runProtoc()
+		backupGameDB()
 		runMigrate()
 		downloadDeps()
 	}