TravBot-v3/src/core/structures.ts

import FileManager from "./storage";
import $, {select, GenericJSON, GenericStructure} from "./lib";
import {watch} from "fs";
import {Guild as DiscordGuild} 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;
	
	constructor(data?: GenericJSON)
	{
		this.money = select(data?.money, 0, Number);
		this.lastReceived = select(data?.lastReceived, -1, Number);
	}
}

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))
			$.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))
			$.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(process.argv[2] === "dev")
{
	watch("data", (event, filename) => {
		$.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
{
	return Storage.getGuild(guild?.id || "N/A").prefix ?? Config.prefix;
}