Equicord/src/equicordplugins/grammarFix/index.ts

/*
 * Vencord, a Discord client mod
 * Copyright (c) 2024 Vendicated and contributors
 * SPDX-License-Identifier: GPL-3.0-or-later
 */

import {
    addPreSendListener,
    removePreSendListener,
    SendListener,
} from "@api/MessageEvents";
import { definePluginSettings } from "@api/Settings";
import { EquicordDevs } from "@utils/constants";
import definePlugin, { OptionType } from "@utils/types";

const settings = definePluginSettings(
    {
        autoCapitalization: {
            type: OptionType.BOOLEAN,
            description: "Auto Capitalization to the first letter"
        },
        autoPunctuation: {
            type: OptionType.BOOLEAN,
            description: "Auto Punctuation at the end of a sentence"
        },
        autoWordReplacement: {
            type: OptionType.BOOLEAN,
            description: "Auto Capitalizes the first letter"
        }
    }
);

const getPresend = dictionary => {
    const presendObject: SendListener = (_, msg) => {
        msg.content = msg.content.trim();
        if (!msg.content.includes("```") && /\w/.test(msg.content.charAt(0))) {
            if (settings.store.autoWordReplacement) {
                const re = new RegExp(
                    `(^|(?<=[^A-Z0-9]+))(${Object.keys(dictionary)
                        .map(k => k.replace(/[.*+?^${}()|[\]\\]/g, "\\$&"))
                        .join("|")})((?=[^A-Z0-9]+)|$)`,
                    "gi"
                );
                if (re !== null) {
                    msg.content = msg.content.replace(re, match => {
                        return dictionary[match.toLowerCase()] || match;
                    });
                }
            }

            if (settings.store.autoPunctuation) {
                if (/[A-Z0-9]/i.test(msg.content.charAt(msg.content.length - 1))) {
                    if (!msg.content.startsWith("http", msg.content.lastIndexOf(" ") + 1))
                        msg.content += ".";
                }
            }

            // Ensure sentences are capitalized after punctuation
            if (settings.store.autoCapitalization) {
                msg.content = msg.content.replace(/([.!?])\s*(\w)/g, match =>
                    match.toUpperCase()
                );

                // Ensure the first character of the entire message is capitalized
                if (!msg.content.startsWith("http")) {
                    msg.content = msg.content.charAt(0).toUpperCase() + msg.content.slice(1);
                }
            }
        }
    };
    return presendObject;
};

export default definePlugin({
    name: "GrammarFix",
    description: "Automatic punctuation, capitalization, and word replacement.",
    authors: [EquicordDevs.unstream],
    dependencies: ["MessageEventsAPI"],
    settings,
    async start() {
        let dictionary = await fetch(
            "https://raw.githubusercontent.com/wont-stream/dictionary/3d52fecd9aca5dfee0fcde0df2c2af357f977df7/index.min.json"
        );
        dictionary = await dictionary.json();

        addPreSendListener(getPresend(dictionary));
    },
    stop() {
        removePreSendListener(getPresend({}));
    },
});