summary refs log tree commit diff
path: root/src/util/connections/ConnectionLoader.ts
blob: 04d7752ae7325c8aad35de4c51f315f89c4646be (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
import fs from "fs";
import path from "path";
import { OrmUtils } from "../imports";
import Connection from "./Connection";
import { ConnectionConfig } from "./ConnectionConfig";
import { ConnectionStore } from "./ConnectionStore";

const root = "dist/connections";
let 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) => {
			let 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(id: string, defaults?: any): any {
		let cfg = ConnectionConfig.get()[id];
		if (defaults) {
			if (cfg) cfg = Object.assign({}, defaults, cfg);
			else {
				cfg = defaults;
				this.setConnectionConfig(id, cfg);
			}
		}

		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<any>,
	): Promise<void> {
		if (!config)
			console.log(
				`[ConnectionConfig/WARN] ${id} tried to set config=null!`,
			);

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