New leveling formula, limit, and many more

SlavyanDesu · Feb 5, 2021 · de5d6ac · de5d6ac
1 parent 4052e54
commit de5d6ac
Show file tree

Hide file tree

Showing 16 changed files with 851 additions and 349 deletions.
diff --git a/README.md b/README.md
@@ -217,7 +217,6 @@ If you want to unlock premium commands, please buy me a coffee at least 1 on Ko-
 | Sholat schedule      |      ✔️      |
 | Latest Line stickers |      ✔️      |
 | Check postage        |      ✔️      |
-| Spoiler text         |      ✔️      |
 | Sending email        |      ✔️      |
 | Random quotes        |      ✔️      |
 | Genshin chara info   |      ✔️      |

diff --git a/database/bot/mute.json b/database/bot/mute.json
@@ -1 +1 @@
-[]
+[]
diff --git a/database/bot/registered.json b/database/bot/registered.json
@@ -1 +1 @@
-[]
+[]
diff --git a/database/group/leveling.json b/database/group/leveling.json
@@ -1 +1 @@
-[]
+[]
diff --git a/database/user/card/background.json b/database/user/card/background.json
@@ -1 +1 @@
-[]
+[]
diff --git a/database/user/level.json b/database/user/level.json
@@ -1 +1 @@
-[]
+[]
diff --git a/function/card.js b/function/card.js
@@ -1,31 +1,27 @@
 const fs = require('fs-extra')
 
