ass/backend/UserConfig.ts

import { UserConfiguration, UserConfigTypeChecker, PostgresConfiguration } from 'ass';

import fs from 'fs-extra';
import { path } from '@tycrek/joint';
import { log } from './log.js';
import fetch from "node-fetch";

const FILEPATH = path.join('.ass-data/userconfig.json');

/**
 * Returns a boolean if the provided value is a number
 */
const numChecker = (val: any) => {
	try { return !isNaN(parseInt(val)) && typeof val !== 'string'; }
	catch (err) { return false; }
}

/**
 * Returns a boolean if the provided value is a non-empty string
 */
const basicStringChecker = (val: any) => typeof val === 'string' && val.length > 0;

/**
 * User-config property type checker functions
 */
const Checkers: UserConfigTypeChecker = {
	uploadsDir: (val) => {
		try {
			fs.pathExistsSync(val)
				? fs.accessSync(val)
				: fs.mkdirSync(val, { recursive: true });
			return true;
		}
		catch (err) {
			log.warn('Cannot access directory', `${val}`);
			console.error(err);
			return false;
		}
	},
	idType: (val) => ['random', 'original', 'gfycat', 'timestamp', 'zws'].includes(val),
	idSize: numChecker,
	gfySize: numChecker,
	maximumFileSize: numChecker,
	discordWebhook: (val) => {
		const regex = /^https:\/\/discord\.com\/api\/webhooks\/\d+\/\S+$/;
		return regex.test(val);
	},

	s3: {
		endpoint: basicStringChecker,
		bucket: basicStringChecker,
		region: (val) => val == null || basicStringChecker(val),
		credentials: {
			accessKey: basicStringChecker,
			secretKey: basicStringChecker
		}
	},

	sql: {
		mySql: {
			host: basicStringChecker,
			user: basicStringChecker,
			password: basicStringChecker,
			database: basicStringChecker,
			port: (val) => numChecker(val) && val >= 1 && val <= 65535
		},
		postgres: {
			port: (val) => numChecker(val) && val >= 1 && val <= 65535
		}
	},

	rateLimit: {
		endpoint: (val) => val == null || (val != null && (numChecker(val.requests) && numChecker(val.duration)))
	}
};

export class UserConfig {
	private static _config: UserConfiguration;
	private static _ready = false;

	public static get config() { return UserConfig._config; }
	public static get ready() { return UserConfig._ready; }

	constructor(config?: any) {
		// Typically this would only happen during first-time setup (for now)
		if (config != null) {
			UserConfig._config = UserConfig.parseConfig(config);
			UserConfig._ready = true;
		}
	}

	/**
	 * Ensures that all config options are valid
	 */
	private static parseConfig(c: any) {
		const config = (typeof c === 'string' ? JSON.parse(c) : c) as UserConfiguration;

		// * Base config
		if (!Checkers.uploadsDir(config.uploadsDir)) throw new Error(`Unable to access uploads directory: ${config.uploadsDir}`);
		if (!Checkers.idType(config.idType)) throw new Error(`Invalid ID type: ${config.idType}`);
		if (!Checkers.idSize(config.idSize)) throw new Error('Invalid ID size');
		if (!Checkers.gfySize(config.gfySize)) throw new Error('Invalid Gfy size');
		if (!Checkers.maximumFileSize(config.maximumFileSize)) throw new Error('Invalid maximum file size');
		if (!Checkers.discordWebhook(config.discordWebhook)) throw new Error('Invalid Discord webhook');

		// * Optional S3 config
		if (config.s3 != null) {
			if (!Checkers.s3.endpoint(config.s3.endpoint)) throw new Error('Invalid S3 Endpoint');
			if (!Checkers.s3.bucket(config.s3.bucket)) throw new Error('Invalid S3 Bucket');
			if (!Checkers.s3.region(config.s3.region)) throw new Error('Invalid S3 Region');
			if (!Checkers.s3.credentials.accessKey(config.s3.credentials.accessKey)) throw new Error('Invalid S3 Access key');
			if (!Checkers.s3.credentials.secretKey(config.s3.credentials.secretKey)) throw new Error('Invalid S3 Secret key');
		}

		// * Optional database config(s)
		if (config.database != null) {
			// these both have the same schema so we can just check both
			if (config.database.kind == 'mysql' || config.database.kind == 'postgres') {
				if (config.database.options != undefined) {
					if (!Checkers.sql.mySql.host(config.database.options.host)) throw new Error('Invalid database host');
					if (!Checkers.sql.mySql.user(config.database.options.user)) throw new Error('Invalid databse user');
					if (!Checkers.sql.mySql.password(config.database.options.password)) throw new Error('Invalid database password');
					if (!Checkers.sql.mySql.database(config.database.options.database)) throw new Error('Invalid database');
					if (!Checkers.sql.mySql.port(config.database.options.port)) throw new Error('Invalid database port');
					if (config.database.kind == 'postgres') {
						if (!Checkers.sql.postgres.port((config.database.options as PostgresConfiguration).port)) {
							throw new Error("Invalid database port");
						}
					}
				} else throw new Error('Database options missing');
			}
		}

		// * optional rate limit config
		if (config.rateLimit != null) {
			if (!Checkers.rateLimit.endpoint(config.rateLimit.login)) throw new Error('Invalid Login rate limit configuration');
			if (!Checkers.rateLimit.endpoint(config.rateLimit.upload)) throw new Error('Invalid Upload rate limit configuration');
			if (!Checkers.rateLimit.endpoint(config.rateLimit.api)) throw new Error('Invalid API rate limit configuration');
		}

		// All is fine, carry on!
		return config;
	}

	/**
	 * Save the config file to disk
	 */
	public static saveConfigFile(): Promise<void> {
		return new Promise(async (resolve, reject) => {
			try {

				// Only save is the config has been parsed
				if (!UserConfig._ready) throw new Error('Config not ready to be saved!');

				// Write to file
				await fs.writeFile(FILEPATH, JSON.stringify(UserConfig._config, null, '\t'));

				resolve(void 0);
			} catch (err) {
				log.error('Failed to save config file!');
				reject(err);
			}
		});
	}

	/**
	 * Reads the config file from disk
	 */
	public static readConfigFile(): Promise<void> {
		return new Promise(async (resolve, reject) => {
			try {

				// Read the file data
				const data = (await fs.readFile(FILEPATH)).toString();

				// Ensure the config is valid
				UserConfig._config = UserConfig.parseConfig(data);
				UserConfig._ready = true;

				resolve(void 0);
			} catch (err) {
				log.error('Failed to read config file!');
				reject(err);
			}
		});
	}
}