Merge pull request #26 from lbedner/db-health-check-integration

DB Health Check Integration

Author: lbedner

…cookiecutter.project_slug}}/.env.example …kiecutter.project_slug}}/.env.example.j2aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/.env.example renamed to aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/.env.example.j2 aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/.env.example renamed to aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/.env.example.j2

@@ -31,27 +31,28 @@ AEGIS_STACK_VERSION=dev
 # AEGIS_STACK_ENV_FILE=.env
 
 # =============================================================================
-# FUTURE COMPONENT SETTINGS
+# COMPONENT SETTINGS
 # =============================================================================
-# These will be uncommented when components are added via `aegis add <component>`
 
-# Database settings (when database component is added)
-# DATABASE_URL=postgresql://user:password@localhost:5432/aegis_stack
-# DATABASE_HOST=localhost
-# DATABASE_PORT=5432
-# DATABASE_NAME=aegis_stack
-# DATABASE_USER=aegis_user
-# DATABASE_PASSWORD=your_secure_password
+{% if cookiecutter.include_database == "yes" %}
+# Database settings
+# DATABASE_URL=sqlite:///./data/app.db  # SQLite database (default)
+# DATABASE_ENGINE_ECHO=false  # Set to true to log all SQL queries (debugging)
+{% endif %}
 
-# Redis settings (when cache component is added)  
+{% if cookiecutter.include_redis == "yes" or cookiecutter.include_cache == "yes" %}
+# Redis settings
 # REDIS_URL=redis://localhost:6379
 # REDIS_HOST=localhost
 # REDIS_PORT=6379
 # REDIS_PASSWORD=
+{% endif %}
 
-# Scheduler settings (when scheduler component is added)
+{% if cookiecutter.include_scheduler == "yes" %}
+# Scheduler settings
 # SCHEDULER_TIMEZONE=UTC
 # SCHEDULER_MAX_WORKERS=10
+{% endif %}
 
 # API Keys and Secrets (add as needed)
 # SECRET_KEY=your-super-secret-key-here

aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/app/components/backend/startup/component_health.py

@@ -180,4 +180,11 @@ async def startup_hook() -> None:
     logger.info("Cache component health check registered")
     {%- endif %}
 
+    {%- if cookiecutter.include_database == "yes" %}
+    # Register database health check (SQLite connectivity and basic operations)
+    from app.services.system.health import check_database_health
+    register_health_check("database", check_database_health)
+    logger.info("Database component health check registered")
+    {%- endif %}
+
     logger.info("✅ Component health detection complete")

aegis/templates/cookiecutter-aegis-project/{{cookiecutter.project_slug}}/app/components/backend/startup/component_health.py.j2

@@ -562,6 +562,103 @@ async def check_cache_health() -> ComponentStatus:
 {% endif %}
 
 
+{% if cookiecutter.include_database == "yes" %}
+async def check_database_health() -> ComponentStatus:
+    """
+    Check database connectivity and basic functionality.
+
+    Returns:
+        ComponentStatus indicating database health
+    """
+    try:
+        # Import db_session from generated project
+        from app.core.db import db_session
+        from pathlib import Path
+        
+        # Check if database file exists for SQLite
+        db_url = settings.DATABASE_URL
+        if db_url.startswith("sqlite:///"):
+            # Extract path from SQLite URL
+            db_path = db_url.replace("sqlite:///", "").lstrip("./")
+            
+            # Check if database file exists
+            if not Path(db_path).exists():
+                return ComponentStatus(
+                    name="database",
+                    status=ComponentStatusType.WARNING,
+                    message="Database not initialized - file does not exist",
+                    response_time_ms=None,
+                    metadata={
+                        "implementation": "sqlite",
+                        "database_exists": False,
+                        "expected_path": db_path,
+                        "url": settings.DATABASE_URL,
+                        "recommendation": "Run database migrations or create database file",
+                    },
+                )
+
+        # Test database connection with simple query
+        with db_session(autocommit=False) as session:
+            # Execute a simple query to test connectivity
+            from sqlalchemy import text
+            session.execute(text("SELECT 1"))
+            # No need to commit since we're just testing connectivity
+
+        return ComponentStatus(
+            name="database",
+            status=ComponentStatusType.HEALTHY,
+            message="Database connection successful",
+            response_time_ms=None,  # Will be set by caller
+            metadata={
+                "implementation": "sqlite",
+                "url": settings.DATABASE_URL,
+                "database_exists": True,
+                "engine_echo": settings.DATABASE_ENGINE_ECHO,
+            },
+        )
+
+    except ImportError:
+        return ComponentStatus(
+            name="database",
+            status=ComponentStatusType.UNHEALTHY,
+            message="Database module not available",
+            response_time_ms=None,
+            metadata={
+                "implementation": "sqlite",
+                "error": "Database module not imported or configured",
+            },
+        )
+    except Exception as e:
+        # Check if it's a file not found error
+        error_str = str(e).lower()
+        if "unable to open database file" in error_str or "no such file" in error_str:
+            return ComponentStatus(
+                name="database",
+                status=ComponentStatusType.WARNING,
+                message="Database file not accessible",
+                response_time_ms=None,
+                metadata={
+                    "implementation": "sqlite",
+                    "url": settings.DATABASE_URL,
+                    "error": str(e),
+                    "recommendation": "Check database file path and permissions",
+                },
+            )
+        
+        return ComponentStatus(
+            name="database",
+            status=ComponentStatusType.UNHEALTHY,
+            message=f"Database connection failed: {str(e)}",
+            response_time_ms=None,
+            metadata={
+                "implementation": "sqlite",
+                "url": settings.DATABASE_URL,
+                "error": str(e),
+            },
+        )
+{% endif %}
+
+
 {% if cookiecutter.include_worker == "yes" %}
 async def check_worker_health() -> ComponentStatus:
     """