selmerBot/commands/db/econ.js

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

let currencySymbol = '$';

//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(interaction, client, server, id, opponent = null, game = null) {
    const db = client.db(String(server));
    const dbo = db.collection(id);

    db.listCollections({name: id})
    .next(function(err, collinfo) {
        if (!collinfo) {
            interaction.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, gamesettings: {battle: {class: 'none', ultimate: true}}, opponent: opponent, state: STATE.IDLE, equipped: { weapons: {main: null, secondary: null}, items: {}}});
        }
    });
}


function addxp(interaction, dbo, amt, xp_list, noPing = false) {
    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}, xp: txp }});

                var user;
                if (interaction.user) {
                    user = interaction.user;
                } else {
                    // This is a message
                    user = interaction.author;
                }

                interaction.channel.send('Congradulations <@' + user.id + '> for reaching rank ' + String(rank) + '!');
            }
        } else {
            if (!noPing) {
                interaction.reply("You've already reached max level!").catch((err) => {
                    interaction.channel.send("You've already reached max level!");
                });
            }
        }

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


function getBalance(dbo, interaction) {
    dbo.find({"balance": {$exists: true}}).toArray(function(err, doc) {
        let bal = 0;
        if (doc[0] && doc[0].balance) {
            bal = doc[0].balance;
        }
        return interaction.reply(`<@${interaction.user.id}>, your current balance is ${currencySymbol}${bal}`)
        .catch((err) => {
            interaction.channel.send(`<@${interaction.user.id}>, your current balance is ${currencySymbol}${bal}`);
        });
    });
}


function rank(dbo, interaction, 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);

        interaction.channel.send(`<@${interaction.user.id}'> you are currently at rank ${next-1} and have ${doc[0].xp}xp. You need ${needed - doc[0].xp} more xp to get to rank ${next}`);
    });
}


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


function checkAndUpdateBal(dbo, item, interaction, amt) {
    return new Promise(function(resolve, reject) {
        dbo.find({"balance": {$exists: true}}).toArray(b = function(err, doc) {
            if (!String(doc)) {
                interaction.reply("Your account doesn't exist, please contact the mods for support").catch(() => {
                    interaction.channel.send("Your account doesn't exist, please contact the mods for support");
                });
                return false;
            }

            const icost = amt * item.cost;
            if (doc[0].balance < icost) {
                interaction.reply("Insufficient funds!").catch(() => { interaction.channel.send("Insufficient funds!"); });
                resolve(false);
            } else {
                let temp = doc[0];
                dbo.updateOne({balance: temp.balance, rank: temp.rank, lastdayworked: temp.lastdayworked}, { $set: { balance: doc[0].balance -= icost }});
                interaction.reply(`You have bought ${item.name} for ${currencySymbol}${icost}!`).catch(() => {
                    interaction.channel.send(`You have bought ${item.name} for ${currencySymbol}${icost}!`);
                });
                resolve(true);
            }
        });
    });
}


