subline/controllers/transactionData.js

const Transaction = require("../models/transaction");
const Order = require("../models/order");
const Merchant = require("../models/merchant");

module.exports = {
    //POST - returns all transactions for a merchant between given dates
    //Inputs:
    //  req.body.from: start date
    //  req.body.to: end date
    //Returns:
    //  List of transactions
    getTransactions: function(req, res){
        if(!req.session.user){
            req.session.error = "You must be logged in to view that page";
            return res.redirect("/");
        }

        let date = new Date();
        let firstDay, lastDay;

        if(req.body.from && req.body.to){
            firstDay = new Date(req.body.from);
            lastDay = new Date(req.body.to);
        }else{
            firstDay = new Date(date.getFullYear(), date.getMonth(), 1);
            lastDay = new Date(date.getFullYear(), date.getMonth() + 1, 0);
        }

        Transaction.find({merchant: req.session.user, date: {$gte: firstDay, $lt: lastDay}},
            {_id: 0, date: 1, recipes: 1},
            {sort: {date: 1}})
            .then((transactions)=>{
                return res.json(transactions);
            })
            .catch((err)=>{
                return res.json("Error: could not retrieve sales data");
            });
    },

    getOrders: function(req, res){
        if(!req.session.user){
            req.session.error = "You must be logged in to view that page";
            return res.redirect("/");
        }

        Order.find({merchant: req.session.user})
            .then((orders)=>{
                return res.json(orders);
            })
            .catch((err)=>{
                return res.json("Error: could not retrieve order data");
            })
    },

    //POST - Update non-pos merchant inventory and create a transaction
    //Inputs:
    //  recipesSold: list of recipes sold and how much (recipe._id and quantity)
    //Returns:
    //  merchant.inventory: entire merchant inventory after being updated
    createTransaction: function(req, res){
        if(!req.session.user){
            res.session.error = "Must be logged in to do that";
            return res.redirect("/");
        }

        let transaction = new Transaction({
            date: Date.now(),
            merchant: req.session.user,
            recipes: []
        });

        for(let recipe of req.body){
            transaction.recipes.push({
                recipe: recipe.id,
                quantity: recipe.quantity
            });
        }

        //Calculate all ingredients used, store to list
        Merchant.findOne({_id: req.session.user})
            .populate("recipes")
            .then((merchant)=>{
                for(let reqRecipe of req.body){
                    let merchRecipe = merchant.recipes.find(r => r._id.toString() === reqRecipe.id);
                    for(let recipeIngredient of merchRecipe.ingredients){
                        let merchInvIngredient = merchant.inventory.find(i => i.ingredient.toString() === recipeIngredient.ingredient.toString());
                        merchInvIngredient.quantity -= recipeIngredient.quantity * reqRecipe.quantity;
                    }
                }

                merchant.save()
                    .then((merchant)=>{
                        res.json({});
                    })
                    .catch((err)=>{
                        return res.json("Error: unable to save user data");
                    });
            })
            .catch((err)=>{
                return res.json("Error: unable to retrieve user data");
            });

        transaction.save()
            .then((transaction)=>{
                return;
            })
            .catch((err)=>{});
    },

    populate: function(req, res){
        if(!req.session.user){
            res.session.error = "Must be logged in to do that";
            return res.redirect("/");
        }

        function randomDate() {
            let now = new Date();
            let start = new Date();
            start.setFullYear(now.getFullYear() - 1);
            return new Date(start.getTime() + Math.random() * (now.getTime() - start.getTime()));
        }

        Merchant.findOne({_id: req.session.user})
            .then((merchant)=>{
                let newTransactions = [];

                for(let i = 0; i < 5000; i++){
                    let newTransaction = new Transaction({
                        merchant: merchant._id,
                        date: randomDate(),
                        recipes: []
                    });

                    let numberOfRecipes = Math.floor((Math.random() * 5) + 1);

                    for(let j = 0; j < numberOfRecipes; j++){
                        let recipeNumber = Math.floor(Math.random() * merchant.recipes.length);
                        let randQuantity = Math.floor((Math.random() * 3) + 1);

                        newTransaction.recipes.push({
                            recipe: merchant.recipes[recipeNumber],
                            quantity: randQuantity
                        });
                    }

                    newTransactions.push(newTransaction);
                }

                Transaction.create(newTransactions)
                    .then((transactions)=>{
                        console.log("completed");
                        return res.redirect("/data");
                    })
                    .catch((err)=>{
                        console.log(err);
                        return;
                    });
            })
            .catch((err)=>{
                console.log(err);
                return;
            });
        
        
    }
}