Create main.py

main.py

@@ -0,0 +1,472 @@
+"""
+Codesage Backend API
+FastAPI server for GitHub App integration
+Connects desktop app to GitHub securely
+"""
+
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from typing import Optional, List, Dict
+import json
+import os
+from pathlib import Path
+from dotenv import load_dotenv
+
+from github_api import GitHubAppAuth, GitHubInsights, TokenManager
+
+# Load environment variables from .env file
+load_dotenv()
+
+
+# Initialize FastAPI app
+app = FastAPI(
+    title="Codesage Backend API",
+    description="Enterprise-grade GitHub insights for development teams",
+    version="1.0.0"
+)
+
+# CORS configuration (allow desktop app to connect)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # In production, specify your desktop app origins
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Load environment variables
+APP_ID = os.getenv("GITHUB_APP_ID")
+PRIVATE_KEY_PATH = os.getenv("GITHUB_PRIVATE_KEY_PATH", "private-key.pem")
+
+# Initialize GitHub App authentication
+if APP_ID and os.path.exists(PRIVATE_KEY_PATH):
+    with open(PRIVATE_KEY_PATH, "r") as f:
+        private_key = f.read()
+    
+    github_auth = GitHubAppAuth(APP_ID, private_key)
+    token_manager = TokenManager(github_auth)
+else:
+    github_auth = None
+    token_manager = None
+    print("⚠️ WARNING: GitHub App credentials not configured")
+
+# Simple JSON storage for installations
+INSTALLATIONS_FILE = "installations.json"
+
+
+# ============================================================================
+# DATA MODELS
+# ============================================================================
+
+class InstallationCallback(BaseModel):
+    """Data received from GitHub installation callback"""
+    installation_id: int
+    org: str
+    setup_action: Optional[str] = None
+
+
+class InsightsRequest(BaseModel):
+    """Request for repository insights from desktop app"""
+    org: str
+    repo: str
+
+
+class TokenRefreshRequest(BaseModel):
+    """Request to refresh installation token"""
+    installation_id: int
+
+
+# ============================================================================
+# UTILITY FUNCTIONS
+# ============================================================================
+
+def load_installations() -> Dict:
+    """Load installations from JSON file"""
+    if os.path.exists(INSTALLATIONS_FILE):
+        with open(INSTALLATIONS_FILE, "r") as f:
+            return json.load(f)
+    return {}
+
+
+def save_installation(org: str, installation_id: int):
+    """Save installation to JSON file"""
+    installations = load_installations()
+    installations[org] = {
+        "installation_id": installation_id,
+        "installed_at": None  # Could add timestamp
+    }
+    
+    with open(INSTALLATIONS_FILE, "w") as f:
+        json.dump(installations, f, indent=2)
+
+
+def get_installation_id(org: str) -> Optional[int]:
+    """Get installation ID for an organization"""
+    installations = load_installations()
+    org_data = installations.get(org)
+    
+    if org_data:
+        return org_data.get("installation_id")
+    return None
+
+
+# ============================================================================
+# API ENDPOINTS
+# ============================================================================
+
+@app.get("/")
+def root():
+    """Health check endpoint"""
+    return {
+        "service": "Codesage Backend API",
+        "status": "operational",
+        "configured": github_auth is not None
+    }
+
+
+@app.get("/health")
+def health_check():
+    """Detailed health check"""
+    return {
+        "status": "healthy",
+        "github_app_configured": github_auth is not None,
+        "installations_count": len(load_installations())
+    }
+
+
+@app.get("/github/callback")
+async def github_callback(request: Request):
+    """
+    Handle GitHub App installation callback
+    
+    After org installs Codesage, GitHub redirects here with installation_id
+    This is the critical "make our product a member" step
+    """
+    # Get query parameters
+    query_params = dict(request.query_params)
+    
+    installation_id = query_params.get("installation_id")
+    setup_action = query_params.get("setup_action")
+    
+    if not installation_id:
+        return {
+            "success": False,
+            "message": "Missing installation_id parameter",
+            "received_params": query_params
+        }
+    
+    try:
+        installation_id = int(installation_id)
+    except ValueError:
+        return {
+            "success": False,
+            "message": "Invalid installation_id format",
+            "installation_id": installation_id
+        }
+    
+    # Try to get org name if credentials are configured
+    org_name = None
+    
+    if github_auth:
+        try:
+            # Get installation token to verify and fetch org info
+            token_data = github_auth.get_installation_token(installation_id)
+            token = token_data["token"]
+            
+            # Get installation repositories to extract org name
+            repos = github_auth.get_installation_repositories(token)
+            
+            if repos:
+                org_name = repos[0]["owner"]["login"]
+                
+        except Exception as e:
+            print(f"⚠️ Could not authenticate with GitHub: {str(e)}")
+            print(f"   Installation ID {installation_id} captured but org name unknown")
+    
+    # Save with org name if we got it, otherwise use placeholder
+    if org_name:
+        save_installation(org_name, installation_id)
+        return {
+            "success": True,
+            "message": f"✅ Successfully installed Codesage for {org_name}",
+            "org": org_name,
+            "installation_id": installation_id
+        }
+    else:
+        # Save with placeholder - user can update via /github/install
+        placeholder = f"installation_{installation_id}"
+        save_installation(placeholder, installation_id)
+        return {
+            "success": True,
+            "message": "✅ Installation ID captured! Use /github/install to register with org name",
+            "installation_id": installation_id,
+            "note": f"Saved as '{placeholder}'. Update your .env file with correct private key path and use POST /github/install to register with org name"
+        }
+
+
+@app.post("/github/install")
+def register_installation(data: InstallationCallback):
+    """
+    Manually register an installation (alternative to callback)
+    
+    Use this during development or for manual setup
+    """
+    save_installation(data.org, data.installation_id)
+    
+    return {
+        "success": True,
+        "message": f"Installation registered for {data.org}",
+        "installation_id": data.installation_id
+    }
+
+
+@app.get("/installations")
+def list_installations():
+    """
+    List all registered installations
+    Shows which orgs have installed Codesage
+    """
+    installations = load_installations()
+    
+    return {
+        "count": len(installations),
+        "installations": installations
+    }
+
+
+@app.get("/installations/{org}")
+def get_org_installation(org: str):
+    """Get installation details for a specific org"""
+    installation_id = get_installation_id(org)
+    
+    if not installation_id:
+        raise HTTPException(status_code=404, detail=f"No installation found for {org}")
+    
+    return {
+        "org": org,
+        "installation_id": installation_id
+    }
+
+
+@app.get("/repos/{org}")
+def list_org_repositories(org: str):
+    """
+    List all repositories accessible for an org
+    This is what the desktop app calls first
+    """
+    if not github_auth:
+        raise HTTPException(status_code=500, detail="GitHub App not configured")
+    
+    installation_id = get_installation_id(org)
+    if not installation_id:
+        raise HTTPException(status_code=404, detail=f"No installation found for {org}")
+    
+    try:
+        # Get fresh token
+        token = token_manager.get_token(installation_id)
+        
+        # Fetch repositories
+        repos = github_auth.get_installation_repositories(token)
+        
+        # Return simplified list
+        return {
+            "org": org,
+            "count": len(repos),
+            "repositories": [
+                {
+                    "name": repo["name"],
+                    "full_name": repo["full_name"],
+                    "private": repo["private"],
+                    "default_branch": repo.get("default_branch"),
+                    "language": repo.get("language"),
+                    "updated_at": repo.get("updated_at")
+                }
+                for repo in repos
+            ]
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch repositories: {str(e)}")
+
+
+@app.post("/insights/commits")
+def get_commits(data: InsightsRequest):
+    """
+    Get commit history for a repository
+    Desktop app → Backend → GitHub
+    """
+    if not github_auth:
+        raise HTTPException(status_code=500, detail="GitHub App not configured")
+    
+    installation_id = get_installation_id(data.org)
+    if not installation_id:
+        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+    
+    try:
+        token = token_manager.get_token(installation_id)
+        insights = GitHubInsights(token)
+        commits = insights.get_commits(data.org, data.repo)
+        
+        return {
+            "org": data.org,
+            "repo": data.repo,
+            "count": len(commits),
+            "commits": commits
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch commits: {str(e)}")
+
+
+@app.post("/insights/pull-requests")
+def get_pull_requests(data: InsightsRequest):
+    """Get pull requests for a repository"""
+    if not github_auth:
+        raise HTTPException(status_code=500, detail="GitHub App not configured")
+    
+    installation_id = get_installation_id(data.org)
+    if not installation_id:
+        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+    
+    try:
+        token = token_manager.get_token(installation_id)
+        insights = GitHubInsights(token)
+        prs = insights.get_pull_requests(data.org, data.repo)
+        
+        return {
+            "org": data.org,
+            "repo": data.repo,
+            "count": len(prs),
+            "pull_requests": prs
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch pull requests: {str(e)}")
+
+
+@app.post("/insights/issues")
+def get_issues(data: InsightsRequest):
+    """Get issues for a repository"""
+    if not github_auth:
+        raise HTTPException(status_code=500, detail="GitHub App not configured")
+    
+    installation_id = get_installation_id(data.org)
+    if not installation_id:
+        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+    
+    try:
+        token = token_manager.get_token(installation_id)
+        insights = GitHubInsights(token)
+        issues = insights.get_issues(data.org, data.repo)
+        
+        return {
+            "org": data.org,
+            "repo": data.repo,
+            "count": len(issues),
+            "issues": issues
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch issues: {str(e)}")
+
+
+@app.post("/insights/contributors")
+def get_contributors(data: InsightsRequest):
+    """Get contributors and their stats"""
+    if not github_auth:
+        raise HTTPException(status_code=500, detail="GitHub App not configured")
+    
+    installation_id = get_installation_id(data.org)
+    if not installation_id:
+        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+    
+    try:
+        token = token_manager.get_token(installation_id)
+        insights = GitHubInsights(token)
+        contributors = insights.get_contributors(data.org, data.repo)
+        
+        return {
+            "org": data.org,
+            "repo": data.repo,
+            "count": len(contributors),
+            "contributors": contributors
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch contributors: {str(e)}")
+
+
+@app.post("/insights/overview")
+def get_repository_overview(data: InsightsRequest):
+    """
+    Get comprehensive repository overview
+    Combines multiple insights into one powerful response
+    """
+    if not github_auth:
+        raise HTTPException(status_code=500, detail="GitHub App not configured")
+    
+    installation_id = get_installation_id(data.org)
+    if not installation_id:
+        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+    
+    try:
+        token = token_manager.get_token(installation_id)
+        insights = GitHubInsights(token)
+        
+        # Fetch multiple insights
+        repo_info = insights.get_repository_info(data.org, data.repo)
+        languages = insights.get_languages(data.org, data.repo)
+        contributors = insights.get_contributors(data.org, data.repo)
+        recent_commits = insights.get_commits(data.org, data.repo, per_page=10)
+        
+        return {
+            "org": data.org,
+            "repo": data.repo,
+            "overview": {
+                "name": repo_info.get("name"),
+                "description": repo_info.get("description"),
+                "stars": repo_info.get("stargazers_count"),
+                "forks": repo_info.get("forks_count"),
+                "open_issues": repo_info.get("open_issues_count"),
+                "default_branch": repo_info.get("default_branch"),
+                "created_at": repo_info.get("created_at"),
+                "updated_at": repo_info.get("updated_at"),
+                "size": repo_info.get("size"),
+                "languages": languages,
+                "contributors_count": len(contributors),
+                "recent_commits_count": len(recent_commits)
+            }
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch overview: {str(e)}")
+
+
+@app.post("/insights/activity")
+def get_repository_activity(data: InsightsRequest):
+    """Get repository activity stats (code frequency, commit activity)"""
+    if not github_auth:
+        raise HTTPException(status_code=500, detail="GitHub App not configured")
+    
+    installation_id = get_installation_id(data.org)
+    if not installation_id:
+        raise HTTPException(status_code=404, detail=f"No installation found for {data.org}")
+    
+    try:
+        token = token_manager.get_token(installation_id)
+        insights = GitHubInsights(token)
+        
+        code_frequency = insights.get_code_frequency(data.org, data.repo)
+        commit_activity = insights.get_commit_activity(data.org, data.repo)
+        
+        return {
+            "org": data.org,
+            "repo": data.repo,
+            "code_frequency": code_frequency,
+            "commit_activity": commit_activity
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to fetch activity: {str(e)}")
+
+
+# ============================================================================
+# RUN SERVER
+# ============================================================================
+
+
+