Session Via Firebase User Ok!

app.py

@@ -98,11 +98,16 @@ async def creaCliente(
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Ocurrió un error inesperado al procesar la solicitud: {str(e)}")
 
-
 @app.post("/creaLinkSesion/")
 async def creaLinkSesion(
-    price_id: str = Form(...), # Sigue siendo requerido, enviado como Form si la petición es multipart/form-data
-    customer_email: Optional[str] = Form(None), # Ahora opcional, valor por defecto None
-    customer_id: Optional[str] = Form(None)     # Ahora opcional, valor por defecto None
+    price_id: str = Form(...),
+    customer_email: Optional[str] = Form(None),
+    customer_id: Optional[str] = Form(None),
+    firebase_user: Optional[str] = Form(None) # <-- ¡Nuevo parámetro para Firebase User ID!
 ):
-    return funciones.create_checkout_session(price_id, customer_email, customer_id)
+    """
+    Crea una Checkout Session, ahora con la opción de buscar/crear cliente
+    usando firebase_user.
+    """
+    # Llama a la función de Stripe con el nuevo parámetro
+    return funciones.create_checkout_session(price_id, customer_email, customer_id, firebase_user)

funciones.py

@@ -3,53 +3,116 @@ import globales
 
 stripe.api_key = globales.llave
    
-def create_checkout_session(price_id, customer_email=None, customer_id=None):
+def create_checkout_session(price_id, customer_email=None, customer_id=None, firebase_user=None):
     """
     Crea una nueva Checkout Session para un pago único.
+    Busca o crea un cliente en Stripe si no se provee customer_id ni customer_email,
+    usando firebase_user como método de búsqueda principal si está disponible.
 
     Args:
         price_id (str): El ID del precio en Stripe para el artículo que se va a comprar.
         customer_email (str, opcional): Email para pre-rellenar o asociar con un cliente existente.
         customer_id (str, opcional): ID de un cliente existente en Stripe para asociar la sesión.
+        firebase_user (str, opcional): ID de usuario de Firebase. Se usará para buscar
+                                       un cliente de Stripe si customer_id/email no están presentes.
 
     Returns:
         str: La URL de la Checkout Session, o None si hay un error.
     """
     try:
+        # --- Lógica para obtener o crear el Customer de Stripe ---
+        final_customer_id = customer_id # Empezamos con el customer_id si ya lo tenemos
+
+        if not final_customer_id: # Si no tenemos un customer_id directo, intentamos buscar/crear
+            found_customer = None
+            
+            # Prioridad 1: Buscar por firebase_user en metadata
+            if firebase_user:
+                search_query = f"metadata['firebase_user']:'{firebase_user}'"
+                print(f"Buscando cliente por Firebase UID: {firebase_user}")
+                existing_customers_by_firebase = stripe.Customer.search(query=search_query, limit=1)
+                if existing_customers_by_firebase.data:
+                    found_customer = existing_customers_by_firebase.data[0]
+                    print(f"Cliente encontrado por Firebase UID: {found_customer.id}")
+                    final_customer_id = found_customer.id
+
+            # Prioridad 2: Si no se encontró por firebase_user, y tenemos un email, buscar por email
+            if not final_customer_id and customer_email:
+                print(f"Cliente no encontrado por Firebase UID. Buscando por email: {customer_email}")
+                existing_customers_by_email = stripe.Customer.list(email=customer_email, limit=1)
+                if existing_customers_by_email.data:
+                    found_customer = existing_customers_by_email.data[0]
+                    print(f"Cliente encontrado por email: {found_customer.id}")
+                    final_customer_id = found_customer.id
+                    
+                    # Opcional: Si lo encontraste por email pero falta el firebase_user, actualízalo
+                    if firebase_user and (not found_customer.metadata or found_customer.metadata.get('firebase_user') != firebase_user):
+                        print(f"Actualizando firebase_user para cliente {found_customer.id}")
+                        updated_metadata = found_customer.metadata.copy() if found_customer.metadata is not None else {}
+                        updated_metadata['firebase_user'] = firebase_user
+                        stripe.Customer.modify(found_customer.id, metadata=updated_metadata)
+
+            # Si después de las búsquedas, todavía no tenemos un customer_id, y tenemos email, creamos uno nuevo
+            if not final_customer_id and customer_email:
+                print(f"Cliente no encontrado. Creando nuevo cliente con email: {customer_email}")
+                customer_metadata = {}
+                if firebase_user:
+                    customer_metadata['firebase_user'] = firebase_user
+                
+                new_customer = stripe.Customer.create(
+                    email=customer_email,
+                    metadata=customer_metadata if customer_metadata else None # Asegura que no se envíe un diccionario vacío si no hay metadata
+                )
+                final_customer_id = new_customer.id
+                print(f"Nuevo cliente creado: {final_customer_id}")
+            elif not final_customer_id and not customer_email and firebase_user:
+                # Si no hay email, pero hay firebase_user, también podríamos crear un cliente
+                # Esto es menos común pero posible si el email no es obligatorio para ti.
+                print(f"Cliente no encontrado. Creando nuevo cliente (sin email) con Firebase UID: {firebase_user}")
+                customer_metadata = {'firebase_user': firebase_user}
+                new_customer = stripe.Customer.create(
+                    metadata=customer_metadata
+                )
+                final_customer_id = new_customer.id
+                print(f"Nuevo cliente creado (sin email): {final_customer_id}")
+            
+            # Nota: Si no hay customer_id, customer_email, ni firebase_user, final_customer_id seguirá siendo None
+            # y la sesión de checkout se creará sin un cliente asociado explícitamente. Stripe puede crearlo
+            # automáticamente en el checkout si se recopila el email.
+        
+        print("Estoy por entrar a checar el customer id que es: ", final_customer_id)
+
+        # --- Parámetros de la Checkout Session ---
         session_params = {
             'line_items': [{
                 'price': price_id,
                 'quantity': 1,
             }],
-            'mode': 'payment',  #payment Para pagos únicos (one-time payment) o #subscription
-            'success_url': 'https://app.splashmix.ink/', #'https://tudominio.com/gracias?session_id={CHECKOUT_SESSION_ID}'
+            'mode': 'payment',
+            'success_url': 'https://app.splashmix.ink/',
             'cancel_url': 'https://app.splashmix.ink/',
-            'locale': 'es', #'es' # Establece el idioma del checkout a español
-            'client_reference_id': 'HERC'
+            'locale': 'es',
+            'client_reference_id': 'HERC',
+            'metadata': {} # Inicializamos metadata para añadir el customer_id y firebase_user si existen
         }
 
-        print("Estoy por entrar a checar el customer id que es: ", customer_id)
-
-        # Opciones para el cliente
-        if customer_id:            
-            session_params['customer'] = customer_id            
+        # Asociar el cliente encontrado/creado a la sesión
+        if final_customer_id:
+            session_params['customer'] = final_customer_id
+            session_params['metadata']['stripe_customer_id'] = final_customer_id # Útil para webhooks
+        
+        # Pre-rellenar el email si está disponible y no se asoció un customer_id directamente
+        # (Stripe recomienda usar 'customer' en lugar de 'customer_email' si ya tienes el ID del cliente)
         elif customer_email:
-            session_params['customer_email'] = customer_email
-            print("Salí del elif")
-        else:
-            print("Entré al else...")
-            # Si no se provee ni customer_id ni customer_email, Stripe puede crear uno si es necesario
-            # O podrías pedir el email en el formulario de Checkout.
-            pass
-
-        # Opcional: Agregar información de envío si aplica (para productos físicos)
-        # session_params['shipping_address_collection'] = {'allowed_countries': ['US', 'CA', 'MX']}
-
-        # Opcional: Permitir códigos promocionales
-        # session_params['allow_promotion_codes'] = True
-
-        # Opcional: Agregar metadata a la sesión (no al pago final directamente, pero útil para webhooks)
-        # session_params['metadata'] = {'order_id': 'ORD12345', 'user_id': 'user_abc'}
+             session_params['customer_email'] = customer_email
+
+        # Añadir firebase_user a la metadata de la sesión si está presente
+        if firebase_user:
+            session_params['metadata']['firebase_user'] = firebase_user
+
+        # Limpiar metadata si quedó vacía (Stripe no acepta diccionarios vacíos)
+        if not session_params['metadata']:
+            del session_params['metadata']
 
         session = stripe.checkout.Session.create(**session_params)