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

codeshift/api/routers/billing.py

@@ -1,395 +0,0 @@
-"""Billing router for the PyResolve API."""
-
-import stripe
-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.billing import (
-    BillingOverview,
-    CheckoutSessionRequest,
-    CheckoutSessionResponse,
-    PaymentMethodInfo,
-    PortalSessionResponse,
-    SubscriptionInfo,
-    TierInfo,
-)
-
-router = APIRouter()
-
-# Tier definitions
-TIERS: dict[str, TierInfo] = {
-    "free": TierInfo(
-        name="free",
-        display_name="Free",
-        price_monthly=0,
-        files_per_month=100,
-        llm_calls_per_month=50,
-        features=[
-            "100 file migrations/month",
-            "50 LLM-assisted migrations/month",
-            "5 supported libraries",
-            "Community support",
-        ],
-    ),
-    "pro": TierInfo(
-        name="pro",
-        display_name="Pro",
-        price_monthly=1900,  # $19.00
-        files_per_month=1000,
-        llm_calls_per_month=500,
-        features=[
-            "1,000 file migrations/month",
-            "500 LLM-assisted migrations/month",
-            "All supported libraries",
-            "Priority support",
-            "Custom knowledge bases",
-        ],
-    ),
-    "unlimited": TierInfo(
-        name="unlimited",
-        display_name="Unlimited",
-        price_monthly=4900,  # $49.00
-        files_per_month=999999999,
-        llm_calls_per_month=999999999,
-        features=[
-            "Unlimited file migrations",
-            "Unlimited LLM-assisted migrations",
-            "All supported libraries",
-            "Priority support",
-            "Custom knowledge bases",
-            "Usage analytics",
-        ],
-    ),
-    "enterprise": TierInfo(
-        name="enterprise",
-        display_name="Enterprise",
-        price_monthly=0,  # Custom pricing
-        files_per_month=999999999,
-        llm_calls_per_month=999999999,
-        features=[
-            "Unlimited migrations",
-            "Dedicated support",
-            "Custom integrations",
-            "SLA guarantees",
-            "Self-hosted option",
-            "SSO/SAML",
-        ],
-    ),
-}
-
-
-def get_stripe_client() -> stripe:
-    """Get configured Stripe client."""
-    settings = get_settings()
-    stripe.api_key = settings.stripe_secret_key
-    return stripe
-
-
-@router.get("/tiers", response_model=list[TierInfo])
-async def list_tiers() -> list[TierInfo]:
-    """List all available pricing tiers."""
-    return list(TIERS.values())
-
-
-@router.get("/tiers/{tier_name}", response_model=TierInfo)
-async def get_tier(tier_name: str) -> TierInfo:
-    """Get details about a specific tier."""
-    tier = TIERS.get(tier_name)
-    if not tier:
-        raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND,
-            detail=f"Tier '{tier_name}' not found",
-        )
-    return tier
-
-
-@router.get("/subscription", response_model=SubscriptionInfo)
-async def get_subscription(user: CurrentUser) -> SubscriptionInfo:
-    """Get the current user's subscription information."""
-    db = get_database()
-    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",
-        )
-
-    # If user has a Stripe subscription, fetch details
-    subscription_id = profile.get("stripe_subscription_id")
-    if subscription_id:
-        try:
-            stripe_client = get_stripe_client()
-            subscription = stripe_client.Subscription.retrieve(subscription_id)
-
-            return SubscriptionInfo(
-                tier=profile.get("tier", "free"),
-                status=subscription.status,
-                stripe_subscription_id=subscription_id,
-                current_period_start=subscription.current_period_start,
-                current_period_end=subscription.current_period_end,
-                cancel_at_period_end=subscription.cancel_at_period_end,
-            )
-        except stripe.error.StripeError:
-            pass  # Fall through to basic response
-
-    return SubscriptionInfo(
-        tier=profile.get("tier", "free"),
-        status="active" if profile.get("tier", "free") != "free" else "free",
-    )
-
-
-@router.get("/overview", response_model=BillingOverview)
-async def get_billing_overview(user: CurrentUser) -> BillingOverview:
-    """Get complete billing overview for the current user."""
-    db = get_database()
-    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")
-    tier_info = TIERS.get(tier, TIERS["free"])
-
-    subscription = SubscriptionInfo(
-        tier=tier,
-        status="active" if tier != "free" else "free",
-    )
-
-    payment_method: PaymentMethodInfo | None = None
-
-    # Fetch Stripe details if available
-    subscription_id = profile.get("stripe_subscription_id")
-
-    if subscription_id:
-        try:
-            stripe_client = get_stripe_client()
-            sub = stripe_client.Subscription.retrieve(subscription_id)
-
-            subscription = SubscriptionInfo(
-                tier=tier,
-                status=sub.status,
-                stripe_subscription_id=subscription_id,
-                current_period_start=sub.current_period_start,
-                current_period_end=sub.current_period_end,
-                cancel_at_period_end=sub.cancel_at_period_end,
-            )
-
-            # Get payment method
-            if sub.default_payment_method:
-                pm = stripe_client.PaymentMethod.retrieve(sub.default_payment_method)
-                if pm.type == "card" and pm.card:
-                    payment_method = PaymentMethodInfo(
-                        id=pm.id,
-                        type=pm.type,
-                        card_brand=pm.card.brand,
-                        card_last4=pm.card.last4,
-                        card_exp_month=pm.card.exp_month,
-                        card_exp_year=pm.card.exp_year,
-                    )
-        except stripe.error.StripeError:
-            pass
-
-    return BillingOverview(
-        subscription=subscription,
-        tier_info=tier_info,
-        payment_method=payment_method,
-    )
-
-
-@router.post("/checkout", response_model=CheckoutSessionResponse)
-async def create_checkout_session(
-    request: CheckoutSessionRequest,
-    user: CurrentUser,
-) -> CheckoutSessionResponse:
-    """Create a Stripe checkout session for upgrading subscription."""
-    settings = get_settings()
-    stripe_client = get_stripe_client()
-
-    # Get or create Stripe customer
-    db = get_database()
-    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",
-        )
-
-    customer_id = profile.get("stripe_customer_id")
-
-    if not customer_id:
-        # Create Stripe customer
-        customer = stripe_client.Customer.create(
-            email=profile["email"],
-            metadata={
-                "user_id": user.user_id,
-            },
-        )
-        customer_id = customer.id
-
-        # Update profile with customer ID
-        db.update_profile(user.user_id, {"stripe_customer_id": customer_id})
-
-    # Get the price ID for the requested tier
-    if request.tier == "pro":
-        price_id = settings.stripe_price_id_pro
-    elif request.tier == "unlimited":
-        price_id = settings.stripe_price_id_unlimited
-    else:
-        raise HTTPException(
-            status_code=status.HTTP_400_BAD_REQUEST,
-            detail=f"Invalid tier: {request.tier}",
-        )
-
-    if not price_id:
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail="Stripe price not configured",
-        )
-
-    # Create checkout session
-    success_url = request.success_url or f"{settings.codeshift_api_url}/billing/success"
-    cancel_url = request.cancel_url or f"{settings.codeshift_api_url}/billing/cancel"
-
-    try:
-        session = stripe_client.checkout.Session.create(
-            customer=customer_id,
-            mode="subscription",
-            line_items=[
-                {
-                    "price": price_id,
-                    "quantity": 1,
-                }
-            ],
-            success_url=success_url + "?session_id={CHECKOUT_SESSION_ID}",
-            cancel_url=cancel_url,
-            metadata={
-                "user_id": user.user_id,
-                "tier": request.tier,
-            },
-        )
-    except stripe.error.StripeError as e:
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail=f"Failed to create checkout session: {str(e)}",
-        ) from e
-
-    return CheckoutSessionResponse(
-        checkout_url=session.url,
-        session_id=session.id,
-    )
-
-
-@router.get("/portal", response_model=PortalSessionResponse)
-async def create_portal_session(user: CurrentUser) -> PortalSessionResponse:
-    """Create a Stripe billing portal session for managing subscription."""
-    settings = get_settings()
-    stripe_client = get_stripe_client()
-
-    db = get_database()
-    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",
-        )
-
-    customer_id = profile.get("stripe_customer_id")
-
-    if not customer_id:
-        raise HTTPException(
-            status_code=status.HTTP_400_BAD_REQUEST,
-            detail="No billing account found. Please subscribe first.",
-        )
-
-    try:
-        session = stripe_client.billing_portal.Session.create(
-            customer=customer_id,
-            return_url=f"{settings.codeshift_api_url}/billing",
-        )
-    except stripe.error.StripeError as e:
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail=f"Failed to create portal session: {str(e)}",
-        ) from e
-
-    return PortalSessionResponse(portal_url=session.url)
-
-
-@router.post("/cancel")
-async def cancel_subscription(user: CurrentUser) -> dict:
-    """Cancel the current subscription at end of billing period."""
-    stripe_client = get_stripe_client()
-
-    db = get_database()
-    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",
-        )
-
-    subscription_id = profile.get("stripe_subscription_id")
-
-    if not subscription_id:
-        raise HTTPException(
-            status_code=status.HTTP_400_BAD_REQUEST,
-            detail="No active subscription found",
-        )
-
-    try:
-        stripe_client.Subscription.modify(
-            subscription_id,
-            cancel_at_period_end=True,
-        )
-    except stripe.error.StripeError as e:
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail=f"Failed to cancel subscription: {str(e)}",
-        ) from e
-
-    return {"message": "Subscription will be canceled at end of billing period"}
-
-
-@router.post("/reactivate")
-async def reactivate_subscription(user: CurrentUser) -> dict:
-    """Reactivate a subscription that was set to cancel."""
-    stripe_client = get_stripe_client()
-
-    db = get_database()
-    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",
-        )
-
-    subscription_id = profile.get("stripe_subscription_id")
-
-    if not subscription_id:
-        raise HTTPException(
-            status_code=status.HTTP_400_BAD_REQUEST,
-            detail="No subscription found",
-        )
-
-    try:
-        stripe_client.Subscription.modify(
-            subscription_id,
-            cancel_at_period_end=False,
-        )
-    except stripe.error.StripeError as e:
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail=f"Failed to reactivate subscription: {str(e)}",
-        ) from e
-
-    return {"message": "Subscription reactivated"}

