subline/controllers/ingredientData.js

const Ingredient = require("../models/ingredient");
const InventoryAdjustment = require("../models/inventoryAdjustment.js");

const helper = require("./helper.js");

const xlsx = require("xlsx");
const fs = require("fs");

module.exports = {
    /*
    POST - create a single ingredient and then add to the merchant
    req.body = {
        ingredient: {
            name: name of ingredient,
            category: category of ingredient,
            unitType: category for the unit (mass, volume, length)
        },
        quantity: quantity of ingredient for current merchant,
        defaultUnit: default unit of measurement to display
    }
    Returns:
        Same as above, with the _id
    */
    createIngredient: function(req, res){
        let newIngredient = {...req.body};
        if(req.body.defaultUnit === "bottle"){
            newIngredient.ingredient.unitSize = helper.convertQuantityToBaseUnit(newIngredient.ingredient.unitSize, newIngredient.ingredient.unitType);
        }

        newIngredient = new Ingredient(newIngredient.ingredient);
        newIngredient.ingredients = [];
        
        newIngredient.save()
            .then((ingredient)=>{
                newIngredient = {
                    ingredient: ingredient,
                    defaultUnit: req.body.defaultUnit
                }

                newIngredient.quantity = helper.convertQuantityToBaseUnit(req.body.quantity, req.body.defaultUnit);

                res.locals.merchant.inventory.push(newIngredient);

                return res.locals.merchant.save();
            })
            .then((response)=>{
                return res.json(newIngredient);
            })
            .catch((err)=>{
                if(typeof(err) === "string"){
                    return res.json(err);
                }
                if(err.name === "ValidationError"){
                    return res.json(err.errors[Object.keys(err.errors)[0]].properties.message);
                }
                return res.json("ERROR: UNABLE TO CREATE THE INGREDIENT");
            });
    },

    /*
    POST: Updates data for a single ingredient
    req.body = {
        id: id of the ingredient,
        name: new name of the ingredient,
        quantity: new quantity of the unit (in grams),
        category: new category of the unit,
        unit: new default unit of the ingredient,
        ingredients: [{
            ingredient: String (Id),
            quantity: Number
        }]
    }
    response = Ingredient
    */
    updateIngredient: function(req, res){
        let popMerchant = res.locals.merchant.populate("inventory.ingredient").execPopulate();

        Promise.all([Ingredient.findOne({_id: req.body.id}), popMerchant])
            .then((response)=>{
                response[0].name = req.body.name;
                response[0].category = req.body.category;
                response[0].ingredients = [];

                // Check ingredients for circular references
                let isCircular = (ingredient, original)=>{
                    if(ingredient.ingredients.length === 0) return false;

                    for(let i = 0; i < ingredient.ingredients.length; i++){
                        for(let j = 0; j < res.locals.merchant.inventory.length; j++){
                            if(res.locals.merchant.inventory[j].ingredient._id.toString() === ingredient.ingredients[i].ingredient.toString()){
                                let next = res.locals.merchant.inventory[j].ingredient;
                                if(next._id.toString() === original._id.toString()) return true;
                                return isCircular(next, original);
                            }
                        }
                    }
                }
                
                for(let i = 0; i < req.body.ingredients.length; i++){
                    for(let j = 0; j < res.locals.merchant.inventory.length; j++){
                        if(res.locals.merchant.inventory[j].ingredient._id.toString() === req.body.ingredients[i].ingredient){
                            let ingredient = res.locals.merchant.inventory[j].ingredient;
                            if(ingredient._id.toString() === req.body.id) throw "YOU HAVE CIRCULAR REFERENCES IN YOUR INGREDIENTS1";
                            if(isCircular(response[0], response[0]) === false){
                                response[0].ingredients.push({
                                    ingredient: req.body.ingredients[i].ingredient,
                                    quantity: req.body.ingredients[i].quantity
                                });
                            }else{
                                throw "YOU HAVE CIRCULAR REFERENCES IN YOUR INGREDIENTS";
                            }
                            break;
                        }
                    }
                }

                //find and update ingredient on merchant
                for(let i = 0; i < res.locals.merchant.inventory.length; i++){
                    if(res.locals.merchant.inventory[i].ingredient.toString() === req.body.id){
                        res.locals.merchant.inventory[i].defaultUnit = req.body.unit;

                        if(res.locals.merchant.inventory[i].quantity !== req.body.quantity){
                            new InventoryAdjustment({
                                date: new Date(),
                                merchant: req.session.owner,
                                ingredient: req.body.id,
                                quantity: req.body.quantity - res.locals.merchant.inventory[i].quantity
                            }).save().catch(()=>{});

                            res.locals.merchant.inventory[i].quantity = req.body.quantity;
                        }
                        
                        break;
                    }
                }
                return Promise.all([response[0].save(), res.locals.merchant.save()])
            })
            .then((response)=>{
                return res.json({
                    ingredient: response[0],
                    quantity: req.body.quantity,
                    defaultUnit: req.body.unit
                });
            })
            .catch((err)=>{
                if(typeof(err) === "string"){
                    return res.json(err);
                }
                if(err.name === "ValidationError"){
                    return res.json(err.errors[Object.keys(err.errors)[0]].properties.message);
                }
                return res.json("ERROR: UNABLE TO UPDATE DATA");
            });
    },

    createFromSpreadsheet: function(req, res){
        //read file, get the correct sheet, create array from sheet
        let workbook = xlsx.readFile(req.file.path);
        fs.unlink(req.file.path, ()=>{});

        let sheets = Object.keys(workbook.Sheets);
        let sheet = {};
        for(let i = 0; i < sheets.length; i++){
            let str = sheets[i].toLowerCase();
            if(str === "ingredient" || str === "ingredients"){
                sheet = workbook.Sheets[sheets[i]];
            }
        }
        const array = xlsx.utils.sheet_to_json(sheet, {
            header: 1
        });

        //get property locations
        let locations = {};
        for(let i = 0; i < array[0].length; i++){
            switch(array[0][i].toLowerCase()){
                case "name": locations.name = i; break;
                case "category": locations.category = i; break;
                case "quantity": locations.quantity = i; break;
                case "unit": locations.unit = i; break;
                case "bottle size": locations.bottleSize = i; break;
                case "bottle unit": locations.bottleUnit = i; break;
            }
        }

        //Create ingredients
        let ingredients = [];
        let merchantData = [];
        for(let i = 1; i < array.length; i++){
            let ingredient = new Ingredient({
                name: array[i][locations.name],
                category: array[i][locations.category],
                unitType: helper.getUnitType(array[i][locations.unit].toLowerCase())
            });

            if(array[i][locations.unit] === "bottle"){
                ingredient.unitType = array[i][locations.bottleUnit];
                ingredient.unitSize = helper.convertQuantityToBaseUnit(array[i][locations.bottleSize], array[i][locations.bottleUnit]);
            }

            let merchantItem = {
                ingredient: ingredient,
                quantity: helper.convertQuantityToBaseUnit(array[i][locations.quantity], array[i][locations.unit]),
                defaultUnit: array[i][locations.unit]
            }
            
            merchantData.push(merchantItem);
            ingredients.push(ingredient);
        }

        for(let i = 0; i < merchantData.length; i++){
            res.locals.merchant.inventory.push(merchantData[i]);
        }

        //Update the database
        Promise.all([Ingredient.create(ingredients), res.locals.merchant.save()])
            .then((response)=>{
                return res.json(merchantData);
            })
            .catch((err)=>{
                if(typeof(err) === "string"){
                    return res.json(err);
                }
                if(err.name === "ValidationError"){
                    return res.json(err.errors[Object.keys(err.errors)[0]].properties.message);
                }
                return "ERROR: UNABLE TO CREATE YOUR INGREDIENTS";
            });
    },

    spreadsheetTemplate: function(req, res){
        let workbook = xlsx.utils.book_new();
        workbook.SheetNames.push("Ingredients");
        let workbookData = [];

        workbookData.push(["Name", "Category", "Quantity", "Unit", "Bottle Size", "Bottle Unit"]);
        workbookData.push(["Example Ingredient 1", "Produce", 100, "lbs"]);
        workbookData.push(["Example Ingredient 2", "Fruit", 3.24, "kg"]);
        workbookData.push(["Example Ingredient 3", "Beverage", 5, "bottle", 750, "ml"]);

        workbook.Sheets.Ingredients = xlsx.utils.aoa_to_sheet(workbookData);
        xlsx.writeFile(workbook, "SublineIngredients.xlsx");
        return res.download("SublineIngredients.xlsx", (err)=>{
            fs.unlink("SublineIngredients.xlsx", ()=>{});
        });
    },

    //DELETE - Removes an ingredient from the merchant's inventory
    removeIngredient: function(req, res){
        for(let i = 0; i < res.locals.merchant.inventory.length; i++){
            if(req.params.id === res.locals.merchant.inventory[i].ingredient._id.toString()){
                res.locals.merchant.inventory.splice(i, 1);
                break;
            }
        }

        Promise.all([res.locals.merchant.save(), Ingredient.deleteOne({_id: req.params.id})])
            .then((response)=>{
                return res.json({});
            })
            .catch((err)=>{
                if(typeof(err) === "string"){
                    return res.json(err);
                }
                if(err.name === "ValidationError"){
                    return res.json(err.errors[Object.keys(err.errors)[0]].properties.message);
                }
                return res.json("ERROR: UNABLE TO RETRIEVE DATA");
            });
    }
}