codeshift 0.3.7__py3-none-any.whl → 0.5.0__py3-none-any.whl

codeshift/api/main.py

@@ -1,103 +0,0 @@
-"""FastAPI application for Codeshift billing API."""
-
-from collections.abc import AsyncGenerator
-from contextlib import asynccontextmanager
-
-from fastapi import FastAPI, Request
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
-
-from codeshift import __version__
-from codeshift.api.config import get_settings
-from codeshift.api.routers import auth, billing, migrate, usage, webhooks
-
-
-@asynccontextmanager
-async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
-    """Application lifespan handler."""
-    # Startup
-    settings = get_settings()
-    print(f"Codeshift API starting (environment: {settings.environment})")
-    yield
-    # Shutdown
-    print("Codeshift API shutting down")
-
-
-app = FastAPI(
-    title="Codeshift API",
-    description="Billing and authentication API for Codeshift CLI",
-    version=__version__,
-    lifespan=lifespan,
-    docs_url="/docs" if not get_settings().is_production else None,
-    redoc_url="/redoc" if not get_settings().is_production else None,
-)
-
-# CORS configuration
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=[
-        "https://codeshift.dev",
-        "https://www.codeshift.dev",
-        "http://localhost:3000",  # Local development
-    ],
-    allow_credentials=True,
-    allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
-    allow_headers=["*"],
-)
-
-
-# Include routers
-app.include_router(auth.router, prefix="/auth", tags=["Authentication"])
-app.include_router(billing.router, prefix="/billing", tags=["Billing"])
-app.include_router(migrate.router, tags=["Migration"])
-app.include_router(usage.router, prefix="/usage", tags=["Usage"])
-app.include_router(webhooks.router, prefix="/webhooks", tags=["Webhooks"])
-
-
-@app.get("/")
-async def root() -> dict:
-    """API root endpoint."""
-    return {
-        "name": "Codeshift API",
-        "version": __version__,
-        "status": "healthy",
-    }
-
-
-@app.get("/health")
-async def health_check() -> dict:
-    """Health check endpoint."""
-    return {"status": "healthy", "version": __version__}
-
-
-@app.exception_handler(Exception)
-async def global_exception_handler(request: Request, exc: Exception) -> JSONResponse:
-    """Global exception handler."""
-    settings = get_settings()
-
-    # Log the error
-    print(f"Unhandled exception: {exc}")
-
-    # Return appropriate response based on environment
-    if settings.is_production:
-        return JSONResponse(
-            status_code=500,
-            content={"detail": "Internal server error"},
-        )
-    else:
-        return JSONResponse(
-            status_code=500,
-            content={"detail": str(exc), "type": type(exc).__name__},
-        )
-
-
-# For running with uvicorn directly
-if __name__ == "__main__":
-    import uvicorn
-
-    uvicorn.run(
-        "codeshift.api.main:app",
-        host="0.0.0.0",
-        port=8000,
-        reload=True,
-    )

codeshift/api/models/__init__.py

@@ -1,55 +0,0 @@
-"""Pydantic models for the PyResolve API."""
-
-from codeshift.api.models.auth import (
-    APIKey,
-    APIKeyCreate,
-    APIKeyResponse,
-    TokenResponse,
-    UserInfo,
-)
-from codeshift.api.models.billing import (
-    CheckoutSessionRequest,
-    CheckoutSessionResponse,
-    PortalSessionResponse,
-    SubscriptionInfo,
-    TierInfo,
-)
-from codeshift.api.models.migrate import (
-    ExplainChangeRequest,
-    ExplainChangeResponse,
-    MigrateCodeRequest,
-    MigrateCodeResponse,
-)
-from codeshift.api.models.usage import (
-    QuotaInfo,
-    UsageEvent,
-    UsageEventCreate,
-    UsageResponse,
-    UsageSummary,
-)
-
-__all__ = [
-    # Auth models
-    "APIKey",
-    "APIKeyCreate",
-    "APIKeyResponse",
-    "TokenResponse",
-    "UserInfo",
-    # Billing models
-    "CheckoutSessionRequest",
-    "CheckoutSessionResponse",
-    "PortalSessionResponse",
-    "SubscriptionInfo",
-    "TierInfo",
-    # Migrate models
-    "ExplainChangeRequest",
-    "ExplainChangeResponse",
-    "MigrateCodeRequest",
-    "MigrateCodeResponse",
-    # Usage models
-    "QuotaInfo",
-    "UsageEvent",
-    "UsageEventCreate",
-    "UsageResponse",
-    "UsageSummary",
-]

codeshift/api/models/auth.py