-/**
- * Add background card.
- * @param {String} userId 
- * @param {Object} _dir 
- */
-const addBg = (userId, _dir) => {
-    const obj = { id: userId, link: 'https://i.ibb.co/tYf3jmz/amos-yan-no-entry-1.jpg' }
-    _dir.push(obj)
-    fs.writeFileSync('./database/user/card/background.json', JSON.stringify(_dir))
-}
-
 /**
  * Get background.
  * @param {String} userId 
  * @param {Object} _dir 
  * @returns {String}
  */
 const getBg = (userId, _dir) => {
-    let position = null
+    let pos = null
+    let found = false
     Object.keys(_dir).forEach((i) => {
         if (_dir[i].id === userId) {
-            position = i
+            pos = i
+            found = true
         }
     })
-    if (position !== null) {
-        return _dir[position].link
+    if (found === false && pos === null) {
+        const obj = { id: userId, link: 'https://i.ibb.co/tYf3jmz/amos-yan-no-entry-1.jpg' }
+        _dir.push(obj)
+        fs.writeFileSync('./database/user/card/background.json', JSON.stringify(_dir))
+        return 'https://i.ibb.co/tYf3jmz/amos-yan-no-entry-1.jpg'
+    } else {
+        return _dir[pos].link
     }
 }
 
@@ -49,7 +45,6 @@ const replaceBg = (userId, link, _dir) => {
 }
 
 module.exports = {
-    addBg,
     getBg,
     replaceBg
 }
diff --git a/function/daily.js b/function/daily.js
@@ -0,0 +1,36 @@
+const fs = require('fs-extra')
+
+/**
+ * Add limit.
+ * @param {String} userId 
+ * @param {String} dir 
+ * @param {Object} _dir 
+ */
+const addLimit = (userId, _dir) => {
+    const obj = { id: userId, time: Date.now() }
+    _dir.push(obj)
+    fs.writeFileSync('./database/user/daily.json', JSON.stringify(_dir))
+}
+
+/**
+ * Get time limit.
+ * @param {String} userId 
+ * @param {Object} _dir 
+ * @returns {Number}
+ */
+const getLimit = (userId, _dir) => {
+    let position = null
+    Object.keys(_dir).forEach((i) => {
+        if (_dir[i].id === userId) {
+            position = i
+        }
+    })
+    if (position !== null) {
+        return _dir[position].time
+    }
+}
+
+module.exports = {
+    addLimit,
+    getLimit
+}
diff --git a/function/index.js b/function/index.js
@@ -1,4 +1,5 @@
 exports.level = require('./level')
+exports.daily = require('./daily')
 exports.limit = require('./limit')
 exports.card = require('./card')
 exports.register = require('./register')

diff --git a/function/level.js b/function/level.js
@@ -1,4 +1,6 @@
 const fs = require('fs-extra')
+const { getBg } = require('./card')
+const _bg = JSON.parse(fs.readFileSync('./database/user/card/background.json'))
 
 /**
  * Get user ID from db.
@@ -7,14 +9,22 @@ const fs = require('fs-extra')
  * @returns {String}
  */
 const getLevelingId = (userId, _dir) => {
-    let position = null
+    let pos = null
+    let found = false
+    getBg(userId, _bg)
     Object.keys(_dir).forEach((i) => {
         if (_dir[i].id === userId) {
-            position = i
+            pos = i
+            found = true
         }
     })
-    if (position !== null) {
-        return _dir[position].id
+    if (found === false && pos === null) {
+        const obj = { id: userId, xp: 0, level: 1 }
+        _dir.push(obj)
+        fs.writeFileSync('./database/user/level.json', JSON.stringify(_dir))
+        return userId
+    } else {
+        return _dir[pos].id
     }
 } 
 
@@ -25,14 +35,22 @@ const getLevelingId = (userId, _dir) => {
  * @returns {Number}
  */
 const getLevelingLevel = (userId, _dir) => {
-    let position = null
+    let pos = null
+    let found = false
+    getBg(userId, _bg)
     Object.keys(_dir).forEach((i) => {
         if (_dir[i].id === userId) {
-            position = i
+            pos = i
+            found = true
         }
     })
-    if (position !== null) {
-        return _dir[position].level
+    if (found === false && pos === null) {
+        const obj = { id: userId, xp: 0, level: 1 }
+        _dir.push(obj)
+        fs.writeFileSync('./database/user/level.json', JSON.stringify(_dir))
+        return 1
+    } else {
+        return _dir[pos].level
     }
 }
 
@@ -43,28 +61,25 @@ const getLevelingLevel = (userId, _dir) => {
  * @returns {Number}
  */
 const getLevelingXp = (userId, _dir) => {
-    let position = null
+    let pos = null
+    let found = false
+    getBg(userId, _bg)
     Object.keys(_dir).forEach((i) => {
         if (_dir[i].id === userId) {
-            position = i
+            pos = i
+            found = true
         }
     })
-    if (position !== null) {
-        return _dir[position].xp
+    if (found === false && pos === null) {
+        const obj = { id: userId, xp: 0, level: 1 }
+        _dir.push(obj)
+        fs.writeFileSync('./database/user/level.json', JSON.stringify(_dir))
+        return 0
+    } else {
+        return _dir[pos].xp
     }
 }
 
-/**
- * Add user to db.
- * @param {String} userId 
- * @param {Object} _dir 
- */
-const addLevelingId = (userId, _dir) => {
-    const obj = { id: userId, xp: 0, level: 1 }
-    _dir.push(obj)
-    fs.writeFileSync('./database/user/level.json', JSON.stringify(_dir))
-}
-
 /**
  * Add user level to db.
  * @param {String} userId 
@@ -103,11 +118,26 @@ const addLevelingXp = (userId, amount, _dir) => {
     }
 }
 
+// Cooldown XP gains to prevent spam
+const xpGain = new Set()
+
+const isGained = (from) => {
+    return !!xpGain.has(from)
+}
+
+const addCooldown = (from) => {
+    xpGain.add(from)
+    setTimeout(() => {
+        return xpGain.delete(from)
+    }, 60000) // Each minute
+}
+
 module.exports = {
     getLevelingId,
     getLevelingLevel,
     getLevelingXp,
-    addLevelingId,
     addLevelingLevel,
-    addLevelingXp
+    addLevelingXp,
+    isGained,
+    addCooldown
 }
diff --git a/function/limit.js b/function/limit.js
@@ -1,36 +1,86 @@
 const fs = require('fs-extra')
 
 /**
- * Add limit.
+ * Check limit.
  * @param {String} userId 
- * @param {String} dir 
  * @param {Object} _dir 
+ * @param {Number} limitCount 
+ * @param {Boolean} isPremium 
+ * @param {Boolean} isOwner 
+ * @returns {Boolean}
  */
-const addLimit = (userId, _dir) => {
-    const obj = { id: userId, time: Date.now() }
-    _dir.push(obj)
-    fs.writeFileSync('./database/user/daily.json', JSON.stringify(_dir))
+const isLimit = (userId, _dir, limitCount, isPremium, isOwner) => {
+    if (isPremium || isOwner) return false
+    let found = false
+    for (let i of _dir) {
+        if (i.id === userId) {
+            if (i.limit <= 0) {
+                found = true
+                return true
+            } else {
+                found = true
+                return false
+            }
+        }
+    }
+    if (found === false) {
+        const obj = { id: userId, limit: limitCount }
+        _dir.push(obj)
+        fs.writeFileSync('./database/user/limit.json', JSON.stringify(_dir))
+        return false
+    }
+}
+
+/**
+ * Add limit to user.
+ * @param {String} userId 
+ * @param {Object} _dir 
+ * @param {Boolean} isPremium 
+ * @param {Boolean} isOwner 
+ * @returns {*}
+ */
+const addLimit = (userId, _dir, isPremium, isOwner) => {
+    if (isPremium || isOwner) return false
+    let pos = null
+    Object.keys(_dir).forEach((i) => {
+        if (_dir[i].id === userId) {
+            pos = i
+        }
+    })
+    if (pos !== null) {
+        _dir[pos].limit -= 1
+        fs.writeFileSync('./database/user/limit.json', JSON.stringify(_dir))
+    }
 }
 
 /**
- * Get time limit.
+ * Get user's limit.
  * @param {String} userId 
  * @param {Object} _dir 
+ * @param {Number} limitCount 
  * @returns {Number}
  */
-const getLimit = (userId, _dir) => {
-    let position = null
+const getLimit = (userId, _dir, limitCount) => {
+    let pos = null
+    let found = false
     Object.keys(_dir).forEach((i) => {
         if (_dir[i].id === userId) {
-            position = i
+            pos = i
+            found = true
         }
     })
-    if (position !== null) {
-        return _dir[position].time
+    if (found === false && pos === null) {
+        const obj = { id: userId, limit: limitCount }
+        _dir.push(obj)
+        fs.writeFileSync('./database/user/limit.json', JSON.stringify(_dir))
+        return limitCount
+    } else {
+        return _dir[pos].limit
     }
 }
 
 module.exports = {
+    isLimit,
     addLimit,
     getLimit
 }
diff --git a/function/loader.js b/function/loader.js
@@ -1,6 +1,13 @@
+/* eslint-disable no-empty-function */
 const fs = require('fs-extra')
 const { color } = require('../tools')
 
+/**
+ * Returns an array of files.
+ * @param {*} dirPath 
+ * @param {String[]} arrayOfFiles 
+ * @returns {String[]}
+ */
 const getAllDirFiles = (dirPath, arrayOfFiles) => {
     const files = fs.readdirSync(dirPath)
     arrayOfFiles = arrayOfFiles || []
@@ -14,6 +21,10 @@ const getAllDirFiles = (dirPath, arrayOfFiles) => {
     return arrayOfFiles
 } 
 
+/**
+ * Uncache a changes.
+ * @param {*} module 
+ */
 const uncache = (module = '.') => {
     return new Promise((resolve, reject) => {
         try {
@@ -25,7 +36,12 @@ const uncache = (module = '.') => {
     })
 }
 
-const nocache = (module, call = () => { }) => {
+/**
+ * Delete file cache.
+ * @param {*} module 
+ * @param {*} call 
+ */
+const nocache = (module, call = () => {}) => {
     console.log(color('[WATCH]', 'orange'), color(`=> '${module}'`, 'yellow'), 'file is now being watched by me!')
     fs.watchFile(require.resolve(module), async () => {
         await uncache(require.resolve(module))