harmony/src/managers/base.ts

56 lines
1.5 KiB
TypeScript

import { Client } from '../models/client.ts'
import { Collection } from '../utils/collection.ts'
export class BaseManager<T, T2> {
client: Client
/** Cache Name or Key used to differentiate caches */
cacheName: string
/** Which data type does this cache have */
DataType: any
constructor(client: Client, cacheName: string, DataType: any) {
this.client = client
this.cacheName = cacheName
this.DataType = DataType
}
async _get(key: string): Promise<T | undefined> {
return this.client.cache.get(this.cacheName, key)
}
async get(key: string): Promise<T2 | undefined> {
const raw = await this._get(key)
if (raw === undefined) return
return new this.DataType(this.client, raw)
}
async set(key: string, value: T): Promise<any> {
return this.client.cache.set(this.cacheName, key, value)
}
async delete(key: string): Promise<boolean> {
return this.client.cache.delete(this.cacheName, key)
}
async array(): Promise<undefined | T2[]> {
let arr = await (this.client.cache.array(this.cacheName) as T[])
if (arr === undefined) arr = []
return arr.map((e) => new this.DataType(this.client, e)) as any
}
async collection(): Promise<Collection<string, T2>> {
const arr = await this.array()
if (arr === undefined) return new Collection()
const collection = new Collection()
for (const elem of arr) {
// @ts-expect-error
collection.set(elem.id, elem)
}
return collection
}
flush(): any {
return this.client.cache.deleteCache(this.cacheName)
}
}