koishi-plugin-ai-video 0.0.2 → 0.0.3

package/lib/index.js

@@ -39,12 +39,12 @@ exports.Config = koishi_1.Schema.intersect([
     }).description('基本设置'),
     koishi_1.Schema.object({
         proxyEnabled: koishi_1.Schema.boolean().default(false).description('启用代理'),
-        proxyProtocol: koishi_1.Schema.union([koishi_1.Schema.const('http'), koishi_1.Schema.const('https')]).default('http'),
-        proxyHost: koishi_1.Schema.string().default(''),
-        proxyPort: koishi_1.Schema.number().default(8080),
-        proxyAuth: koishi_1.Schema.boolean().default(false),
-        proxyUsername: koishi_1.Schema.string().default(''),
-        proxyPassword: koishi_1.Schema.string().role('secret').default(''),
+        proxyProtocol: koishi_1.Schema.union([koishi_1.Schema.const('http'), koishi_1.Schema.const('https')]).default('http').description('代理协议'),
+        proxyHost: koishi_1.Schema.string().default('').description('代理主机地址'),
+        proxyPort: koishi_1.Schema.number().default(8080).description('代理端口'),
+        proxyAuth: koishi_1.Schema.boolean().default(false).description('启用代理认证'),
+        proxyUsername: koishi_1.Schema.string().default('').description('代理用户名'),
+        proxyPassword: koishi_1.Schema.string().role('secret').default('').description('代理密码'),
     }).description('代理设置'),
     koishi_1.Schema.object({
         useCustomApi: koishi_1.Schema.boolean().default(false).description('使用自定义API'),
@@ -52,26 +52,30 @@ exports.Config = koishi_1.Schema.intersect([
         apiKey: koishi_1.Schema.string().role('secret').default('').description('API密钥'),
         model: koishi_1.Schema.string().default('agnes-video-v2.0').description('模型'),
         img2videoModel: koishi_1.Schema.string().default('').description('图生视频模型'),
-        videoDuration: koishi_1.Schema.number().default(0).description('时长'),
-        videoResolution: koishi_1.Schema.string().default('').description('分辨率'),
+        videoDuration: koishi_1.Schema.number().default(0).description('默认视频时长（秒）'),
+        videoResolution: koishi_1.Schema.string().default('').description('默认视频分辨率（宽x高）'),
         txt2videoPrompt: koishi_1.Schema.string().default('').description('文生视频提示模板'),
         img2videoPrompt: koishi_1.Schema.string().default('').description('图生视频提示模板'),
-        customHeaders: koishi_1.Schema.string().role('textarea').default('{}').description('自定义请求头'),
+        customHeaders: koishi_1.Schema.string().role('textarea').default('{}').description('自定义请求头(JSON)'),
     }).description('内置API'),
     koishi_1.Schema.object({
-        apiStrategy: koishi_1.Schema.union([koishi_1.Schema.const('sequence'), koishi_1.Schema.const('roundrobin')]).default('roundrobin'),
+        apiStrategy: koishi_1.Schema.union([
+            koishi_1.Schema.const('sequence').description('顺序使用'),
+            koishi_1.Schema.const('roundrobin').description('轮询使用')
+        ]).default('roundrobin').description('API调用策略'),
         customApiList: koishi_1.Schema.array(koishi_1.Schema.object({
-            enable: koishi_1.Schema.boolean().default(true),
-            endpoint: koishi_1.Schema.string().default('https://apihub.agnes-ai.com/v1/videos'),
-            apiKey: koishi_1.Schema.string().role('secret').default(''),
-            model: koishi_1.Schema.string().default('agnes-video-v2.0'),
-            img2videoModel: koishi_1.Schema.string().default(''),
-            videoDuration: koishi_1.Schema.number().default(0),
-            videoResolution: koishi_1.Schema.string().default(''),
-            txt2videoPrompt: koishi_1.Schema.string().default(''),
-            img2videoPrompt: koishi_1.Schema.string().default(''),
+            enable: koishi_1.Schema.boolean().default(true).description('启用'),
+            endpoint: koishi_1.Schema.string().default('https://apihub.agnes-ai.com/v1/videos').description('API端点'),
+            apiKey: koishi_1.Schema.string().role('secret').default('').description('API密钥'),
+            model: koishi_1.Schema.string().default('agnes-video-v2.0').description('模型'),
+            img2videoModel: koishi_1.Schema.string().default('').description('图生视频模型'),
+            videoDuration: koishi_1.Schema.number().default(0).description('默认视频时长（秒）'),
+            videoResolution: koishi_1.Schema.string().default('').description('默认视频分辨率（宽x高）'),
+            txt2videoPrompt: koishi_1.Schema.string().default('').description('文生视频提示模板'),
+            img2videoPrompt: koishi_1.Schema.string().default('').description('图生视频提示模板'),
             customHeaders: koishi_1.Schema.string().role('textarea')
-                .default('{"Authorization":"Bearer {apiKey}","Content-Type":"application/json"}'),
+                .default('{"Authorization":"Bearer {apiKey}","Content-Type":"application/json"}')
+                .description('自定义请求头(JSON)'),
             bodyTemplate: koishi_1.Schema.string().role('textarea')
                 .default(JSON.stringify({
                 txt2videoBody: { model: '{model}', prompt: '{prompt}', height: 768, width: 1152, num_frames: 121, frame_rate: 24 },
@@ -80,11 +84,11 @@ exports.Config = koishi_1.Schema.intersect([
                 pollUrlTemplate: 'https://apihub.agnes-ai.com/agnesapi?video_id={task_id}',
                 taskIdPath: 'video_id',
             }, null, 2))
-                .description('自定义请求体'),
+                .description('自定义请求体(JSON模板)'),
         })).default([]).description('自定义API列表'),
     }).description('自定义API配置'),
     koishi_1.Schema.object({
-        blacklistAdmins: koishi_1.Schema.array(String).default([]).description('管理员QQ'),
+        blacklistAdmins: koishi_1.Schema.array(String).default([]).description('管理员QQ号'),
     }).description('权限管理'),
     koishi_1.Schema.object({
         messages: koishi_1.Schema.object({
@@ -147,8 +151,6 @@ async function apply(ctx, cfg) {
         return apiCallTimestamps.length + 1 <= cfg.rateLimit;
     }
     function recordApiCall() { apiCallTimestamps.push(Date.now()); }
-    const BUILTIN_TXT2VIDEO = { model: '{model}', prompt: '{prompt}', duration: '{duration}', size: '{size}' };
-    const BUILTIN_IMG2VIDEO = { model: '{model}', prompt: '{prompt}', duration: '{duration}', size: '{size}', image_url: '{url}' };
     function parseApiEntry(entry) {
         if (!entry.endpoint)
             return null;
@@ -167,8 +169,8 @@ async function apply(ctx, cfg) {
         if (entry.bodyTemplate) {
             try {
                 const tmpl = JSON.parse(entry.bodyTemplate);
-                txt2videoBody = tmpl.txt2videoBody || BUILTIN_TXT2VIDEO;
-                img2videoBody = tmpl.img2videoBody || BUILTIN_IMG2VIDEO;
+                txt2videoBody = tmpl.txt2videoBody || { model: '{model}', prompt: '{prompt}', duration: '{duration}', size: '{size}' };
+                img2videoBody = tmpl.img2videoBody || { model: '{model}', prompt: '{prompt}', duration: '{duration}', size: '{size}', image_url: '{url}' };
                 responseVideoPath = tmpl.responseVideoPath || 'video_url';
                 pollUrlTemplate = tmpl.pollUrlTemplate || '';
                 taskIdPath = tmpl.taskIdPath || '';
@@ -178,8 +180,8 @@ async function apply(ctx, cfg) {
             }
         }
         else {
-            txt2videoBody = BUILTIN_TXT2VIDEO;
-            img2videoBody = BUILTIN_IMG2VIDEO;
+            txt2videoBody = { model: '{model}', prompt: '{prompt}', duration: '{duration}', size: '{size}' };
+            img2videoBody = { model: '{model}', prompt: '{prompt}', duration: '{duration}', size: '{size}', image_url: '{url}' };
             responseVideoPath = 'video_url';
             pollUrlTemplate = '';
             taskIdPath = '';
@@ -215,8 +217,8 @@ async function apply(ctx, cfg) {
         return {
             endpoint: cfg.apiEndpoint || 'https://apihub.agnes-ai.com/v1/videos',
             headers,
-            txt2videoBody: BUILTIN_TXT2VIDEO,
-            img2videoBody: BUILTIN_IMG2VIDEO,
+            txt2videoBody: { model: '{model}', prompt: '{prompt}', duration: '{duration}', size: '{size}' },
+            img2videoBody: { model: '{model}', prompt: '{prompt}', duration: '{duration}', size: '{size}', image_url: '{url}' },
             responseVideoPath: 'video_url',
             pollUrlTemplate: '',
             taskIdPath: '',
@@ -249,15 +251,27 @@ async function apply(ctx, cfg) {
             return buildBuiltinApi();
         }
     }
-    function resolveTemplate(template, vars) {
-        const jsonStr = JSON.stringify(template);
-        let processed = jsonStr;
-        for (const [key, value] of Object.entries(vars)) {
-            if (value === undefined || value === null)
-                continue;
-            processed = processed.replace(new RegExp(`\\{${key}\\}`, 'g'), JSON.stringify(String(value)).slice(1, -1));
-        }
-        return JSON.parse(processed);
+    function resolveTemplate(templateObj, vars) {
+        if (templateObj === null || templateObj === undefined)
+            return templateObj;
+        if (typeof templateObj === 'string') {
+            const match = templateObj.match(/^\{(\w+)\}$/);
+            if (match && vars.hasOwnProperty(match[1])) {
+                return vars[match[1]];
+            }
+            return templateObj;
+        }
+        if (Array.isArray(templateObj)) {
+            return templateObj.map(item => resolveTemplate(item, vars));
+        }
+        if (typeof templateObj === 'object') {
+            const result = {};
+            for (const key of Object.keys(templateObj)) {
+                result[key] = resolveTemplate(templateObj[key], vars);
+            }
+            return result;
+        }
+        return templateObj;
     }
     function getValueByPath(obj, path) {
         if (!obj || !path)
@@ -333,7 +347,7 @@ async function apply(ctx, cfg) {
             logger.error('发送失败', e);
         }
     }
-    async function sendVideo(session, url) {
+    async function sendSingleVideo(session, url) {
         const mode = cfg.videoSendMode;
         if (mode === 'url')
             await safeSend(session, url);
@@ -366,7 +380,12 @@ async function apply(ctx, cfg) {
         if (promptTemplate)
             finalPrompt = promptTemplate.replace('{prompt}', prompt).replace('{url}', imageUrl);
         const bodyTemplate = isImg2Video ? api.img2videoBody : api.txt2videoBody;
-        const vars = { model, prompt: finalPrompt, duration: String(duration), size: resolution };
+        const vars = {
+            model,
+            prompt: finalPrompt,
+            duration,
+            size: resolution,
+        };
         if (isImg2Video)
             vars.url = imageUrl;
         let body;
@@ -374,10 +393,13 @@ async function apply(ctx, cfg) {
             body = resolveTemplate(bodyTemplate, vars);
         }
         catch {
-            return safeSend(session, cfg.messages.fail + cfg.messages.templateError);
+            await safeSend(session, cfg.messages.fail + cfg.messages.templateError);
+            return null;
+        }
+        if (!validateEndpoint(api.endpoint)) {
+            await safeSend(session, cfg.messages.fail + '（端点无效）');
+            return null;
         }
-        if (!validateEndpoint(api.endpoint))
-            return safeSend(session, cfg.messages.fail + '（端点无效）');
         const sensitive = api.headers?.Authorization?.split(' ')[1] || '';
         if (debug)
             logger.info('API请求', JSON.stringify(sanitizeForLog(body, sensitive)));
@@ -393,32 +415,62 @@ async function apply(ctx, cfg) {
             if (cfg.pollEnabled && api.taskIdPath) {
                 const taskId = getValueByPath(res, api.taskIdPath);
                 if (taskId) {
-                    const pollUrl = api.pollUrlTemplate.replace('{endpoint}', api.endpoint).replace('{task_id}', taskId);
-                    safeSend(session, cfg.messages.pollWaiting);
+                    const pollUrl = api.pollUrlTemplate.replace('{task_id}', taskId);
+                    await safeSend(session, cfg.messages.pollWaiting);
                     res = await pollForResult(pollUrl, api.headers, cfg.pollInterval, cfg.pollTimeout);
                 }
             }
             const videoUrl = getValueByPath(res, api.responseVideoPath) || findVideoUrl(res);
-            if (videoUrl)
-                await sendVideo(session, videoUrl);
-            else
-                await safeSend(session, cfg.messages.fail + cfg.messages.noContent);
-            const userId = `${session.guildId || 'private'}-${session.userId}`;
-            lastTaskMap.set(userId, { prompt, imageUrl, isImg2Video, model });
+            return videoUrl || null;
         }
         catch (err) {
             logger.error('视频生成失败', err);
-            safeSend(session, cfg.messages.fail + ' [' + (err.message?.slice(0, 100) || '未知错误') + ']');
+            await safeSend(session, cfg.messages.fail + ' [' + (err.message?.slice(0, 100) || '未知错误') + ']');
+            return null;
         }
     }
-    async function generateVideo(session, prompt, imageUrl = '') {
-        if (!checkRateLimit())
-            return safeSend(session, cfg.messages.rateLimit);
-        const api = getApi();
-        if (!api)
-            return safeSend(session, cfg.messages.noApi);
-        recordApiCall();
-        return customGenerateVideo(session, api, prompt, imageUrl);
+    async function generateVideos(session, prompt, imageUrl, count) {
+        const videoUrls = [];
+        for (let i = 0; i < count; i++) {
+            if (!checkRateLimit()) {
+                await safeSend(session, cfg.messages.rateLimit);
+                break;
+            }
+            const api = getApi();
+            if (!api) {
+                await safeSend(session, cfg.messages.noApi);
+                break;
+            }
+            recordApiCall();
+            const url = await customGenerateVideo(session, api, prompt, imageUrl);
+            if (url)
+                videoUrls.push(url);
+            if (i < count - 1)
+                await new Promise(r => setTimeout(r, 1000));
+        }
+        if (videoUrls.length === 0) {
+            await safeSend(session, cfg.messages.fail + cfg.messages.noContent);
+            return;
+        }
+        if (videoUrls.length === 1 || !cfg.enableForward) {
+            for (const url of videoUrls) {
+                await sendSingleVideo(session, url);
+            }
+        }
+        else {
+            const children = videoUrls.map(url => (0, koishi_1.h)('message', koishi_1.h.video(url)));
+            try {
+                await safeSend(session, (0, koishi_1.h)('message', { forward: true }, ...children));
+            }
+            catch {
+                for (const url of videoUrls)
+                    await sendSingleVideo(session, url);
+            }
+        }
+        const userId = `${session.guildId || 'private'}-${session.userId}`;
+        if (videoUrls.length > 0) {
+            lastTaskMap.set(userId, { prompt, imageUrl, isImg2Video: !!imageUrl, model: 'multiple' });
+        }
     }
     function startTimer(session, key, collect) {
         return setTimeout(() => {
@@ -459,6 +511,8 @@ async function apply(ctx, cfg) {
         await safeSend(session, cfg.messages.enterCollect);
     });
     ctx.on('message', async (session) => {
+        if (session.command)
+            return;
         if (!session || !session.elements)
             return;
         const key = `${session.guildId || 'private'}-${session.userId}`;
@@ -478,7 +532,7 @@ async function apply(ctx, cfg) {
             if (!collect.prompt && !collect.imageUrl)
                 return safeSend(session, cfg.messages.empty);
             await safeSend(session, cfg.messages.generating);
-            return generateVideo(session, collect.prompt || '默认', collect.imageUrl);
+            return generateVideos(session, collect.prompt || '默认', collect.imageUrl, cfg.maxVideos);
         }
         if (imgs.length > 0 && !collect.imageUrl) {
             const assets = ctx.assets;
@@ -515,7 +569,7 @@ async function apply(ctx, cfg) {
         if (last.isImg2Video)
             return safeSend(session, cfg.messages.redrawImg2Video);
         await safeSend(session, cfg.messages.redrawing);
-        return generateVideo(session, last.prompt);
+        return generateVideos(session, last.prompt, '', cfg.maxVideos);
     });
     function isValidQQ(id) { return /^\d{5,11}$/.test(id); }
     async function isBlacklisted(userId) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-ai-video",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "description": "Koishi AI 视频生成插件，支持文生视频、图生视频",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",