From f8bd1ddb43f05bc9adac4b8aea64a185cf88162b Mon Sep 17 00:00:00 2001
From: Austin Kelsay <53542748+AustinKelsay@users.noreply.github.com>
Date: Mon, 12 May 2025 09:36:43 -0500
Subject: [PATCH] Update src/hooks/courses/useCoursePayment.js

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
---
 src/hooks/courses/useCoursePayment.js | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/src/hooks/courses/useCoursePayment.js b/src/hooks/courses/useCoursePayment.js
index 4743cf1..c2740b8 100644
--- a/src/hooks/courses/useCoursePayment.js
+++ b/src/hooks/courses/useCoursePayment.js
@@ -34,11 +34,23 @@ const useCoursePayment = (course) => {
   
   // Handler for successful payment
   const handlePaymentSuccess = useCallback(async (response) => {
-    if (response && response?.preimage) {
-      // Update session to reflect purchase
-      const updated = await update();
-      showToast('success', 'Payment Success', 'You have successfully purchased this course');
-      return true;
+    if (response?.preimage) {
+      try {
+        await update(); // refresh session
+        showToast(
+          'success',
+          'Payment Success',
+          'You have successfully purchased this course'
+        );
+        return true;
+      } catch (err) {
+        showToast(
+          'warn',
+          'Session Refresh Failed',
+          'Purchase succeeded but we could not refresh your session automatically. Please reload the page.'
+        );
+        return false;
+      }
     } else {
       showToast('error', 'Error', 'Failed to purchase course. Please try again.');
       return false;