From cd3f83ffd70b47530911ee1dbc1117c70d9756a5 Mon Sep 17 00:00:00 2001
From: pixtaded <pixtaded@noreply.localhost>
Date: Sat, 31 Aug 2024 17:50:27 +0300
Subject: [PATCH] Add prompt filtering

---
 config.example.json |  3 ++-
 index.js            | 10 +++++++---
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/config.example.json b/config.example.json
index 7ba9b36..07d4c7a 100644
--- a/config.example.json
+++ b/config.example.json
@@ -3,5 +3,6 @@
     "ALLOWED_IDS": ["1266775948548903026", "1151129960141037659", "1275008490959667234", "1234866477086019648", "1277611369578889277", "1242022836529266708", "1275098630059200586", "1277311957405405296", "1277936618749300760", "1275899048171409528"],
     "admin": "593524756393754635",
     "shapeGuild": "1277635303363973173",
-    "shapeChannel": "1277652579547086860"
+    "shapeChannel": "1277652579547086860",
+    "shapeUserId": "1277634934781120542"
 }
\ No newline at end of file
diff --git a/index.js b/index.js
index 4cf283c..7066d28 100644
--- a/index.js
+++ b/index.js
@@ -39,21 +39,25 @@ async function getMessageContent(message) {
             const buffer = Buffer.from(response.data, "binary");
             const file = new MessageAttachment(buffer, attachment.name);
 
-            if (message.content) return { content: `${message.author.displayName}: ${message.content} `, files: [file] };
+            if (message.content) return { content: filterPrompt(`${message.author.displayName}: ${message.content}`), files: [file] };
             else return {files: [file]};
         } catch (error) {
             console.error("Ошибка при получении файла:", error.message);
             return "Я нихуя не понял.";
         }
     } else {
-        if (!message.reference) return message.author.displayName + ": " + message.content;
+        if (!message.reference) return filterPrompt(`${message.author.displayName}: ${message.content}`);
         else {
             let reply = message.channel.messages.fetch(message.reference.messageId);
-            return `${message.author.displayName}: ${message.content} \`Ответ на: ${reply.author.displayName}: + ${reply.content}\``;
+            return filterPrompt(`${message.author.displayName}: ${message.content} \`Ответ на: ${reply.author.displayName}: ${reply.content}\``);
         } 
     }
 }
 
+function filterPrompt(string) {
+    return string.replace(client.user.id, config.shapeUserId).replace(client.guilds.cache.get(config.shapeGuild).members.cache.get(config.shapeUserId));
+}
+
 async function sendToAI(message) {
     if (message.author.id == client.user.id) return;