BetterDiscordAddons/Plugins/MessageUtilities/MessageUtilities.plugin.js

/**
 * @name MessageUtilities
 * @authorId 278543574059057154
 * @invite Jx3TjNS
 * @donate https://www.paypal.me/MircoWittrien
 * @patreon https://www.patreon.com/MircoWittrien
 * @website https://github.com/mwittrien/BetterDiscordAddons/tree/master/Plugins/MessageUtilities
 * @source https://raw.githubusercontent.com/mwittrien/BetterDiscordAddons/master/Plugins/MessageUtilities/MessageUtilities.plugin.js
 * @updateUrl https://raw.githubusercontent.com/mwittrien/BetterDiscordAddons/master/Plugins/MessageUtilities/MessageUtilities.plugin.js
 */

module.exports = (_ => {
	const config = {
		"info": {
			"name": "MessageUtilities",
			"author": "DevilBro",
			"version": "1.8.1",
			"description": "Offer a number of useful message options. Remap the keybindings in the settings"
		},
		"changeLog": {
			"added": {
				"Replies": "Added option for replies",
				"Quotes": "Now requires CustomQuoter plugin since Discord replaces Quotes with Replies"
			},
			"fixed": {
				"Settings": "Fixed Settings"
			}
		}
	};

	return !window.BDFDB_Global || (!window.BDFDB_Global.loaded && !window.BDFDB_Global.started) ? class {
		getName () {return config.info.name;}
		getAuthor () {return config.info.author;}
		getVersion () {return config.info.version;}
		getDescription () {return config.info.description;}
		
		load() {
			if (!window.BDFDB_Global || !Array.isArray(window.BDFDB_Global.pluginQueue)) window.BDFDB_Global = Object.assign({}, window.BDFDB_Global, {pluginQueue: []});
			if (!window.BDFDB_Global.downloadModal) {
				window.BDFDB_Global.downloadModal = true;
				BdApi.showConfirmationModal("Library Missing", `The library plugin needed for ${config.info.name} is missing. Please click "Download Now" to install it.`, {
					confirmText: "Download Now",
					cancelText: "Cancel",
					onCancel: _ => {delete window.BDFDB_Global.downloadModal;},
					onConfirm: _ => {
						delete window.BDFDB_Global.downloadModal;
						require("request").get("https://mwittrien.github.io/BetterDiscordAddons/Library/0BDFDB.plugin.js", (e, r, b) => {
							if (!e && b && b.indexOf(`* @name BDFDB`) > -1) require("fs").writeFile(require("path").join(BdApi.Plugins.folder, "0BDFDB.plugin.js"), b, _ => {});
							else BdApi.alert("Error", "Could not download BDFDB library plugin, try again some time later.");
						});
					}
				});
			}
			if (!window.BDFDB_Global.pluginQueue.includes(config.info.name)) window.BDFDB_Global.pluginQueue.push(config.info.name);
		}
		start() {this.load();}
		stop() {}
	} : (([Plugin, BDFDB]) => {
		const clickMap = ["CLICK", "DBLCLICK"];
		var firedEvents = [], clickTimeout;
		var settings = {}, bindings = {}, enabledBindings = {}, toasts = {};
		
		return class MessageUtilities extends Plugin {
			onLoad() {
				this.defaults = {
					settings: {
						"addHints":					{value: true, 	description: "Add keycombo hints to contextmenus: "},
						"clearOnEscape":			{value: true, 	description: "Clear chat input when Escape is pressed: "}
					},
					toasts: {},
					bindings: {
						"Edit_Message":				{name: "Edit Message",			func: this.doEdit,		value: {click: 1, 	keycombo: []}		},
						"Delete_Message":			{name: "Delete Message",		func: this.doDelete,	value: {click: 0, 	keycombo: [46]}		},
						"Pin/Unpin_Message":		{name: "Pin/Unpin Message",		func: this.doPinUnPin,	value: {click: 0, 	keycombo: [17]}		},
						"Reply_to_Message":			{name: "Reply to Message",		func: this.doReply,		value: {click: 0, 	keycombo: [17,84]}	},
						"React_to_Message":			{name: "Open React Menu",		func: this.doOpenReact,	value: {click: 0, 	keycombo: [17,83]}	},
						"Copy_Raw":					{name: "Copy raw Message",		func: this.doCopyRaw,	value: {click: 0, 	keycombo: [17,68]}	},
						"Copy_Link":				{name: "Copy Message Link",		func: this.doCopyLink,	value: {click: 0, 	keycombo: [17,81]}	},
						"__Quote_Message":			{name: "Quote Message",			func: this.doQuote,		value: {click: 0, 	keycombo: [17,87]}, plugin: "CustomQuoter"},
						"__Note_Message":			{name: "Note Message",			func: this.doNote,		value: {click: 0, 	keycombo: [16]}, 	plugin: "PersonalPins"},
						"__Translate_Message":		{name: "Translate Message",		func: this.doTranslate,	value: {click: 0, 	keycombo: [20]}, 	plugin: "GoogleTranslateOption"},
						"__Reveal_Spoilers":		{name: "Reveal All Spoilers",	func: this.doReveal,	value: {click: 0, 	keycombo: [17,74]}, plugin: "RevealAllSpoilersOption"}
					}
				};
				
				this.patchedModules = {
					before: {
						Menu: "default"
					}
				};
				
				for (let type in this.defaults.bindings) {
					let nativeAction = type.indexOf("__") != 0;
					this.defaults.settings[type] = {value: nativeAction};
					if (nativeAction) this.defaults.toasts[type] = {value: type != "Edit_Message" && type != "React_to_Message" && type != "Quote_Message"};
				}
			}
			
			onStart() {
				BDFDB.ListenerUtils.add(this, document, "click", BDFDB.dotCNC.message + BDFDB.dotCN.searchresultsgroupcozy, e => {
					if (!BDFDB.DOMUtils.getParent(BDFDB.dotCN.messagetoolbarbutton, e.target)) this.onClick(e, 0, "onSglClick");
				});
				BDFDB.ListenerUtils.add(this, document, "dblclick", BDFDB.dotCNC.message + BDFDB.dotCN.searchresultsgroupcozy, e => {
					if (!BDFDB.DOMUtils.getParent(BDFDB.dotCN.messagetoolbarbutton, e.target)) this.onClick(e, 1, "onDblClick");
				});
				BDFDB.ListenerUtils.add(this, document, "keydown", e => {
					if (BDFDB.DOMUtils.getParent(BDFDB.dotCN.textareawrapchat, document.activeElement)) this.onKeyDown(document.activeElement, e.which, "onKeyDown");
				});
				
				this.forceUpdateAll();
			}
			
			onStop() {
				this.forceUpdateAll();
			}

			getSettingsPanel (collapseStates = {}) {
				let settingsPanel;
				return settingsPanel = BDFDB.PluginUtils.createSettingsPanel(this, {
					collapseStates: collapseStates,
					children: _ => {
						let settingsItems = [];
						
						for (let key in settings) if (this.defaults.settings[key].description) settingsItems.push(BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.SettingsSaveItem, {
							type: "Switch",
							plugin: this,
							keys: ["settings", key],
							label: this.defaults.settings[key].description,
							value: settings[key]
						}));
						settingsItems.push(BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.FormComponents.FormDivider, {
							className: BDFDB.disCN.marginbottom8
						}));
						for (let action in bindings) if (!this.defaults.bindings[action].plugin || BDFDB.BDUtils.isPluginEnabled(this.defaults.bindings[action].plugin)) {
							settingsItems.push(BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.Flex, {
								className: BDFDB.disCN.marginbottom8,
								align: BDFDB.LibraryComponents.Flex.Align.CENTER,
								direction: BDFDB.LibraryComponents.Flex.Direction.HORIZONTAL,
								children: [
									BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.SettingsLabel, {
										label: this.defaults.bindings[action].name + (this.defaults.bindings[action].plugin ? ` (${this.defaults.bindings[action].plugin})` : "")
									}),
									toasts[action] != undefined ? BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.SettingsSaveItem, {
										type: "Switch",
										mini: true,
										plugin: this,
										keys: ["toasts", action],
										grow: 0,
										label: "Toast:",
										value: toasts[action]
									}) : null
								].filter(n => n)
							}));
							settingsItems.push(BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.SettingsSaveItem, {
								type: "Switch",
								dividerBottom: true,
								mini: true,
								plugin: this,
								keys: ["settings", action],
								value: settings[action],
								labelchildren: BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.Flex, {
									direction: BDFDB.LibraryComponents.Flex.Direction.HORIZONTAL,
									children: [
										BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.KeybindRecorder, {
											defaultValue: bindings[action].keycombo.filter(n => n),
											reset: true,
											onChange: keycombo => {
												bindings[action].keycombo = keycombo;
												BDFDB.DataUtils.save(bindings, this, "bindings");
												this.SettingsUpdated = true;
											}
										}),
										BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.Select, {
											value: bindings[action].click,
											options: clickMap.map((label, i) => {return {value: i, label: label}}),
											onChange: choice => {
												bindings[action].click = choice.value;
												BDFDB.DataUtils.save(bindings, this, "bindings");
												this.SettingsUpdated = true;
											}
										})
									]
								})
							}));
						}
						settingsItems.push(BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.SettingsItem, {
							type: "Button",
							color: BDFDB.LibraryComponents.Button.Colors.RED,
							label: "Reset all Key Bindings",
							onClick: (e, instance) => {
								BDFDB.ModalUtils.confirm(this, "Are you sure you want to reset all Key Bindings?", _ => {
									BDFDB.DataUtils.remove(this, "bindings");
									settingsPanel.parentElement.appendChild(this.getSettingsPanel());
									settingsPanel.remove();
									this.SettingsUpdated = true;
								});
							},
							children: BDFDB.LanguageUtils.LanguageStrings.RESET
						}));
						
						return settingsItems;
					}
				});
			}

			onSettingsClosed () {
				if (this.SettingsUpdated) {
					delete this.SettingsUpdated;
					this.forceUpdateAll();
				}
			}
			
			forceUpdateAll () {
				settings = BDFDB.DataUtils.get(this, "settings");
				bindings = BDFDB.DataUtils.get(this, "bindings");
				enabledBindings = BDFDB.ObjectUtils.filter(bindings, action => settings[action], true);
				toasts = BDFDB.DataUtils.get(this, "toasts");
				
				BDFDB.PatchUtils.forceAllUpdates(this);
			}
		
			processMenu (e) {
				let contextMenu = BDFDB.ReactUtils.findChild(e.instance, {props: "navId"});
				if (contextMenu && BDFDB.ArrayUtils.is(contextMenu.props.children)) for (let group of contextMenu.props.children) {
					if (group && group.type == BDFDB.LibraryComponents.MenuItems.MenuGroup && BDFDB.ArrayUtils.is(group.props.children)) for (let item of group.props.children) {
						if (item && item.props && item.props.id && !item.props.hint && !item.props.children) {
							let hint, action;
							if (item.props.id == "mark-unread") hint = settings.addHints && `${BDFDB.LibraryModules.KeyCodeUtils.getString(18)}+${clickMap[0]}`;
							else {
								switch (item.props.id) {
									case "copy-link":
										action = "Copy_Link";
										break;
									case "edit":
										action = "Edit_Message";
										break;
									case "pin":
									case "unpin":
										action = "Pin/Unpin_Message";
										break;
									case "delete":
										action = "Delete_Message";
										break;
									case "quote":
										action = "Quote_Message";
										break;
								}
								if (action) hint = this.getActiveShortcutString(action);
							}
							if (hint) item.props.hint = _ => {
								return BDFDB.ReactUtils.createElement(BDFDB.LibraryComponents.MenuItems.MenuHint, {
									hint: hint
								});
							};
						}
					}
				}
			}

			onClick (e, click, name) {
				if (!firedEvents.includes(name)) {
					firedEvents.push(name);
					let priorityAction = null;
					for (let action in enabledBindings) {
						let binding = enabledBindings[action];
						let priorityBinding = enabledBindings[priorityAction];
						if (this.checkIfBindingIsValid(binding, click) && (!enabledBindings[priorityAction] || binding.click > priorityBinding.click || binding.keycombo.length > priorityBinding.keycombo.length)) priorityAction = action;
					}
					if (priorityAction) {
						let {messageDiv, message} = this.getMessageData(e.currentTarget);
						if (messageDiv && message) {
							BDFDB.ListenerUtils.stopEvent(e);
							BDFDB.TimeUtils.clear(clickTimeout);
							if (!this.hasDoubleClickOverwrite(enabledBindings[priorityAction])) {
								this.defaults.bindings[priorityAction].func.apply(this, [{messageDiv, message}, priorityAction]);
							}
							else clickTimeout = BDFDB.TimeUtils.timeout(_ => {
								this.defaults.bindings[priorityAction].func.apply(this, [{messageDiv, message}, priorityAction]);
							}, 500);
						}
					}
					BDFDB.TimeUtils.timeout(_ => {BDFDB.ArrayUtils.remove(firedEvents, name, true)});
				}
			}

			checkIfBindingIsValid (binding, doneclick) {
				if (binding.click != doneclick) return false;
				for (let key of binding.keycombo) if (!BDFDB.ListenerUtils.isPressed(key)) return false;
				return true;
			}

			hasDoubleClickOverwrite (binding) {
				if (binding.click == 1) return false;
				let dblBindings = BDFDB.ObjectUtils.filter(enabledBindings, bndg => {return bndg.click == 1});
				for (let dblAction in dblBindings) {
					let dblBinding = dblBindings[dblAction];
					let overwrite = true;
					if (BDFDB.equals(binding.keycombo, dblBinding.keycombo)) return true;
				}
				return false;
			}

			doDelete ({messageDiv, message}, action) {
				let deleteLink = messageDiv.parentElement.querySelector(BDFDB.dotCNS.messagelocalbotoperations + BDFDB.dotCN.anchor);
				if (deleteLink) deleteLink.click();
				else {
					let channel = BDFDB.LibraryModules.ChannelStore.getChannel(message.channel_id);
					if ((channel && BDFDB.UserUtils.can("MANAGE_MESSAGES")) || message.author.id == BDFDB.UserUtils.me.id && message.type != 1 && message.type != 2 && message.type != 3) {
						BDFDB.LibraryModules.MessageUtils.deleteMessage(message.channel_id, message.id, message.state != "SENT");
						if (toasts[action]) BDFDB.NotificationUtils.toast("Message has been deleted", {type: "success"});
					}
				}
			}

			doEdit ({messageDiv, message}, action) {
				if (message.author.id == BDFDB.UserUtils.me.id && !messageDiv.querySelector("textarea")) {
					BDFDB.LibraryModules.MessageUtils.startEditMessage(message.channel_id, message.id, message.content);
					if (toasts[action]) BDFDB.NotificationUtils.toast("Started editing.", {type: "success"});
				}
			}

			doOpenReact ({messageDiv, message}, action) {
				let reactButton = messageDiv.querySelector(`${BDFDB.dotCN.messagetoolbarbutton}[aria-label="${BDFDB.LanguageUtils.LanguageStrings.ADD_REACTION}"]`);
				if (reactButton) {
					reactButton.click();
					if (toasts[action]) BDFDB.NotificationUtils.toast("Reaction popout has been opened", {type: "success"});
				}
			}

			doPinUnPin ({messageDiv, message}, action) {
				if (message.state == "SENT") {
					let channel = BDFDB.LibraryModules.ChannelStore.getChannel(message.channel_id);
					if (channel && (channel.type == 1 || channel.type == 3 || BDFDB.UserUtils.can("MANAGE_MESSAGES")) && message.type == 0) {
						if (message.pinned) {
							BDFDB.LibraryModules.MessagePinUtils.unpinMessage(channel, message.id);
							if (toasts[action]) BDFDB.NotificationUtils.toast("Message has been unpinned", {type: "error"});
						}
						else {
							BDFDB.LibraryModules.MessagePinUtils.pinMessage(channel, message.id);
							if (toasts[action]) BDFDB.NotificationUtils.toast("Message has been pinned", {type: "success"});
						}
					}
				}
			}
			
			doReply ({messageDiv, message}, action) {
				if (message.state == "SENT") {
					let channel = BDFDB.LibraryModules.ChannelStore.getChannel(message.channel_id);
					if (channel && BDFDB.UserUtils.can("SEND_MESSAGES") && message.type == 0) {
						BDFDB.LibraryModules.MessageManageUtils.replyToMessage(channel, message);
						if (toasts[action]) BDFDB.NotificationUtils.toast("Added reply to message", {type: "success"});
					}
				}
			}

			doCopyRaw ({messageDiv, message}, action) {
				if (message.content) {
					BDFDB.LibraryRequires.electron.clipboard.write({text: message.content});
					if (toasts[action]) BDFDB.NotificationUtils.toast("Raw message content has been copied", {type: "success"});
				}
			}

			doCopyLink ({messageDiv, message}, action) {
				let channel = BDFDB.LibraryModules.ChannelStore.getChannel(message.channel_id);
				if (channel) {
					BDFDB.LibraryModules.MessageManageUtils.copyLink(channel, message);
					if (toasts[action]) BDFDB.NotificationUtils.toast("Messagelink has been copied", {type: "success"});
				}
			}

			doQuote ({messageDiv, message}, action) {
				if (BDFDB.BDUtils.isPluginEnabled(this.defaults.bindings.__Quote_Message.plugin)) {
					let channel = BDFDB.LibraryModules.ChannelStore.getChannel(message.channel_id);
					if (channel) BDFDB.BDUtils.getPlugin(this.defaults.bindings.__Quote_Message.plugin).quote(channel, message);
				}
			}

			doNote ({messageDiv, message}, action) {
				if (BDFDB.BDUtils.isPluginEnabled(this.defaults.bindings.__Note_Message.plugin)) {
					let channel = BDFDB.LibraryModules.ChannelStore.getChannel(message.channel_id);
					if (channel) BDFDB.BDUtils.getPlugin(this.defaults.bindings.__Note_Message.plugin).addMessageToNotes(message, channel);
				}
			}

			doTranslate ({messageDiv, message}, action) {
				if (BDFDB.BDUtils.isPluginEnabled(this.defaults.bindings.__Translate_Message.plugin)) {
					let channel = BDFDB.LibraryModules.ChannelStore.getChannel(message.channel_id);
					if (channel) BDFDB.BDUtils.getPlugin(this.defaults.bindings.__Translate_Message.plugin).translateMessage(message, channel);
				}
			}

			doReveal ({messageDiv, message}, action) {
				if (BDFDB.BDUtils.isPluginEnabled(this.defaults.bindings.__Reveal_Spoilers.plugin)) {
					BDFDB.BDUtils.getPlugin(this.defaults.bindings.__Reveal_Spoilers.plugin).revealAllSpoilers(messageDiv);
				}
			}

			onKeyDown (target, key, name) {
				if (!firedEvents.includes(name)) {
					firedEvents.push(name);
					if (key == 27 && settings.clearOnEscape) {
						let chatform = BDFDB.DOMUtils.getParent(BDFDB.dotCN.chatform, target);
						if (chatform) {
							let instance = BDFDB.ReactUtils.findOwner(chatform, {name: "ChannelTextAreaForm"}) || BDFDB.ReactUtils.findOwner(chatform, {name: "ChannelTextAreaForm", up: true});
							if (instance) instance.setState({textValue: "", richValue: BDFDB.LibraryModules.SlateUtils.deserialize("")});
						}
					}
					BDFDB.TimeUtils.timeout(_ => {BDFDB.ArrayUtils.remove(firedEvents, name, true)});
				}
			}

			getActiveShortcutString (action) {
				if (!action) return null;
				let str = [];
				if (settings.addHints && settings[action] && enabledBindings[action]) {
					if (enabledBindings[action].keycombo.length) str.push(BDFDB.LibraryModules.KeyCodeUtils.getString(enabledBindings[action].keycombo));
					str.push(clickMap[enabledBindings[action].click]);
				}
				return str.join("+").replace(/ /g, "");
			}

			getMessageData (target) {
				let messageDiv = BDFDB.DOMUtils.getParent(BDFDB.dotCNC.message + BDFDB.dotCN.searchresultsgroupcozy, target);
				if (messageDiv && messageDiv.querySelector(BDFDB.dotCN.textarea)) return {messageDiv: null, message: null};
				let instance = BDFDB.ReactUtils.getInstance(messageDiv);
				let message = instance && (BDFDB.ReactUtils.findValue(instance, "baseMessage") || BDFDB.ReactUtils.findValue(instance, "message"));
				return {messageDiv, message};
			}
		};
	})(window.BDFDB_Global.PluginUtils.buildPlugin(config));
})();