Newer
Older
percord / src / util / connections / ConnectionLoader.ts
@Rory& Rory& on 19 Feb 2024 2 KB Add nix flake to repo (#1111)
/*
	Spacebar: A FOSS re-implementation and extension of the Discord.com backend.
	Copyright (C) 2023 Spacebar and Spacebar Contributors
	
	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU Affero General Public License as published
	by the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.
	
	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU Affero General Public License for more details.
	
	You should have received a copy of the GNU Affero General Public License
	along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

import { Connection } from "@spacebar/util";
import fs from "fs";
import path from "path";
import { ConnectionConfig } from "./ConnectionConfig";
import { ConnectionStore } from "./ConnectionStore";

const root = path.join(__dirname, "..", "..", "connections");
const connectionsLoaded = false;

export class ConnectionLoader {
	public static async loadConnections() {
		if (connectionsLoaded) return;
		ConnectionConfig.init();
		const dirs = fs.readdirSync(root).filter((x) => {
			try {
				fs.readdirSync(path.join(root, x));
				return true;
			} catch (e) {
				return false;
			}
		});

		dirs.forEach(async (x) => {
			const modPath = path.resolve(path.join(root, x));
			const mod = new (require(modPath).default)() as Connection;
			ConnectionStore.connections.set(mod.id, mod);

			mod.init();
			// console.log(`[Connections] Loaded connection '${mod.id}'`);
		});
	}

	public static getConnectionConfig<T>(id: string, defaults?: unknown): T {
		let cfg = ConnectionConfig.get()[id];
		if (defaults) {
			if (cfg) cfg = Object.assign({}, defaults, cfg);
			else {
				cfg = defaults;
				this.setConnectionConfig(id, cfg);
			}
		}

		if (cfg?.enabled) console.log(`[Connections] ${id} enabled`);

		// if (!cfg)
		// 	console.log(
		// 		`[ConnectionConfig/WARN] Getting connection settings for '${id}' returned null! (Did you forget to add settings?)`,
		// 	);
		return cfg;
	}

	public static async setConnectionConfig(
		id: string,
		config: Partial<unknown>,
	): Promise<void> {
		if (!config)
			console.warn(`[Connections/WARN] ${id} tried to set config=null!`);

		await ConnectionConfig.set({
			[id]: Object.assign(
				config,
				ConnectionLoader.getConnectionConfig(id) || {},
			),
		});
	}
}