const SubscriptionButton = (props: Props) => {
  const [loading, setLoading] = React.useState(false);
  const router = useRouter();
  
  const handleSubscription = async () => {
    try {
      setLoading(true);
      const response = await fetch("/api/stripe");
      const data = await response.json();
      
      // Redirect to Stripe
      router.push(data.url);
    } catch (error) {
      console.error(error);
    } finally {
      setLoading(false);
    }
  };
  
  return (
    <Button disabled={loading} onClick={handleSubscription}>
      {props.isPro ? "Manage Subscriptions" : "Get Pro"}
    </Button>
  );
};

export const checkSubscription = async () => {
  const { userId } = await auth();
  if (!userId) return false;

  const _userSubscriptions = await db
    .select()
    .from(userSubscriptions)
    .where(eq(userSubscriptions.userId, userId));

  if (!_userSubscriptions[0]) return false;

  const userSubscription = _userSubscriptions[0];

  // Check if subscription is valid
  const isValid = 
    userSubscription.stripePriceId && 
    userSubscription.stripeCurrentPeriodEnd?.getTime()! + 
    1000 * 60 * 60 * 24 > Date.now();

  return !!isValid;
};

// For existing Pro users - go to billing portal
if (_userSubscriptions[0] && _userSubscriptions[0].stripeCustomerId) {
  const stripeSession = await stripe.billingPortal.sessions.create({
    customer: _userSubscriptions[0].stripeCustomerId,
    return_url,
  });
  return NextResponse.json({ url: stripeSession.url });
}

// For new subscribers - create checkout session
const stripeSession = await stripe.checkout.sessions.create({
  success_url: return_url,
  cancel_url: return_url,
  payment_method_types: ["card"],
  mode: "subscription",
  billing_address_collection: "auto",
  customer_email: user?.emailAddresses[0].emailAddress,
  line_items: [
    {
      price_data: {
        currency: "INR",
        product_data: {
          name: "Ai-PDF Pro",
          description: "countless hours of time saved",
        },
        unit_amount: 160000, // ₹1,600 in paisa
        recurring: {
          interval: "month",
        },
      },
      quantity: 1,
    },
  ],
  metadata: {
    userId,
  },
});

export const userSubscriptions = pgTable("user_subscriptions", {
  id: serial("id").primaryKey(),
  userId: varchar("user_id", { length: 256 }).notNull().unique(),
  stripeCustomerId: varchar("stripe_customer_id", { length: 256 })
    .notNull()
    .unique(),
  stripeSubscriptionId: varchar("stripe_subscription_id", {
    length: 256,
  }).unique(),
  stripePriceId: varchar("stripe_price_id", { length: 256 }),
  stripeCurrentPeriodEnd: timestamp("stripe_current_period_ended_at"),
});