selmerBot/commands/db/econ.js

const { MongoClient, ServerApiVersion } = require('mongodb');
// const { update } = require('apt');
const { Collection, Client, Formatters, Intents } = require('discord.js');
const { CLIENT_ODBC } = require('mysql/lib/protocol/constants/client');

//Declair an "enum" to help with BASE calculations
const BASE = {
    PAY: 5,
    HP: 5,
    MP: 10,
    XP: 5
}

const STATE = {
    IDLE: 0,
    FIGHTING: 1,
    DEFENDING: 2,
    PRONE: 3,
    WAITING: 4 //For items ONLY
}
//Note that leveling up to the next level takes 10% more xp than the previous one


//#region functions

function isNum(arg) {
    return (!isNaN(arg) && Number.isSafeInteger(Number(arg)));
};


function CreateNewCollection(message, client, server, id, opponent = null, game = null) {
    client.connect(err => {
        const db = client.db(String(server) + "[ECON]");
        const dbo = db.collection(id);
        if (err) { return console.log(err); }
        db.listCollections({name: id})
        .next(function(err, collinfo) {
            if (err) { return console.log(err); }
            if (!collinfo) {
                message.reply("You didn't have a place in my databases, so I created one for you!\nPlease try your command again!")
                let hp_mp = {maxhp: BASE.HP, hp: BASE.HP, maxmp: BASE.MP, mp: BASE.MP}
                dbo.insertOne({balance: 10, rank: 1, lastdayworked: 0, xp: 0, hpmp: hp_mp, game: game, opponent: opponent, state: STATE.IDLE, equipped: { weapons: {main: null, secondary: null}, items: {}}});
            }
        });
    });

    client.close();
}


function addxp(message, dbo, amt, xp_list) {
    if (!isNum(amt)) { return console.log("This isn't a number...."); }

    dbo.find({"balance": {$exists: true}}).toArray(function(err, doc) {
        if (!String(doc)) { return console.log("ERROR!\nThis account does not exist!"); }

        temp = doc[0];
        let rank = temp.rank + 1; //The table starts at rank 0, the user starts at rank 1
        const txp =  amt; /*temp.xp + amt; // This part was used before the xp check was made in the 'work' function */
        //If the rank is less than 100, you can still advance
        if (rank < 101) {
            let needed = xp_list.get(rank);
            if (txp >= needed) {
                //Get to the max level possible with the current xp (may skip)
                while (txp >= needed) {
                    rank ++;
                    needed = xp_list.get(rank);
                }
                rank --; //Maybe?

                let newhp;
                if (newhp < 200) {
                    newhp = BASE.HP * rank;
                } else {
                    newhp = temp.hpmp.hp + 50;
                }

                let newmp = temp.mp + 5;
                
                dbo.updateOne({balance: temp.balance, rank: temp.rank, lastdayworked: temp.lastdayworked}, { $set: { rank: rank, hpmp: {maxhp: newhp, maxmp: newmp} }});
                message.channel.send('Congradulations <@' + message.author.id + '> for reaching rank ' + String(rank) + '!');
            }
        } else {
            message.reply("You've already reached max level!");
        }

        dbo.updateOne({balance: temp.balance}, { $set: { xp: txp}});
    });
}


function getBalance(dbo, message) {
    dbo.find({"balance": {$exists: true}}).toArray(function(err, doc) {
        return message.reply('Your current balance is $' + String(doc[0].balance));
    });
}


function rank(dbo, message, xp_list) {
    dbo.find({"balance": {$exists: true}}).toArray(function(err, doc) {
        if (!String(doc)) { return console.log("ERROR!\nThis account does not exist!"); }

        let next = doc[0].rank + 1;
        let needed = xp_list.get(next);

        message.channel.send('<@' + message.author.id + '> you are currently at rank ' + String(next-1) + ' and have ' + String(doc[0].xp) + 'xp. You need ' + String(needed - doc[0].xp) + ' more xp to get to rank ' + String(next));
    });
}


//Changes one type of currency for another
function convertCurrency(id, amt, dbo) {
    
}