@@ -1,108 +0,0 @@
-"""Authentication models for the PyResolve API."""
-
-from datetime import datetime
-
-from pydantic import BaseModel, EmailStr, Field
-
-
-class UserInfo(BaseModel):
-    """User profile information."""
-
-    id: str
-    email: EmailStr
-    full_name: str | None = None
-    tier: str = "free"
-    stripe_customer_id: str | None = None
-    created_at: datetime
-
-    class Config:
-        from_attributes = True
-
-
-class APIKeyCreate(BaseModel):
-    """Request to create a new API key."""
-
-    name: str = Field(default="CLI Key", min_length=1, max_length=100)
-    scopes: list[str] = Field(default=["read", "write"])
-
-
-class APIKey(BaseModel):
-    """API key information (without the secret)."""
-
-    id: str
-    name: str
-    key_prefix: str
-    scopes: list[str]
-    revoked: bool = False
-    last_used_at: datetime | None = None
-    expires_at: datetime | None = None
-    created_at: datetime
-
-    class Config:
-        from_attributes = True
-
-
-class APIKeyResponse(BaseModel):
-    """Response when creating a new API key (includes full key once)."""
-
-    id: str
-    name: str
-    key: str  # Full API key - only shown once
-    key_prefix: str
-    scopes: list[str]
-    created_at: datetime
-
-
-class TokenResponse(BaseModel):
-    """Response for authentication token."""
-
-    access_token: str
-    token_type: str = "bearer"
-    expires_in: int
-    refresh_token: str | None = None
-
-
-class LoginRequest(BaseModel):
-    """Request for CLI login."""
-
-    email: EmailStr
-    password: str = Field(min_length=6)
-
-
-class LoginResponse(BaseModel):
-    """Response for CLI login."""
-
-    api_key: str
-    user: UserInfo
-    message: str = "Login successful"
-
-
-class DeviceCodeRequest(BaseModel):
-    """Request to initiate device code flow."""
-
-    client_id: str = "codeshift-cli"
-
-
-class DeviceCodeResponse(BaseModel):
-    """Response with device code for authentication."""
-
-    device_code: str
-    user_code: str
-    verification_uri: str
-    expires_in: int = 900  # 15 minutes
-    interval: int = 5  # Polling interval in seconds
-
-
-class DeviceTokenRequest(BaseModel):
-    """Request to exchange device code for token."""
-
-    device_code: str
-    client_id: str = "codeshift-cli"
-
-
-class RegisterRequest(BaseModel):
-    """Request for new user registration."""
-
-    email: EmailStr
-    password: str = Field(min_length=8, description="Password must be at least 8 characters")
-    full_name: str | None = Field(default=None, max_length=100)

codeshift/api/models/billing.py

@@ -1,92 +0,0 @@
-"""Billing models for the PyResolve API."""
-
-from datetime import datetime
-
-from pydantic import BaseModel, Field
-
-
-class TierInfo(BaseModel):
-    """Information about a pricing tier."""
-
-    name: str
-    display_name: str
-    price_monthly: int  # In cents
-    files_per_month: int
-    llm_calls_per_month: int
-    features: list[str]
-
-
-class SubscriptionInfo(BaseModel):
-    """Current subscription information."""
-
-    tier: str
-    status: str  # active, canceled, past_due, etc.
-    stripe_subscription_id: str | None = None
-    current_period_start: datetime | None = None
-    current_period_end: datetime | None = None
-    cancel_at_period_end: bool = False
-
-
-class CheckoutSessionRequest(BaseModel):
-    """Request to create a Stripe checkout session."""
-
-    tier: str = Field(..., pattern="^(pro|unlimited)$")
-    success_url: str | None = None
-    cancel_url: str | None = None
-
-
-class CheckoutSessionResponse(BaseModel):
-    """Response with Stripe checkout session."""
-
-    checkout_url: str
-    session_id: str
-
-
-class PortalSessionResponse(BaseModel):
-    """Response with Stripe billing portal URL."""
-
-    portal_url: str
-
-
-class PriceInfo(BaseModel):
-    """Stripe price information."""
-
-    id: str
-    product_id: str
-    unit_amount: int
-    currency: str
-    recurring_interval: str
-
-
-class InvoiceInfo(BaseModel):
-    """Invoice information."""
-
-    id: str
-    status: str
-    amount_due: int
-    amount_paid: int
-    currency: str
-    created: datetime
-    due_date: datetime | None = None
-    hosted_invoice_url: str | None = None
-    pdf_url: str | None = None
-
-
-class PaymentMethodInfo(BaseModel):
-    """Payment method information."""
-
-    id: str
-    type: str  # card, bank_account, etc.
-    card_brand: str | None = None
-    card_last4: str | None = None
-    card_exp_month: int | None = None
-    card_exp_year: int | None = None
-
-
-class BillingOverview(BaseModel):
-    """Complete billing overview for a user."""
-
-    subscription: SubscriptionInfo
-    tier_info: TierInfo
-    payment_method: PaymentMethodInfo | None = None
-    upcoming_invoice: InvoiceInfo | None = None

codeshift/api/models/migrate.py

