From 383bd15263641d9e87a0fe4b63c0b3fff297e3e9 Mon Sep 17 00:00:00 2001
From: nicoagostini <nic.voy@hotmail.com>
Date: Fri, 9 May 2025 14:03:35 -0700
Subject: [PATCH] Issue 19: Questionnaire more than once, fixed

---
 src/router/user.js | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/router/user.js b/src/router/user.js
index 9286457..e4511ad 100644
--- a/src/router/user.js
+++ b/src/router/user.js
@@ -163,7 +163,7 @@ module.exports = (middleware, users, plans, assets) => {
     });
 
     router.get('/newPlan', (req, res) => {
-        if (!req.session.user.financialData) {
+        if (!req.session.user.financialData || !req.session.user) {
             req.session.errMessage = "Please complete your financial data before creating a plan.";
             return res.status(status.Unauthorized).redirect('/questionnaire');
         }
@@ -299,7 +299,13 @@ module.exports = (middleware, users, plans, assets) => {
 
             req.session.user.financialData = true;
             req.session.errMessage = "";
-            res.status(status.Ok).redirect("/home");
+
+            req.session.save(err => {
+                if (err) {
+                    res.status(status.InternalServerError).redirect("/plans");
+                }
+                res.status(status.Ok).redirect("/plans");
+            });
 
         }).catch(err => {
             console.error("Error updating questionnaire in database:", err);