mirror of
https://github.com/ION606/selmerBot.git
synced 2026-05-14 21:26:54 +00:00
ION606
360 lines
13 KiB
JavaScript
360 lines
13 KiB
JavaScript
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!"); }
|
|
}
|
|
|
|
*/ |