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;
}