stripe integration

tools/create_invoice.py

@@ -1,5 +1,72 @@
+import os
+import stripe
+from dotenv import load_dotenv
+
+load_dotenv()  # Load environment variables from a .env file
+
+stripe.api_key = os.getenv("STRIPE_API_KEY", "YOUR_DEFAULT_KEY")
+
+
+def ensure_customer_exists(
+    customer_id: str = None, email: str = "default@example.com"
+) -> str:
+    """Ensure a Stripe customer exists; create one if not."""
+    if customer_id:
+        try:
+            stripe.Customer.retrieve(customer_id)
+            return customer_id
+        except stripe.error.InvalidRequestError:
+            # Customer ID is invalid or doesn't exist
+            pass
+
+    # Create a new customer if no valid customer_id
+    customer = stripe.Customer.create(email=email)
+    return customer.id
+
+
 def create_invoice(args: dict) -> dict:
-    # e.g. amount, flight details, etc.
+    """Create and finalize a Stripe invoice."""
+    # Find or create customer
+    customer_id = ensure_customer_exists(
+        args.get("customer_id"), args.get("email", "default@example.com")
+    )
+
+    # Get amount and convert to cents
+    amount = args.get("amount", 200.00)  # Default to $200.00
+    try:
+        amount_cents = int(float(amount) * 100)
+    except (TypeError, ValueError):
+        return {"error": "Invalid amount provided. Please confirm the amount."}
+
+    # Create an invoice item
+    stripe.InvoiceItem.create(
+        customer=customer_id,
+        amount=amount_cents,
+        currency="usd",
+        description=args.get("flightDetails", "Service Invoice"),
+    )
+
+    # Create and finalize the invoice
+    invoice = stripe.Invoice.create(
+        customer=customer_id,
+        collection_method="send_invoice",  # Invoice is sent to the customer
+        days_until_due=args.get("days_until_due", 7),  # Default due date: 7 days
+        pending_invoice_items_behavior="include",  # No pending invoice items
+    )
+    finalized_invoice = stripe.Invoice.finalize_invoice(invoice.id)
+
+    return {
+        "invoiceStatus": finalized_invoice.status,
+        "invoiceURL": finalized_invoice.hosted_invoice_url,
+        "reference": finalized_invoice.number,
+    }
+
+
+def create_invoice_example(args: dict) -> dict:
+    """
+    This is an example implementation of the CreateInvoice tool
+    Doesn't call any external services, just returns a dummy response
+    """
     print("[CreateInvoice] Creating invoice with:", args)
     return {
         "invoiceStatus": "generated",