re-add unoptimised code for now

bot/util/redis.js

@@ -1,43 +1,198 @@
+'use strict'
+
+// TODO IN REDIS DATABASE MODULE: Find a better way to store arrays/objects
+
+const Database = require('../../base/database')
+
 const redis = require('redis')
+const { promisify } = require('util')
 const generators = require('redis-async-gen')
 
-class Redis {
+class RedisDatabase extends Database {
-  constructor(client) {
+    constructor(client) {
-    this.client = client;
+        super(client)
-  }
+        // Config
+        let conf = this.client.config.redis
 
-  init () {
+        // Create redis client
-    // Create redis client
+        this.global = redis.createClient(conf)
-    this.global = redis.createClient(this.client.config.redis)
+        this.guild = this.global.duplicate({ db: 1 })
-    this.server = this.global.duplicate({ db: 1 })
+        this.member = this.global.duplicate({ db: 2 })
-    this.member = this.global.duplicate({ db: 2 })
+        this.user = this.global.duplicate({ db: 3 })
-    this.user = this.global.duplicate({ db: 3 })
-  }
 
-  // Deletes specified guild entry
+        // Async
-  async deleteGuild (id) {
+        this.guildGetAsync = promisify(this.guild.get).bind(this.guild)
-    this.server.del(id)
+        this.memberGetAsync = promisify(this.member.get).bind(this.member)
-    var { keysMatching } = await generators.using(this.member)
+        this.userGetAsync = promisify(this.user.get).bind(this.user)
-    // eslint-disable-next-line no-unused-vars
+
-    for await (const key of keysMatching(id + '-*')) {
+        this.guildSetAsync = promisify(this.guild.set).bind(this.guild)
-      this.member.del(key)
+        this.memberSetAsync = promisify(this.member.set).bind(this.member)
+        this.userSetAsync = promisify(this.user.set).bind(this.user)
+
+        // Generators
+        this.guildGenerators = generators.using(this.guild)
+        this.memberGenerators = generators.using(this.member)
+        this.userGenerators = generators.using(this.user)
     }
-  }
 
-  // Deletes specified user and their member entries in guilds
+    async userExists(id) {
-  async deleteUser (id) {
+        for await (const key of this.userGenerators.keysMatching(id + '-*')) {
-    this.user.del(id)
+            return true
-    var { keysMatching } = await generators.using(this.member)
+        }
-    // eslint-disable-next-line no-unused-vars
+
-    for await (const key of keysMatching('*-' + id)) {
+        for await (const key of this.memberGenerators.keysMatching('*-' + id + '-*')) {
-      this.member.del(key)
+            return true
+        }
+
+        return false
     }
-  }
 
-  // Deletes member of user in specified guild
+    async memberExists(guildId, id) {
-  async deleteMember (guildId, id) {
+        for await (const key of this.memberGenerators.keysMatching(guildId + '-' + id + '-*')) {
-    this.member.del(guildId + '-' + id)
+            return true
-  }
+        }
+
+        return false
+    }
+
+    async guildExists(id) {
+        for await (const key of this.guildGenerators.keysMatching(id + '-*')) {
+            return true
+        }
+
+        return false
+    }
+
+    async createUser(id) {
+        // No need to create users with all the data in Redis
+    }
+
+    async createMember(guildId, id) {
+        // No need to create members with all the data in Redis
+    }
+
+    async createGuild(id) {
+        // No need to create guilds with all the data in Redis
+    }
+
+    async getUser(id) {
+        let db = this
+
+        return new function() {
+            this.id = id
+            this.db = db
+
+            this.get = async function(key) {
+                let result = await this.db.userGetAsync(this.id + '-' + key)
+
+                if(result === null) {
+                    return this.db.client.config.defaultUserData[key]
+                } else {
+                    if(String(result).startsWith('[') || String(result).startsWith('{')) return JSON.parse(result)
+
+                    return result
+                }
+            }
+
+            this.set = async function(key, value) {
+                if(typeof(value) === 'object') value = JSON.stringify(value)
+
+                return await this.db.userSetAsync(this.id + '-' + key, value)
+            }
+
+            this.delete = async function() {
+                return await this.db.deleteUser(this.id)
+            }
+        }
+    }
+
+    async getMember(guildId, id) {
+        let db = this
+
+        return new function() {
+            this.guildId = guildId
+            this.id = id
+            this.db = db
+
+            this.get = async function(key) {
+                let result = await this.db.memberGetAsync(this.guildId + '-' + this.id + '-' + key)
+
+                if(result === null) {
+                    return this.db.client.config.defaultMemberData[key]
+                } else {
+                    if(String(result).startsWith('[') || String(result).startsWith('{')) return JSON.parse(result)
+
+                    return result
+                }
+            }
+
+            this.set = async function(key, value) {
+                if(typeof(value) === 'object') value = JSON.stringify(value)
+
+                return await this.db.memberSetAsync(this.guildId + '-' + this.id + '-' + key, value)
+            }
+
+            this.delete = async function() {
+                return await this.db.deleteMember(this.guildId, this.id)
+            }
+        }
+    }
+
+    async getGuild(id) {
+        let db = this
+
+        return new function() {
+            this.id = id
+            this.db = db
+
+            this.get = async function(key) {
+                let result = await this.db.guildGetAsync(this.id + '-' + key)
+
+                if(result === null) {
+                    return this.db.client.config.defaultGuildData[key]
+                } else {
+                    if(String(result).startsWith('[') || String(result).startsWith('{')) return JSON.parse(result)
+
+                    return result
+                }
+            }
+
+            this.set = async function(key, value) {
+                if(typeof(value) === 'object') value = JSON.stringify(value)
+
+                return await this.db.guildSetAsync(this.id + '-' + key, value)
+            }
+
+            this.delete = async function() {
+                return await this.db.deleteGuild(this.id)
+            }
+        }
+    }
+
+    // Deletes specified guild entry
+    async deleteGuild(id) {
+        for await (const key of this.guildGenerators.keysMatching(id + '-*')) {
+            this.guild.del(key)
+        }
+    }
+
+    // Deletes specified user and their member entries in guilds
+    async deleteUser(id) {
+        for await (const key of this.userGenerators.keysMatching(id + '-*')) {
+            this.user.del(key)
+        }
+
+        for await (const key of this.memberGenerators.keysMatching('*-' + id + '-*')) {
+            this.member.del(key)
+        }
+    }
+
+    // Deletes member of user in specified guild
+    async deleteMember(guildId, id) {
+        for await (const key of this.memberGenerators.keysMatching(guildId + '-' + id + '-*')) {
+            this.member.del(key)
+        }
+    }
 }
 
-module.exports = Redis 
+module.exports = RedisDatabase