From 3f4c60ab67a64079d8af3f6dc9adea2d013bc067 Mon Sep 17 00:00:00 2001
From: nmemmert <nmemmert@example.com>
Date: Mon, 27 Oct 2025 11:31:56 -0400
Subject: [PATCH] Restore Firestore save logic in SetupWizard with proper user
 authentication

---
 src/app/page.tsx               |  2 +-
 src/components/SetupWizard.tsx | 11 +++++++++--
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/app/page.tsx b/src/app/page.tsx
index 89b60f9..c719c13 100644
--- a/src/app/page.tsx
+++ b/src/app/page.tsx
@@ -567,7 +567,7 @@ export default function BudgetDashboard() {
 
       {/* Modals and Components */}
       {showAuthModal && <AuthModal isOpen={showAuthModal} onClose={() => setShowAuthModal(false)} onAuthSuccess={handleAuthSuccess} />}
-      {showSetupWizard && <SetupWizard onComplete={handleSetupComplete} onSkip={() => setShowSetupWizard(false)} />}
+      {showSetupWizard && <SetupWizard onComplete={handleSetupComplete} onSkip={() => setShowSetupWizard(false)} userId={user?.uid || ''} />}
       {editingTransaction && (
         <TransactionEdit
           transaction={editingTransaction}
diff --git a/src/components/SetupWizard.tsx b/src/components/SetupWizard.tsx
index 5ff8cb5..7ae97ce 100644
--- a/src/components/SetupWizard.tsx
+++ b/src/components/SetupWizard.tsx
@@ -28,11 +28,12 @@ interface Envelope {
 interface SetupWizardProps {
   onComplete: (accounts: Account[], envelopes: Envelope[]) => void;
   onSkip: () => void;
+  userId: string;
 }
 
 type WizardStep = 'welcome' | 'data-source' | 'accounts' | 'envelopes' | 'income-setup' | 'get-paid-setup' | 'complete';
 
-export default function SetupWizard({ onComplete, onSkip }: SetupWizardProps) {
+export default function SetupWizard({ onComplete, onSkip, userId }: SetupWizardProps) {
   const [currentStep, setCurrentStep] = useState<WizardStep>('welcome');
   const [dataSource, setDataSource] = useState<'import' | 'manual' | null>(null);
   const [accounts, setAccounts] = useState<Account[]>([]);
@@ -147,9 +148,15 @@ export default function SetupWizard({ onComplete, onSkip }: SetupWizardProps) {
         return account;
       });
 
+      await DataService.saveUserData(userId, {
+        accounts: accountsWithDefaults,
+        envelopes: envelopesWithAllocations,
+        transactions: [],
+        setupCompleted: true,
+      });
       onComplete(accountsWithDefaults, envelopesWithAllocations);
     } catch (error) {
-      console.error('Error completing setup:', error);
+      console.error('Error saving setup data:', error);
     }
   };