feat: auto-seed in dev mode on startup if DB empty

src/tracker/app.py

@@ -64,6 +64,15 @@ async def lifespan(app: FastAPI):
             await conn.run_sync(Base.metadata.create_all)
         logger.info("Database tables ensured.")
 
+        # Auto-seed if DB is empty (dev only)
+        async with async_session() as session:
+            result = await session.execute(select(Member).limit(1))
+            if not result.scalar_one_or_none():
+                logger.info("Empty DB detected in dev mode — running seed...")
+                from tracker.init_db import seed_dev_data
+                await seed_dev_data(session)
+                logger.info("Dev seed complete.")
+
     heartbeat_task = asyncio.create_task(heartbeat_monitor())
     yield
     heartbeat_task.cancel()

src/tracker/init_db.py

@@ -1,11 +1,11 @@
-"""Initialize database — drop all and create fresh tables + seed data."""
+"""Initialize database — dev seed data."""
 
 import asyncio
 import hashlib
 import logging
-import uuid
 
-from tracker.database import engine, async_session
+from sqlalchemy.ext.asyncio import AsyncSession
+
 from tracker.enums import AuthMethod, ChatKind, MemberRole, MemberStatus, MemberType, ProjectStatus
 from tracker.models import Base, Member, Chat, Project, ProjectMember, AgentConfig
 
@@ -16,16 +16,9 @@ def hash_password(password: str) -> str:
     return hashlib.sha256(password.encode()).hexdigest()
 
 
-async def init_db():
+async def seed_dev_data(session: AsyncSession):
-    """Drop all tables and recreate from models. Seed with admin + lobby."""
+    """Seed dev data into an existing session. Expects empty DB."""
-    async with engine.begin() as conn:
+    # Admin (owner)
-        logger.info("Dropping all tables...")
-        await conn.run_sync(Base.metadata.drop_all)
-        logger.info("Creating all tables...")
-        await conn.run_sync(Base.metadata.create_all)
-
-    async with async_session() as session:
-        # Create admin (owner)
     admin = Member(
         name="Admin",
         slug="admin",
@@ -37,7 +30,7 @@ async def init_db():
     )
     session.add(admin)
 
-        # Create coder agent
+    # Coder agent (for dev/testing)
     coder = Member(
         name="Coder",
         slug="coder",
@@ -48,11 +41,9 @@ async def init_db():
         status=MemberStatus.OFFLINE,
     )
     session.add(coder)
-
-        # Flush to get member IDs
     await session.flush()
 
-        # Create team-board project
+    # Team-board project
     project = Project(
         name="Team Board",
         slug="team-board",
@@ -62,39 +53,35 @@ async def init_db():
     session.add(project)
     await session.flush()
 
-        # Create project chat
+    # Project chat
-        project_chat = Chat(
+    project_chat = Chat(kind=ChatKind.PROJECT, project_id=project.id)
-            kind=ChatKind.PROJECT,
-            project_id=project.id,
-        )
     session.add(project_chat)
 
-        # Create project members
+    # Project members
-        admin_member = ProjectMember(
+    session.add(ProjectMember(project_id=project.id, member_id=admin.id, role=MemberRole.OWNER))
-            project_id=project.id,
+    session.add(ProjectMember(project_id=project.id, member_id=coder.id, role=MemberRole.MEMBER))
-            member_id=admin.id,
-            role=MemberRole.OWNER,
-        )
-        session.add(admin_member)
 
-        coder_member = ProjectMember(
+    # Lobby chat
-            project_id=project.id,
+    lobby = Chat(kind=ChatKind.LOBBY, project_id=None)
-            member_id=coder.id,
-            role=MemberRole.MEMBER,
-        )
-        session.add(coder_member)
-
-        # Create lobby chat
-        lobby = Chat(
-            kind=ChatKind.LOBBY,
-            project_id=None,
-        )
     session.add(lobby)
 
     await session.commit()
-        logger.info("Seed data created: admin user, coder agent, team-board project (id=%s), lobby chat (id=%s)", project.id, lobby.id)
+    logger.info("Dev seed: admin, coder, project team-board (id=%s), lobby", project.id)
+
+
+async def reset_db():
+    """Drop all tables, recreate, and seed. For dev use only."""
+    from tracker.database import engine, async_session
+
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.drop_all)
+        await conn.run_sync(Base.metadata.create_all)
+    logger.info("Tables recreated.")
+
+    async with async_session() as session:
+        await seed_dev_data(session)
 
 
 if __name__ == "__main__":
     logging.basicConfig(level=logging.INFO)
-    asyncio.run(init_db())
+    asyncio.run(reset_db())