function checkAndUpdateBal(dbo, item, message, args) {
    let b = false;
    dbo.find({"balance": {$exists: true}}).toArray(b = function(err, doc) {
        if (!String(doc)) {
            message.reply("Your account doesn't exist, please contact the mods for support");
            return false;
        }

        const icost = args[0] * item.cost;
        if (doc[0].balance < icost) {
            message.reply("Insufficient funds!");
            return false;
        } else {
            let temp = doc[0];
            dbo.updateOne({balance: temp.balance, rank: temp.rank, lastdayworked: temp.lastdayworked}, { $set: { balance: doc[0].balance -= icost }});
            message.reply("You have bought " + item.name + " for $" + icost + "!");
            return true;
        }
    });

    return b;
}


function buy(id, message, args, dbo, shop, xp_list) {
    if (args.length < 2) { return; }
    if (!isNum(args[0])) { return message.reply("Please enter a number for query 2"); }

    let query = args[1];
    let item = shop.filter(function (item) { return item.name.toLowerCase() == query.toLowerCase(); });

    if (!String(item)) { return message.reply("This item does not exist!"); }

    let success = Boolean(checkAndUpdateBal(dbo, item[0], message, args));
    if (!success) { return; }

    var newObj = { name: item[0].name, cost: item[0].cost, icon: item[0].icon, sect: item[0].sect};

    addxp(message, dbo, Math.ceil(item[0].cost * 1.2), xp_list);
    
    dbo.find(newObj, {$exists: true}).toArray(function(err, doc) {
        if(String(doc)) {
            let newnum = doc[0].num + Number(args[0]);
            dbo.updateOne({ name: item[0].name }, {$set: {num: newnum}});
        } else {
            dbo.insertOne({ name: item[0].name, cost: item[0].cost, icon: item[0].icon, sect: item[0].sect, num: Number(args[0])});
        }
    });
};


function sell(id, message, args, dbo, shop, xp_list) {
    if (args.length < 2) { return; }
    if (!isNum(args[0])) { return message.reply("Please enter a number for query 1"); }
    
    let query = args[1];
    var newObj = { name: query };

    let item = shop.filter(function (titem) { return titem.name.toLowerCase() == query.toLowerCase(); });
    if (!String(item)) { return message.reply("This item does not exist!"); }

    item[0] = {name: item[0].name, cost: item[0].cost, icon: item[0].icon, sect: item[0].sect};

    let functional_item = item[0];

    dbo.find(functional_item, {$exists: true}).toArray(function(err, doc) {
        if(String(doc)) {

            //Make sure you don't sell more than you have
            let num = Number(args[0]);
            if (num < doc[0].num) {
                let newNum = doc[0].num - num;
                dbo.updateOne({ name: item[0].name }, {$set: {num: newNum}});
            } else {
                num = doc[0].num;
                dbo.deleteOne({ name: item[0].name });
            }

            //Update the balance
            let amountSoldFor = functional_item.cost * num;

            dbo.find({"balance": {$exists: true}}).toArray(function(err, doc) { 
                let currentBal = doc[0].balance;
                dbo.updateOne({"balance": {$exists: true}}, { $set: { balance: currentBal + amountSoldFor }});
            });

            addxp(message, dbo, Math.ceil(functional_item.cost * 1.2), xp_list);

            message.reply(`You've sold ${num} ${String(functional_item.name)} for $${amountSoldFor}`);
        } else {
            message.reply("You don't own this item!");
        }
    });
}


function work(dbo, message, xp_list) {
    let fulldate = new Date();
    let date = fulldate.getDate();
    dbo.find({"lastdayworked": {$exists: true}}).toArray(function(err, doc) {
        if (!String(doc)) { return message.reply("Your account doesn't exist, please contact the mods for support"); }
        if (doc[0].lastdayworked == 111111) {//date
            message.reply("You've already worked today, try again tomorrow!");
        } else {
            //Amount to be paid
            let amt = 0;
            amt = (BASE.PAY * doc[0].rank);
            let xp_earned = doc[0].xp + Math.ceil(amt*1.5);

            //Update the amount to the new TOTAL balance
            dbo.updateOne({"balance": {$exists: true}}, { $set: { balance: doc[0].balance + amt, lastdayworked: date }});
            addxp(message, dbo, xp_earned, xp_list);
            message.channel.send('<@' + message.author.id + '> worked and earned $' + amt +' and ' + String(xp_earned) + ' xp!');
        }
    });
}


