1 files changed, 50 insertions, 0 deletions
diff --git a/src/lib/extensions/BotMessage.ts b/src/lib/extensions/BotMessage.ts
new file mode 100644
index 0000000..85c2721
--- /dev/null
+++ b/src/lib/extensions/BotMessage.ts
@@ -0,0 +1,50 @@
+import {
+	TextChannel,
+	NewsChannel,
+	DMChannel,
+	Message,
+	Structures
+} from 'discord.js';
+import { BotClient } from './BotClient';
+import { Guild as GuildModel } from '../types/Models';
+import { BotGuild } from './BotGuild';
+
+export class GuildSettings {
+	private message: BotMessage;
+	constructor(message: BotMessage) {
+		this.message = message;
+	}
+	public async getPrefix(): Promise<string> {
+		return await GuildModel.findByPk(this.message.guild.id).then(
+			(gm) => gm?.prefix || this.message.client.config.prefix
+		);
+	}
+	public async setPrefix(value: string): Promise<void> {
+		let entry = await GuildModel.findByPk(this.message.guild.id);
+		if (!entry) {
+			entry = GuildModel.build({
+				id: this.message.guild.id,
+				prefix: value
+			});
+		} else {
+			entry.prefix = value;
+		}
+		await entry.save();
+	}
+}
+
+export class BotMessage extends Message {
+	constructor(
+		client: BotClient,
+		data: Record<string, unknown>,
+		channel: TextChannel | DMChannel | NewsChannel
+	) {
+		super(client, data, channel);
+	}
+	public guild: BotGuild;
+	public client: BotClient;
+	static install(): void {
+		Structures.extend('Message', () => BotMessage);
+	}
+	public settings = new GuildSettings(this);
+}