koishi-plugin-ai-video 0.0.2 → 0.0.4

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,41 +84,41 @@ 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({
-            generating: koishi_1.Schema.string().default('视频生成中，请耐心等待...'),
-            enterCollect: koishi_1.Schema.string().default('已进入收集模式，请继续发送图片/文字。发送「开始」触发生成，发送「取消」退出。当前已收集: 0 张图片, 0 段文字'),
-            collectUpdate: koishi_1.Schema.string().default('当前已收集: {images} 张图片, 文字已更新'),
-            collectTimeout: koishi_1.Schema.string().default('收集超时，已自动退出'),
-            empty: koishi_1.Schema.string().default('[提示] 请输入提示词或上传图片'),
-            noApi: koishi_1.Schema.string().default('[提示] 未配置可用API'),
-            fail: koishi_1.Schema.string().default('[提示] 视频生成失败'),
-            noContent: koishi_1.Schema.string().default('（未返回任何视频内容）'),
-            templateError: koishi_1.Schema.string().default('（模板配置错误）'),
-            txt2videoDisabled: koishi_1.Schema.string().default('[提示] 文生视频功能未启用'),
-            img2videoDisabled: koishi_1.Schema.string().default('[提示] 图生视频功能未启用'),
-            rateLimit: koishi_1.Schema.string().default('[提示] 调用次数已达上限'),
-            needAssets: koishi_1.Schema.string().default('[提示] 图生视频需要正确配置 assets 服务'),
-            blacklisted: koishi_1.Schema.string().default('[提示] 你已被加入黑名单'),
-            noPermission: koishi_1.Schema.string().default('[提示] 无权限管理黑名单'),
-            blacklistAddSuccess: koishi_1.Schema.string().default('已将 {targets} 加入黑名单'),
-            blacklistRemoveSuccess: koishi_1.Schema.string().default('已将 {targets} 移出黑名单'),
-            blacklistAddFail: koishi_1.Schema.string().default('{targets} 已在黑名单或无效'),
-            blacklistRemoveFail: koishi_1.Schema.string().default('{targets} 不在黑名单'),
-            invalidUserId: koishi_1.Schema.string().default('无效QQ号：{targets}'),
-            blacklistListEmpty: koishi_1.Schema.string().default('黑名单为空'),
-            blacklistListTitle: koishi_1.Schema.string().default('当前黑名单：'),
-            noLastTask: koishi_1.Schema.string().default('没有上一次记录，无法重绘'),
-            redrawing: koishi_1.Schema.string().default('正在重绘上一次文生视频...'),
-            redrawImg2Video: koishi_1.Schema.string().default('[提示] 重绘仅支持文生视频'),
-            cancelCollect: koishi_1.Schema.string().default('已取消收集模式'),
-            pollWaiting: koishi_1.Schema.string().default('视频生成中（异步轮询），请稍后...'),
+            generating: koishi_1.Schema.string().default('视频生成中，请耐心等待...').description('生成中提示'),
+            enterCollect: koishi_1.Schema.string().default('已进入收集模式，请继续发送图片/文字。发送「开始」触发生成，发送「取消」退出。当前已收集: 0 张图片, 0 段文字').description('进入收集模式提示'),
+            collectUpdate: koishi_1.Schema.string().default('当前已收集: {images} 张图片, 文字已更新').description('收集更新提示'),
+            collectTimeout: koishi_1.Schema.string().default('收集超时，已自动退出').description('收集超时提示'),
+            empty: koishi_1.Schema.string().default('[提示] 请输入提示词或上传图片').description('空输入提示'),
+            noApi: koishi_1.Schema.string().default('[提示] 未配置可用API').description('无API提示'),
+            fail: koishi_1.Schema.string().default('[提示] 视频生成失败').description('生成失败提示'),
+            noContent: koishi_1.Schema.string().default('（未返回任何视频内容）').description('无内容提示'),
+            templateError: koishi_1.Schema.string().default('（模板配置错误）').description('模板错误提示'),
+            txt2videoDisabled: koishi_1.Schema.string().default('[提示] 文生视频功能未启用').description('文生视频禁用提示'),
+            img2videoDisabled: koishi_1.Schema.string().default('[提示] 图生视频功能未启用').description('图生视频禁用提示'),
+            rateLimit: koishi_1.Schema.string().default('[提示] 调用次数已达上限').description('频率限制提示'),
+            needAssets: koishi_1.Schema.string().default('[提示] 图生视频需要正确配置 assets 服务').description('需要Assets提示'),
+            blacklisted: koishi_1.Schema.string().default('[提示] 你已被加入黑名单').description('黑名单提示'),
+            noPermission: koishi_1.Schema.string().default('[提示] 无权限管理黑名单').description('无权限提示'),
+            blacklistAddSuccess: koishi_1.Schema.string().default('已将 {targets} 加入黑名单').description('添加黑名单成功提示'),
+            blacklistRemoveSuccess: koishi_1.Schema.string().default('已将 {targets} 移出黑名单').description('移除黑名单成功提示'),
+            blacklistAddFail: koishi_1.Schema.string().default('{targets} 已在黑名单或无效').description('添加黑名单失败提示'),
+            blacklistRemoveFail: koishi_1.Schema.string().default('{targets} 不在黑名单').description('移除黑名单失败提示'),
+            invalidUserId: koishi_1.Schema.string().default('无效QQ号：{targets}').description('无效用户ID提示'),
+            blacklistListEmpty: koishi_1.Schema.string().default('黑名单为空').description('黑名单为空提示'),
+            blacklistListTitle: koishi_1.Schema.string().default('当前黑名单：').description('黑名单标题提示'),
+            noLastTask: koishi_1.Schema.string().default('没有上一次记录，无法重绘').description('无上一次任务提示'),
+            redrawing: koishi_1.Schema.string().default('正在重绘上一次文生视频...').description('重绘中文生视频提示'),
+            redrawImg2Video: koishi_1.Schema.string().default('[提示] 重绘仅支持文生视频').description('重绘图生视频提示'),
+            cancelCollect: koishi_1.Schema.string().default('已取消收集模式').description('取消收集提示'),
+            pollWaiting: koishi_1.Schema.string().default('视频生成中（异步轮询），请稍后...').description('轮询等待提示'),
         }).description('消息文本'),
     }).description('消息文本'),
 ]);
@@ -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.4",
   "description": "Koishi AI 视频生成插件，支持文生视频、图生视频",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",