Update app.py

app.py

@@ -43,7 +43,7 @@ app.add_middleware(
 
 # Load environment variables
 APP_ID = os.getenv("GITHUB_APP_ID")
-PRIVATE_KEY_PATH = os.getenv("GITHUB_PRIVATE_KEY_PATH")
+PRIVATE_KEY_PATH = os.getenv("GITHUB_PRIVATE_KEY_PATH", "private-key.pem")
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_SECRET_KEY = os.getenv("SUPABASE_SECRET_KEY")
 DEFAULT_FIREBASE_ID = os.getenv("DEFAULT_FIREBASE_ID", "JDfZoVuGJhdLBEvR7rVCQKBG9r02")
@@ -125,7 +125,7 @@ def save_installation(org: str, installation_id: int):
 
 
 def get_installation_id(org: str) -> Optional[int]:
-    """Get installation ID for an organization"""
+    """Get installation ID for an organization (legacy from file system)"""
     installations = load_installations()
     org_data = installations.get(org)
     
@@ -134,6 +134,22 @@ def get_installation_id(org: str) -> Optional[int]:
     return None
 
 
+def get_installation_id_for_user(firebase_id: str) -> Optional[int]:
+    """Get installation ID for a user from Supabase Users table"""
+    if not supabase:
+        logger.error("Supabase not configured")
+        return None
+    
+    try:
+        result = supabase.table("Users").select("installation_id").eq("firebase_id", firebase_id).limit(1).execute()
+        if result.data and result.data[0].get("installation_id"):
+            return result.data[0]["installation_id"]
+    except Exception as e:
+        logger.error(f"Failed to get installation_id for user {firebase_id}: {str(e)}")
+    
+    return None
+
+
 def ensure_user_exists(firebase_id: str):
     if not supabase:
         raise HTTPException(status_code=500, detail="Supabase not configured")
@@ -437,6 +453,13 @@ async def github_callback(request: Request, background_tasks: BackgroundTasks):
         save_installation(org_name, installation_id)
         sync_result = None
         if supabase:
+            # Save installation_id to Users table
+            try:
+                supabase.table("Users").update({"installation_id": installation_id}).eq("firebase_id", firebase_id).execute()
+                logger.info(f"Saved installation_id {installation_id} for user {firebase_id}")
+            except Exception as e:
+                logger.error(f"Failed to save installation_id to Users table: {str(e)}")
+            
             # Run heavy sync in background so callback returns immediately
             background_tasks.add_task(
                 sync_installation_to_supabase,
@@ -563,9 +586,9 @@ def get_commits(data: InsightsRequest):
     if not supabase:
         raise HTTPException(status_code=500, detail="Supabase not configured")
     
-    installation_id = get_installation_id(data.org)
+    installation_id = get_installation_id_for_user(data.firebase_id)
     if not installation_id:
-        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+        raise HTTPException(status_code=404, detail=f"No GitHub installation found for user {data.firebase_id}")
     
     try:
         # Fetch commits from GitHub
@@ -624,9 +647,9 @@ def get_pull_requests(data: InsightsRequest):
     if not supabase:
         raise HTTPException(status_code=500, detail="Supabase not configured")
     
-    installation_id = get_installation_id(data.org)
+    installation_id = get_installation_id_for_user(data.firebase_id)
     if not installation_id:
-        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+        raise HTTPException(status_code=404, detail=f"No GitHub installation found for user {data.firebase_id}")
     
     try:
         # Fetch pull requests from GitHub
@@ -697,9 +720,9 @@ def get_issues(data: InsightsRequest):
     if not supabase:
         raise HTTPException(status_code=500, detail="Supabase not configured")
     
-    installation_id = get_installation_id(data.org)
+    installation_id = get_installation_id_for_user(data.firebase_id)
     if not installation_id:
-        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+        raise HTTPException(status_code=404, detail=f"No GitHub installation found for user {data.firebase_id}")
     
     try:
         # Fetch issues from GitHub
@@ -767,9 +790,9 @@ def sync_github_to_supabase(data: GitHubSyncRequest):
     if not supabase:
         raise HTTPException(status_code=500, detail="Supabase not configured")
 
-    installation_id = get_installation_id(data.org)
+    installation_id = get_installation_id_for_user(data.firebase_id)
     if not installation_id:
-        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+        raise HTTPException(status_code=404, detail=f"No GitHub installation found for user {data.firebase_id}")
 
     try:
         synced = sync_installation_to_supabase(data.firebase_id, data.org, installation_id)
@@ -792,9 +815,9 @@ def get_contributors(data: InsightsRequest):
     if not supabase:
         raise HTTPException(status_code=500, detail="Supabase not configured")
     
-    installation_id = get_installation_id(data.org)
+    installation_id = get_installation_id_for_user(data.firebase_id)
     if not installation_id:
-        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+        raise HTTPException(status_code=404, detail=f"No GitHub installation found for user {data.firebase_id}")
     
     try:
         # Fetch contributors from GitHub
@@ -848,9 +871,9 @@ def get_repository_overview(data: InsightsRequest):
     if not github_auth:
         raise HTTPException(status_code=500, detail="GitHub App not configured")
     
-    installation_id = get_installation_id(data.org)
+    installation_id = get_installation_id_for_user(data.firebase_id)
     if not installation_id:
-        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+        raise HTTPException(status_code=404, detail=f"No GitHub installation found for user {data.firebase_id}")
     
     try:
         token = token_manager.get_token(installation_id)
@@ -890,9 +913,9 @@ def get_repository_activity(data: InsightsRequest):
     if not github_auth:
         raise HTTPException(status_code=500, detail="GitHub App not configured")
     
-    installation_id = get_installation_id(data.org)
+    installation_id = get_installation_id_for_user(data.firebase_id)
     if not installation_id:
-        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+        raise HTTPException(status_code=404, detail=f"No GitHub installation found for user {data.firebase_id}")
     
     try:
         logger.info(f"Fetching activity for {data.org}/{data.repo}")
@@ -926,11 +949,11 @@ def test_sync(data: GitHubSyncRequest):
     """
     logger.info(f"Test sync called for firebase_id={data.firebase_id}, org={data.org}")
     
-    installation_id = get_installation_id(data.org)
+    installation_id = get_installation_id_for_user(data.firebase_id)
     if not installation_id:
         return {
             "success": False,
-            "error": f"No installation found for {data.org}",
+            "error": f"No GitHub installation found for user {data.firebase_id}",
             "available_orgs": list(load_installations().keys())
         }
 
@@ -952,4 +975,7 @@ def test_sync(data: GitHubSyncRequest):
         }
 
 
+# ============================================================================
+# RUN SERVER
+# ============================================================================