Add subcription feature

services/auth/app/api/users.py

@@ -2,7 +2,7 @@
 User management API routes
 """
 
-from fastapi import APIRouter, Depends, HTTPException, status, BackgroundTasks, Path
+from fastapi import APIRouter, Depends, HTTPException, status, BackgroundTasks, Path, Body
 from sqlalchemy.ext.asyncio import AsyncSession
 from typing import Dict, Any
 import structlog
@@ -223,7 +223,9 @@ async def get_user_by_id(
             created_at=user.created_at,
             last_login=user.last_login,
             role=user.role,
-            tenant_id=None
+            tenant_id=None,
+            payment_customer_id=user.payment_customer_id,
+            default_payment_method_id=user.default_payment_method_id
         )
 
     except HTTPException:
@@ -481,3 +483,71 @@ async def get_user_activity(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail="Failed to get user activity information"
         )
+
+
+@router.patch("/api/v1/auth/users/{user_id}/tenant")
+async def update_user_tenant(
+    user_id: str = Path(..., description="User ID"),
+    tenant_data: Dict[str, Any] = Body(..., description="Tenant data containing tenant_id"),
+    db: AsyncSession = Depends(get_db)
+):
+    """
+    Update user's tenant_id after tenant registration
+
+    This endpoint is called by the tenant service after a user creates their tenant.
+    It links the user to their newly created tenant.
+    """
+    try:
+        # Log the incoming request data for debugging
+        logger.debug("Received tenant update request",
+                    user_id=user_id,
+                    tenant_data=tenant_data)
+
+        tenant_id = tenant_data.get("tenant_id")
+
+        if not tenant_id:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="tenant_id is required"
+            )
+
+        logger.info("Updating user tenant_id",
+                   user_id=user_id,
+                   tenant_id=tenant_id)
+
+        user_service = UserService(db)
+        user = await user_service.get_user_by_id(uuid.UUID(user_id))
+
+        if not user:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail="User not found"
+            )
+
+        # Update user's tenant_id
+        user.tenant_id = uuid.UUID(tenant_id)
+        user.updated_at = datetime.now(timezone.utc)
+
+        await db.commit()
+        await db.refresh(user)
+
+        logger.info("Successfully updated user tenant_id",
+                   user_id=user_id,
+                   tenant_id=tenant_id)
+
+        return {
+            "success": True,
+            "user_id": str(user.id),
+            "tenant_id": str(user.tenant_id)
+        }
+
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error("Failed to update user tenant_id",
+                    user_id=user_id,
+                    error=str(e))
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail="Failed to update user tenant_id"
+        )