From 917c1727ab2a91a058c4b61a897f09e9690d6019 Mon Sep 17 00:00:00 2001
From: jie65535 <jie65535@qq.com>
Date: Sun, 22 Dec 2024 15:05:44 +0800
Subject: [PATCH] Add subject and sender replacement

---
 src/main/kotlin/JChatGPT.kt | 25 +++++++++++++++++++++----
 1 file changed, 21 insertions(+), 4 deletions(-)

diff --git a/src/main/kotlin/JChatGPT.kt b/src/main/kotlin/JChatGPT.kt
index 66940bc..3275fe8 100644
--- a/src/main/kotlin/JChatGPT.kt
+++ b/src/main/kotlin/JChatGPT.kt
@@ -135,12 +135,29 @@ object JChatGPT : KotlinPlugin(
         startChat(context)
     }
 
-    private fun getSystemPrompt(): String {
+    private fun MessageEvent.getSystemPrompt(): String {
         val now = OffsetDateTime.now()
-        return PluginConfig.prompt.replace(
-            "{time}",
+        val prompt = StringBuilder(PluginConfig.prompt)
+        fun replace(target: String, replacement: ()->String) {
+            val i = prompt.indexOf(target)
+            if (i != -1) {
+                prompt.replace(i, i + target.length, replacement())
+            }
+        }
+        replace("{time}") {
             "$now ${now.dayOfWeek.getDisplayName(TextStyle.FULL, Locale.CHINA)}"
-        )
+        }
+        replace("{subject}") {
+            if (this is GroupMessageEvent) {
+                "《${subject.name}》群聊中"
+            } else {
+                "私聊中"
+            }
+        }
+        replace("{sender}") {
+            senderName
+        }
+        return prompt.toString()
     }
 
     private suspend fun MessageEvent.startChat(context: List<ChatMessage>? = null) {