ensure can reach stackcoin

Author: jackharrhy

lucky_pot.py

@@ -1,4 +1,6 @@
 import asyncio
+import random
+
 import hikari
 from loguru import logger
 
@@ -17,6 +19,14 @@
 from luckypot.discord.commands import register_commands
 from luckypot.discord.scheduler import run_daily_draw_loop
 
+# Startup retry parameters for reaching StackCoin. Tuned so that luckypot waits
+# roughly long enough for stackcoin to come up after a simultaneous restart
+# (~16s in practice), but gives up well inside Docker's automatic-restart window
+# so the whole container gets recycled by `restart: unless-stopped`.
+STACKCOIN_CONNECT_MAX_ATTEMPTS = 12
+STACKCOIN_CONNECT_BASE_DELAY = 1.0
+STACKCOIN_CONNECT_MAX_DELAY = 30.0
+
 
 logger.add("lucky_pot.log", rotation="1 day", retention="7 days", level="INFO")
 
@@ -43,6 +53,44 @@ def _task_done_callback(task: asyncio.Task) -> None:
         logger.error(f"Background task {task.get_name()} failed: {exc!r}")
 
 
+async def _fetch_stackcoin_discord_id_or_die() -> str:
+    """Retry ``fetch_stackcoin_discord_id`` against a not-yet-ready StackCoin.
+
+    stackcoin-python normalises transport faults into ``StackCoinError`` (since
+    v0.1.6), so we only catch that. If we exhaust all attempts we raise — the
+    caller in :func:`on_started` shuts the whole bot down by closing the hikari
+    bot, which exits the process non-zero and lets Docker's
+    ``restart: unless-stopped`` policy recycle the container. By the next boot
+    StackCoin has been up for ages.
+    """
+    last_exc: BaseException | None = None
+    for attempt in range(1, STACKCOIN_CONNECT_MAX_ATTEMPTS + 1):
+        try:
+            discord_id = await stk.fetch_stackcoin_discord_id()
+            if discord_id is not None:
+                return discord_id
+            # Defence in depth: SDK returned None without raising — treat as a
+            # soft miss and retry, since the only None path in stk.py is the
+            # StackCoinError swallow at stk.py:40-42.
+            last_exc = RuntimeError("fetch_stackcoin_discord_id returned None")
+        except stackcoin.StackCoinError as e:
+            last_exc = e
+        delay = min(
+            STACKCOIN_CONNECT_MAX_DELAY,
+            STACKCOIN_CONNECT_BASE_DELAY * (2 ** (attempt - 1)),
+        ) + random.random()
+        logger.warning(
+            f"StackCoin not ready (attempt {attempt}/"
+            f"{STACKCOIN_CONNECT_MAX_ATTEMPTS}), retrying in {delay:.1f}s: "
+            f"{last_exc!r}"
+        )
+        await asyncio.sleep(delay)
+    raise RuntimeError(
+        f"Could not reach StackCoin after {STACKCOIN_CONNECT_MAX_ATTEMPTS} "
+        f"attempts; refusing to start half-functional. Last error: {last_exc!r}"
+    )
+
+
 @bot.listen()
 async def on_started(_event: hikari.StartedEvent) -> None:
     global _gateway
@@ -64,7 +112,12 @@ def persist_event_id(event_id: int) -> None:
         finally:
             c.close()
 
-    await stk.fetch_stackcoin_discord_id()
+    try:
+        await _fetch_stackcoin_discord_id_or_die()
+    except RuntimeError as e:
+        logger.error(str(e))
+        await bot.close()
+        return
 
     _gateway = stackcoin.Gateway(
         ws_url=settings.stackcoin_ws_url,

pyproject.toml

@@ -9,7 +9,7 @@ dependencies = [
     "hikari-lightbulb>=3.1.1",
     "loguru>=0.7.3",
     "pydantic-settings>=2.0",
-    "stackcoin>=0.1.5",
+    "stackcoin>=0.1.6",
 ]
 
 [build-system]