aboutsummaryrefslogtreecommitdiff
path: root/src/plugins/textReplace.tsx
diff options
context:
space:
mode:
Diffstat (limited to 'src/plugins/textReplace.tsx')
-rw-r--r--src/plugins/textReplace.tsx240
1 files changed, 240 insertions, 0 deletions
diff --git a/src/plugins/textReplace.tsx b/src/plugins/textReplace.tsx
new file mode 100644
index 0000000..73ac1f2
--- /dev/null
+++ b/src/plugins/textReplace.tsx
@@ -0,0 +1,240 @@
+/*
+ * Vencord, a modification for Discord's desktop app
+ * Copyright (c) 2023 Vendicated and contributors
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <https://www.gnu.org/licenses/>.
+*/
+
+import { DataStore } from "@api/index";
+import { addPreSendListener, removePreSendListener } from "@api/MessageEvents";
+import { definePluginSettings } from "@api/settings";
+import { Flex } from "@components/Flex";
+import { Devs } from "@utils/constants";
+import Logger from "@utils/Logger";
+import { useForceUpdater } from "@utils/misc";
+import definePlugin, { OptionType } from "@utils/types";
+import { Button, Forms, React, TextInput, useState } from "@webpack/common";
+
+const STRING_RULES_KEY = "TextReplace_rulesString";
+const REGEX_RULES_KEY = "TextReplace_rulesRegex";
+
+type Rule = Record<"find" | "replace" | "onlyIfIncludes", string>;
+
+interface TextReplaceProps {
+ title: string;
+ rulesArray: Rule[];
+ rulesKey: string;
+}
+
+const makeEmptyRule: () => Rule = () => ({
+ find: "",
+ replace: "",
+ onlyIfIncludes: ""
+});
+const makeEmptyRuleArray = () => [makeEmptyRule()];
+
+let stringRules = makeEmptyRuleArray();
+let regexRules = makeEmptyRuleArray();
+
+const settings = definePluginSettings({
+ replace: {
+ type: OptionType.COMPONENT,
+ description: "",
+ component: () =>
+ <>
+ <TextReplace
+ title="Using String"
+ rulesArray={stringRules}
+ rulesKey={STRING_RULES_KEY}
+ />
+ <TextReplace
+ title="Using Regex"
+ rulesArray={regexRules}
+ rulesKey={REGEX_RULES_KEY}
+ />
+ </>
+ },
+});
+
+function stringToRegex(str: string) {
+ const match = str.match(/^(\/)?(.+?)(?:\/([gimsuy]*))?$/); // Regex to match regex
+ return match
+ ? new RegExp(
+ match[2], // Pattern
+ match[3]
+ ?.split("") // Remove duplicate flags
+ .filter((char, pos, flagArr) => flagArr.indexOf(char) === pos)
+ .join("")
+ ?? "g"
+ )
+ : new RegExp(str); // Not a regex, return string
+}
+
+function renderFindError(find: string) {
+ try {
+ stringToRegex(find);
+ return null;
+ } catch (e) {
+ return (
+ <span style={{ color: "var(--text-danger)" }}>
+ {String(e)}
+ </span>
+ );
+ }
+}
+
+function Input({ initialValue, onChange, placeholder }: {
+ placeholder: string;
+ initialValue: string;
+ onChange(value: string): void;
+}) {
+ const [value, setValue] = useState(initialValue);
+ return (
+ <TextInput
+ placeholder={placeholder}
+ value={value}
+ onChange={setValue}
+ spellCheck={false}
+ onBlur={() => value !== initialValue && onChange(value)}
+ />
+ );
+}
+
+function TextReplace({ title, rulesArray, rulesKey }: TextReplaceProps) {
+ const isRegexRules = title === "Using Regex";
+
+ const update = useForceUpdater();
+
+ async function onClickRemove(index: number) {
+ rulesArray.splice(index, 1);
+
+ await DataStore.set(rulesKey, rulesArray);
+ update();
+ }
+
+ async function onChange(e: string, index: number, key: string) {
+ if (index === rulesArray.length - 1)
+ rulesArray.push(makeEmptyRule());
+
+ rulesArray[index][key] = e;
+
+ if (rulesArray[index].find === "" && rulesArray[index].replace === "" && rulesArray[index].onlyIfIncludes === "" && index !== rulesArray.length - 1)
+ rulesArray.splice(index, 1);
+
+ await DataStore.set(rulesKey, rulesArray);
+ update();
+ }
+
+ return (
+ <>
+ <Forms.FormTitle tag="h4">{title}</Forms.FormTitle>
+ <Flex flexDirection="column" style={{ gap: "0.5em" }}>
+ {
+ rulesArray.map((rule, index) =>
+ <React.Fragment key={`${rule.find}-${index}`}>
+ <Flex flexDirection="row" style={{ gap: 0 }}>
+ <Flex flexDirection="row" style={{ flexGrow: 1, gap: "0.5em" }}>
+ <Input
+ placeholder="Find"
+ initialValue={rule.find}
+ onChange={e => onChange(e, index, "find")}
+ />
+ <Input
+ placeholder="Replace"
+ initialValue={rule.replace}
+ onChange={e => onChange(e, index, "replace")}
+ />
+ <Input
+ placeholder="Only if includes"
+ initialValue={rule.onlyIfIncludes}
+ onChange={e => onChange(e, index, "onlyIfIncludes")}
+ />
+ </Flex>
+ <Button
+ size={Button.Sizes.MIN}
+ onClick={() => onClickRemove(index)}
+ style={{
+ background: "none",
+ ...(index === rulesArray.length - 1
+ ? {
+ visibility: "hidden",
+ pointerEvents: "none"
+ }
+ : {}
+ )
+ }}
+ >
+ <svg width="24" height="24" viewBox="0 0 24 24">
+ <title>Delete Rule</title>
+ <path fill="var(--status-danger)" d="M15 3.999V2H9V3.999H3V5.999H21V3.999H15Z" />
+ <path fill="var(--status-danger)" d="M5 6.99902V18.999C5 20.101 5.897 20.999 7 20.999H17C18.103 20.999 19 20.101 19 18.999V6.99902H5ZM11 17H9V11H11V17ZM15 17H13V11H15V17Z" />
+ </svg>
+ </Button>
+ </Flex>
+ {isRegexRules && renderFindError(rule.find)}
+ </React.Fragment>
+ )
+ }
+ </Flex>
+ </>
+ );
+}
+
+export default definePlugin({
+ name: "TextReplace",
+ description: "Replace text in your messages",
+ authors: [Devs.Samu, Devs.AutumnVN],
+ dependencies: ["MessageEventsAPI"],
+
+ settings,
+
+ async start() {
+ stringRules = await DataStore.get(STRING_RULES_KEY) ?? makeEmptyRuleArray();
+ regexRules = await DataStore.get(REGEX_RULES_KEY) ?? makeEmptyRuleArray();
+
+ this.preSend = addPreSendListener((_, msg) => {
+ // pad so that rules can use " word " to only match whole "word"
+ msg.content = " " + msg.content + " ";
+
+ if (stringRules) {
+ for (const rule of stringRules) {
+ if (!rule.find || !rule.replace) continue;
+ if (rule.onlyIfIncludes && !msg.content.includes(rule.onlyIfIncludes)) continue;
+
+ msg.content = msg.content.replaceAll(rule.find, rule.replace);
+ }
+ }
+
+ if (regexRules) {
+ for (const rule of regexRules) {
+ if (!rule.find || !rule.replace) continue;
+ if (rule.onlyIfIncludes && !msg.content.includes(rule.onlyIfIncludes)) continue;
+
+ try {
+ const regex = stringToRegex(rule.find);
+ msg.content = msg.content.replace(regex, rule.replace);
+ } catch (e) {
+ new Logger("TextReplace").error(`Invalid regex: ${rule.find}`);
+ }
+ }
+ }
+
+ msg.content = msg.content.trim();
+ });
+ },
+
+ stop() {
+ removePreSendListener(this.preSend);
+ }
+});