import stripe
from pydantic import BaseModel

stripe.api_key = os.getenv("STRIPE_SECRET_KEY")

class CheckoutRequest(BaseModel):
    amount_cents: int
    customer_email: str
    invoice_id: str

async def create_checkout_session(request: CheckoutRequest):
    """Create Stripe checkout session for invoice payment."""
    session = stripe.checkout.Session.create(
        payment_method_types=['card'],
        line_items=[{
            'price_data': {
                'currency': 'usd',
                'product_data': {'name': f'Invoice #{request.invoice_id}'},
                'unit_amount': request.amount_cents,
            },
            'quantity': 1,
        }],
        mode='payment',
        success_url=f'{BASE_URL}/payment/success?session_id={{CHECKOUT_SESSION_ID}}',
        cancel_url=f'{BASE_URL}/payment/cancel',
        customer_email=request.customer_email,
        metadata={'invoice_id': request.invoice_id}
    )
    return session.url

WEBHOOK_SECRET = os.getenv("STRIPE_WEBHOOK_SECRET")

@app.post("/webhooks/stripe")
async def stripe_webhook(request: Request):
    payload = await request.body()
    sig = request.headers.get("Stripe-Signature")

    # ALWAYS verify signature
    try:
        event = stripe.Webhook.construct_event(payload, sig, WEBHOOK_SECRET)
    except stripe.error.SignatureVerificationError:
        raise HTTPException(status_code=400, detail="Invalid signature")

    # Handle events idempotently
    event_id = event["id"]
    if await is_processed(event_id):
        return {"status": "duplicate"}

    match event["type"]:
        case "payment_intent.succeeded":
            await handle_payment_success(event["data"]["object"])
        case "payment_intent.payment_failed":
            await handle_payment_failure(event["data"]["object"])
        case "customer.subscription.deleted":
            await handle_subscription_cancel(event["data"]["object"])

    await mark_processed(event_id)
    return {"status": "ok"}

async def handle_payment_success(payment_intent):
    """Sync successful payment to QuickBooks."""
    invoice_id = payment_intent["metadata"]["invoice_id"]

    # Mark invoice as paid in QBO
    await qbo_mark_invoice_paid(invoice_id, payment_intent["id"])

    # Log activity
    await log_activity("payment_received", {
        "invoice_id": invoice_id,
        "amount": payment_intent["amount"] / 100,
        "stripe_id": payment_intent["id"]
    })

    # Send confirmation email
    await send_payment_confirmation(payment_intent)

async def sync_payment_to_qbo(payment_intent):
    """Record Stripe payment in QuickBooks."""
    invoice_id = payment_intent["metadata"]["invoice_id"]

    payment = {
        "TotalAmt": payment_intent["amount"] / 100,
        "CustomerRef": {"value": payment_intent["customer"]},
        "Line": [{
            "Amount": payment_intent["amount"] / 100,
            "LinkedTxn": [{
                "TxnId": invoice_id,
                "TxnType": "Invoice"
            }]
        }],
        "PrivateNote": f"Stripe: {payment_intent['id']}"
    }

    return await qbo_create_payment(payment)

# .env.example
STRIPE_SECRET_KEY=sk_test_...
STRIPE_PUBLISHABLE_KEY=pk_test_...
STRIPE_WEBHOOK_SECRET=whsec_...