Base code

2026-01-08 22:39:53 +07:00
parent 697115c61a
commit c35fa52117
2169 changed files with 626670 additions and 0 deletions
--- a/backend/api/user_environment.py
+++ b/backend/api/user_environment.py
@@ -0,0 +1,140 @@
+"""
+User Environment API endpoints
+Handles user-specific environment setup and management.
+"""
+
+from fastapi import APIRouter, HTTPException, Depends
+from typing import Dict, Any, Optional
+from loguru import logger
+
+from services.progressive_setup_service import ProgressiveSetupService
+from services.database import get_db_session
+from middleware.auth_middleware import get_current_user
+
+router = APIRouter(prefix="/api/user-environment", tags=["user-environment"])
+
+@router.post("/initialize")
+async def initialize_user_environment(current_user: Dict[str, Any] = Depends(get_current_user)):
+    """Initialize user environment based on onboarding progress."""
+    try:
+        user_id = str(current_user.get('id'))
+        db_session = get_db_session()
+        
+        if not db_session:
+            raise HTTPException(status_code=500, detail="Database connection failed")
+        
+        setup_service = ProgressiveSetupService(db_session)
+        result = setup_service.initialize_user_environment(user_id)
+        
+        return {
+            "message": "User environment initialized successfully",
+            "data": result
+        }
+        
+    except Exception as e:
+        logger.error(f"Error initializing user environment: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error initializing user environment: {str(e)}")
+    finally:
+        if db_session:
+            db_session.close()
+
+@router.get("/status")
+async def get_user_environment_status(current_user: Dict[str, Any] = Depends(get_current_user)):
+    """Get current user environment status."""
+    try:
+        user_id = str(current_user.get('id'))
+        db_session = get_db_session()
+        
+        if not db_session:
+            raise HTTPException(status_code=500, detail="Database connection failed")
+        
+        setup_service = ProgressiveSetupService(db_session)
+        status = setup_service.get_user_environment_status(user_id)
+        
+        return status
+        
+    except Exception as e:
+        logger.error(f"Error getting user environment status: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error getting user environment status: {str(e)}")
+    finally:
+        if db_session:
+            db_session.close()
+
+@router.post("/upgrade")
+async def upgrade_user_environment(
+    new_step: int,
+    current_user: Dict[str, Any] = Depends(get_current_user)
+):
+    """Upgrade user environment when progressing in onboarding."""
+    try:
+        user_id = str(current_user.get('id'))
+        db_session = get_db_session()
+        
+        if not db_session:
+            raise HTTPException(status_code=500, detail="Database connection failed")
+        
+        setup_service = ProgressiveSetupService(db_session)
+        result = setup_service.upgrade_user_environment(user_id, new_step)
+        
+        return {
+            "message": "User environment upgraded successfully",
+            "data": result
+        }
+        
+    except Exception as e:
+        logger.error(f"Error upgrading user environment: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error upgrading user environment: {str(e)}")
+    finally:
+        if db_session:
+            db_session.close()
+
+@router.delete("/cleanup")
+async def cleanup_user_environment(current_user: Dict[str, Any] = Depends(get_current_user)):
+    """Clean up user environment (for account deletion)."""
+    try:
+        user_id = str(current_user.get('id'))
+        db_session = get_db_session()
+        
+        if not db_session:
+            raise HTTPException(status_code=500, detail="Database connection failed")
+        
+        setup_service = ProgressiveSetupService(db_session)
+        success = setup_service.cleanup_user_environment(user_id)
+        
+        if success:
+            return {"message": "User environment cleaned up successfully"}
+        else:
+            raise HTTPException(status_code=500, detail="Failed to cleanup user environment")
+        
+    except Exception as e:
+        logger.error(f"Error cleaning up user environment: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error cleaning up user environment: {str(e)}")
+    finally:
+        if db_session:
+            db_session.close()
+
+@router.get("/workspace")
+async def get_user_workspace_info(current_user: Dict[str, Any] = Depends(get_current_user)):
+    """Get user workspace information."""
+    try:
+        user_id = str(current_user.get('id'))
+        db_session = get_db_session()
+        
+        if not db_session:
+            raise HTTPException(status_code=500, detail="Database connection failed")
+        
+        setup_service = ProgressiveSetupService(db_session)
+        workspace_manager = setup_service.workspace_manager
+        workspace = workspace_manager.get_user_workspace(user_id)
+        
+        if not workspace:
+            raise HTTPException(status_code=404, detail="User workspace not found")
+        
+        return workspace
+        
+    except Exception as e:
+        logger.error(f"Error getting user workspace: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error getting user workspace: {str(e)}")
+    finally:
+        if db_session:
+            db_session.close()