koishi-plugin-imgdraw-selfuse 0.0.7 → 0.1.0

package/lib/index.d.ts

@@ -26,6 +26,26 @@ export declare const Config: Schema<Schemastery.ObjectS<{
     txt2imgModel: Schema<string, string>;
     img2imgModel: Schema<string, string>;
     maxImages: Schema<number, number>;
+    enableImgCompress: Schema<boolean, boolean>;
+    imgMaxWidth: Schema<number, number>;
+    imgMaxHeight: Schema<number, number>;
+    imgQuality: Schema<number, number>;
+    imgMaxFileSize: Schema<number, number>;
+    enableImg2ImgBase64: Schema<boolean, boolean>;
+    enablePresets: Schema<boolean, boolean>;
+    presets: Schema<Schemastery.ObjectS<{
+        enable: Schema<boolean, boolean>;
+        text: Schema<string, string>;
+        command: Schema<string, string>;
+        keyword: Schema<string, string>;
+        enableKeywordMatch: Schema<boolean, boolean>;
+    }>[], Schemastery.ObjectT<{
+        enable: Schema<boolean, boolean>;
+        text: Schema<string, string>;
+        command: Schema<string, string>;
+        keyword: Schema<string, string>;
+        enableKeywordMatch: Schema<boolean, boolean>;
+    }>[]>;
     apiList: Schema<Schemastery.ObjectS<{
         enable: Schema<boolean, boolean>;
         apiKey: Schema<string, string>;
@@ -105,6 +125,26 @@ export declare const Config: Schema<Schemastery.ObjectS<{
     txt2imgModel: Schema<string, string>;
     img2imgModel: Schema<string, string>;
     maxImages: Schema<number, number>;
+    enableImgCompress: Schema<boolean, boolean>;
+    imgMaxWidth: Schema<number, number>;
+    imgMaxHeight: Schema<number, number>;
+    imgQuality: Schema<number, number>;
+    imgMaxFileSize: Schema<number, number>;
+    enableImg2ImgBase64: Schema<boolean, boolean>;
+    enablePresets: Schema<boolean, boolean>;
+    presets: Schema<Schemastery.ObjectS<{
+        enable: Schema<boolean, boolean>;
+        text: Schema<string, string>;
+        command: Schema<string, string>;
+        keyword: Schema<string, string>;
+        enableKeywordMatch: Schema<boolean, boolean>;
+    }>[], Schemastery.ObjectT<{
+        enable: Schema<boolean, boolean>;
+        text: Schema<string, string>;
+        command: Schema<string, string>;
+        keyword: Schema<string, string>;
+        enableKeywordMatch: Schema<boolean, boolean>;
+    }>[]>;
     apiList: Schema<Schemastery.ObjectS<{
         enable: Schema<boolean, boolean>;
         apiKey: Schema<string, string>;

package/lib/index.js

@@ -10,6 +10,14 @@ const axios_1 = __importDefault(require("axios"));
 const yaml_1 = __importDefault(require("yaml"));
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
+// 尝试导入 sharp，如果未安装则给出提示
+let sharp;
+try {
+    sharp = require('sharp');
+}
+catch {
+    sharp = null;
+}
 exports.name = 'ai-image';
 exports.inject = {
     required: ['console', 'i18n', 'database'],
@@ -17,6 +25,14 @@ exports.inject = {
 };
 const logger = new koishi_1.Logger('ai-image');
 // ==================== 配置 Schema ====================
+// 预置提示词单项配置
+const PresetItem = koishi_1.Schema.object({
+    enable: koishi_1.Schema.boolean().default(true).description('启用此预置提示词'),
+    text: koishi_1.Schema.string().default('').description('预置提示词文本（将自动添加到 prompt 前）'),
+    command: koishi_1.Schema.string().default('').description('触发指令（如 draw0，留空则自动生成 drawN）'),
+    keyword: koishi_1.Schema.string().default('').description('匹配关键词（如 猫娘，留空则不启用关键词匹配）'),
+    enableKeywordMatch: koishi_1.Schema.boolean().default(false).description('启用关键词匹配（用户 prompt 包含关键词时自动添加）'),
+}).description('预置提示词配置项');
 exports.Config = koishi_1.Schema.object({
     debug: koishi_1.Schema.boolean().default(false).description('开启调试模式，输出完整请求日志'),
     apiStrategy: koishi_1.Schema.union([
@@ -30,6 +46,20 @@ exports.Config = koishi_1.Schema.object({
     txt2imgModel: koishi_1.Schema.string().default('').description('文生图专用模型，留空则使用通用模型'),
     img2imgModel: koishi_1.Schema.string().default('').description('图生图专用模型，留空则使用通用模型'),
     maxImages: koishi_1.Schema.number().default(5).description('图生图最大支持图片数量'),
+    // ==================== 图片压缩配置 ====================
+    enableImgCompress: koishi_1.Schema.boolean().default(true).description('启用图生图图片压缩（推荐开启，可防止大图超时）'),
+    imgMaxWidth: koishi_1.Schema.number().default(1536).description('图片压缩最大宽度（像素）'),
+    imgMaxHeight: koishi_1.Schema.number().default(1536).description('图片压缩最大高度（像素）'),
+    imgQuality: koishi_1.Schema.number().default(85).description('JPEG 压缩质量 1-100（越高越清晰，建议 80-90）'),
+    imgMaxFileSize: koishi_1.Schema.number().default(3).description('图片最大体积（MB），超过会进一步压缩'),
+    // ==========================================================
+    // ==================== 新增：图生图 base64 转换开关 ====================
+    enableImg2ImgBase64: koishi_1.Schema.boolean().default(true).description('图生图将图片转换为 base64（关闭则直接传 URL，部分 API 不需要 base64）'),
+    // ==========================================================
+    // ==================== 预置提示词配置 ====================
+    enablePresets: koishi_1.Schema.boolean().default(false).description('启用预置提示词功能（仅用于文生图）'),
+    presets: koishi_1.Schema.array(PresetItem).default([]).description('预置提示词列表（可添加多个，支持指令触发和关键词匹配）'),
+    // ==========================================================
     apiList: koishi_1.Schema.array(koishi_1.Schema.object({
         enable: koishi_1.Schema.boolean().default(true).description('启用此 API'),
         apiKey: koishi_1.Schema.string().description('API Key'),
@@ -37,7 +67,7 @@ exports.Config = koishi_1.Schema.object({
     })).default([]).description('API 配置列表（支持多账号负载）'),
     enableTxt2Img: koishi_1.Schema.boolean().default(true).description('启用文生图'),
     enableImg2Img: koishi_1.Schema.boolean().default(true).description('启用图生图'),
-    command: koishi_1.Schema.string().default('draw').description('文生图指令'),
+    command: koishi_1.Schema.string().default('draw').description('文生图主指令'),
     aliases: koishi_1.Schema.array(String).default([]).description('文生图指令别名'),
     img2imgCommand: koishi_1.Schema.string().default('imgdraw').description('图生图指令'),
     img2imgAliases: koishi_1.Schema.array(String).default([]).description('图生图指令别名'),
@@ -74,6 +104,11 @@ exports.Config = koishi_1.Schema.object({
 // ==================== 主函数 ====================
 async function apply(ctx, cfg) {
     const debug = cfg.debug;
+    // 检查 sharp 是否安装
+    if (cfg.enableImgCompress && !sharp) {
+        logger.warn('图片压缩已启用，但未检测到 sharp 库。请运行：npm install sharp');
+        logger.warn('在未安装 sharp 的情况下，将跳过压缩，可能导致大图超时');
+    }
     // 加载本地化文件
     try {
         const loc = path_1.default.join(__dirname, 'locales', 'zh-CN.yml');
@@ -126,11 +161,11 @@ async function apply(ctx, cfg) {
         if (!str)
             return '';
         // 1. 清理标准 HTML 标签
-        let cleaned = str.replace(/<<[^>]+>/g, ' ');
+        let cleaned = str.replace(/<[^>]+>/g, ' ');
         // 2. 清理 QQ XML 图片标签（如 <img src="..." file="..."/>）
-        cleaned = cleaned.replace(/<<img\s+[^>]+\/>/gi, ' ');
+        cleaned = cleaned.replace(/<img\s+[^>]+\/>/gi, ' ');
         // 3. 清理其他 XML 标签
-        cleaned = cleaned.replace(/<<[^>]+>/g, ' ');
+        cleaned = cleaned.replace(/<[^>]+>/g, ' ');
         // 4. 清理多余空格和换行
         cleaned = cleaned.replace(/\s+/g, ' ').trim();
         return cleaned;
@@ -153,27 +188,104 @@ async function apply(ctx, cfg) {
             return markdownMatch[1];
         return null;
     }
-    // ==================== 新增：URL 转 base64 ====================
-    async function urlToBase64(url) {
+    // ==================== 图片压缩函数 ====================
+    async function compressImage(buffer) {
+        if (!sharp || !cfg.enableImgCompress) {
+            return buffer;
+        }
+        try {
+            let image = sharp(buffer);
+            const metadata = await image.metadata();
+            // 计算缩放比例
+            let width = metadata.width || cfg.imgMaxWidth;
+            let height = metadata.height || cfg.imgMaxHeight;
+            const ratio = Math.min(cfg.imgMaxWidth / width, cfg.imgMaxHeight / height, 1 // 不放大
+            );
+            if (ratio < 1) {
+                width = Math.round(width * ratio);
+                height = Math.round(height * ratio);
+                image = image.resize(width, height, { fit: 'inside' });
+            }
+            // 压缩并转换格式
+            let quality = Math.max(1, Math.min(100, cfg.imgQuality));
+            let compressed = await image
+                .jpeg({ quality, progressive: true, mozjpeg: true })
+                .toBuffer();
+            // 如果还超过限制，进一步降低质量
+            const maxBytes = cfg.imgMaxFileSize * 1024 * 1024;
+            while (compressed.length > maxBytes && quality > 40) {
+                quality -= 5;
+                compressed = await image
+                    .jpeg({ quality, progressive: true, mozjpeg: true })
+                    .toBuffer();
+            }
+            if (debug) {
+                logger.info(`图片压缩: ${buffer.length} -> ${compressed.length} bytes ` +
+                    `(${Math.round(compressed.length / buffer.length * 100)}%), ` +
+                    `尺寸: ${Math.round(width)}x${Math.round(height)}, 质量: ${quality}`);
+            }
+            return compressed;
+        }
+        catch (e) {
+            logger.error('图片压缩失败，使用原图', e);
+            return buffer;
+        }
+    }
+    async function compressBase64Image(base64Url) {
+        if (!sharp || !cfg.enableImgCompress) {
+            return base64Url;
+        }
+        try {
+            const base64Data = base64Url.split(',')[1];
+            const buffer = Buffer.from(base64Data, 'base64');
+            const compressed = await compressImage(buffer);
+            return `data:image/jpeg;base64,${compressed.toString('base64')}`;
+        }
+        catch (e) {
+            logger.error('base64 图片压缩失败，使用原图', e);
+            return base64Url;
+        }
+    }
+    // ==========================================================
+    // ==================== 修改：URL 转 base64（支持开关）====================
+    /**
+     * 处理图片 URL，根据配置决定返回 base64 还是原始 URL
+     * @param url 图片 URL
+     * @param forceBase64 强制使用 base64（覆盖全局配置）
+     */
+    async function processImageUrl(url, forceBase64) {
         if (!url)
             return null;
+        // 如果已经是 base64，检查是否需要压缩
         if (url.startsWith('data:image/')) {
+            if (cfg.enableImgCompress && sharp) {
+                return await compressBase64Image(url);
+            }
             return url;
         }
+        // 判断是否转换为 base64
+        const needBase64 = forceBase64 !== undefined ? forceBase64 : cfg.enableImg2ImgBase64;
+        if (!needBase64) {
+            // 直接返回 URL（不转换 base64）
+            if (debug)
+                logger.info('图生图使用原始 URL:', url.slice(0, 100));
+            return url;
+        }
+        // 转换为 base64
         try {
             const res = await axios_1.default.get(url, {
                 responseType: 'arraybuffer',
                 timeout: 30000,
             });
-            const base64 = Buffer.from(res.data).toString('base64');
-            const mime = res.headers['content-type'] || 'image/jpeg';
-            return `data:${mime};base64,${base64}`;
+            const compressed = await compressImage(Buffer.from(res.data));
+            return `data:image/jpeg;base64,${compressed.toString('base64')}`;
         }
         catch (e) {
             logger.error('图片转 base64 失败', e);
             throw new Error('图片下载失败，请检查 selfUrl 是否可访问');
         }
     }
+    // ==========================================================
     // 统一发送图片函数
     async function sendImage(session, imgUrl) {
         const trimmed = imgUrl.trim();
@@ -318,6 +430,68 @@ async function apply(ctx, cfg) {
         }
         return { success, fail };
     }
+    // ==================== 预置提示词处理函数 ====================
+    /**
+     * 获取所有启用的预置提示词配置
+     */
+    function getEnabledPresets() {
+        if (!cfg.enablePresets || !cfg.presets)
+            return [];
+        return cfg.presets.filter((p) => p.enable && p.text);
+    }
+    /**
+     * 根据指令名查找匹配的预置提示词
+     * @param cmd 指令名，如 "draw0"
+     */
+    function findPresetByCommand(cmd) {
+        const presets = getEnabledPresets();
+        for (const preset of presets) {
+            const presetCmd = preset.command?.trim();
+            if (presetCmd && presetCmd === cmd) {
+                return preset;
+            }
+        }
+        return null;
+    }
+    /**
+     * 根据用户 prompt 关键词匹配预置提示词
+     * @param prompt 用户输入的 prompt
+     */
+    function findPresetsByKeyword(prompt) {
+        const presets = getEnabledPresets();
+        const matched = [];
+        if (!prompt)
+            return matched;
+        const lowerPrompt = prompt.toLowerCase();
+        for (const preset of presets) {
+            if (preset.enableKeywordMatch && preset.keyword) {
+                const keywords = preset.keyword.split(/[,，|\/\s]+/).map((k) => k.trim().toLowerCase()).filter(Boolean);
+                for (const kw of keywords) {
+                    if (lowerPrompt.includes(kw)) {
+                        matched.push(preset);
+                        break;
+                    }
+                }
+            }
+        }
+        return matched;
+    }
+    /**
+     * 拼接预置提示词和用户 prompt
+     * @param prompt 用户输入的原始 prompt
+     * @param presets 要应用的预置提示词列表
+     */
+    function buildPromptWithPresets(prompt, presets) {
+        if (!presets || presets.length === 0)
+            return prompt;
+        const presetTexts = presets.map((p) => p.text.trim()).filter(Boolean);
+        if (presetTexts.length === 0)
+            return prompt;
+        const combinedPreset = presetTexts.join('，');
+        // 预置提示词放在前面，用户 prompt 在后面
+        return `${combinedPreset}，${prompt}`;
+    }
+    // ==========================================================
     // ==================== 核心生成函数 ====================
     async function generate(session, prompt, imageUrl, modelOverride) {
         if (!checkRateLimit()) {
@@ -334,14 +508,14 @@ async function apply(ctx, cfg) {
         const model = modelOverride || cfg.model;
         let content;
         if (imageUrl) {
-            const base64Url = await urlToBase64(imageUrl);
-            if (!base64Url) {
-                await safeSend(session, cfg.messages.fail + '（图片转换失败）');
+            const processedUrl = await processImageUrl(imageUrl);
+            if (!processedUrl) {
+                await safeSend(session, cfg.messages.fail + '（图片处理失败）');
                 return;
             }
             content = [
                 { type: 'text', text: prompt },
-                { type: 'image_url', image_url: { url: base64Url } },
+                { type: 'image_url', image_url: { url: processedUrl } },
             ];
         }
         else {
@@ -400,14 +574,14 @@ async function apply(ctx, cfg) {
         }
         const model = modelOverride || cfg.model;
         const finalPrompt = prompt.replace('{url}', imageUrls.join(', '));
-        const base64Urls = (await Promise.all(imageUrls.map(url => urlToBase64(url)))).filter((url) => url !== null);
-        if (base64Urls.length === 0) {
-            await safeSend(session, cfg.messages.fail + '（图片转换失败）');
+        const processedUrls = (await Promise.all(imageUrls.map(url => processImageUrl(url)))).filter((url) => url !== null);
+        if (processedUrls.length === 0) {
+            await safeSend(session, cfg.messages.fail + '（图片处理失败）');
             return;
         }
         const content = [
             { type: 'text', text: finalPrompt },
-            ...base64Urls.map(url => ({ type: 'image_url', image_url: { url } })),
+            ...processedUrls.map(url => ({ type: 'image_url', image_url: { url } })),
         ];
         const body = {
             model,
@@ -494,30 +668,90 @@ async function apply(ctx, cfg) {
         }
         return cleanHtmlTags(textParts.join(' '));
     }
+    // ==================== 新增：文生图核心逻辑（支持预置提示词）====================
+    async function doTxt2Img(session, rawPrompt, explicitPresets) {
+        if (!session)
+            return;
+        if (await isBlacklisted(session.userId)) {
+            await safeSend(session, cfg.messages.blacklisted);
+            return;
+        }
+        if (!cfg.enableTxt2Img) {
+            await safeSend(session, cfg.messages.txt2imgDisabled);
+            return;
+        }
+        let prompt = cleanHtmlTags(rawPrompt || '');
+        if (!prompt) {
+            await safeSend(session, cfg.messages.empty);
+            return;
+        }
+        // 处理预置提示词
+        const appliedPresets = [];
+        // 1. 显式指定的预置提示词（通过指令触发）
+        if (explicitPresets && explicitPresets.length > 0) {
+            appliedPresets.push(...explicitPresets);
+        }
+        // 2. 关键词匹配（仅在未通过指令触发时，或允许叠加时）
+        if (cfg.enablePresets) {
+            const keywordPresets = findPresetsByKeyword(prompt);
+            for (const kp of keywordPresets) {
+                // 避免重复添加
+                if (!appliedPresets.some((p) => p.text === kp.text)) {
+                    appliedPresets.push(kp);
+                }
+            }
+        }
+        // 拼接预置提示词
+        if (appliedPresets.length > 0) {
+            prompt = buildPromptWithPresets(prompt, appliedPresets);
+            if (debug) {
+                logger.info(`应用预置提示词: ${appliedPresets.map((p) => p.command || 'keyword').join(', ')}`);
+                logger.info(`最终 prompt: ${prompt.slice(0, 200)}...`);
+            }
+        }
+        await safeSend(session, cfg.messages.generating);
+        const finalPrompt = cfg.txt2imgPrompt.replace('{prompt}', prompt);
+        const model = cfg.txt2imgModel || cfg.model;
+        await generate(session, finalPrompt, undefined, model);
+    }
+    // ==========================================================
     // ==================== 命令注册 ====================
+    // 主文生图指令
     const cmd = ctx.command(`${cfg.command} <raw:text>`, 'draw');
     cfg.aliases.forEach((alias) => cmd.alias(alias));
     cmd.action(async ({ session }, raw) => {
         try {
-            if (!session)
-                return;
-            if (await isBlacklisted(session.userId))
-                return safeSend(session, cfg.messages.blacklisted);
-            if (!cfg.enableTxt2Img)
-                return safeSend(session, cfg.messages.txt2imgDisabled);
-            const prompt = cleanHtmlTags(raw || '');
-            if (!prompt)
-                return safeSend(session, cfg.messages.empty);
-            await safeSend(session, cfg.messages.generating);
-            const finalPrompt = cfg.txt2imgPrompt.replace('{prompt}', prompt);
-            const model = cfg.txt2imgModel || cfg.model;
-            await generate(session, finalPrompt, undefined, model);
+            await doTxt2Img(session, raw);
         }
         catch (e) {
             logger.error('文生图命令异常', e);
             await safeSend(session, cfg.messages.fail);
         }
     });
+    // ==================== 新增：预置提示词指令注册 ====================
+    if (cfg.enablePresets && cfg.presets) {
+        const presets = getEnabledPresets();
+        for (let i = 0; i < presets.length; i++) {
+            const preset = presets[i];
+            const presetCmd = preset.command?.trim() || `draw${i}`;
+            // 注册指令
+            const pCmd = ctx.command(`${presetCmd} <raw:text>`, `使用预置提示词: ${preset.text.slice(0, 20)}...`);
+            pCmd.action(async ({ session }, raw) => {
+                try {
+                    if (debug)
+                        logger.info(`预置指令触发: ${presetCmd}, 预置文本: ${preset.text.slice(0, 50)}`);
+                    await doTxt2Img(session, raw, [preset]);
+                }
+                catch (e) {
+                    logger.error(`预置指令 ${presetCmd} 异常`, e);
+                    await safeSend(session, cfg.messages.fail);
+                }
+            });
+            if (debug)
+                logger.info(`注册预置提示词指令: ${presetCmd}`);
+        }
+    }
+    // ==========================================================
     // ==================== 修改：图生图命令支持合并消息 ====================
     const imgCmd = ctx.command(`${cfg.img2imgCommand} <raw:text>`, 'imgdraw');
     cfg.img2imgAliases.forEach((alias) => imgCmd.alias(alias));

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-imgdraw-selfuse",
-  "description": "修改自ai-image的画图插件,支持openai兼容api,增加base64转换以适配GPTimg2，支持发送合并消息图生图",
-  "version": "0.0.7",
+  "description": "修改自ai-image的画图插件,支持openai兼容api,增加base64转换自由开启，增加图文合并消息图生图，增加预置提示词",
+  "version": "0.1.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [