chore: bold flags

index.tsx

@@ -1,9 +1,136 @@
+/*
+ * Vencord, a Discord client mod
+ * Copyright (c) 2024 Vendicated and contributors
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+import { ApplicationCommandInputType, ApplicationCommandOptionType, Argument, CommandContext, sendBotMessage } from "@api/Commands";
+import { DataStore } from "@api/index";
+import { addAccessory } from "@api/MessageAccessories";
 import { Devs } from "@utils/constants";
 import definePlugin from "@utils/types";
+import { React, Text } from "@webpack/common";
+import { Message } from "discord-types/general";
+
+const flags = new Map<string, Flag>();
+type Flag = {
+    type: "danger" | "warning" | "info" | "positive";
+    text: string;
+};
+
+function Flag({ id }: { id: string; }) {
+    return (
+        <div>
+            <Text
+                variant="text-md/bold"
+                style={{ color: flags.get(id)?.type === "danger" ? "#FF7473" : flags.get(id)?.type === "warning" ? "#ffb02e" : flags.get(id)?.type === "info" ? "#62A8FF" : "#62FF74" }}
+            >
+                {flags.get(id)?.type === "danger" ? "🛑" : flags.get(id)?.type === "warning" ? "⚠️" : flags.get(id)?.type === "info" ? "🔵" : "✅"} {flags.get(id)?.text}
+            </Text>
+        </div>
+    );
+}
 
 export default definePlugin({
     name: "UserFlags",
-    description: "Add "flags" to users that will always show under their messages",
+    description: `Add "flags" to users that will always show under their messages`,
     authors: [Devs.nin0dev],
+    dependencies: ["MessageAccessoriesAPI"],
+    async start() {
+        const savedFlags = await DataStore.get("USERFLAGS");
+        if (savedFlags) {
+            const parsedFlags = new Map<string, Flag>(JSON.parse(savedFlags));
+            parsedFlags.forEach((value, key) => {
+                flags.set(key, value);
+            });
+        }
+        addAccessory("flag", (props: Record<string, any>) => {
+            if (flags.has((props.message as Message).author.id)) {
+                return <Flag id={props.message.author.id} />;
+            }
+            return null;
+        }, 4);
+    },
+    commands: [
+        {
+            name: "flag set",
+            description: "Set a flag on a user",
+            inputType: ApplicationCommandInputType.BOT,
+            options: [
+                {
+                    name: "user",
+                    type: ApplicationCommandOptionType.USER,
+                    description: "The user to set a flag to",
+                    required: true
+                },
+                {
+                    "name": "type",
+                    "type": ApplicationCommandOptionType.STRING,
+                    "description": "The type of flag to add",
+                    choices: [
+                        {
+                            label: "Danger",
+                            value: "danger",
+                            name: "Danger"
+                        },
+                        {
+                            label: "Warning",
+                            value: "warning",
+                            name: "Warning"
+                        },
+                        {
+                            label: "Info",
+                            value: "info",
+                            name: "Info"
+                        },
+                        {
+                            label: "Positive",
+                            value: "positive",
+                            name: "Positive"
+                        }
+                    ],
+                    required: true
+                },
+                {
+                    name: "message",
+                    type: ApplicationCommandOptionType.STRING,
+                    description: "The flag content",
+                    required: true
+                },
+            ],
+            execute: async (args: Argument[], ctx: CommandContext) => {
+                flags.set(args[0].value, {
+                    type: args[1].value as "danger" | "warning" | "info" | "positive",
+                    text: args[2].value
+                });
+                sendBotMessage(ctx.channel.id, {
+                    content: `Flag set on <@${args[0].value}> with content \`${args[2].value}\`!`
+                });
+                await DataStore.set("USERFLAGS", JSON.stringify(Array.from(flags.entries())));
+                return;
+            }
+        },
+        {
+            name: "flag delete",
+            description: "Delete the flag from a user",
+            inputType: ApplicationCommandInputType.BOT,
+            options: [
+                {
+                    name: "user",
+                    type: ApplicationCommandOptionType.USER,
+                    description: "The user to delete the flag from",
+                    required: true
+                }
+            ],
+            execute: async (args: Argument[], ctx: CommandContext) => {
+                flags.delete(args[0].value);
+                sendBotMessage(ctx.channel.id, {
+                    content: `Flag removed from <@${args[0].value}>`
+                });
+                await DataStore.set("USERFLAGS", JSON.stringify(Array.from(flags.entries())));
+                return;
+            }
+        }
+    ]
 });