feat: add Stripe customer creation during user initialization

Co-Authored-By: WeWrite <getwewrite@gmail.com>

Author: devin-ai-integration[bot]

app/api/payments/create-customer/route.js

@@ -1,11 +1,34 @@
 import { NextResponse } from 'next/server';
 import Stripe from 'stripe';
+import { getDatabase } from '@/firebase/database';
 
 const stripe = new Stripe(process.env.STRIPE_SECRET_KEY);
+const db = getDatabase();
 
-export async function POST() {
+export async function POST(request) {
   try {
-    const customer = await stripe.customers.create();
+    const { userId, email, name } = await request.json();
+
+    if (!userId) {
+      return NextResponse.json(
+        { error: 'User ID is required' },
+        { status: 400 }
+      );
+    }
+
+    // Create a Stripe customer with user information
+    const customer = await stripe.customers.create({
+      email,
+      name,
+      metadata: {
+        userId,
+      },
+    });
+
+    // Update user record with Stripe customer ID
+    await db.ref(`users/${userId}`).update({
+      stripeCustomerId: customer.id,
+    });
 
     return NextResponse.json({
       customerId: customer.id

app/providers/AuthProvider.js

@@ -2,8 +2,10 @@
 
 import { useEffect, useState, createContext, useContext } from "react";
 import { auth, onAuthStateChanged } from "../firebase/auth";
+import { getDatabase } from "../firebase/database";
 
 export const AuthContext = createContext();
+const db = getDatabase();
 
 export const useAuth = () => {
   const context = useContext(AuthContext);
@@ -18,6 +20,32 @@ export const AuthProvider = ({ children }) => {
   const [loading, setLoading] = useState(true);
   const [error, setError] = useState(null);
 
+  const createStripeCustomer = async (userData) => {
+    try {
+      const response = await fetch('/api/payments/create-customer', {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({
+          userId: userData.uid,
+          email: userData.email,
+          name: userData.displayName,
+        }),
+      });
+
+      if (!response.ok) {
+        throw new Error('Failed to create Stripe customer');
+      }
+
+      const { customerId } = await response.json();
+      return customerId;
+    } catch (error) {
+      console.error('Error creating Stripe customer:', error);
+      return null;
+    }
+  };
+
   useEffect(() => {
     if (typeof window === 'undefined') {
       setLoading(false);
@@ -26,15 +54,24 @@ export const AuthProvider = ({ children }) => {
 
     let unsubscribe;
     try {
-      unsubscribe = onAuthStateChanged(auth, (user) => {
+      unsubscribe = onAuthStateChanged(auth, async (authUser) => {
         try {
-          if (user) {
+          if (authUser) {
+            const userSnapshot = await db.ref(`users/${authUser.uid}`).get();
+            const userData = userSnapshot.val() || {};
+
+            let stripeCustomerId = userData.stripeCustomerId;
+            if (!stripeCustomerId) {
+              stripeCustomerId = await createStripeCustomer(authUser);
+            }
+
             setUser({
-              uid: user.uid,
-              email: user.email,
-              displayName: user.displayName,
-              username: user.username || user.displayName,
-              groups: user.groups || ['default-group']
+              uid: authUser.uid,
+              email: authUser.email,
+              displayName: authUser.displayName,
+              username: userData.username || authUser.displayName,
+              groups: userData.groups || ['default-group'],
+              stripeCustomerId,
             });
           } else {
             setUser(null);