economy functions

2 files changed, 52 insertions, 1 deletions

diff --git a/Plugins/economy.js b/Plugins/economy.js
new file mode 100644
index 0000000..aa0bb24
--- /dev/null
+++ b/Plugins/economy.js
@@ -0,0 +1,51 @@
+/** **************************************
+ *
+ *   Economy: Plugin for PokeBot that enables economy features.
+ *   Copyright (C) 2018 TheEdge, jtsshieh, Alee
+ *
+ *   This program is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   This program is distributed in the hope that it will be useful,
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ * *************************************/
+
+const db = require('quick.db');
+exports.get = async (userid) => {
+  const amount = await db.fetch(`money_${userid}`);
+  if (amount) {
+    return amount;
+  }
+  else {
+    await db.set(`money_${userid}`, 0);
+    return await db.fetch(`money_${userid}`);
+  }
+};
+
+exports.add = async (userid, money) => {
+  const amount = await db.fetch(`money_${userid}`);
+  if (amount) {
+    await db.set(`money_${userid}`, amount + money);
+  }
+  else {
+    await db.set(`money_${userid}`, money);
+  }
+};
+
+exports.subtract = async (userid, money) => {
+  const amount = await db.fetch(`money_${userid}`);
+  if (amount) {
+    await db.set(`money_${userid}`, amount - money);
+  }
+  else {
+    await db.set(`money_${userid}`, 0 - money);
+  }
+};
diff --git a/bot.js b/bot.js
index 414eb76..f97343f 100644
--- a/bot.js
+++ b/bot.js
@@ -7,7 +7,7 @@ bot.commands = new Discord.Collection();
 bot.aliases = new Discord.Collection();
 bot.categories = new Discord.Collection();
 bot.queue = new Discord.Collection();
-bot.plugins = { music : require('./Plugins/Music.js') };
+bot.plugins = { music : require('./Plugins/Music.js') , economy : require('./Plugins/Economy.js')};
 cmdLoader();
 
 async function cmdLoader() {