mirror of
https://github.com/discordeno/discordeno.git
synced 2026-06-02 17:00:08 +00:00
104 lines
2.9 KiB
TypeScript
104 lines
2.9 KiB
TypeScript
import type { Channel } from "../structures/channel.ts";
|
|
import type { Guild } from "../structures/guild.ts";
|
|
import type { Message } from "../structures/message.ts";
|
|
import type { PresenceUpdatePayload } from "../types/discord.ts";
|
|
import { cache } from "../utils/cache.ts";
|
|
import type { Collection } from "../utils/collection.ts";
|
|
|
|
export type TableName =
|
|
| "guilds"
|
|
| "channels"
|
|
| "messages"
|
|
| "presences"
|
|
| "unavailableGuilds";
|
|
|
|
function set(
|
|
table: "guilds",
|
|
key: string,
|
|
value: Guild,
|
|
): Promise<Collection<string, Guild>>;
|
|
function set(
|
|
table: "channels",
|
|
key: string,
|
|
value: Channel,
|
|
): Promise<Collection<string, Channel>>;
|
|
function set(
|
|
table: "messages",
|
|
key: string,
|
|
value: Message,
|
|
): Promise<Collection<string, Message>>;
|
|
function set(
|
|
table: "presences",
|
|
key: string,
|
|
value: PresenceUpdatePayload,
|
|
): Promise<Collection<string, PresenceUpdatePayload>>;
|
|
function set(
|
|
table: "unavailableGuilds",
|
|
key: string,
|
|
value: number,
|
|
): Promise<Collection<string, number>>;
|
|
async function set(table: TableName, key: string, value: any) {
|
|
return cache[table].set(key, value);
|
|
}
|
|
|
|
function get(table: "guilds", key: string): Promise<Guild | undefined>;
|
|
function get(table: "channels", key: string): Promise<Channel | undefined>;
|
|
function get(table: "messages", key: string): Promise<Message | undefined>;
|
|
function get(
|
|
table: "presences",
|
|
key: string,
|
|
): Promise<PresenceUpdatePayload | undefined>;
|
|
function get(
|
|
table: "unavailableGuilds",
|
|
key: string,
|
|
): Promise<Guild | undefined>;
|
|
async function get(table: TableName, key: string) {
|
|
return cache[table].get(key);
|
|
}
|
|
|
|
function forEach(
|
|
table: "guilds",
|
|
callback: (value: Guild, key: string, map: Map<string, Guild>) => unknown,
|
|
): void;
|
|
function forEach(
|
|
table: "unavailableGuilds",
|
|
callback: (value: Guild, key: string, map: Map<string, Guild>) => unknown,
|
|
): void;
|
|
function forEach(
|
|
table: "channels",
|
|
callback: (value: Channel, key: string, map: Map<string, Channel>) => unknown,
|
|
): void;
|
|
function forEach(
|
|
table: "messages",
|
|
callback: (value: Message, key: string, map: Map<string, Message>) => unknown,
|
|
): void;
|
|
function forEach(
|
|
table: TableName,
|
|
callback: (value: any, key: string, map: Map<string, any>) => unknown,
|
|
) {
|
|
return cache[table].forEach(callback);
|
|
}
|
|
|
|
export let cacheHandlers = {
|
|
/** Deletes all items from the cache */
|
|
clear: async function (table: TableName) {
|
|
return cache[table].clear();
|
|
},
|
|
/** Deletes 1 item from cache using the key */
|
|
delete: async function (table: TableName, key: string) {
|
|
return cache[table].delete(key);
|
|
},
|
|
/** Check if something exists in cache with a key */
|
|
has: async function (table: TableName, key: string) {
|
|
return cache[table].has(key);
|
|
},
|
|
|
|
// Done differently to have overloads
|
|
/** Add a key value pair to the cache */
|
|
set,
|
|
/** Get the value from the cache using its key */
|
|
get,
|
|
/** Run a function on all items in this cache */
|
|
forEach,
|
|
};
|