Backend Foundation - Docker Infrastructure & Mock API Endpoints (Tasks B2 & B3)

.github/workflows/ci.yml

@@ -95,6 +95,7 @@ jobs:
     - name: Run tests with coverage
       run: pytest --cov=. --cov-report=xml --cov-report=html
       env:
+        TESTING: true
         DATABASE_URL: ${{ secrets.DATABASE_URL || 'sqlite:///test.db' }}
         OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY || 'test-key' }}
 

backend/Dockerfile.celery

@@ -0,0 +1,27 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    gcc \
+    libpq-dev \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements
+COPY requirements.txt requirements-dev.txt ./
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements-dev.txt
+
+# Copy the application
+COPY . .
+
+# Set environment variables
+ENV PYTHONPATH=/app
+ENV PYTHONUNBUFFERED=1
+
+# Default command (overridden in docker-compose)
+CMD ["celery", "-A", "celery_app", "worker", "--loglevel=info"] 

backend/api/__init__.py

@@ -0,0 +1 @@
+# API package for SmartQuery backend

backend/api/auth.py

@@ -0,0 +1,120 @@
+import os
+import uuid
+from datetime import datetime, timedelta
+from typing import Any, Dict
+
+import jwt
+from fastapi import APIRouter, Depends, HTTPException
+from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
+
+from models.response_schemas import (
+    ApiResponse,
+    AuthResponse,
+    LoginRequest,
+    RefreshTokenRequest,
+    User,
+)
+
+router = APIRouter(prefix="/auth", tags=["authentication"])
+security = HTTPBearer()
+
+# Mock user database
+MOCK_USERS = {
+    "google_user_123": {
+        "id": "user_001",
+        "email": "john.doe@example.com",
+        "name": "John Doe",
+        "avatar_url": "https://lh3.googleusercontent.com/a/default-user",
+        "created_at": "2025-01-01T00:00:00Z",
+        "last_sign_in_at": "2025-01-01T12:00:00Z",
+    }
+}
+
+# Mock JWT settings
+JWT_SECRET = os.getenv("JWT_SECRET", "mock_secret_key_for_development")
+ALGORITHM = "HS256"
+ACCESS_TOKEN_EXPIRE_MINUTES = 60
+
+
+def create_access_token(data: Dict[str, Any]) -> str:
+    """Create JWT access token"""
+    to_encode = data.copy()
+    expire = datetime.utcnow() + timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
+    to_encode.update({"exp": expire})
+    return jwt.encode(to_encode, JWT_SECRET, algorithm=ALGORITHM)
+
+
+def verify_token(credentials: HTTPAuthorizationCredentials = Depends(security)) -> str:
+    """Verify JWT token and return user_id"""
+    try:
+        payload = jwt.decode(
+            credentials.credentials, JWT_SECRET, algorithms=[ALGORITHM]
+        )
+        user_id: str = payload.get("sub")
+        if user_id is None:
+            raise HTTPException(status_code=401, detail="Invalid token")
+        return user_id
+    except jwt.PyJWTError:
+        raise HTTPException(status_code=401, detail="Invalid token")
+
+
+@router.post("/google")
+async def login_with_google(request: LoginRequest) -> ApiResponse[AuthResponse]:
+    """Mock Google OAuth login"""
+    # Mock Google token validation
+    if not request.google_token.startswith("mock_google_token"):
+        raise HTTPException(status_code=401, detail="Invalid Google token")
+
+    # Mock user from Google token
+    user_data = MOCK_USERS["google_user_123"]
+    user = User(**user_data)
+
+    # Create JWT tokens
+    access_token = create_access_token(data={"sub": user.id})
+    refresh_token = str(uuid.uuid4())
+
+    auth_response = AuthResponse(
+        user=user,
+        access_token=access_token,
+        refresh_token=refresh_token,
+        expires_in=ACCESS_TOKEN_EXPIRE_MINUTES * 60,
+    )
+
+    return ApiResponse(success=True, data=auth_response)
+
+
+@router.get("/me")
+async def get_current_user(user_id: str = Depends(verify_token)) -> ApiResponse[User]:
+    """Get current user information"""
+    # Mock user lookup
+    for mock_user in MOCK_USERS.values():
+        if mock_user["id"] == user_id:
+            user = User(**mock_user)
+            return ApiResponse(success=True, data=user)
+
+    raise HTTPException(status_code=404, detail="User not found")
+
+
+@router.post("/logout")
+async def logout(user_id: str = Depends(verify_token)) -> ApiResponse[Dict[str, str]]:
+    """Logout current user"""
+    return ApiResponse(success=True, data={"message": "Logged out successfully"})
+
+
+@router.post("/refresh")
+async def refresh_token(request: RefreshTokenRequest) -> ApiResponse[Dict[str, Any]]:
+    """Refresh access token"""
+    # Mock refresh token validation
+    if not request.refresh_token:
+        raise HTTPException(status_code=401, detail="Invalid refresh token")
+
+    # Create new access token
+    new_access_token = create_access_token(data={"sub": "user_001"})
+
+    return ApiResponse(
+        success=True,
+        data={
+            "access_token": new_access_token,
+            "expires_in": ACCESS_TOKEN_EXPIRE_MINUTES * 60,
+        },
+    )