init

2026-06-15 16:53:25 +03:30
commit 390ef96e7e
34 changed files with 4463 additions and 0 deletions
@@ -0,0 +1,97 @@
+package store
+
+import (
+	"database/sql"
+	"embed"
+	"fmt"
+	"log/slog"
+	"strings"
+
+	_ "modernc.org/sqlite" // درایور SQLite خالص Go (بدون cgo)
+)
+
+//go:embed migrations/*.sql
+var migrationsFS embed.FS
+
+// Store دسترسی به دیتابیس را کپسوله می‌کند.
+type Store struct {
+	DB *sql.DB
+}
+
+// Open اتصال SQLite را باز کرده و migrations را اجرا می‌کند.
+func Open(path string) (*Store, error) {
+	// pragmaها برای کارایی و یکپارچگی روی سرور کوچک
+	dsn := fmt.Sprintf("file:%s?_pragma=journal_mode(WAL)&_pragma=busy_timeout(5000)&_pragma=foreign_keys(ON)", path)
+	db, err := sql.Open("sqlite", dsn)
+	if err != nil {
+		return nil, err
+	}
+	// SQLite تک‌نویسنده است؛ pool کوچک نگه می‌داریم.
+	db.SetMaxOpenConns(1)
+	if err := db.Ping(); err != nil {
+		return nil, err
+	}
+	s := &Store{DB: db}
+	if err := s.migrate(); err != nil {
+		return nil, err
+	}
+	return s, nil
+}
+
+func (s *Store) migrate() error {
+	// جدول ردیابی تا هر migration فقط یک‌بار اجرا شود.
+	if _, err := s.DB.Exec(
+		`CREATE TABLE IF NOT EXISTS schema_migrations (
+			name       TEXT PRIMARY KEY,
+			applied_at TEXT NOT NULL DEFAULT (datetime('now'))
+		)`); err != nil {
+		return fmt.Errorf("create schema_migrations: %w", err)
+	}
+
+	entries, err := migrationsFS.ReadDir("migrations")
+	if err != nil {
+		return fmt.Errorf("read migrations: %w", err)
+	}
+	for _, e := range entries {
+		if e.IsDir() {
+			continue
+		}
+		name := e.Name()
+
+		var applied int
+		if err := s.DB.QueryRow(
+			`SELECT COUNT(*) FROM schema_migrations WHERE name = ?`, name).Scan(&applied); err != nil {
+			return err
+		}
+		if applied > 0 {
+			continue
+		}
+
+		b, err := migrationsFS.ReadFile("migrations/" + name)
+		if err != nil {
+			return err
+		}
+		if _, err := s.DB.Exec(string(b)); err != nil {
+			// دیتابیس توسعه‌ای که قبل از افزودن ردیابی، این migration را اعمال کرده
+			// (ستون/جدول از قبل هست) ⇒ به‌عنوان اعمال‌شده علامت می‌زنیم.
+			if alreadyApplied(err) {
+				slog.Warn("migration schema already present; marking applied", "name", name)
+			} else {
+				return fmt.Errorf("apply %s: %w", name, err)
+			}
+		}
+		if _, err := s.DB.Exec(
+			`INSERT INTO schema_migrations (name) VALUES (?)`, name); err != nil {
+			return fmt.Errorf("record %s: %w", name, err)
+		}
+	}
+	return nil
+}
+
+// alreadyApplied تشخیص می‌دهد خطا ناشی از وجود قبلی همان تغییر است.
+func alreadyApplied(err error) bool {
+	m := strings.ToLower(err.Error())
+	return strings.Contains(m, "duplicate column") || strings.Contains(m, "already exists")
+}
+
+func (s *Store) Close() error { return s.DB.Close() }
@@ -0,0 +1,40 @@
+package store
+
+import (
+	"path/filepath"
+	"testing"
+)
+
+// TestMigrateIdempotent بررسی می‌کند باز کردن مکرر یک دیتابیس (مثل ری‌استارت سرور)
+// migration‌ها را دوباره اجرا نکرده و خطای «duplicate column» ندهد.
+func TestMigrateIdempotent(t *testing.T) {
+	path := filepath.Join(t.TempDir(), "mig.db")
+	for i := 1; i <= 3; i++ {
+		s, err := Open(path)
+		if err != nil {
+			t.Fatalf("open #%d: %v", i, err)
+		}
+		s.Close()
+	}
+}
+
+// TestMigrateLegacyDB سناریوی دیتابیسِ از قبل migrate‌شده ولی بدون جدول ردیابی
+// (مثل DB فعلیِ توسعه) را شبیه‌سازی می‌کند: نباید خطای duplicate column بدهد.
+func TestMigrateLegacyDB(t *testing.T) {
+	path := filepath.Join(t.TempDir(), "legacy.db")
+	s, err := Open(path)
+	if err != nil {
+		t.Fatalf("first open: %v", err)
+	}
+	// حذف جدول ردیابی ⇒ سرور فکر می‌کند هیچ migration اجرا نشده، در حالی که ستون‌ها هستند.
+	if _, err := s.DB.Exec(`DROP TABLE schema_migrations`); err != nil {
+		t.Fatal(err)
+	}
+	s.Close()
+
+	s2, err := Open(path)
+	if err != nil {
+		t.Fatalf("reopen legacy db: %v", err)
+	}
+	s2.Close()
+}
@@ -0,0 +1,43 @@
+-- اسکیمای اولیه حکم‌شو (SQLite)
+
+CREATE TABLE IF NOT EXISTS users (
+    id          INTEGER PRIMARY KEY AUTOINCREMENT,
+    mobile      TEXT    NOT NULL UNIQUE,
+    first_name  TEXT,
+    last_name   TEXT,
+    avatar      TEXT,
+    coins       INTEGER NOT NULL DEFAULT 0,
+    is_admin    INTEGER NOT NULL DEFAULT 0,
+    created_at  TEXT    NOT NULL DEFAULT (datetime('now'))
+);
+
+CREATE TABLE IF NOT EXISTS otp_tokens (
+    id          INTEGER PRIMARY KEY AUTOINCREMENT,
+    user_id     INTEGER NOT NULL,
+    token       TEXT    NOT NULL,
+    created_at  TEXT    NOT NULL DEFAULT (datetime('now')),
+    FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE
+);
+
+CREATE INDEX IF NOT EXISTS idx_otp_user ON otp_tokens(user_id);
+
+CREATE TABLE IF NOT EXISTS wallet_tx (
+    id          INTEGER PRIMARY KEY AUTOINCREMENT,
+    user_id     INTEGER NOT NULL,
+    amount      INTEGER NOT NULL,        -- مثبت=واریز، منفی=برداشت
+    reason      TEXT    NOT NULL,        -- مثل: game_win, kot, purchase
+    ref         TEXT,                    -- شناسه میز/تراکنش مرجع
+    created_at  TEXT    NOT NULL DEFAULT (datetime('now')),
+    FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE
+);
+
+CREATE INDEX IF NOT EXISTS idx_wallet_user ON wallet_tx(user_id);
+
+CREATE TABLE IF NOT EXISTS game_history (
+    id           INTEGER PRIMARY KEY AUTOINCREMENT,
+    room         TEXT    NOT NULL,
+    players_json TEXT    NOT NULL,
+    winner_team  INTEGER NOT NULL,
+    kot          INTEGER NOT NULL DEFAULT 0,
+    created_at   TEXT    NOT NULL DEFAULT (datetime('now'))
+);
@@ -0,0 +1,50 @@
+-- اقتصاد: کیف‌پول، بلیط، XP/سطح، VIP، اسکین کارت، پاداش تبلیغ، خریدهای IAP
+
+ALTER TABLE users ADD COLUMN tickets        INTEGER NOT NULL DEFAULT 0;
+ALTER TABLE users ADD COLUMN xp             INTEGER NOT NULL DEFAULT 0;
+ALTER TABLE users ADD COLUMN trophies       INTEGER NOT NULL DEFAULT 0;
+ALTER TABLE users ADD COLUMN vip_until      TEXT;            -- زمان پایان VIP (UTC) یا NULL
+ALTER TABLE users ADD COLUMN selected_card  TEXT NOT NULL DEFAULT 'simple';
+ALTER TABLE users ADD COLUMN last_daily_at  TEXT;            -- آخرین دریافت سکه روزانه
+ALTER TABLE users ADD COLUMN xp_boost_mult  INTEGER NOT NULL DEFAULT 1;
+ALTER TABLE users ADD COLUMN xp_boost_until TEXT;            -- زمان پایان بوستر XP
+
+-- ستون نوع ارز در دفتر تراکنش (coin | ticket)
+ALTER TABLE wallet_tx ADD COLUMN currency TEXT NOT NULL DEFAULT 'coin';
+
+-- اسکین کارت‌های متعلق به کاربر (کارت ساده پیش‌فرض و رایگان است)
+CREATE TABLE IF NOT EXISTS user_cards (
+    user_id    INTEGER NOT NULL,
+    card_id    TEXT    NOT NULL,
+    created_at TEXT    NOT NULL DEFAULT (datetime('now')),
+    PRIMARY KEY (user_id, card_id),
+    FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE
+);
+
+-- پاداش تبلیغ rewarded؛ token یکتا برای جلوگیری از دریافت تکراری
+CREATE TABLE IF NOT EXISTS ad_rewards (
+    id         INTEGER PRIMARY KEY AUTOINCREMENT,
+    user_id    INTEGER NOT NULL,
+    token      TEXT    NOT NULL UNIQUE,
+    amount     INTEGER NOT NULL,
+    created_at TEXT    NOT NULL DEFAULT (datetime('now')),
+    FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE
+);
+
+CREATE INDEX IF NOT EXISTS idx_ad_rewards_user ON ad_rewards(user_id, created_at);
+
+-- خریدهای درون‌برنامه‌ای (Cafe Bazaar / Myket)؛ purchase_token یکتا به ازای فروشگاه
+CREATE TABLE IF NOT EXISTS purchases (
+    id             INTEGER PRIMARY KEY AUTOINCREMENT,
+    user_id        INTEGER NOT NULL,
+    store          TEXT    NOT NULL,   -- bazaar | myket
+    product_id     TEXT    NOT NULL,   -- SKU بسته
+    purchase_token TEXT    NOT NULL,
+    status         TEXT    NOT NULL,   -- pending | verified | failed
+    coins          INTEGER NOT NULL DEFAULT 0,
+    tickets        INTEGER NOT NULL DEFAULT 0,
+    vip_days       INTEGER NOT NULL DEFAULT 0,
+    created_at     TEXT    NOT NULL DEFAULT (datetime('now')),
+    UNIQUE (store, purchase_token),
+    FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE
+);