mirror of
https://github.com/keanuplayz/TravBot-v3.git
synced 2024-08-15 02:33:12 +00:00
141 lines
4.7 KiB
TypeScript
141 lines
4.7 KiB
TypeScript
import FileManager from "./modules/storage";
|
|
import {select, GenericJSON, GenericStructure} from "./lib";
|
|
import {watch} from "fs";
|
|
import {Guild as DiscordGuild, Snowflake} from "discord.js";
|
|
|
|
class ConfigStructure extends GenericStructure {
|
|
public token: string;
|
|
public prefix: string;
|
|
public owner: string;
|
|
public admins: string[];
|
|
public support: string[];
|
|
|
|
constructor(data: GenericJSON) {
|
|
super("config");
|
|
this.token = select(data.token, "<ENTER YOUR TOKEN HERE>", String);
|
|
this.prefix = select(data.prefix, "$", String);
|
|
this.owner = select(data.owner, "", String);
|
|
this.admins = select(data.admins, [], String, true);
|
|
this.support = select(data.support, [], String, true);
|
|
}
|
|
}
|
|
|
|
class User {
|
|
public money: number;
|
|
public lastReceived: number;
|
|
public lastMonday: number;
|
|
public timezone: number | null; // This is for the standard timezone only, not the daylight savings timezone
|
|
public daylightSavingsRegion: "na" | "eu" | "sh" | null;
|
|
|
|
constructor(data?: GenericJSON) {
|
|
this.money = select(data?.money, 0, Number);
|
|
this.lastReceived = select(data?.lastReceived, -1, Number);
|
|
this.lastMonday = select(data?.lastMonday, -1, Number);
|
|
this.timezone = data?.timezone ?? null;
|
|
this.daylightSavingsRegion = /^((na)|(eu)|(sh))$/.test(data?.daylightSavingsRegion)
|
|
? data?.daylightSavingsRegion
|
|
: null;
|
|
}
|
|
}
|
|
|
|
class Guild {
|
|
public prefix: string | null;
|
|
|
|
constructor(data?: GenericJSON) {
|
|
this.prefix = select(data?.prefix, null, String);
|
|
}
|
|
}
|
|
|
|
class StorageStructure extends GenericStructure {
|
|
public users: {[id: string]: User};
|
|
public guilds: {[id: string]: Guild};
|
|
|
|
constructor(data: GenericJSON) {
|
|
super("storage");
|
|
this.users = {};
|
|
this.guilds = {};
|
|
|
|
for (let id in data.users) if (/\d{17,19}/g.test(id)) this.users[id] = new User(data.users[id]);
|
|
|
|
for (let id in data.guilds) if (/\d{17,19}/g.test(id)) this.guilds[id] = new Guild(data.guilds[id]);
|
|
}
|
|
|
|
/** Gets a user's profile if they exist and generate one if not. */
|
|
public getUser(id: string): User {
|
|
if (!/\d{17,19}/g.test(id))
|
|
console.warn(`"${id}" is not a valid user ID! It will be erased when the data loads again.`);
|
|
|
|
if (id in this.users) return this.users[id];
|
|
else {
|
|
const user = new User();
|
|
this.users[id] = user;
|
|
return user;
|
|
}
|
|
}
|
|
|
|
/** Gets a guild's settings if they exist and generate one if not. */
|
|
public getGuild(id: string): Guild {
|
|
if (!/\d{17,19}/g.test(id))
|
|
console.warn(`"${id}" is not a valid guild ID! It will be erased when the data loads again.`);
|
|
|
|
if (id in this.guilds) return this.guilds[id];
|
|
else {
|
|
const guild = new Guild();
|
|
this.guilds[id] = guild;
|
|
return guild;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Exports instances. Don't worry, importing it from different files will load the same instance.
|
|
export let Config = new ConfigStructure(FileManager.read("config"));
|
|
export let Storage = new StorageStructure(FileManager.read("storage"));
|
|
|
|
// This part will allow the user to manually edit any JSON files they want while the program is running which'll update the program's cache.
|
|
// However, fs.watch is a buggy mess that should be avoided in production. While it helps test out stuff for development, it's not a good idea to have it running outside of development as it causes all sorts of issues.
|
|
if (IS_DEV_MODE) {
|
|
watch("data", (event, filename) => {
|
|
console.debug("File Watcher:", event, filename);
|
|
const header = filename.substring(0, filename.indexOf(".json"));
|
|
|
|
switch (header) {
|
|
case "config":
|
|
Config = new ConfigStructure(FileManager.read("config"));
|
|
break;
|
|
case "storage":
|
|
Storage = new StorageStructure(FileManager.read("storage"));
|
|
break;
|
|
}
|
|
});
|
|
}
|
|
|
|
export function getPrefix(guild: DiscordGuild | null): string {
|
|
let prefix = Config.prefix;
|
|
|
|
if (guild) {
|
|
const possibleGuildPrefix = Storage.getGuild(guild.id).prefix;
|
|
|
|
// Here, lossy comparison works in our favor because you wouldn't want an empty string to trigger the prefix.
|
|
if (possibleGuildPrefix) {
|
|
prefix = possibleGuildPrefix;
|
|
}
|
|
}
|
|
|
|
return prefix;
|
|
}
|
|
|
|
export interface EmoteRegistryDumpEntry {
|
|
ref: string;
|
|
id: Snowflake;
|
|
name: string;
|
|
requires_colons: boolean;
|
|
animated: boolean;
|
|
url: string;
|
|
guild_id: Snowflake;
|
|
guild_name: string;
|
|
}
|
|
|
|
export interface EmoteRegistryDump {
|
|
version: number;
|
|
list: EmoteRegistryDumpEntry[];
|
|
}
|