import { Router, Request, Response } from "express";
import Stripe from "stripe";
import { createProKey, revokeByCustomer } from "../services/keys.js";

let _stripe: Stripe | null = null;
function getStripe(): Stripe {
  if (!_stripe) {
    const key = process.env.STRIPE_SECRET_KEY;
    if (!key) throw new Error("STRIPE_SECRET_KEY not configured");
    _stripe = new Stripe(key, { apiVersion: "2025-01-27.acacia" as any });
  }
  return _stripe;
}

const router = Router();

// Create a Stripe Checkout session for Pro subscription
router.post("/checkout", async (_req: Request, res: Response) => {
  try {
    const priceId = await getOrCreateProPrice();

    const session = await getStripe().checkout.sessions.create({
      mode: "subscription",
      payment_method_types: ["card"],
      line_items: [{ price: priceId, quantity: 1 }],
      success_url: `${process.env.BASE_URL || "https://docfast.dev"}/v1/billing/success?session_id={CHECKOUT_SESSION_ID}`,
      cancel_url: `${process.env.BASE_URL || "https://docfast.dev"}/#pricing`,
    });

    res.json({ url: session.url });
  } catch (err: any) {
    console.error("Checkout error:", err.message);
    res.status(500).json({ error: "Failed to create checkout session" });
  }
});

// Success page — provision Pro API key after checkout
router.get("/success", async (req: Request, res: Response) => {
  const sessionId = req.query.session_id as string;
  if (!sessionId) {
    res.status(400).json({ error: "Missing session_id" });
    return;
  }

  try {
    const session = await getStripe().checkout.sessions.retrieve(sessionId);
    const customerId = session.customer as string;
    const email = session.customer_details?.email || "unknown@docfast.dev";

    if (!customerId) {
      res.status(400).json({ error: "No customer found" });
      return;
    }

    const keyInfo = createProKey(email, customerId);

    // Return a nice HTML page instead of raw JSON
    res.send(`<!DOCTYPE html>
<html><head><title>Welcome to DocFast Pro!</title>
<style>
body { font-family: system-ui; background: #0a0a0a; color: #e8e8e8; display: flex; align-items: center; justify-content: center; min-height: 100vh; margin: 0; }
.card { background: #141414; border: 1px solid #222; border-radius: 16px; padding: 48px; max-width: 500px; text-align: center; }
h1 { color: #4f9; margin-bottom: 8px; }
.key { background: #1a1a1a; border: 1px solid #333; border-radius: 8px; padding: 16px; margin: 24px 0; font-family: monospace; font-size: 0.9rem; word-break: break-all; cursor: pointer; }
.key:hover { border-color: #4f9; }
p { color: #888; line-height: 1.6; }
a { color: #4f9; }
</style></head><body>
<div class="card">
<h1>🎉 Welcome to Pro!</h1>
<p>Your API key:</p>
<div class="key" onclick="navigator.clipboard.writeText('${keyInfo.key}')" title="Click to copy">${keyInfo.key}</div>
<p><strong>Save this key!</strong> It won't be shown again.</p>
<p>10,000 PDFs/month • All endpoints • Priority support</p>
<p><a href="/docs">View API docs →</a></p>
</div></body></html>`);
  } catch (err: any) {
    console.error("Success page error:", err.message);
    res.status(500).json({ error: "Failed to retrieve session" });
  }
});

// Stripe webhook for subscription lifecycle events
router.post("/webhook", async (req: Request, res: Response) => {
  const sig = req.headers["stripe-signature"] as string;
  const webhookSecret = process.env.STRIPE_WEBHOOK_SECRET;

  let event: Stripe.Event;

  if (webhookSecret && sig) {
    try {
      event = getStripe().webhooks.constructEvent(req.body, sig, webhookSecret);
    } catch (err: any) {
      console.error("Webhook signature verification failed:", err.message);
      res.status(400).json({ error: "Invalid signature" });
      return;
    }
  } else {
    event = req.body as Stripe.Event;
  }

  switch (event.type) {
    case "customer.subscription.deleted": {
      const sub = event.data.object as Stripe.Subscription;
      const customerId = sub.customer as string;
      revokeByCustomer(customerId);
      console.log(`Subscription cancelled for ${customerId}, key revoked`);
      break;
    }
    default:
      break;
  }

  res.json({ received: true });
});

// --- Price management ---
let cachedPriceId: string | null = null;

async function getOrCreateProPrice(): Promise<string> {
  if (cachedPriceId) return cachedPriceId;

  const products = await getStripe().products.search({ query: "name:'DocFast Pro'" });
  let productId: string;

  if (products.data.length > 0) {
    productId = products.data[0].id;
    const prices = await getStripe().prices.list({ product: productId, active: true, limit: 1 });
    if (prices.data.length > 0) {
      cachedPriceId = prices.data[0].id;
      return cachedPriceId;
    }
  } else {
    const product = await getStripe().products.create({
      name: "DocFast Pro",
      description: "Unlimited PDF conversions via API. HTML, Markdown, and URL to PDF.",
    });
    productId = product.id;
  }

  const price = await getStripe().prices.create({
    product: productId,
    unit_amount: 900,
    currency: "usd",
    recurring: { interval: "month" },
  });

  cachedPriceId = price.id;
  return cachedPriceId;
}

export { router as billingRouter };
