From bb932e68d3fb629d7fa817ada73fc307cc162f88 Mon Sep 17 00:00:00 2001
From: Mikkel Georgsen <msgeorgsen@gmail.com>
Date: Fri, 16 Jan 2026 15:14:56 +0000
Subject: [PATCH] feat(01-03): create database module with async session
 management

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
---
 src/moai/core/database.py | 106 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 106 insertions(+)
 create mode 100644 src/moai/core/database.py

diff --git a/src/moai/core/database.py b/src/moai/core/database.py
new file mode 100644
index 0000000..e3b14ec
--- /dev/null
+++ b/src/moai/core/database.py
@@ -0,0 +1,106 @@
+"""Database session management for MoAI.
+
+Provides async session management using SQLAlchemy 2.0 async support.
+
+Usage pattern:
+    from moai.core.database import init_db, create_tables, get_session, close_db
+
+    # Initialize on startup
+    init_db("sqlite+aiosqlite:///./moai.db")
+    await create_tables()
+
+    # Use sessions for database operations
+    async with get_session() as session:
+        project = Project(name="My Project")
+        session.add(project)
+        # Auto-commits on context exit, rollback on exception
+
+    # Cleanup on shutdown
+    await close_db()
+"""
+
+from collections.abc import AsyncGenerator
+from contextlib import asynccontextmanager
+
+from sqlalchemy.ext.asyncio import (
+    AsyncEngine,
+    AsyncSession,
+    async_sessionmaker,
+    create_async_engine,
+)
+
+# Module-level state
+DATABASE_URL: str = "sqlite+aiosqlite:///./moai.db"
+engine: AsyncEngine | None = None
+async_session_factory: async_sessionmaker[AsyncSession] | None = None
+
+
+def init_db(url: str | None = None) -> None:
+    """Initialize the database engine and session factory.
+
+    Args:
+        url: Database URL. Defaults to module-level DATABASE_URL if not provided.
+             Use "sqlite+aiosqlite:///:memory:" for in-memory testing.
+    """
+    global engine, async_session_factory, DATABASE_URL
+
+    if url is not None:
+        DATABASE_URL = url
+
+    engine = create_async_engine(DATABASE_URL, echo=False)
+    async_session_factory = async_sessionmaker(engine, expire_on_commit=False)
+
+
+async def create_tables() -> None:
+    """Create all database tables defined in models.
+
+    Must be called after init_db(). Creates tables if they don't exist.
+    """
+    from moai.core.models import Base
+
+    if engine is None:
+        raise RuntimeError("Database not initialized. Call init_db() first.")
+
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
+
+
+@asynccontextmanager
+async def get_session() -> AsyncGenerator[AsyncSession, None]:
+    """Async context manager providing a database session.
+
+    Yields:
+        AsyncSession: Database session for operations.
+
+    The session auto-commits on successful context exit.
+    On exception, the session is rolled back automatically.
+
+    Example:
+        async with get_session() as session:
+            project = Project(name="Test")
+            session.add(project)
+            # Commits automatically on exit
+    """
+    if async_session_factory is None:
+        raise RuntimeError("Database not initialized. Call init_db() first.")
+
+    async with async_session_factory() as session:
+        try:
+            yield session
+            await session.commit()
+        except Exception:
+            await session.rollback()
+            raise
+
+
+async def close_db() -> None:
+    """Dispose of the database engine and release connections.
+
+    Should be called during application shutdown.
+    """
+    global engine, async_session_factory
+
+    if engine is not None:
+        await engine.dispose()
+        engine = None
+        async_session_factory = None