feat: backfill preauth requests for existing auto-enter users on startup (temporary)

Author: jackharrhy

lucky_pot.py

@@ -5,7 +5,7 @@
 from luckypot import db
 from luckypot.config import settings
 import stackcoin
-from luckypot.game import on_request_accepted, on_request_denied
+from luckypot.game import on_request_accepted, on_request_denied, backfill_preauths_for_auto_enter_users
 from luckypot import stk
 from luckypot.stk import get_client as get_stk_client
 from luckypot.discord.bot import (
@@ -87,6 +87,14 @@ async def handle_denied(event: stackcoin.RequestDeniedEvent):
     background_tasks.append(gateway_task)
     logger.info("StackCoin gateway started")
 
+    # TEMPORARY: backfill preauths for existing auto-enter users.
+    # Remove once all users have been sent preauth requests.
+    backfill_task = asyncio.create_task(
+        backfill_preauths_for_auto_enter_users(),
+        name="preauth-backfill",
+    )
+    backfill_task.add_done_callback(_task_done_callback)
+
     draw_task = asyncio.create_task(
         run_daily_draw_loop(
             announce=announce,

luckypot/db.py

@@ -320,6 +320,12 @@ def get_auto_enter_users(conn, guild_id: str) -> list[str]:
     return [row["discord_id"] for row in cursor.fetchall()]
 
 
+def get_all_auto_enter_users(conn) -> list[dict]:
+    """Return all auto-enter users across all guilds."""
+    cursor = conn.execute("SELECT discord_id, guild_id FROM auto_enter_users")
+    return [{"discord_id": row["discord_id"], "guild_id": row["guild_id"]} for row in cursor.fetchall()]
+
+
 def get_auto_enter_status(conn, discord_id: str, guild_id: str) -> bool:
     """Return True if the user is opted in to auto-enter for this guild."""
     cursor = conn.execute(

luckypot/game.py

@@ -470,6 +470,69 @@ async def daily_pot_draw(
                 conn.close()
 
 
+# ── TEMPORARY: preauth migration for existing auto-enter users ──────────
+# Remove this function and its call in lucky_pot.py once all existing
+# auto-enter users have been sent preauth requests.
+
+
+async def backfill_preauths_for_auto_enter_users() -> None:
+    """Request preauths for existing auto-enter users who don't have one yet.
+
+    TEMPORARY — safe to remove after rollout once preauths have been sent.
+    """
+    conn = db.get_connection()
+    try:
+        all_users = db.get_all_auto_enter_users(conn)
+    finally:
+        conn.close()
+
+    if not all_users:
+        logger.info("Preauth backfill: no auto-enter users found")
+        return
+
+    # Deduplicate by discord_id (a user may be auto-entered in multiple guilds,
+    # but preauths are per-user not per-guild)
+    seen: set[str] = set()
+    unique_users: list[str] = []
+    for entry in all_users:
+        if entry["discord_id"] not in seen:
+            seen.add(entry["discord_id"])
+            unique_users.append(entry["discord_id"])
+
+    logger.info(f"Preauth backfill: checking {len(unique_users)} auto-enter user(s)")
+    requested = 0
+
+    for discord_id in unique_users:
+        try:
+            stk_user = await stk.get_user_by_discord_id(discord_id)
+            if stk_user is None:
+                continue
+
+            preauths = await stk.get_preauths(user_id=stk_user["id"])
+            has_active_or_pending = any(
+                p.get("status") in ("active", "pending") for p in preauths
+            )
+            if has_active_or_pending:
+                continue
+
+            result = await stk.create_preauth(
+                user_id=stk_user["id"],
+                max_amount=10,
+                window_hours=24,
+            )
+            if result:
+                requested += 1
+                logger.info(
+                    f"Preauth backfill: requested preauth for discord_id={discord_id} (user {stk_user['id']})"
+                )
+        except Exception:
+            logger.exception(
+                f"Preauth backfill: failed for discord_id={discord_id}"
+            )
+
+    logger.info(f"Preauth backfill: done, requested {requested} new preauth(s)")
+
+
 async def on_request_accepted(
     event_data: RequestAcceptedData, announce: RawAnnounceFn = None
 ):