Add forecasting service

services/forecasting/app/core/auth.py

@@ -1,22 +1,48 @@
+# ================================================================
+# services/forecasting/app/core/auth.py
+# ================================================================
 """
-Authentication configuration for forecasting service
+Authentication utilities for forecasting service
 """
 
-from shared.auth.jwt_handler import JWTHandler
-from shared.auth.decorators import require_auth, require_role
-from app.core.config import settings
+import structlog
+from fastapi import HTTPException, status, Request
+from typing import Dict, Any, Optional
 
-# Initialize JWT handler
-jwt_handler = JWTHandler(
-    secret_key=settings.JWT_SECRET_KEY,
-    algorithm=settings.JWT_ALGORITHM,
-    access_token_expire_minutes=settings.JWT_ACCESS_TOKEN_EXPIRE_MINUTES
-)
+logger = structlog.get_logger()
 
-# Export commonly used functions
-verify_token = jwt_handler.verify_token
-create_access_token = jwt_handler.create_access_token
-get_current_user = jwt_handler.get_current_user
+async def get_current_user_from_headers(request: Request) -> Dict[str, Any]:
+    """
+    Get current user from gateway headers
+    Gateway middleware adds user context to headers after JWT verification
+    """
+    
+    try:
+        # Extract user information from headers set by API Gateway
+        user_id = request.headers.get("X-User-ID")
+        user_email = request.headers.get("X-User-Email")
+        tenant_id = request.headers.get("X-Tenant-ID")
+        user_roles = request.headers.get("X-User-Roles", "").split(",")
+        
+        if not user_id or not tenant_id:
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                detail="Authentication required"
+            )
+        
+        return {
+            "user_id": user_id,
+            "email": user_email,
+            "tenant_id": tenant_id,
+            "roles": [role.strip() for role in user_roles if role.strip()]
+        }
+        
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error("Error extracting user from headers", error=str(e))
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Invalid authentication"
+        )
 
-# Export decorators
-__all__ = ['verify_token', 'create_access_token', 'get_current_user', 'require_auth', 'require_role']

services/forecasting/app/core/database.py

@@ -1,12 +1,73 @@
+# ================================================================
+# services/forecasting/app/core/database.py
+# ================================================================
 """
 Database configuration for forecasting service
 """
 
-from shared.database.base import DatabaseManager
+import structlog
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
+from sqlalchemy.pool import NullPool
+from sqlalchemy import text
+
 from app.core.config import settings
+from shared.database.base import Base
 
-# Initialize database manager
-database_manager = DatabaseManager(settings.DATABASE_URL)
+logger = structlog.get_logger()
 
-# Alias for convenience
-get_db = database_manager.get_db
+# Create async engine
+engine = create_async_engine(
+    settings.DATABASE_URL,
+    poolclass=NullPool,
+    echo=settings.DEBUG,
+    future=True
+)
+
+# Create session factory
+AsyncSessionLocal = async_sessionmaker(
+    engine,
+    class_=AsyncSession,
+    expire_on_commit=False,
+    autoflush=False,
+    autocommit=False
+)
+
+class DatabaseManager:
+    """Database management operations"""
+    
+    async def create_tables(self):
+        """Create database tables"""
+        async with engine.begin() as conn:
+            await conn.run_sync(Base.metadata.create_all)
+        logger.info("Forecasting database tables created successfully")
+    
+    async def get_session(self) -> AsyncSession:
+        """Get database session"""
+        async with AsyncSessionLocal() as session:
+            try:
+                yield session
+                await session.commit()
+            except Exception as e:
+                await session.rollback()
+                logger.error(f"Database session error: {e}")
+                raise
+            finally:
+                await session.close()
+
+# Global database manager instance
+database_manager = DatabaseManager()
+
+async def get_db() -> AsyncSession:
+    """Database dependency"""
+    async for session in database_manager.get_session():
+        yield session
+
+async def get_db_health() -> bool:
+    """Check database health"""
+    try:
+        async with AsyncSessionLocal() as session:
+            await session.execute(text("SELECT 1"))
+            return True
+    except Exception as e:
+        logger.error(f"Database health check failed: {e}")
+        return False