function buy(id, interaction, dbo, shop, xp_list) {
    const args = interaction.options.data;

    //REAPPLY THIS TO OTHER FUNCTIONS
    let query = args.filter((arg) => { return (arg.name == 'item'); })[0].value;
    let amt = args.filter((arg) => { return (arg.name == 'amount'); })[0].value;
    let item = shop.filter(function (item) { return item.name.toLowerCase() == query.toLowerCase(); })[0];

    if (!String(item)) { return interaction.reply("This item does not exist!").catch(() => { interaction.channel.send("This item does not exist!"); }); }

    checkAndUpdateBal(dbo, item, interaction, amt).then((success) => {
        //The message is handled in the CheckAndUpdateBal() function
        if (!success) { return }

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

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


//FIXME
function sell(id, interaction, dbo, shop, xp_list) {
    const args = interaction.options.data;
    const query = args.filter((arg) => { return (arg.name == 'item'); })[0].value;
    var num = args.filter((arg) => { return (arg.name == 'amount'); })[0].value;

    let item = shop.filter(function (titem) { return titem.name.toLowerCase() == query.toLowerCase(); });
    if (!String(item)) {
        return interaction.reply("This item does not exist!").catch((err) => {
            interaction.channel.send("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
            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(interaction, dbo, Math.ceil(functional_item.cost * 1.2), xp_list);

            interaction.reply(`You've sold ${num} ${String(functional_item.name)} for ${currencySymbol}${amountSoldFor}`)
            .catch((err) => {
                interaction.channel.send(`You've sold ${num} ${String(functional_item.name)} for ${currencySymbol}${amountSoldFor}`);
            });
        } else {
            interaction.reply("You don't own this item!").catch((err) => {
                interaction.channel.send(`You've sold ${num} ${String(functional_item.name)} for ${currencySymbol}${amountSoldFor}`);
            });
        }
    });
}


function work(dbo, interaction, xp_list) {
    let fulldate = new Date();
    let date = fulldate.getDate();
    dbo.find({"lastdayworked": {$exists: true}}).toArray(function(err, doc) {
        if (!String(doc)) {
            return interaction.reply("Your account doesn't exist, please contact the mods for support").catch((err) => {
                interaction.channel.send("Your account doesn't exist, please contact the mods for support");
            });
        }
        if (doc[0].lastdayworked == date) {//date
            interaction.reply("You've already worked today, try again tomorrow!").catch((err) => {
                interaction.channel.send("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(interaction, dbo, xp_earned, xp_list);
            interaction.reply(`<@${interaction.user.id}> worked and earned ${currencySymbol}${amt} and ${xp_earned} xp!`).catch((err) => {
                interaction.channel.send(`<@${interaction.user.id}> worked and earned ${currencySymbol}${amt} and ${xp_earned} xp!`)
            });
        }
    });
}


function printInventory(dbo, interaction) {
    let tempstring = "";
    dbo.find().toArray(function(err, docs){
        docs.forEach(val => {
            if (!val.balance && val.name != undefined) {
                tempstring += String(val.num) + " " + val.name + " (" + val.icon + ")\n";
            }
        });

        if (tempstring == "") { tempstring += "You have nothing in your inventory!"; }
        interaction.reply(tempstring).catch((err) => {
            interaction.channel.send(tempstring);
        });
    });
}


function getShop(interaction, items, bot) {
    const args = interaction.options.data;
    const type = args.filter((arg) => { return (arg.name == 'type'); })[0].value.toLowerCase();

    // 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) {
        const amt = args.filter((arg) => { return (arg.name == 'page'); })[0].value;
        if (amt.value < (items.length / 9)) {
            ind = Number(amt);
        } else {
            return interaction.reply("That number is too large").catch(() => { interaction.channel.send("That number is too large"); });
        }
    } else {
        noinp = true;
    }

    const items2 = items.filter(function(f) { return (f.sect.toLowerCase() == type) }).slice((ind - 1)*10, (ind - 1)*10+10);
    newText = Formatters.codeBlock(items2.map(i => `${i.icon} (${i.name}): $${i.cost}`).join('\n')); //${currencySymbol} doesn't owrk for some reason

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

    return interaction.reply(newText).catch(() => { interaction.channel.send(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, interaction, Discord, mongouri, items, xp_list) {
        //Set Discord vars
        const id = interaction.user.id;
        const server = interaction.guildId;
        const command = interaction.commandName;

        // 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!");
        //  }


        bot.mongoconnection.then(async (client) => {

            //Initialize if necessary
            CreateNewCollection(interaction, client, server, id);
            
            const db = client.db(String(server));
            const dbo = db.collection(id);

            currencySymbol = bot.currencysymbolmmain;

            /*/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, interaction, dbo, items, xp_list);
            } else if (command == 'shop') {
                getShop(interaction, items, bot);
            } else if (command == 'work') {
                work(dbo, interaction, xp_list);
            } else if (command == 'rank') {
                rank(dbo, interaction, xp_list);
            } else if (command == 'inventory') {
                printInventory(dbo, interaction);
            } else if (command == 'balance') {
                getBalance(dbo, interaction);
            } else if (command == 'sell') {
                sell(id, interaction, dbo, items, xp_list);
            } else {
                interaction.reply(`${command} is not a command`).catch((err) => {
                    interaction.channel.send(`${command} is not a command`);
                });
            }
                
        });
    },

    //Battle Updating stuff
    addxp, checkAndUpdateBal, CreateNewCollection, econHelp, BASE, STATE,
    options: []
}