const { Message, BaseInteraction, User } = require("discord.js");

User.prototype.getUsername = function () {
	return this.discriminator === "0" ? this.username : this.tag;
};

BaseInteraction.prototype.getLocale = function () {
	return this.guild ? this.data?.guild?.language : "en-US";
};

BaseInteraction.prototype.translate = function (key, args) {
	const language = this.client.translations.get(this.getLocale());
	if (!language) throw "Interaction: Invalid language set in data.";

	return language(key, args);
};

BaseInteraction.prototype.replyT = async function (key, args, options = {}) {
	const translated = this.translate(key, args, this.getLocale());
	const string = options.prefixEmoji ? `${this.client.customEmojis[options.prefixEmoji]} | ${translated}` : translated;

	if (options.edit) return await this.editReply({ content: string, ephemeral: options.ephemeral || false });
	else return await this.reply({ content: string, ephemeral: options.ephemeral || false });
};

BaseInteraction.prototype.success = async function (key, args, options = {}) {
	options.prefixEmoji = "success";

	return await this.replyT(key, args, options);
};

BaseInteraction.prototype.error = async function (key, args, options = {}) {
	options.prefixEmoji = "error";

	return await this.replyT(key, args, options);
};

Message.prototype.getLocale = function () {
	return this.guild ? this.data?.guild?.language : "en-US";
};

Message.prototype.translate = function (key, args) {
	const language = this.client.translations.get(this.getLocale());
	if (!language) throw "Message: Invalid language set in data.";

	return language(key, args);
};

Message.prototype.replyT = async function (key, args, options = {}) {
	const translated = this.translate(key, args, this.getLocale());
	const string = options.prefixEmoji ? `${this.client.customEmojis[options.prefixEmoji]} | ${translated}` : translated;

	if (options.edit) return await this.edit({ content: string, allowedMentions: { repliedUser: options.mention ? true : false } });
	else return await this.reply({ content: string, allowedMentions: { repliedUser: options.mention ? true : false } });
};

Message.prototype.success = async function (key, args, options = {}) {
	options.prefixEmoji = "success";

	return await this.replyT(key, args, options);
};

Message.prototype.error = async function (key, args, options = {}) {
	options.prefixEmoji = "error";

	return await this.replyT(key, args, options);
};