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

codeshift/api/routers/usage.py

@@ -1,291 +0,0 @@
-"""Usage tracking router for the PyResolve API."""
-
-from datetime import datetime, timezone
-
-from fastapi import APIRouter, HTTPException, status
-
-from codeshift.api.auth import CurrentUser
-from codeshift.api.config import get_settings
-from codeshift.api.database import get_database
-from codeshift.api.models.usage import (
-    QuotaCheckRequest,
-    QuotaCheckResponse,
-    QuotaInfo,
-    UsageEvent,
-    UsageEventCreate,
-    UsageResponse,
-    UsageSummary,
-)
-
-router = APIRouter()
-
-
-@router.get("/quota", response_model=QuotaInfo)
-async def get_quota(user: CurrentUser) -> QuotaInfo:
-    """Get current quota information for the authenticated user."""
-    db = get_database()
-    settings = get_settings()
-
-    # Get user profile for tier info
-    profile = db.get_profile_by_id(user.user_id)
-    if not profile:
-        raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND,
-            detail="User profile not found",
-        )
-
-    tier = profile.get("tier", "free")
-    limits = settings.get_tier_limits(tier)
-
-    # Get current billing period
-    billing_period = datetime.now(timezone.utc).strftime("%Y-%m")
-
-    # Get usage for current period
-    usage = db.get_usage_for_period(user.user_id, billing_period)
-
-    files_migrated = usage.get("file_migrated", 0)
-    llm_calls = usage.get("llm_call", 0)
-
-    return QuotaInfo.from_usage(
-        tier=tier,
-        billing_period=billing_period,
-        files_migrated=files_migrated,
-        llm_calls=llm_calls,
-        files_limit=limits["files_per_month"],
-        llm_calls_limit=limits["llm_calls_per_month"],
-    )
-
-
-@router.get("/", response_model=UsageResponse)
-async def get_usage(
-    user: CurrentUser,
-    billing_period: str | None = None,
-    limit: int = 20,
-) -> UsageResponse:
-    """Get usage summary and recent events."""
-    db = get_database()
-    settings = get_settings()
-
-    # Get user profile for tier info
-    profile = db.get_profile_by_id(user.user_id)
-    if not profile:
-        raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND,
-            detail="User profile not found",
-        )
-
-    tier = profile.get("tier", "free")
-    limits = settings.get_tier_limits(tier)
-
-    # Default to current billing period
-    if not billing_period:
-        billing_period = datetime.now(timezone.utc).strftime("%Y-%m")
-
-    # Get usage summary
-    usage = db.get_usage_for_period(user.user_id, billing_period)
-    files_migrated = usage.get("file_migrated", 0)
-    llm_calls = usage.get("llm_call", 0)
-
-    quota = QuotaInfo.from_usage(
-        tier=tier,
-        billing_period=billing_period,
-        files_migrated=files_migrated,
-        llm_calls=llm_calls,
-        files_limit=limits["files_per_month"],
-        llm_calls_limit=limits["llm_calls_per_month"],
-    )
-
-    # Get recent events
-    events_data = db.get_usage_events(user.user_id, billing_period, limit=limit)
-
-    recent_events = [
-        UsageEvent(
-            id=e["id"],
-            user_id=e["user_id"],
-            event_type=e["event_type"],
-            library=e.get("library"),
-            quantity=e["quantity"],
-            metadata=e.get("metadata", {}),
-            billing_period=e["billing_period"],
-            created_at=e["created_at"],
-        )
-        for e in events_data
-    ]
-
-    return UsageResponse(quota=quota, recent_events=recent_events)
-
-
-@router.post("/", response_model=UsageEvent)
-async def record_usage(request: UsageEventCreate, user: CurrentUser) -> UsageEvent:
-    """Record a usage event.
-
-    This is called by the CLI after performing migrations.
-    """
-    db = get_database()
-    settings = get_settings()
-
-    # Get user's tier and limits
-    profile = db.get_profile_by_id(user.user_id)
-    if not profile:
-        raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND,
-            detail="User profile not found",
-        )
-
-    tier = profile.get("tier", "free")
-    limits = settings.get_tier_limits(tier)
-
-    # Get current usage
-    billing_period = datetime.now(timezone.utc).strftime("%Y-%m")
-    usage = db.get_usage_for_period(user.user_id, billing_period)
-
-    # Check quota for file_migrated and llm_call events
-    if request.event_type == "file_migrated":
-        current = usage.get("file_migrated", 0)
-        limit = limits["files_per_month"]
-        if current + request.quantity > limit:
-            raise HTTPException(
-                status_code=status.HTTP_429_TOO_MANY_REQUESTS,
-                detail=f"File migration quota exceeded. Used: {current}, Limit: {limit}",
-            )
-    elif request.event_type == "llm_call":
-        current = usage.get("llm_call", 0)
-        limit = limits["llm_calls_per_month"]
-        if current + request.quantity > limit:
-            raise HTTPException(
-                status_code=status.HTTP_429_TOO_MANY_REQUESTS,
-                detail=f"LLM call quota exceeded. Used: {current}, Limit: {limit}",
-            )
-
-    # Record the event
-    result = db.record_usage_event(
-        user_id=user.user_id,
-        event_type=request.event_type,
-        library=request.library,
-        quantity=request.quantity,
-        metadata=request.metadata,
-    )
-
-    return UsageEvent(
-        id=result["id"],
-        user_id=result["user_id"],
-        event_type=result["event_type"],
-        library=result.get("library"),
-        quantity=result["quantity"],
-        metadata=result.get("metadata", {}),
-        billing_period=result["billing_period"],
-        created_at=result["created_at"],
-    )
-
-
-@router.post("/check", response_model=QuotaCheckResponse)
-async def check_quota(request: QuotaCheckRequest, user: CurrentUser) -> QuotaCheckResponse:
-    """Check if an operation is within quota before performing it.
-
-    This is used by the CLI to pre-check before starting a migration.
-    """
-    db = get_database()
-    settings = get_settings()
-
-    # Get user's tier and limits
-    profile = db.get_profile_by_id(user.user_id)
-    if not profile:
-        raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND,
-            detail="User profile not found",
-        )
-
-    tier = profile.get("tier", "free")
-    limits = settings.get_tier_limits(tier)
-
-    # Get current usage
-    billing_period = datetime.now(timezone.utc).strftime("%Y-%m")
-    usage = db.get_usage_for_period(user.user_id, billing_period)
-
-    # Determine current usage and limit based on event type
-    if request.event_type == "file_migrated":
-        current_usage = usage.get("file_migrated", 0)
-        limit = limits["files_per_month"]
-        remaining = max(0, limit - current_usage)
-        allowed = current_usage + request.quantity <= limit
-        message = (
-            None if allowed else f"Would exceed file migration quota ({current_usage}/{limit})"
-        )
-    elif request.event_type == "llm_call":
-        current_usage = usage.get("llm_call", 0)
-        limit = limits["llm_calls_per_month"]
-        remaining = max(0, limit - current_usage)
-        allowed = current_usage + request.quantity <= limit
-        message = None if allowed else f"Would exceed LLM call quota ({current_usage}/{limit})"
-    else:
-        # scan and apply have no limits
-        current_usage = usage.get(request.event_type, 0)
-        limit = 999999999
-        remaining = limit
-        allowed = True
-        message = None
-
-    return QuotaCheckResponse(
-        allowed=allowed,
-        current_usage=current_usage,
-        limit=limit,
-        remaining=remaining,
-        message=message,
-    )
-
-
-@router.get("/summary", response_model=UsageSummary)
-async def get_usage_summary(
-    user: CurrentUser,
-    billing_period: str | None = None,
-) -> UsageSummary:
-    """Get usage summary for a billing period."""
-    db = get_database()
-
-    # Default to current billing period
-    if not billing_period:
-        billing_period = datetime.now(timezone.utc).strftime("%Y-%m")
-
-    usage = db.get_usage_for_period(user.user_id, billing_period)
-
-    return UsageSummary(
-        billing_period=billing_period,
-        files_migrated=usage.get("file_migrated", 0),
-        llm_calls=usage.get("llm_call", 0),
-        scans=usage.get("scan", 0),
-        applies=usage.get("apply", 0),
-        total_events=sum(usage.values()),
-    )
-
-
-@router.get("/history", response_model=list[UsageSummary])
-async def get_usage_history(
-    user: CurrentUser,
-    months: int = 6,
-) -> list[UsageSummary]:
-    """Get usage history for past months."""
-    from datetime import timedelta
-
-    db = get_database()
-    summaries = []
-
-    # Calculate billing periods for past N months
-    now = datetime.now(timezone.utc)
-    for i in range(months):
-        date = now - timedelta(days=30 * i)
-        billing_period = date.strftime("%Y-%m")
-
-        usage = db.get_usage_for_period(user.user_id, billing_period)
-
-        summaries.append(
-            UsageSummary(
-                billing_period=billing_period,
-                files_migrated=usage.get("file_migrated", 0),
-                llm_calls=usage.get("llm_call", 0),
-                scans=usage.get("scan", 0),
-                applies=usage.get("apply", 0),
-                total_events=sum(usage.values()),
-            )
-        )
-
-    return summaries

