embee/woomy-v2
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

re-add unoptimised code for now

Browse source
This commit is contained in:
Emily 2020-10-08 13:09:41 +11:00
parent 5f6c498cb2
commit ebb39b997d
1 changed files with 187 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

219
bot/util/redis.js
View file

@ -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 {
constructor(client) {
this.client = client;
}
class RedisDatabase extends Database {
constructor(client) {
super(client)
// Config
let conf = this.client.config.redis
init () {
// Create redis client
this.global = redis.createClient(this.client.config.redis)
this.server = this.global.duplicate({ db: 1 })
this.member = this.global.duplicate({ db: 2 })
this.user = this.global.duplicate({ db: 3 })
}
// Create redis client
this.global = redis.createClient(conf)
this.guild = this.global.duplicate({ db: 1 })
this.member = this.global.duplicate({ db: 2 })
this.user = this.global.duplicate({ db: 3 })
// Deletes specified guild entry
async deleteGuild (id) {
this.server.del(id)
var { keysMatching } = await generators.using(this.member)
// eslint-disable-next-line no-unused-vars
for await (const key of keysMatching(id + '-*')) {
this.member.del(key)
// Async
this.guildGetAsync = promisify(this.guild.get).bind(this.guild)
this.memberGetAsync = promisify(this.member.get).bind(this.member)
this.userGetAsync = promisify(this.user.get).bind(this.user)
this.guildSetAsync = promisify(this.guild.set).bind(this.guild)
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 deleteUser (id) {
this.user.del(id)
var { keysMatching } = await generators.using(this.member)
// eslint-disable-next-line no-unused-vars
for await (const key of keysMatching('*-' + id)) {
this.member.del(key)
async userExists(id) {
for await (const key of this.userGenerators.keysMatching(id + '-*')) {
return true
}
for await (const key of this.memberGenerators.keysMatching('*-' + id + '-*')) {
return true
}
return false
}
}
// Deletes member of user in specified guild
async deleteMember (guildId, id) {
this.member.del(guildId + '-' + id)
}
async memberExists(guildId, id) {
for await (const key of this.memberGenerators.keysMatching(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