@@ -1,42 +0,0 @@
-"""Migration models for the PyResolve API."""
-
-from typing import Any
-
-from pydantic import BaseModel, Field
-
-
-class MigrateCodeRequest(BaseModel):
-    """Request to migrate code using LLM."""
-
-    code: str = Field(..., description="Source code to migrate")
-    library: str = Field(..., description="Library being upgraded (e.g., 'pydantic')")
-    from_version: str = Field(..., description="Current version (e.g., '1.10.0')")
-    to_version: str = Field(..., description="Target version (e.g., '2.5.0')")
-    context: str | None = Field(None, description="Optional context about the migration")
-
-
-class MigrateCodeResponse(BaseModel):
-    """Response from LLM migration."""
-
-    success: bool
-    migrated_code: str
-    original_code: str
-    error: str | None = None
-    usage: dict[str, Any] = Field(default_factory=dict)
-    cached: bool = False
-
-
-class ExplainChangeRequest(BaseModel):
-    """Request to explain a migration change."""
-
-    original_code: str = Field(..., description="Original code before migration")
-    transformed_code: str = Field(..., description="Transformed code after migration")
-    library: str = Field(..., description="Library being upgraded")
-
-
-class ExplainChangeResponse(BaseModel):
-    """Response with explanation of changes."""
-
-    success: bool
-    explanation: str | None = None
-    error: str | None = None

codeshift/api/models/usage.py

@@ -1,116 +0,0 @@
-"""Usage tracking models for the PyResolve API."""
-
-from datetime import datetime
-from typing import Any
-
-from pydantic import BaseModel, Field
-
-
-class UsageEventCreate(BaseModel):
-    """Request to record a usage event."""
-
-    event_type: str = Field(..., pattern="^(file_migrated|llm_call|scan|apply)$")
-    library: str | None = None
-    quantity: int = Field(default=1, ge=1)
-    metadata: dict[str, Any] = Field(default_factory=dict)
-
-
-class UsageEvent(BaseModel):
-    """A recorded usage event."""
-
-    id: str
-    user_id: str
-    event_type: str
-    library: str | None = None
-    quantity: int
-    metadata: dict[str, Any]
-    billing_period: str
-    created_at: datetime
-
-    class Config:
-        from_attributes = True
-
-
-class UsageSummary(BaseModel):
-    """Summary of usage for a billing period."""
-
-    billing_period: str
-    files_migrated: int = 0
-    llm_calls: int = 0
-    scans: int = 0
-    applies: int = 0
-    total_events: int = 0
-
-
-class QuotaInfo(BaseModel):
-    """Current quota information for a user."""
-
-    tier: str
-    billing_period: str
-
-    # Current usage
-    files_migrated: int = 0
-    llm_calls: int = 0
-
-    # Limits
-    files_limit: int
-    llm_calls_limit: int
-
-    # Calculated fields
-    files_remaining: int
-    llm_calls_remaining: int
-    files_percentage: float
-    llm_calls_percentage: float
-
-    @classmethod
-    def from_usage(
-        cls,
-        tier: str,
-        billing_period: str,
-        files_migrated: int,
-        llm_calls: int,
-        files_limit: int,
-        llm_calls_limit: int,
-    ) -> "QuotaInfo":
-        """Create QuotaInfo from usage data."""
-        files_remaining = max(0, files_limit - files_migrated)
-        llm_calls_remaining = max(0, llm_calls_limit - llm_calls)
-
-        return cls(
-            tier=tier,
-            billing_period=billing_period,
-            files_migrated=files_migrated,
-            llm_calls=llm_calls,
-            files_limit=files_limit,
-            llm_calls_limit=llm_calls_limit,
-            files_remaining=files_remaining,
-            llm_calls_remaining=llm_calls_remaining,
-            files_percentage=round(files_migrated / files_limit * 100, 1) if files_limit > 0 else 0,
-            llm_calls_percentage=(
-                round(llm_calls / llm_calls_limit * 100, 1) if llm_calls_limit > 0 else 0
-            ),
-        )
-
-
-class UsageResponse(BaseModel):
-    """Response for usage queries."""
-
-    quota: QuotaInfo
-    recent_events: list[UsageEvent] = Field(default_factory=list)
-
-
-class QuotaCheckRequest(BaseModel):
-    """Request to check if an operation is within quota."""
-
-    event_type: str = Field(..., pattern="^(file_migrated|llm_call|scan|apply)$")
-    quantity: int = Field(default=1, ge=1)
-
-
-class QuotaCheckResponse(BaseModel):
-    """Response for quota check."""
-
-    allowed: bool
-    current_usage: int
-    limit: int
-    remaining: int
-    message: str | None = None

codeshift/api/routers/__init__.py

@@ -1,5 +0,0 @@
-"""API routers for PyResolve."""
-
-from codeshift.api.routers import auth, billing, migrate, usage, webhooks
-
-__all__ = ["auth", "billing", "migrate", "usage", "webhooks"]