feat: add frame

internal/economy/frame.go

@@ -0,0 +1,118 @@
+package economy
+
+import "context"
+
+// Frame یک قابِ آواتار (کازمتیک). Owned یعنی قابلِ انتخاب. سبکِ رنگی سمتِ کلاینت است.
+type Frame struct {
+	ID         string `json:"id"`
+	Title      string `json:"title"`
+	PriceCoins int64  `json:"price_coins"`
+	VIP        bool   `json:"vip"`
+	Owned      bool   `json:"owned"`
+}
+
+// Frames قاب‌های فعال را با وضعیتِ مالکیتِ کاربر برمی‌گرداند.
+func (s *Service) Frames(ctx context.Context, userID int64) ([]Frame, error) {
+	rows, err := s.db.QueryContext(ctx,
+		`SELECT id, title, price_coins, vip FROM avatar_frames WHERE enabled = 1 ORDER BY sort`)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var frames []Frame
+	for rows.Next() {
+		var f Frame
+		var vip int
+		if err := rows.Scan(&f.ID, &f.Title, &f.PriceCoins, &vip); err != nil {
+			return nil, err
+		}
+		f.VIP = vip == 1
+		frames = append(frames, f)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+
+	owned, err := s.ownedFrames(ctx, userID)
+	if err != nil {
+		return nil, err
+	}
+	isVIP := s.isVIP(ctx, userID)
+	for i := range frames {
+		f := &frames[i]
+		f.Owned = (f.PriceCoins == 0 && !f.VIP) || (f.VIP && isVIP) || owned[f.ID]
+	}
+	return frames, nil
+}
+
+func (s *Service) ownedFrames(ctx context.Context, userID int64) (map[string]bool, error) {
+	rows, err := s.db.QueryContext(ctx,
+		`SELECT frame_id FROM user_frames WHERE user_id = ?`, userID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	owned := map[string]bool{}
+	for rows.Next() {
+		var id string
+		if err := rows.Scan(&id); err != nil {
+			return nil, err
+		}
+		owned[id] = true
+	}
+	return owned, rows.Err()
+}
+
+// BuyFrame یک قاب را با سکه باز می‌کند.
+func (s *Service) BuyFrame(ctx context.Context, userID int64, frameID string) error {
+	var price int64
+	var vip, enabled int
+	err := s.db.QueryRowContext(ctx,
+		`SELECT price_coins, vip, enabled FROM avatar_frames WHERE id = ?`, frameID).
+		Scan(&price, &vip, &enabled)
+	if err != nil || enabled != 1 {
+		return ErrNotFound
+	}
+	tx, err := s.db.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	var exists int
+	_ = tx.QueryRowContext(ctx,
+		`SELECT 1 FROM user_frames WHERE user_id = ? AND frame_id = ?`, userID, frameID).Scan(&exists)
+	if exists == 1 {
+		return ErrAlreadyOwned
+	}
+	if price > 0 {
+		if err := adjustTx(ctx, tx, userID, CurrencyCoin, -price, "buy_frame", frameID); err != nil {
+			return err
+		}
+	}
+	if _, err := tx.ExecContext(ctx,
+		`INSERT INTO user_frames (user_id, frame_id) VALUES (?, ?)`, userID, frameID); err != nil {
+		return err
+	}
+	return tx.Commit()
+}
+
+// SelectFrame قابِ انتخابیِ کاربر را تنظیم می‌کند (باید مالکش باشد یا رایگان/VIP).
+func (s *Service) SelectFrame(ctx context.Context, userID int64, frameID string) error {
+	frames, err := s.Frames(ctx, userID)
+	if err != nil {
+		return err
+	}
+	for _, f := range frames {
+		if f.ID == frameID {
+			if !f.Owned {
+				return ErrNotOwned
+			}
+			_, err := s.db.ExecContext(ctx,
+				`UPDATE users SET selected_frame = ? WHERE id = ?`, frameID, userID)
+			return err
+		}
+	}
+	return ErrNotFound
+}