fix: Fix the session state

Author: LubomirGeorgiev

src/app/(dashboard)/dashboard/billing/_components/credit-packages.tsx

@@ -37,9 +37,9 @@ export function CreditPackages() {
   const [isDialogOpen, setIsDialogOpen] = useState(false);
   const [selectedPackage, setSelectedPackage] = useState<CreditPackage | null>(null);
   const [clientSecret, setClientSecret] = useState<string | null>(null);
-  const session = useSessionStore((state) => state.session);
+  const session = useSessionStore((state) => state);
   const transactionsRefresh = useTransactionStore((state) => state.triggerRefresh);
-  const sessionIsLoading = useSessionStore((state) => state.isLoading);
+  const sessionIsLoading = session?.isLoading;
 
   const handlePurchase = async (pkg: CreditPackage) => {
     try {
@@ -78,7 +78,7 @@ export function CreditPackages() {
                 </>
               ) : (
                 <div className="text-3xl font-bold">
-                  {session?.user.currentCredits.toLocaleString()} credits
+                  {session?.session?.user?.currentCredits.toLocaleString()} credits
                 </div>
               )}
             </div>

src/state/session.ts

@@ -8,11 +8,19 @@ export const useSessionStore = create(
       session: null as SessionValidationResult | null,
       isLoading: true,
       lastFetched: null as Date | null,
+      isInitialized: false,
     },
     (set) => ({
-      setSession: (session: SessionValidationResult) => set({ session, isLoading: false, lastFetched: new Date() }),
-      clearSession: () => set({ session: null, isLoading: false, lastFetched: null }),
-      refetchSession: () => set({ isLoading: true, lastFetched: null }),
+      setSession: (session: SessionValidationResult) =>
+        set({ session, isLoading: false, lastFetched: new Date(), isInitialized: true }),
+      clearSession: () =>
+        set({ session: null, isLoading: false, lastFetched: null, isInitialized: true }),
+      refetchSession: () =>
+        set((state) => ({
+          ...state,
+          isLoading: !state.isInitialized,
+          lastFetched: null
+        })),
     })
   )
 )