feat: 绘画预设支持识别用户昵称以制作meme

Author: misaka20002

apps/Doubao.js

@@ -243,7 +243,7 @@ export class Doubao extends plugin {
         }
 
         // 处理预设
-        const presetResult = applyPresets(msg, Config.getConfig("presets"))
+        const presetResult = applyPresets(msg, Config.getConfig("presets"), e)
         msg = presetResult.processedText
 
         // 处理 msg，使用 handleParam 解析参数

apps/Jimeng.js

@@ -96,7 +96,7 @@ export class Jimeng extends plugin {
         }
 
         // 处理预设
-        const presetResult = applyPresets(msg, Config.getConfig("presets"))
+        const presetResult = applyPresets(msg, Config.getConfig("presets"), e)
         msg = presetResult.processedText
 
         // 处理 msg

apps/SF_Painting.js

@@ -589,7 +589,7 @@ export class SF_Painting extends plugin {
         let msg = e.msg.replace(/^#(flux|FLUX|(sf|SF)(画图|绘图|绘画))(\n*)?/, '').trim()
 
         // 处理预设
-        const presetResult = applyPresets(msg, Config.getConfig("presets"))
+        const presetResult = applyPresets(msg, Config.getConfig("presets"), e)
         msg = presetResult.processedText
 
         // 处理 msg
@@ -821,7 +821,7 @@ export class SF_Painting extends plugin {
 
         // 处理预设
         if (paintModel) {
-            const presetResult = applyPresets(toAiMessage, Config.getConfig("presets"))
+            const presetResult = applyPresets(toAiMessage, Config.getConfig("presets"), e)
             toAiMessage = presetResult.processedText
             msg = presetResult.originalText
             // 处理 msg
@@ -1586,7 +1586,7 @@ ${e.sfRuntime.isgeneratePrompt === undefined ? "Tags中可用：--自动提示
 
         // 处理预设
         if (paintModel) {
-            const presetResult = applyPresets(toAiMessage, Config.getConfig("presets"))
+            const presetResult = applyPresets(toAiMessage, Config.getConfig("presets"), e)
             toAiMessage = presetResult.processedText
             msg = presetResult.originalText
             // 处理 msg

guoba.support.js

@@ -851,7 +851,7 @@ export function supportGuoba() {
                 field: "prompt",
                 label: "预设文本",
                 component: "InputTextArea",
-                bottomHelpMessage: "支持的固定参数: 横图, 竖图, 方图, --1:1, --16:9, --9:16, --upimgs 2, reference_strength = 0.8 等",
+                bottomHelpMessage: "1.支持的固定参数: 横图, 竖图, 方图, --1:1, --16:9, --9:16, --upimgs 2, reference_strength = 0.8 等；2.meme制作：支持将预设文本中的 _sender_name_ 替换为 被At的用户或当前用户昵称； _sender_id_ 替换为 被At的用户或当前用户qq； _sender_groupid_ 替换为 当前群号； _date_ 替换为 当前日期； _time_ 替换为 当前时间；",
               },
             ],
           },

utils/applyPresets.js

@@ -1,14 +1,16 @@
+import _ from 'lodash'
+
 /**
  * 应用预设到文本中
  * @param {string} text - 原始输入文本
  * @param {object} config - 配置对象,包含 presets 数组
- * @param {Array} config.presets - 预设数组,每个预设包含 name 和 prompt
+ * @param {object} e 需要 e.sender
  * @returns {object} 返回对象包含:
  *   - processedText: 处理后的文本(预设名替换为预设文本)
  *   - usedPresets: 使用过的预设数组 [{name, prompt}]
  *   - originalText: 处理后的文本(预设名替换为占位符： {sf预设: ${presetName}} )
  */
-export function applyPresets(text, config) {
+export function applyPresets(text, config, e = {}) {
     const originalTextInput = text || '';
     if (!text || typeof text !== 'string') {
         return {
@@ -94,6 +96,9 @@ export function applyPresets(text, config) {
         originalText = originalText.slice(0, start) + `{sf预设: ${preset.name}}` + originalText.slice(end)
     }
 
+    // 应用设定拓展
+    processedText = replacePromptForSenderMsg(e, processedText);
+
     return {
         processedText: processedText.trim(),
         usedPresets,
@@ -122,4 +127,46 @@ export function generatePresetInfo(usedPresets) {
 
     const presetNames = usedPresets.map(p => p.name).join('、')
     return `\n使用预设：${presetNames}`
+}
+
+/** 在 prompt 中替换文本使用 e.sender 信息 */
+function replacePromptForSenderMsg(e, systemMsg = "") {
+    if (!e.sender || !e.message)
+        return systemMsg;
+
+    /** at或当前用户昵称 */
+    let name = ''
+    if (e.message.filter(m => m.type === 'at').length > 0) {
+        name = _.trim(e.message.filter(m => m.type === 'at')[0].text, '@')
+
+        systemMsg = systemMsg.replace(/_sender_id_/igm, e.self_id)
+    } else {
+        name = e.sender.card || e.sender.nickname
+
+        systemMsg = systemMsg.replace(/_sender_id_/igm, e.sender.user_id)
+        systemMsg = systemMsg.replace(/_sender_gender_/igm, e.sender.sex)
+        systemMsg = systemMsg.replace(/_sender_age_/igm, e.sender.age)
+        systemMsg = systemMsg.replace(/_sender_area_/igm, e.sender.area)
+        systemMsg = systemMsg.replace(/_sender_role_/igm, `${e.sender.role == "owner" ? '群主' : `${e.sender.role == "admin" ? '管理员' : ''}`}`)
+        systemMsg = systemMsg.replace(/_sender_title_/igm, e.sender.title)
+    }
+
+    const getCurrentDate = () => {
+        const date = new Date();
+        const year = date.getFullYear();
+        const month = String(date.getMonth() + 1).padStart(2, '0');
+        const day = String(date.getDate()).padStart(2, '0');
+        return `${year}-${month}-${day}`;
+    };
+    const getCurrentTime = () => {
+        const date = new Date();
+        const hours = String(date.getHours()).padStart(2, '0');
+        const minutes = String(date.getMinutes()).padStart(2, '0');
+        return `${hours}:${minutes}`;
+    };
+    systemMsg = systemMsg.replace(/_sender_name_/igm, name)
+    systemMsg = systemMsg.replace(/_date_/igm, getCurrentDate())
+    systemMsg = systemMsg.replace(/_time_/igm, getCurrentTime())
+    systemMsg = systemMsg.replace(/_sender_groupid_/igm, e.group_id)
+    return systemMsg;
 }