Merge pull request #4 from JoaquinPar/feature/calculations

Feature/calculations

app.js

@@ -38,6 +38,7 @@ async function initDatabase() {
     
     // For any collection, init here
     users = await getCollection(db, "users");
+    plans = await getCollection(db, "plans");
 }
 
 /*** ROUTINGS ***/
@@ -78,8 +79,8 @@ initDatabase().then(() => {
     app.use(require("./src/auth/authentication")(users));
 
     // Import middleware & apply to user routes
-    const middleware = require("./src/auth/middleware")(users);
-    app.use(require('./src/router/user')(middleware, users));
+    const middleware = require("./src/auth/middleware")(users, plans);
+    app.use(require('./src/router/user')(middleware, users, plans));
 
     // 404 handler
     app.get('/*splat', (req, res) => {

src/auth/authentication.js

@@ -40,6 +40,7 @@ module.exports = (users) => {
             }
 
             req.session.authenticated = true;
+            req.session.userId = user._id;
             req.session.email = req.body.email;
             req.session.errMessage = "";
             res.redirect("/home");
@@ -84,6 +85,7 @@ module.exports = (users) => {
 
             req.session.authenticated = true;
             req.session.email = req.body.email;
+            req.session.userId = results.insertedId;
 
             req.session.errMessage = "";
             return res.status(status.Ok).redirect("/home");

src/router/user.js

@@ -3,7 +3,7 @@ const bcrypt = require('bcrypt');
 const joi = require("joi");
 const salt = 12;
 
-module.exports = (middleware, users) => {
+module.exports = (middleware, users, plans) => {
     const router = require("express").Router();
     
     router.use(middleware);
@@ -18,11 +18,78 @@ module.exports = (middleware, users) => {
         return res.status(status.Ok);
     });
 
-    router.get('/plans', (req, res) => {
-        res.render('plans', { user: req.user });
-        return res.status(status.Ok);
+    router.get('/plans', async (req, res) => {
+        if (!req.session.email) {
+            return res.status(status.Unauthorized).redirect('/login');
+        }
+    
+        try {
+
+            // console.log(req.user.email);
+            const userPlans = await plans.find({userEmail: req.user.email }).toArray();
+            // console.log(userPlans);
+            res.render('plans', {
+                user: req.user,
+                plans: userPlans
+            });
+    
+        } catch (err) {
+            console.error("Error fetching plans:", err);
+            req.session.errMessage = "Could not load your plans. Please try again.";
+            res.status(status.InternalServerError).redirect('/home');
+        }
     });
 
+    router.get('/newPlan', (req, res) => {
+        const errMessage = req.session.errMessage;
+        req.session.errMessage = ""; 
+        res.render('newPlan', { user: req.user, errMessage: errMessage });
+    });
+
+    router.post('/newPlan', async (req, res) => {
+        if (!req.session.email) {
+            return res.status(status.Unauthorized).redirect('/login');
+        }
+
+            const planSchema = joi.object({
+                name: joi.string().min(3).max(100).required(),
+                retirementAge: joi.number().min(18).max(120).required(),
+                retirementExpenses: joi.number().min(0).required(),
+                retirementAssets: joi.number().min(0).required(),
+                retirementLiabilities: joi.number().min(0).required(),
+            });
+
+            const validationOptions = { convert: true, abortEarly: false }; 
+            const { error, value } = planSchema.validate(req.body, validationOptions);
+
+            if (error) { 
+                console.error("Plan validation error:", error.details);
+                req.session.errMessage = "Invalid input: " + error.details.map(d => d.message.replace(/"/g, '')).join(', '); 
+                res.status(status.BadRequest).redirect("/newPlan"); 
+                return; 
+            }
+            const newPlan = {
+                userEmail: req.user.email,
+                name: value.name,
+                retirementAge: value.retirementAge,
+                retirementExpenses: value.retirementExpenses,
+                retirementAssets: value.retirementAssets,
+                retirementLiabilities: value.retirementLiabilities,
+                progress: "0%"
+            };
+
+            try{
+                await plans.insertOne(newPlan);
+                req.session.errMessage = ""; 
+                res.redirect('/plans'); 
+            }
+            catch(err){
+                console.error("Error saving plan:", err);
+                req.session.errMessage = "An error occurred while saving your plan. Please try again.";
+                res.status(status.InternalServerError).redirect("/newPlan");
+            }
+        });
+
     router.get('/more', (req, res) => {
         res.render('more', { user: req.user });
         return res.status(status.Ok);
@@ -45,6 +112,8 @@ module.exports = (middleware, users) => {
     });
 
     router.post('/questionnaire', (req, res) => {
+        // console.log("Questionnaire POST body:", req.body);
+  
         const questionnaireSchema = joi.object({
             dob: joi.date().required(),
             education: joi.string().valid('primary', 'secondary', 'tertiary', 'postgraduate').required(),
@@ -53,10 +122,6 @@ module.exports = (middleware, users) => {
             expenses: joi.number().min(0).required(),
             assets: joi.number().min(0).required(),
             liabilities: joi.number().min(0).required(),
-            retirementAge: joi.number().min(18).max(120).required(),
-            retirementExpenses: joi.number().min(0).required(),
-            retirementAssets: joi.number().min(0).required(),
-            retirementLiabilities: joi.number().min(0).required(),
         });
     
         const validationOptions = { convert: true, abortEarly: false }; 
@@ -81,10 +146,6 @@ module.exports = (middleware, users) => {
                     expenses: value.expenses,
                     assets: value.assets,
                     liabilities: value.liabilities,
-                    retirementAge: value.retirementAge,
-                    retirementExpenses: value.retirementExpenses,
-                    retirementAssets: value.retirementAssets,
-                    retirementLiabilities: value.retirementLiabilities,
                 } 
             }
         ).then((result) => { 

src/views/newPlan.ejs

@@ -0,0 +1,40 @@
+<%- include("./partials/fileHeader") %>
+<%- include("./partials/header") %>
+
+<main class="container mx-auto p-4">
+    <h2 class="text-xl font-semibold mb-4">Welcome: <%= user.email %></h2>
+    <div class="max-w-md mx-auto bg-white p-8 mb-10 rounded-lg shadow-md">
+        <h3 class="text-lg font-medium mb-6">Financial Questionnaire</h3>
+        <form action="/newPlan" method="post" class="space-y-4">
+            <div>
+                <label for="name" class="block text-sm font-medium text-gray-700">Plan Name</label>
+                <input type="text" name="name" placeholder="e.g., Retirement Plan" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
+            </div>
+            <div>
+                <label for="retirementAge" class="block text-sm font-medium text-gray-700">Desired Retirement Age</label>
+                <input type="number" name="retirementAge" placeholder="e.g., 65" min="18" max="120" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
+            </div>
+
+            <div>
+                <label for="retirementExpenses" class="block text-sm font-medium text-gray-700">Estimated Monthly Retirement Expenses</label>
+                <input type="number" name="retirementExpenses" min="0" placeholder="e.g., 3000" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
+            </div>
+
+            <div>
+                <label for="retirementAssets" class="block text-sm font-medium text-gray-700">Estimated Retirement Assets</label>
+                <input type="number" name="retirementAssets" min="0" placeholder="e.g., 500000" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
+            </div>
+
+            <div>
+                <label for="retirementLiabilities" class="block text-sm font-medium text-gray-700">Estimated Retirement Liabilities</label>
+                <input type="number" name="retirementLiabilities" min="0" placeholder="e.g., 50000" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
+            </div>
+            
+            <div>
+                <button type="submit" class="w-full flex justify-center py-2 px-4 border border-transparent rounded-md shadow-sm text-sm font-medium text-white bg-indigo-600 hover:bg-indigo-700 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-indigo-500">Save Plan</button>
+            </div>
+        </form>
+    </div>
+</main>
+
+<%- include("./partials/footer") %>

src/views/plans.ejs

@@ -1,8 +1,23 @@
 <%- include("./partials/fileHeader") %>
 <%- include("./partials/header") %>
 
-<main>
-    The Plans Page
+<main class="container mx-auto p-4">
+    <h2 class="text-xl text-white font-semibold mb-4">Welcome: <%= user.email %></h2>
+    <div class="max-w-md mx-auto bg-white p-8 mb-10 rounded-lg shadow-md space-y-4">
+        <h3 class="text-lg font-medium mb-6">My Retirement Plans</h3>
+        <a href="/newPlan" class="block max-w-sm p-2"><div class="border-2 text-center border-gray-600 bg-blue-600 text-white py-1 w-full rounded-md hover:bg-transparent hover:text-indigo-700 font-semibold">New Plan</div></a>
+        <hr class="mt-3">
+        <% plans.forEach(plan => { %>
+            <a href="<%= plan.link %>" class="block max-w-sm p-6 bg-white border border-gray-200 rounded-lg shadow-sm hover:bg-gray-100 dark:bg-gray-800 dark:border-gray-700 dark:hover:bg-gray-700">
+                <h5 class="mb-2 text-2xl font-bold tracking-tight text-gray-900 dark:text-white"><%= plan.name %></h5>
+                <div class="w-full bg-gray-200 rounded-full h-2.5 mb-4 dark:bg-gray-700">
+                    <div class="bg-green-600 h-2.5 rounded-full dark:bg-green-500" style="width: <%= plan.progress %> "></div>
+                </div>
+                <p class="font-normal text-gray-700 dark:text-gray-400"><%= plan.description %></p>
+            </a>
+        <% }) %>
+    </div>
+    
 </main>
 
 <%- include("./partials/navBar") %>

src/views/questionnaire.ejs

@@ -60,26 +60,6 @@
                 <label for="liabilities" class="block text-sm font-medium text-gray-700">Liabilities</label>
                 <input type="number" name="liabilities" min="0" placeholder="Estimated Liabilities" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
             </div>
-        
-            <div>
-                <label for="retirementAge" class="block text-sm font-medium text-gray-700">Desired Retirement Age</label>
-                <input type="number" name="retirementAge" placeholder="e.g., 65" min="18" max="120" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
-            </div>
-
-            <div>
-                <label for="retirementExpenses" class="block text-sm font-medium text-gray-700">Estimated Monthly Retirement Expenses</label>
-                <input type="number" name="retirementExpenses" min="0" placeholder="e.g., 3000" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
-            </div>
-
-            <div>
-                <label for="retirementAssets" class="block text-sm font-medium text-gray-700">Estimated Retirement Assets</label>
-                <input type="number" name="retirementAssets" min="0" placeholder="e.g., 500000" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
-            </div>
-
-            <div>
-                <label for="retirementLiabilities" class="block text-sm font-medium text-gray-700">Estimated Retirement Liabilities</label>
-                <input type="number" name="retirementLiabilities" min="0" placeholder="e.g., 50000" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm">
-            </div>
             
             <div>
                 <button type="submit" class="w-full flex justify-center py-2 px-4 border border-transparent rounded-md shadow-sm text-sm font-medium text-white bg-indigo-600 hover:bg-indigo-700 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-indigo-500">Submit Questionnaire</button>