function printInventory(dbo, message) {
    let tempstring = "";
    dbo.find().toArray(function(err, docs){
        docs.forEach(val => {
            if (!val.balance) {
                tempstring += String(val.num) + " " + val.name + " (" + val.icon + ")\n";
            }
        });
        if (tempstring == "") { tempstring += "You have nothing in your inventory!"; }
        message.reply(tempstring);
    });
}


function getShop(message, args, items, bot) {
    if (args.length == 0) {
        let temp = Formatters.codeBlock(items.map(i => `${i.sect}`).join(' '));
        temp = [...new Set(temp.split(' '))];

        return message.reply(`Please use the format ${bot.prefix}shop [type] [page number]\nTypes are: ${temp}`);
    }

    let ind = 1;
    let noinp = false;
    if (args.length > 1) {
        if (args[1] < (items.length / 9)) {
            ind = Number(args[1]);
        } else {
            return message.reply("That number is too large");
        }
    } else {
        noinp = true;
    }

    const items2 = items.filter(function(f) { return (f.sect.toLowerCase() == args[0].toLowerCase()) }).slice((ind - 1)*10, (ind - 1)*10+10);
    newText = Formatters.codeBlock(items2.map(i => `${i.icon} (${i.name}): \$${i.cost}`).join('\n'));

    if (noinp) {
        newText += `(Use ${bot.prefix}shop [type] [page number] to access other pages)`;
    }

    return message.reply(newText);
}


function econHelp() {
    let l = ["buy", 'shop', 'work', 'rank', 'inventory', 'balance', 'sell']
    
    return l.join(", ");
}
//#endregion

//Main Code
module.exports = {
    name: 'econ',
    description: 'ECON',
    async execute(bot, message, args, command, Discord, mongouri, items, xp_list) {
        //Set Discord vars
        const id = message.author.id;
        const server = message.guild.id;

        const client = new MongoClient(mongouri, { useNewUrlParser: true, useUnifiedTopology: true, serverApi: ServerApiVersion.v1 });
        if (client.writeConcern || client.writeConcern) { 
            client.close();
            return message.reply("Something went wrong with the database, please try again later and contact support if this problem persists!");
         }
        //Initialize if necessary
        CreateNewCollection(message, client, server, id);

        client.connect(err => {
            const db = client.db(String(server) + "[ECON]");
            const dbo = db.collection(id);
            if (err) { return console.log(err); }

            //test area
            if (command == 'xp' || command == 'adbal') {
                //Selmer Dev only command
                if (message.member.roles.cache.has('944048889038774302')) {
                    if (command == 'xp') {
                        return addxp(message, dbo, Number(args[0]), xp_list);
                    }
                }
            }

            //Command Area
            if(command == 'init') {
                //Add security check here
                // init.execute(bot, message, args, command, dbo, Discord, connect);
                return;
            } else if (command == 'buy') {
                buy(id, message, args, dbo, items, xp_list);
            } else if (command == 'shop') {
                getShop(message, args, items, bot);
            } else if (command == 'work') {
                work(dbo, message, xp_list);
            } else if (command == 'rank') {
                rank(dbo, message, xp_list);
            } else if (command == 'inventory') {
                printInventory(dbo, message);
            } else if (command == 'balance') {
                getBalance(dbo, message);
            } else if (command == 'sell') {
                sell(id, message, args, dbo, items, xp_list);
            } else {
                message.channel.send("'" + message.content + "' is not a command!");
            }
                
        });

        //Close the database
        client.close();
    },

    //Battle Updating stuff
    addxp, checkAndUpdateBal, CreateNewCollection, econHelp, addxp, BASE, STATE
}



/*
?????????????? What did I need this for?
else if (command == 'checkinv') {
                const req = dbo.findOne({ id: message.guild.id });
                if (!req) { return message.reply("Doc doesn't exist!"); }
            }

*/