codeshift/api/routers/webhooks.py

@@ -1,289 +0,0 @@
-"""Webhooks router for the PyResolve API."""
-
-import stripe
-from fastapi import APIRouter, Header, HTTPException, Request, status
-
-from codeshift.api.config import get_settings
-from codeshift.api.database import get_database
-
-router = APIRouter()
-
-
-def get_stripe_client() -> stripe:
-    """Get configured Stripe client."""
-    settings = get_settings()
-    stripe.api_key = settings.stripe_secret_key
-    return stripe
-
-
-@router.post("/stripe")
-async def handle_stripe_webhook(
-    request: Request,
-    stripe_signature: str = Header(None, alias="Stripe-Signature"),
-) -> dict:
-    """Handle Stripe webhook events.
-
-    Handles the following events:
-    - checkout.session.completed: User completed checkout, activate subscription
-    - customer.subscription.created: New subscription created
-    - customer.subscription.updated: Subscription updated (upgrade/downgrade)
-    - customer.subscription.deleted: Subscription canceled
-    - invoice.paid: Invoice paid successfully
-    - invoice.payment_failed: Payment failed
-    """
-    settings = get_settings()
-    stripe_client = get_stripe_client()
-
-    # Get raw body for signature verification
-    payload = await request.body()
-
-    if not stripe_signature:
-        raise HTTPException(
-            status_code=status.HTTP_400_BAD_REQUEST,
-            detail="Missing Stripe-Signature header",
-        )
-
-    try:
-        event = stripe_client.Webhook.construct_event(
-            payload,
-            stripe_signature,
-            settings.stripe_webhook_secret,
-        )
-    except ValueError as e:
-        raise HTTPException(
-            status_code=status.HTTP_400_BAD_REQUEST,
-            detail="Invalid payload",
-        ) from e
-    except stripe.error.SignatureVerificationError as e:
-        raise HTTPException(
-            status_code=status.HTTP_400_BAD_REQUEST,
-            detail="Invalid signature",
-        ) from e
-
-    # Handle the event
-    event_type = event["type"]
-    data = event["data"]["object"]
-
-    try:
-        if event_type == "checkout.session.completed":
-            await handle_checkout_completed(data)
-        elif event_type == "customer.subscription.created":
-            await handle_subscription_created(data)
-        elif event_type == "customer.subscription.updated":
-            await handle_subscription_updated(data)
-        elif event_type == "customer.subscription.deleted":
-            await handle_subscription_deleted(data)
-        elif event_type == "invoice.paid":
-            await handle_invoice_paid(data)
-        elif event_type == "invoice.payment_failed":
-            await handle_invoice_payment_failed(data)
-        else:
-            # Log unhandled events for debugging
-            print(f"Unhandled webhook event: {event_type}")
-    except Exception as e:
-        # Log error but don't fail the webhook
-        print(f"Error handling webhook {event_type}: {e}")
-        # Still return 200 to prevent Stripe retries for non-critical errors
-
-    return {"received": True}
-
-
-async def handle_checkout_completed(session: dict) -> None:
-    """Handle successful checkout session completion."""
-    db = get_database()
-
-    # Get user ID from metadata
-    user_id = session.get("metadata", {}).get("user_id")
-    tier = session.get("metadata", {}).get("tier", "pro")
-
-    if not user_id:
-        print("Checkout session missing user_id in metadata")
-        return
-
-    # Get subscription ID
-    subscription_id = session.get("subscription")
-    customer_id = session.get("customer")
-
-    # Update user's profile
-    db.update_profile(
-        user_id,
-        {
-            "tier": tier,
-            "stripe_customer_id": customer_id,
-            "stripe_subscription_id": subscription_id,
-        },
-    )
-
-    print(f"User {user_id} upgraded to {tier}")
-
-
-async def handle_subscription_created(subscription: dict) -> None:
-    """Handle new subscription creation."""
-    db = get_database()
-
-    customer_id = subscription.get("customer")
-    subscription_id = subscription.get("id")
-
-    # Find user by customer ID
-    result = (
-        db.client.table("profiles").select("id").eq("stripe_customer_id", customer_id).execute()
-    )
-
-    if not result.data:
-        print(f"No user found for customer {customer_id}")
-        return
-
-    user_id = result.data[0]["id"]
-
-    # Determine tier from price
-    items = subscription.get("items", {}).get("data", [])
-    if items:
-        price_id = items[0].get("price", {}).get("id")
-        settings = get_settings()
-
-        if price_id == settings.stripe_price_id_pro:
-            tier = "pro"
-        elif price_id == settings.stripe_price_id_unlimited:
-            tier = "unlimited"
-        else:
-            tier = "pro"  # Default
-    else:
-        tier = "pro"
-
-    # Update profile
-    db.update_profile(
-        user_id,
-        {
-            "tier": tier,
-            "stripe_subscription_id": subscription_id,
-            "billing_period_start": subscription.get("current_period_start"),
-            "billing_period_end": subscription.get("current_period_end"),
-        },
-    )
-
-    print(f"Subscription created for user {user_id}: {tier}")
-
-
-async def handle_subscription_updated(subscription: dict) -> None:
-    """Handle subscription updates (upgrades/downgrades)."""
-    db = get_database()
-
-    customer_id = subscription.get("customer")
-    subscription_id = subscription.get("id")
-    status_value = subscription.get("status")
-
-    # Find user by customer ID
-    result = (
-        db.client.table("profiles").select("id").eq("stripe_customer_id", customer_id).execute()
-    )
-
-    if not result.data:
-        print(f"No user found for customer {customer_id}")
-        return
-
-    user_id = result.data[0]["id"]
-
-    # Determine tier from price
-    items = subscription.get("items", {}).get("data", [])
-    if items:
-        price_id = items[0].get("price", {}).get("id")
-        settings = get_settings()
-
-        if price_id == settings.stripe_price_id_pro:
-            tier = "pro"
-        elif price_id == settings.stripe_price_id_unlimited:
-            tier = "unlimited"
-        else:
-            tier = "pro"
-    else:
-        tier = "pro"
-
-    # Handle canceled subscriptions
-    if status_value in ("canceled", "unpaid"):
-        tier = "free"
-
-    # Update profile
-    db.update_profile(
-        user_id,
-        {
-            "tier": tier,
-            "stripe_subscription_id": subscription_id if tier != "free" else None,
-            "billing_period_start": subscription.get("current_period_start"),
-            "billing_period_end": subscription.get("current_period_end"),
-        },
-    )
-
-    print(f"Subscription updated for user {user_id}: {tier} ({status_value})")
-
-
-async def handle_subscription_deleted(subscription: dict) -> None:
-    """Handle subscription cancellation/deletion."""
-    db = get_database()
-
-    customer_id = subscription.get("customer")
-
-    # Find user by customer ID
-    result = (
-        db.client.table("profiles").select("id").eq("stripe_customer_id", customer_id).execute()
-    )
-
-    if not result.data:
-        print(f"No user found for customer {customer_id}")
-        return
-
-    user_id = result.data[0]["id"]
-
-    # Downgrade to free tier
-    db.update_profile(
-        user_id,
-        {
-            "tier": "free",
-            "stripe_subscription_id": None,
-            "billing_period_start": None,
-            "billing_period_end": None,
-        },
-    )
-
-    print(f"Subscription deleted for user {user_id}, downgraded to free")
-
-
-async def handle_invoice_paid(invoice: dict) -> None:
-    """Handle successful invoice payment."""
-    customer_id = invoice.get("customer")
-    amount_paid = invoice.get("amount_paid", 0)
-
-    print(f"Invoice paid for customer {customer_id}: ${amount_paid / 100:.2f}")
-
-    # Could send receipt email, update analytics, etc.
-
-
-async def handle_invoice_payment_failed(invoice: dict) -> None:
-    """Handle failed invoice payment."""
-    db = get_database()
-
-    customer_id = invoice.get("customer")
-    attempt_count = invoice.get("attempt_count", 0)
-
-    print(f"Invoice payment failed for customer {customer_id} (attempt {attempt_count})")
-
-    # Find user
-    result = (
-        db.client.table("profiles")
-        .select("id, email")
-        .eq("stripe_customer_id", customer_id)
-        .execute()
-    )
-
-    if not result.data:
-        return
-
-    # After multiple failed attempts, could:
-    # - Send warning email
-    # - Temporarily restrict features
-    # - Eventually downgrade to free
-
-    if attempt_count >= 3:
-        user_id = result.data[0]["id"]
-        print(f"Multiple payment failures for user {user_id}, consider downgrade")
-        # Uncomment to auto-downgrade after 3 failures:
-        # db.update_profile(user_id, {"tier": "free"})