credit api

app.py

@@ -12,7 +12,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 
 from core.database import init_db
-from routers import auth, blink, general, gemini
+from routers import auth, blink, credits, general, gemini
 from services.drive_service import DriveService
 
 # Configure logging
@@ -86,6 +86,7 @@ app.include_router(general.router)
 app.include_router(auth.router)
 app.include_router(blink.router)
 app.include_router(gemini.router)
+app.include_router(credits.router)
 
 
 @app.exception_handler(Exception)

routers/credits.py

@@ -0,0 +1,121 @@
+"""
+Credits Router - API endpoints for credit management.
+
+Provides endpoints for checking credit balance and viewing credit history.
+"""
+from fastapi import APIRouter, Depends, Query
+from pydantic import BaseModel
+from typing import List, Optional
+from datetime import datetime
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy import select, desc
+
+from core.database import get_db
+from core.models import User, GeminiJob
+from dependencies import get_current_user
+
+router = APIRouter(prefix="/credits", tags=["credits"])
+
+
+# Response Models
+class CreditBalanceResponse(BaseModel):
+    """Response for credit balance endpoint."""
+    user_id: str
+    credits: int
+    last_used_at: Optional[str] = None
+
+
+class CreditHistoryItem(BaseModel):
+    """Single item in credit history."""
+    job_id: str
+    job_type: str
+    status: str
+    credits_reserved: int
+    credits_refunded: bool
+    error_message: Optional[str] = None
+    created_at: str
+    completed_at: Optional[str] = None
+
+
+class CreditHistoryResponse(BaseModel):
+    """Response for credit history endpoint."""
+    user_id: str
+    current_balance: int
+    history: List[CreditHistoryItem]
+    total_count: int
+    page: int
+    limit: int
+
+
+@router.get("", response_model=CreditBalanceResponse)
+async def get_credits(
+    user: User = Depends(get_current_user)
+):
+    """
+    Get current credit balance.
+    
+    Returns the user's current credit balance and last usage time.
+    """
+    return CreditBalanceResponse(
+        user_id=user.user_id,
+        credits=user.credits,
+        last_used_at=user.last_used_at.isoformat() if user.last_used_at else None
+    )
+
+
+@router.get("/history", response_model=CreditHistoryResponse)
+async def get_credit_history(
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+    page: int = Query(1, ge=1, description="Page number"),
+    limit: int = Query(20, ge=1, le=100, description="Items per page")
+):
+    """
+    Get credit usage history.
+    
+    Returns a paginated list of jobs with credit transactions,
+    showing which jobs used credits and which were refunded.
+    
+    Only includes jobs where credits were reserved (credits_reserved > 0).
+    """
+    offset = (page - 1) * limit
+    
+    # Query jobs with credit transactions
+    query = select(GeminiJob).where(
+        GeminiJob.user_id == user.user_id,
+        GeminiJob.credits_reserved > 0  # Only jobs that had credits reserved
+    ).order_by(desc(GeminiJob.created_at)).offset(offset).limit(limit)
+    
+    result = await db.execute(query)
+    jobs = result.scalars().all()
+    
+    # Get total count
+    count_query = select(GeminiJob).where(
+        GeminiJob.user_id == user.user_id,
+        GeminiJob.credits_reserved > 0
+    )
+    count_result = await db.execute(count_query)
+    total_count = len(count_result.scalars().all())
+    
+    # Build history items
+    history = []
+    for job in jobs:
+        history.append(CreditHistoryItem(
+            job_id=job.job_id,
+            job_type=job.job_type,
+            status=job.status,
+            credits_reserved=job.credits_reserved,
+            credits_refunded=job.credits_refunded or False,
+            error_message=job.error_message,
+            created_at=job.created_at.isoformat() if job.created_at else None,
+            completed_at=job.completed_at.isoformat() if job.completed_at else None
+        ))
+    
+    return CreditHistoryResponse(
+        user_id=user.user_id,
+        current_balance=user.credits,
+        history=history,
+        total_count=total_count,
+        page=page,
+        limit=limit
+    )