codeshift/api/routers/migrate.py

@@ -1,304 +0,0 @@
-"""Migration router for LLM-powered code migrations.
-
-This router handles all LLM-powered migrations (Tier 2/3).
-The Anthropic API calls are made server-side, ensuring:
-1. Users don't need their own API keys
-2. Usage can be tracked and billed
-3. The LLM prompts remain server-side
-"""
-
-import re
-from typing import Annotated
-
-from anthropic import Anthropic
-from fastapi import APIRouter, Depends, HTTPException, status
-
-from codeshift.api.auth import AuthenticatedUser, require_tier
-from codeshift.api.config import get_settings
-from codeshift.api.database import get_database
-from codeshift.api.models.migrate import (
-    ExplainChangeRequest,
-    ExplainChangeResponse,
-    MigrateCodeRequest,
-    MigrateCodeResponse,
-)
-
-router = APIRouter(prefix="/migrate", tags=["migrate"])
-
-
-def get_anthropic_client():  # type: ignore[no-untyped-def]
-    """Get the server-side Anthropic client."""
-
-    settings = get_settings()
-    if not settings.anthropic_api_key:
-        raise HTTPException(
-            status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
-            detail="LLM service not configured",
-        )
-
-    return Anthropic(api_key=settings.anthropic_api_key)
-
-
-def check_llm_quota(user: AuthenticatedUser, quantity: int = 1) -> None:
-    """Check if user has remaining LLM quota.
-
-    Raises HTTPException if quota exceeded.
-    """
-    db = get_database()
-    quota = db.get_user_quota(user.user_id)
-
-    if not quota:
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail="Could not retrieve quota information",
-        )
-
-    # Get tier limits
-    tier_limits = {
-        "free": 0,  # Free tier cannot use LLM
-        "pro": 100,
-        "unlimited": 999999,
-        "enterprise": 999999,
-    }
-
-    limit = tier_limits.get(user.tier, 0)
-    current_usage = quota.get("llm_calls", 0)
-
-    if current_usage + quantity > limit:
-        raise HTTPException(
-            status_code=status.HTTP_402_PAYMENT_REQUIRED,
-            detail={
-                "error": "LLM quota exceeded",
-                "current_usage": current_usage,
-                "limit": limit,
-                "tier": user.tier,
-                "upgrade_url": "https://codeshift.dev/pricing",
-            },
-        )
-
-
-def record_llm_usage(user: AuthenticatedUser, library: str, tokens_used: dict) -> None:
-    """Record LLM usage for billing."""
-    db = get_database()
-    db.record_usage_event(
-        user_id=user.user_id,
-        event_type="llm_call",
-        library=library,
-        quantity=1,
-        metadata={
-            "input_tokens": tokens_used.get("input_tokens", 0),
-            "output_tokens": tokens_used.get("output_tokens", 0),
-        },
-    )
-
-
-def extract_code_from_response(content: str) -> str:
-    """Extract Python code from LLM response."""
-    # Try to find code block
-    code_block_pattern = r"```(?:python)?\n(.*?)```"
-    matches = re.findall(code_block_pattern, content, re.DOTALL)
-
-    if matches:
-        # Return the longest code block (likely the full migration)
-        longest_match: str = max(matches, key=len)
-        return longest_match.strip()
-
-    # No code block found, assume the entire response is code
-    content = content.strip()
-    if content.startswith("```"):
-        content = content[3:]
-    if content.endswith("```"):
-        content = content[:-3]
-
-    return content.strip()
-
-
-def validate_syntax(code: str) -> bool:
-    """Quick syntax validation."""
-    try:
-        compile(code, "<string>", "exec")
-        return True
-    except SyntaxError:
-        return False
-
-
-@router.post("/code", response_model=MigrateCodeResponse)
-async def migrate_code(
-    request: MigrateCodeRequest,
-    user: Annotated[AuthenticatedUser, Depends(require_tier("pro"))],
-) -> MigrateCodeResponse:
-    """Migrate code using LLM.
-
-    This endpoint requires Pro tier or higher.
-    The LLM call is made server-side using PyResolve's Anthropic API key.
-    """
-    # Check quota
-    check_llm_quota(user)
-
-    # Get Anthropic client
-    client = get_anthropic_client()
-
-    # Build the prompt
-    system_prompt = f"""You are an expert Python developer specializing in code migrations.
-Your task is to migrate Python code from {request.library} v{request.from_version} to v{request.to_version}.
-
-Guidelines:
-1. Only modify code that needs to change for the migration
-2. Preserve all comments, formatting, and code style where possible
-3. Add brief inline comments explaining non-obvious changes
-4. If you're unsure about a change, add a TODO comment
-5. Return ONLY the migrated code, no explanations before or after
-
-Important {request.library} migration changes to consider:
-- Config class -> model_config = ConfigDict(...)
-- @validator -> @field_validator with @classmethod
-- @root_validator -> @model_validator with @classmethod
-- .dict() -> .model_dump()
-- .json() -> .model_dump_json()
-- .schema() -> .model_json_schema()
-- .parse_obj() -> .model_validate()
-- .parse_raw() -> .model_validate_json()
-- .copy() -> .model_copy()
-- orm_mode -> from_attributes
-- Field(regex=...) -> Field(pattern=...)
-"""
-
-    user_prompt = f"""Migrate the following Python code from {request.library} v{request.from_version} to v{request.to_version}.
-
-{f"Context: {request.context}" if request.context else ""}
-
-Code to migrate:
-```python
-{request.code}
-```
-
-Return only the migrated Python code:"""
-
-    try:
-        response = client.messages.create(
-            model="claude-sonnet-4-20250514",
-            max_tokens=4096,
-            temperature=0.0,
-            system=system_prompt,
-            messages=[{"role": "user", "content": user_prompt}],
-        )
-
-        # Extract content
-        content = ""
-        for block in response.content:
-            if hasattr(block, "text"):
-                content += block.text
-
-        # Extract code from response
-        migrated_code = extract_code_from_response(content)
-
-        # Validate syntax
-        if not validate_syntax(migrated_code):
-            return MigrateCodeResponse(
-                success=False,
-                migrated_code=request.code,
-                original_code=request.code,
-                error="LLM output has syntax errors",
-                usage={
-                    "input_tokens": response.usage.input_tokens,
-                    "output_tokens": response.usage.output_tokens,
-                },
-            )
-
-        # Record usage
-        record_llm_usage(
-            user,
-            request.library,
-            {
-                "input_tokens": response.usage.input_tokens,
-                "output_tokens": response.usage.output_tokens,
-            },
-        )
-
-        return MigrateCodeResponse(
-            success=True,
-            migrated_code=migrated_code,
-            original_code=request.code,
-            usage={
-                "input_tokens": response.usage.input_tokens,
-                "output_tokens": response.usage.output_tokens,
-            },
-        )
-
-    except Exception as e:
-        return MigrateCodeResponse(
-            success=False,
-            migrated_code=request.code,
-            original_code=request.code,
-            error=str(e),
-        )
-
-
-@router.post("/explain", response_model=ExplainChangeResponse)
-async def explain_change(
-    request: ExplainChangeRequest,
-    user: Annotated[AuthenticatedUser, Depends(require_tier("pro"))],
-) -> ExplainChangeResponse:
-    """Explain a migration change using LLM.
-
-    This endpoint requires Pro tier or higher.
-    """
-    # Check quota
-    check_llm_quota(user)
-
-    # Get Anthropic client
-    client = get_anthropic_client()
-
-    system_prompt = """You are an expert Python developer.
-Explain code changes clearly and concisely for other developers.
-Focus on the 'why' not just the 'what'."""
-
-    user_prompt = f"""Explain the following {request.library} migration change:
-
-Original:
-```python
-{request.original_code}
-```
-
-Migrated:
-```python
-{request.transformed_code}
-```
-
-Provide a brief explanation (2-3 sentences) of what changed and why:"""
-
-    try:
-        response = client.messages.create(
-            model="claude-sonnet-4-20250514",
-            max_tokens=500,
-            temperature=0.0,
-            system=system_prompt,
-            messages=[{"role": "user", "content": user_prompt}],
-        )
-
-        # Extract content
-        content = ""
-        for block in response.content:
-            if hasattr(block, "text"):
-                content += block.text
-
-        # Record usage
-        record_llm_usage(
-            user,
-            request.library,
-            {
-                "input_tokens": response.usage.input_tokens,
-                "output_tokens": response.usage.output_tokens,
-            },
-        )
-
-        return ExplainChangeResponse(
-            success=True,
-            explanation=content.strip(),
-        )
-
-    except Exception as e:
-        return ExplainChangeResponse(
-            success=False,
-            error=str(e),
-        )