index.js

const {
	entities: { Plugin },
	webpack: { React, getModule, getModuleByDisplayName },
	util: { getOwnerInstance, getReactInstance, findInReactTree },
	injector: { inject, uninject },
} = require("powercord");

const Settings = require("./components/Settings");
const BetterRepliedMessage = require("./components/BetterRepliedMessage");

const RepliedMessage = getModule(
	(m) => m?.default?.displayName === "RepliedMessage",
	false
);
const ChatChannelMessage = getModule(
	(m) =>
		m.type &&
		(m.__powercordOriginal_type || m.type)
			.toString()
			.indexOf("useContextMenuMessage") !== -1,
	false
);
const SearchChannelMessage = getModule(
	(m) => m?.type?.displayName === "ChannelMessage",
	false
);

module.exports = class BetterReplies extends (
	Plugin
) {
	async startPlugin() {
		this.loadStylesheet("style.scss");

		powercord.api.settings.registerSettings(this.entityID, {
			category: this.entityID,
			label: "Better Replies",
			render: (settings) =>
				React.createElement(Settings, {
					settings,
					plugin: this,
					settingsID: this.entityID,
				}),
		});

		inject(
			"better-replies-chat-channel-message",
			ChatChannelMessage,
			"type",
			this.channelMessage
		);
		ChatChannelMessage.type.displayName = "ChannelMessage";

		inject(
			"better-replies-search-channel-message",
			SearchChannelMessage,
			"type",
			this.channelMessage
		);
		SearchChannelMessage.type.displayName = "ChannelMessage";

		window.KLibrary?.Tools?.ReactTools?.rerenderAllMessages();
	}

	pluginWillUnload() {
		uninject("better-replies-chat-channel-message");
		uninject("better-replies-search-channel-message");
		powercord.api.settings.unregisterSettings(this.entityID);

		window.KLibrary?.Tools?.ReactTools?.rerenderAllMessages();
	}

	channelMessage = (args, res) => {
		const repliedMessage = args?.[0]?.message?.messageReference;
		const depth = parseInt(res.props?.id?.split("depth-")?.[1] ?? 0);
		const messageElement =
			findInReactTree(res, (r) => r.childrenRepliedMessage) ?? res.props;

		if (
			args?.[0]?.message?.messageReference &&
			depth < this.settings.get("max-depth", 2)
		) {
			messageElement.childrenRepliedMessage = React.createElement(
				BetterRepliedMessage,
				{
					referenceMode:
						res.props.referenceMode ??
						this.settings.get("reference-mode", "auto"),
					referenceStyle:
						res.props.referenceStyle ??
						this.settings.get("reference-style", "better"),
					settings: this.settings,
					depth: depth + 1,
					message_id: repliedMessage?.message_id,
					channel_id: repliedMessage?.channel_id,
				}
			);
		}
		return res;
	};
};