npm - koishi-plugin-ai-video - Versions diffs - 0.0.1 → 0.0.3 - Mend

koishi-plugin-ai-video 0.0.1 → 0.0.3

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (3) hide show

package/lib/index.d.ts CHANGED Viewed

@@ -19,6 +19,7 @@ export declare const Config: Schema<Schemastery.ObjectS<{
     pollEnabled: Schema<boolean, boolean>;
     pollInterval: Schema<number, number>;
     pollTimeout: Schema<number, number>;
+    collectTimeout: Schema<number, number>;
 }> | Schemastery.ObjectS<{
     proxyEnabled: Schema<boolean, boolean>;
     proxyProtocol: Schema<"http" | "https", "http" | "https">;
@@ -70,6 +71,9 @@ export declare const Config: Schema<Schemastery.ObjectS<{
 }> | Schemastery.ObjectS<{
     messages: Schema<Schemastery.ObjectS<{
         generating: Schema<string, string>;
+        enterCollect: Schema<string, string>;
+        collectUpdate: Schema<string, string>;
+        collectTimeout: Schema<string, string>;
         empty: Schema<string, string>;
         noApi: Schema<string, string>;
         fail: Schema<string, string>;
@@ -91,8 +95,13 @@ export declare const Config: Schema<Schemastery.ObjectS<{
         noLastTask: Schema<string, string>;
         redrawing: Schema<string, string>;
         redrawImg2Video: Schema<string, string>;
+        cancelCollect: Schema<string, string>;
+        pollWaiting: Schema<string, string>;
     }>, Schemastery.ObjectT<{
         generating: Schema<string, string>;
+        enterCollect: Schema<string, string>;
+        collectUpdate: Schema<string, string>;
+        collectTimeout: Schema<string, string>;
         empty: Schema<string, string>;
         noApi: Schema<string, string>;
         fail: Schema<string, string>;
@@ -114,6 +123,8 @@ export declare const Config: Schema<Schemastery.ObjectS<{
         noLastTask: Schema<string, string>;
         redrawing: Schema<string, string>;
         redrawImg2Video: Schema<string, string>;
+        cancelCollect: Schema<string, string>;
+        pollWaiting: Schema<string, string>;
     }>>;
 }>, {
     debug: boolean;
@@ -129,6 +140,7 @@ export declare const Config: Schema<Schemastery.ObjectS<{
     pollEnabled: boolean;
     pollInterval: number;
     pollTimeout: number;
+    collectTimeout: number;
 } & import("cosmokit").Dict & {
     proxyEnabled: boolean;
     proxyProtocol: "http" | "https";
@@ -168,6 +180,9 @@ export declare const Config: Schema<Schemastery.ObjectS<{
 } & {
     messages: Schemastery.ObjectT<{
         generating: Schema<string, string>;
+        enterCollect: Schema<string, string>;
+        collectUpdate: Schema<string, string>;
+        collectTimeout: Schema<string, string>;
         empty: Schema<string, string>;
         noApi: Schema<string, string>;
         fail: Schema<string, string>;
@@ -189,6 +204,8 @@ export declare const Config: Schema<Schemastery.ObjectS<{
         noLastTask: Schema<string, string>;
         redrawing: Schema<string, string>;
         redrawImg2Video: Schema<string, string>;
+        cancelCollect: Schema<string, string>;
+        pollWaiting: Schema<string, string>;
     }>;
 }>;
 declare module 'koishi' {

package/lib/index.js CHANGED Viewed

@@ -18,181 +18,159 @@ exports.inject = {
 const logger = new koishi_1.Logger('ai-video');
 exports.Config = koishi_1.Schema.intersect([
     koishi_1.Schema.object({
-        debug: koishi_1.Schema.boolean().default(false).description('开启调试模式，输出完整请求日志'),
-        timeout: koishi_1.Schema.number().default(600000).description('接口请求超时时间（毫秒），视频生成较慢，建议加大'),
-        rateLimit: koishi_1.Schema.number().default(50).description('每小时调用次数限制'),
+        debug: koishi_1.Schema.boolean().default(false).description('开启调试模式'),
+        timeout: koishi_1.Schema.number().default(600000).description('超时时间（毫秒）'),
+        rateLimit: koishi_1.Schema.number().default(50).description('每小时调用次数上限'),
         maxVideos: koishi_1.Schema.number().default(1).description('单次生成最多视频数量'),
         videoDuration: koishi_1.Schema.number().default(5).description('默认视频时长（秒）'),
         videoResolution: koishi_1.Schema.string().default('1024x576').description('默认视频分辨率（宽x高）'),
-        enableForward: koishi_1.Schema.boolean().default(true).description('多视频结果是否使用合并转发'),
-        enableTxt2Video: koishi_1.Schema.boolean().default(true).description('启用文生视频功能'),
-        enableImg2Video: koishi_1.Schema.boolean().default(true).description('启用图生视频功能'),
+        enableForward: koishi_1.Schema.boolean().default(true).description('多视频合并转发'),
+        enableTxt2Video: koishi_1.Schema.boolean().default(true).description('启用文生视频'),
+        enableImg2Video: koishi_1.Schema.boolean().default(true).description('启用图生视频'),
         videoSendMode: koishi_1.Schema.union([
-            koishi_1.Schema.const('video').description('仅发送视频文件'),
-            koishi_1.Schema.const('url').description('仅发送视频链接'),
-            koishi_1.Schema.const('both').description('发送视频文件和链接'),
-        ]).default('video').description('生成结果发送方式'),
-        pollEnabled: koishi_1.Schema.boolean().default(false).description('是否启用异步轮询（若 API 返回任务 ID 则自动轮询至完成）'),
+            koishi_1.Schema.const('video').description('仅视频'),
+            koishi_1.Schema.const('url').description('仅链接'),
+            koishi_1.Schema.const('both').description('视频+链接'),
+        ]).default('video').description('发送方式'),
+        pollEnabled: koishi_1.Schema.boolean().default(false).description('启用异步轮询'),
         pollInterval: koishi_1.Schema.number().default(3000).description('轮询间隔（毫秒）'),
-        pollTimeout: koishi_1.Schema.number().default(600000).description('轮询总超时时间（毫秒）'),
+        pollTimeout: koishi_1.Schema.number().default(600000).description('轮询超时（毫秒）'),
+        collectTimeout: koishi_1.Schema.number().default(120).description('收集模式超时（秒）'),
     }).description('基本设置'),
     koishi_1.Schema.object({
-        proxyEnabled: koishi_1.Schema.boolean().default(false).description('是否启用 HTTP/HTTPS 代理'),
-        proxyProtocol: koishi_1.Schema.union([
-            koishi_1.Schema.const('http').description('HTTP'),
-            koishi_1.Schema.const('https').description('HTTPS'),
-        ]).default('http').description('代理协议'),
-        proxyHost: koishi_1.Schema.string().default('').description('代理地址'),
+        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').description('代理协议'),
+        proxyHost: koishi_1.Schema.string().default('').description('代理主机地址'),
         proxyPort: koishi_1.Schema.number().default(8080).description('代理端口'),
-        proxyAuth: koishi_1.Schema.boolean().default(false).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 配置（开启后下方自定义列表生效）'),
-        apiEndpoint: koishi_1.Schema.string().default('https://api.openai.com/v1/video/generations').description('API 端点地址'),
-        apiKey: koishi_1.Schema.string().role('secret').default('').description('API 密钥'),
-        model: koishi_1.Schema.string().default('video-generation-model').description('模型名称'),
-        img2videoModel: koishi_1.Schema.string().default('').description('图生视频专用模型名称（留空则使用上方模型）'),
-        videoDuration: koishi_1.Schema.number().default(0).description('视频时长（秒，留空则使用全局默认）'),
-        videoResolution: koishi_1.Schema.string().default('').description('视频分辨率（留空则使用全局默认）'),
-        txt2videoPrompt: koishi_1.Schema.string().default('').description('文生视频提示词模板。变量：{prompt}（留空则直接使用用户输入）'),
-        img2videoPrompt: koishi_1.Schema.string().default('').description('图生视频提示词模板。变量：{url} {prompt}（留空则直接使用用户输入）'),
-        customHeaders: koishi_1.Schema.string().role('textarea').default('{}').description('自定义请求头 JSON 对象（合并到默认请求头）'),
-    }).description('内置 API 设置'),
+        useCustomApi: koishi_1.Schema.boolean().default(false).description('使用自定义API'),
+        apiEndpoint: 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('{}').description('自定义请求头(JSON)'),
+    }).description('内置API'),
     koishi_1.Schema.object({
         apiStrategy: koishi_1.Schema.union([
-            koishi_1.Schema.const('sequence').description('顺序模式'),
-            koishi_1.Schema.const('roundrobin').description('负载均衡模式'),
-        ]).default('roundrobin').description('API 调度策略'),
+            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).description('是否启用此 API'),
-            endpoint: koishi_1.Schema.string().default('https://api.openai.com/v1/video/generations').description('API 端点地址'),
-            apiKey: koishi_1.Schema.string().role('secret').default('').description('API 密钥'),
-            model: koishi_1.Schema.string().default('video-generation-model').description('模型名称'),
-            img2videoModel: koishi_1.Schema.string().default('').description('图生视频专用模型名称（留空则使用上方模型）'),
-            videoDuration: koishi_1.Schema.number().default(0).description('视频时长（秒，留空使用全局默认）'),
-            videoResolution: koishi_1.Schema.string().default('').description('视频分辨率（留空使用全局默认）'),
-            txt2videoPrompt: koishi_1.Schema.string().default('').description('文生视频提示词模板。变量：{prompt}（留空则直接使用用户输入）'),
-            img2videoPrompt: koishi_1.Schema.string().default('').description('图生视频提示词模板。变量：{url} {prompt}（留空则直接使用用户输入）'),
+            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"}')
-                .description('自定义请求头 JSON，支持 {apiKey} 变量'),
+                .description('自定义请求头(JSON)'),
             bodyTemplate: koishi_1.Schema.string().role('textarea')
                 .default(JSON.stringify({
-                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: '{endpoint}/{task_id}',
-                taskIdPath: 'task_id',
+                txt2videoBody: { model: '{model}', prompt: '{prompt}', height: 768, width: 1152, num_frames: 121, frame_rate: 24 },
+                img2videoBody: { model: '{model}', prompt: '{prompt}', image: '{url}', height: 768, width: 1152, num_frames: 121, frame_rate: 24 },
+                responseVideoPath: 'remixed_from_video_id',
+                pollUrlTemplate: 'https://apihub.agnes-ai.com/agnesapi?video_id={task_id}',
+                taskIdPath: 'video_id',
             }, null, 2))
-                .description('自定义请求体 JSON 模板（高级，留空使用内置格式）。\n支持变量：{model}、{prompt}、{duration}、{size}、{url}。\n轮询配置可选：pollUrlTemplate、taskIdPath'),
-        })).default([]).description('自定义 API 配置列表（仅当"使用自定义 API 配置"开启时生效）'),
-    }).description('自定义 API 配置'),
+                .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('视频生成中，请耐心等待...').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('API 返回空结果时的追加提示'),
-            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 服务（selfUrl 未正确设置或服务未启动）').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('无效 QQ 号的提示'),
-            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('图生视频任务无法重绘时的提示'),
-        }).description('所有提示文案的自定义配置，支持模板变量'),
+            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('视频生成中（异步轮询），请稍后...'),
+        }).description('消息文本'),
     }).description('消息文本'),
 ]);
 async function apply(ctx, cfg) {
     const debug = cfg.debug;
     try {
         const loc = path_1.default.join(__dirname, 'locales', 'zh-CN.yml');
-        if (fs_1.default.existsSync(loc)) {
+        if (fs_1.default.existsSync(loc))
             ctx.i18n.define('zh-CN', yaml_1.default.parse(fs_1.default.readFileSync(loc, 'utf8')));
-        }
     }
     catch { }
+    const collectSessions = new Map();
     const lastTaskMap = new Map();
     let apiRoundRobinIdx = 0;
     const apiCallTimestamps = [];
-    ctx.model.extend('ai_video_blacklist', {
-        id: 'string',
-        createdAt: 'date',
-    }, {
-        primary: 'id',
+    ctx.model.extend('ai_video_blacklist', { id: 'string', createdAt: 'date' }, { primary: 'id' });
+    ctx.on('dispose', () => {
+        for (const [, s] of collectSessions)
+            clearTimeout(s.timer);
+        collectSessions.clear();
     });
     function checkRateLimit() {
         const now = Date.now();
         const oneHourAgo = now - 3600000;
-        let trimIdx = 0;
-        while (trimIdx < apiCallTimestamps.length && apiCallTimestamps[trimIdx] < oneHourAgo) {
-            trimIdx++;
-        }
-        if (trimIdx > 0) {
-            apiCallTimestamps.splice(0, trimIdx);
-        }
+        let i = 0;
+        while (i < apiCallTimestamps.length && apiCallTimestamps[i] < oneHourAgo)
+            i++;
+        if (i > 0)
+            apiCallTimestamps.splice(0, i);
         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 recordApiCall() { apiCallTimestamps.push(Date.now()); }
     function parseApiEntry(entry) {
         if (!entry.endpoint)
             return null;
-        const headers = {
-            'Content-Type': 'application/json'
-        };
-        if (entry.apiKey) {
+        const headers = { 'Content-Type': 'application/json' };
+        if (entry.apiKey)
             headers['Authorization'] = `Bearer ${entry.apiKey}`;
-        }
         if (entry.customHeaders) {
             try {
                 const custom = JSON.parse(entry.customHeaders);
-                for (const [k, v] of Object.entries(custom)) {
+                for (const [k, v] of Object.entries(custom))
                     headers[k] = typeof v === 'string' ? v.replace(/\{apiKey\}/g, entry.apiKey || '') : String(v);
-                }
             }
             catch { }
         }
-        let txt2videoBody;
-        let img2videoBody;
-        let responseVideoPath;
-        let pollUrlTemplate;
-        let taskIdPath;
+        let txt2videoBody, img2videoBody, responseVideoPath, pollUrlTemplate, taskIdPath;
         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 || '';
@@ -202,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 = '';
@@ -211,8 +189,7 @@ async function apply(ctx, cfg) {
         return {
             endpoint: entry.endpoint,
             headers,
-            txt2videoBody,
-            img2videoBody,
+            txt2videoBody, img2videoBody,
             responseVideoPath,
             pollUrlTemplate,
             taskIdPath,
@@ -226,26 +203,22 @@ async function apply(ctx, cfg) {
         };
     }
     function buildBuiltinApi() {
-        const headers = {
-            'Content-Type': 'application/json'
-        };
-        if (cfg.apiKey) {
+        const headers = { 'Content-Type': 'application/json' };
+        if (cfg.apiKey)
             headers['Authorization'] = `Bearer ${cfg.apiKey}`;
-        }
         if (cfg.customHeaders) {
             try {
                 const custom = JSON.parse(cfg.customHeaders);
-                for (const [k, v] of Object.entries(custom)) {
+                for (const [k, v] of Object.entries(custom))
                     headers[k] = typeof v === 'string' ? v.replace(/\{apiKey\}/g, cfg.apiKey || '') : String(v);
-                }
             }
             catch { }
         }
         return {
-            endpoint: cfg.apiEndpoint || 'https://api.openai.com/v1/video/generations',
+            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: '',
@@ -254,19 +227,17 @@ async function apply(ctx, cfg) {
             videoResolution: cfg.videoResolution || '',
             txt2videoPrompt: cfg.txt2videoPrompt || '',
             img2videoPrompt: cfg.img2videoPrompt || '',
-            model: cfg.model || 'video-generation-model',
+            model: cfg.model || 'agnes-video-v2.0',
             img2videoModel: cfg.img2videoModel || '',
         };
     }
     function getApi() {
         if (cfg.useCustomApi) {
-            const entries = cfg.customApiList.filter((e) => e.enable);
-            if (entries.length === 0)
+            const entries = cfg.customApiList.filter(e => e.enable);
+            if (!entries.length)
                 return null;
-            const apis = entries
-                .map((e) => parseApiEntry(e))
-                .filter((a) => a !== null);
-            if (apis.length === 0)
+            const apis = entries.map(parseApiEntry).filter(Boolean);
+            if (!apis.length)
                 return null;
             if (cfg.apiStrategy === 'sequence')
                 return apis[0];
@@ -280,401 +251,393 @@ 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;
-            const regex = new RegExp(`\\{${key}\\}`, 'g');
-            processed = processed.replace(regex, JSON.stringify(String(value)).slice(1, -1));
-        }
-        return JSON.parse(processed);
-    }
-    function getValueByPath(obj, pathStr) {
-        if (!obj || !pathStr)
-            return undefined;
-        const normalized = pathStr.replace(/\[(\d+)\]/g, '.$1');
-        const keys = normalized.split('.').filter(k => k !== '');
-        let current = obj;
-        for (const key of keys) {
-            if (current === undefined || current === null)
-                return undefined;
-            const numKey = /^\d+$/.test(key) ? parseInt(key) : key;
-            current = current[numKey];
-        }
-        return current;
-    }
-    async function pollForResult(pollUrl, headers, path, interval, timeout) {
-        const start = Date.now();
-        while (Date.now() - start < timeout) {
-            try {
-                const res = await axios_1.default.get(pollUrl, { headers, timeout: 10000 });
-                const status = getValueByPath(res.data, 'status');
-                if (status === 'completed' || status === 'succeeded') {
-                    return res.data;
-                }
-                if (status === 'failed' || status === 'error') {
-                    throw new Error('Task failed');
-                }
-                await new Promise(r => setTimeout(r, interval));
-            }
-            catch (e) {
-                if (e.message === 'Task failed')
-                    throw e;
-                await new Promise(r => setTimeout(r, interval));
+    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;
         }
-        throw new Error('Polling timeout');
-    }
-    async function sendVideo(session, url) {
-        const mode = cfg.videoSendMode;
-        if (mode === 'url') {
-            await safeSend(session, url);
+        if (Array.isArray(templateObj)) {
+            return templateObj.map(item => resolveTemplate(item, vars));
         }
-        else {
-            try {
-                await safeSend(session, koishi_1.segment.video(url));
-            }
-            catch {
-                if (mode === 'both')
-                    await safeSend(session, url);
+        if (typeof templateObj === 'object') {
+            const result = {};
+            for (const key of Object.keys(templateObj)) {
+                result[key] = resolveTemplate(templateObj[key], vars);
             }
-            if (mode === 'both')
-                await safeSend(session, url);
+            return result;
         }
+        return templateObj;
     }
-    async function handleVideoResponse(session, responseData, api) {
-        let videoUrl = getValueByPath(responseData, api.responseVideoPath);
-        if (!videoUrl) {
-            const found = findFirstVideoUrl(responseData);
-            if (found)
-                videoUrl = found;
-        }
-        if (videoUrl && typeof videoUrl === 'string') {
-            await sendVideo(session, videoUrl.trim());
-        }
-        else {
-            await safeSend(session, cfg.messages.fail + cfg.messages.noContent);
+    function getValueByPath(obj, path) {
+        if (!obj || !path)
+            return undefined;
+        const keys = path.replace(/\[(\d+)\]/g, '.$1').split('.').filter(k => k);
+        let cur = obj;
+        for (const k of keys) {
+            if (cur == null)
+                return undefined;
+            cur = cur[/^\d+$/.test(k) ? parseInt(k) : k];
         }
+        return cur;
     }
-    function findFirstVideoUrl(obj) {
+    function findVideoUrl(obj) {
         if (!obj)
             return null;
         if (typeof obj === 'string') {
-            const trimmed = obj.trim();
-            if (/^https?:\/\/.+\.(mp4|mov|avi|webm|mkv)(\?.*)?$/i.test(trimmed))
-                return trimmed;
+            const t = obj.trim();
+            if (/^https?:\/\/.+\.(mp4|mov|avi|webm|mkv)(\?.*)?$/i.test(t))
+                return t;
             return null;
         }
         if (Array.isArray(obj)) {
             for (const item of obj) {
-                const found = findFirstVideoUrl(item);
-                if (found)
-                    return found;
+                const f = findVideoUrl(item);
+                if (f)
+                    return f;
             }
         }
         else if (typeof obj === 'object') {
-            for (const key of Object.keys(obj)) {
-                const found = findFirstVideoUrl(obj[key]);
-                if (found)
-                    return found;
+            for (const k of Object.keys(obj)) {
+                const f = findVideoUrl(obj[k]);
+                if (f)
+                    return f;
             }
         }
         return null;
     }
-    function validateEndpointUrl(url) {
+    function validateEndpoint(url) {
         try {
-            const parsed = new URL(url);
-            return parsed.protocol === 'http:' || parsed.protocol === 'https:';
+            const p = new URL(url);
+            return p.protocol === 'http:' || p.protocol === 'https:';
         }
         catch {
             return false;
         }
     }
-    async function safeSend(session, message) {
+    async function pollForResult(pollUrl, headers, interval, timeout) {
+        const start = Date.now();
+        while (Date.now() - start < timeout) {
+            try {
+                const res = await axios_1.default.get(pollUrl, { headers, timeout: 10000 });
+                const status = getValueByPath(res.data, 'status');
+                if (status === 'completed' || status === 'succeeded')
+                    return res.data;
+                if (status === 'failed' || status === 'error')
+                    throw new Error('Task failed');
+                await new Promise(r => setTimeout(r, interval));
+            }
+            catch (e) {
+                if (e.message === 'Task failed')
+                    throw e;
+                await new Promise(r => setTimeout(r, interval));
+            }
+        }
+        throw new Error('Polling timeout');
+    }
+    async function safeSend(session, msg) {
         try {
-            await session.send(message);
+            await session.send(msg);
         }
         catch (e) {
-            logger.error('发送消息失败', e);
+            logger.error('发送失败', e);
         }
     }
-    function getErrorMessage(err) {
-        if (axios_1.default.isAxiosError(err)) {
-            if (err.code === 'ECONNABORTED')
-                return '请求超时';
-            if (err.response) {
-                const status = err.response.status;
-                if (status === 401)
-                    return 'API Key 无效';
-                if (status === 429)
-                    return '请求过于频繁';
-                if (status >= 500)
-                    return '服务器错误';
-                return `HTTP ${status}`;
+    async function sendSingleVideo(session, url) {
+        const mode = cfg.videoSendMode;
+        if (mode === 'url')
+            await safeSend(session, url);
+        else {
+            try {
+                await safeSend(session, koishi_1.segment.video(url));
             }
-            return '网络错误';
+            catch { }
+            if (mode === 'both')
+                await safeSend(session, url);
         }
-        return err.message?.slice(0, 100) || '未知错误';
     }
     function sanitizeForLog(obj, sensitive) {
         if (!sensitive)
             return obj;
         try {
-            const str = JSON.stringify(obj);
-            return JSON.parse(str.split(sensitive).join('***'));
+            return JSON.parse(JSON.stringify(obj).split(sensitive).join('***'));
         }
         catch {
             return obj;
         }
     }
-    function isValidQQ(id) {
-        return /^\d{5,11}$/.test(id);
-    }
-    async function isBlacklisted(userId) {
-        try {
-            const rows = await ctx.database.get('ai_video_blacklist', { id: userId });
-            return rows.length > 0;
-        }
-        catch (e) {
-            return false;
-        }
-    }
-    async function addToBlacklist(ids) {
-        const success = [];
-        const fail = [];
-        const validIds = ids.filter(id => { if (isValidQQ(id))
-            return true; fail.push(id); return false; });
-        if (validIds.length === 0)
-            return { success, fail };
-        const existing = await ctx.database.get('ai_video_blacklist', { id: validIds });
-        const existingSet = new Set(existing.map((e) => e.id));
-        const toCreate = validIds.filter(id => !existingSet.has(id));
-        for (const id of toCreate) {
-            try {
-                await ctx.database.create('ai_video_blacklist', { id, createdAt: new Date() });
-                success.push(id);
-            }
-            catch {
-                fail.push(id);
-            }
-        }
-        for (const entry of existing)
-            fail.push(entry.id);
-        return { success, fail };
-    }
-    async function removeFromBlacklist(ids) {
-        const success = [];
-        const fail = [];
-        const validIds = ids.filter(id => { if (isValidQQ(id))
-            return true; fail.push(id); return false; });
-        if (validIds.length === 0)
-            return { success, fail };
-        const existing = await ctx.database.get('ai_video_blacklist', { id: validIds });
-        const existingSet = new Set(existing.map((e) => e.id));
-        const toRemove = validIds.filter(id => existingSet.has(id));
-        for (const id of toRemove) {
-            try {
-                await ctx.database.remove('ai_video_blacklist', { id });
-                success.push(id);
-            }
-            catch {
-                fail.push(id);
-            }
-        }
-        for (const id of validIds.filter(id => !existingSet.has(id)))
-            fail.push(id);
-        return { success, fail };
-    }
-    async function customGenerateVideo(session, api, prompt, imageUrl = '', modelOverride) {
+    async function customGenerateVideo(session, api, prompt, imageUrl = '') {
         const isImg2Video = !!imageUrl;
-        const model = modelOverride || (isImg2Video ? (api.img2videoModel || api.model) : api.model);
+        const model = isImg2Video ? (api.img2videoModel || api.model) : api.model;
         const duration = api.videoDuration || cfg.videoDuration || 5;
         const resolution = api.videoResolution || cfg.videoResolution || '1024x576';
         const promptTemplate = isImg2Video ? api.img2videoPrompt : api.txt2videoPrompt;
         let finalPrompt = prompt;
-        if (promptTemplate) {
-            finalPrompt = promptTemplate.replace('{prompt}', prompt).replace('{url}', imageUrl || '');
-        }
+        if (promptTemplate)
+            finalPrompt = promptTemplate.replace('{prompt}', prompt).replace('{url}', imageUrl);
         const bodyTemplate = isImg2Video ? api.img2videoBody : api.txt2videoBody;
-        const bodyVars = { model, prompt: finalPrompt, duration: String(duration), size: resolution };
-        if (isImg2Video) {
-            bodyVars['url'] = imageUrl;
-        }
+        const vars = {
+            model,
+            prompt: finalPrompt,
+            duration,
+            size: resolution,
+        };
+        if (isImg2Video)
+            vars.url = imageUrl;
         let body;
         try {
-            body = resolveTemplate(bodyTemplate, bodyVars);
+            body = resolveTemplate(bodyTemplate, vars);
         }
         catch {
             await safeSend(session, cfg.messages.fail + cfg.messages.templateError);
-            return;
+            return null;
         }
-        if (!validateEndpointUrl(api.endpoint)) {
-            await safeSend(session, cfg.messages.fail + '（API端点配置无效）');
-            return;
+        if (!validateEndpoint(api.endpoint)) {
+            await safeSend(session, cfg.messages.fail + '（端点无效）');
+            return null;
         }
         const sensitive = api.headers?.Authorization?.split(' ')[1] || '';
-        if (debug) {
+        if (debug)
             logger.info('API请求', JSON.stringify(sanitizeForLog(body, sensitive)));
-        }
         try {
-            const config = {
-                url: api.endpoint,
-                method: api.method,
-                headers: api.headers,
-                data: body,
-                timeout: cfg.timeout,
-            };
+            const config = { url: api.endpoint, method: api.method, headers: api.headers, data: body, timeout: cfg.timeout };
             if (cfg.proxyEnabled && cfg.proxyHost) {
                 config.proxy = {
-                    protocol: cfg.proxyProtocol,
-                    host: cfg.proxyHost,
-                    port: cfg.proxyPort,
-                    auth: cfg.proxyAuth && cfg.proxyUsername ? { username: cfg.proxyUsername, password: cfg.proxyPassword } : undefined,
+                    protocol: cfg.proxyProtocol, host: cfg.proxyHost, port: cfg.proxyPort,
+                    auth: cfg.proxyAuth && cfg.proxyUsername ? { username: cfg.proxyUsername, password: cfg.proxyPassword } : undefined
                 };
             }
-            let responseData = (await (0, axios_1.default)(config)).data;
+            let res = (await (0, axios_1.default)(config)).data;
             if (cfg.pollEnabled && api.taskIdPath) {
-                const taskId = getValueByPath(responseData, api.taskIdPath);
+                const taskId = getValueByPath(res, api.taskIdPath);
                 if (taskId) {
-                    const pollUrl = api.pollUrlTemplate.replace('{endpoint}', api.endpoint).replace('{task_id}', taskId);
-                    responseData = await pollForResult(pollUrl, api.headers, api.responseVideoPath, cfg.pollInterval, cfg.pollTimeout);
+                    const pollUrl = api.pollUrlTemplate.replace('{task_id}', taskId);
+                    await safeSend(session, cfg.messages.pollWaiting);
+                    res = await pollForResult(pollUrl, api.headers, cfg.pollInterval, cfg.pollTimeout);
                 }
             }
-            await handleVideoResponse(session, responseData, api);
-            const userId = `${session.guildId || 'private'}-${session.userId}`;
-            lastTaskMap.set(userId, {
-                prompt,
-                imageUrl,
-                isImg2Video,
-                model,
-                duration,
-                resolution,
-            });
+            const videoUrl = getValueByPath(res, api.responseVideoPath) || findVideoUrl(res);
+            return videoUrl || null;
         }
         catch (err) {
             logger.error('视频生成失败', err);
-            await safeSend(session, cfg.messages.fail + ` [${getErrorMessage(err)}]`);
+            await safeSend(session, cfg.messages.fail + ' [' + (err.message?.slice(0, 100) || '未知错误') + ']');
+            return null;
         }
     }
-    async function generateVideo(session, prompt, imageUrl = '', modelOverride) {
-        if (!checkRateLimit()) {
-            await safeSend(session, cfg.messages.rateLimit);
+    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;
         }
-        const api = getApi();
-        if (!api) {
-            await safeSend(session, cfg.messages.noApi);
-            return;
+        if (videoUrls.length === 1 || !cfg.enableForward) {
+            for (const url of videoUrls) {
+                await sendSingleVideo(session, url);
+            }
         }
-        recordApiCall();
-        return customGenerateVideo(session, api, prompt, imageUrl, modelOverride);
-    }
-    ctx.command('video <raw:text>', 'AI视频生成（文生视频/图生视频自动识别）')
-        .action(async ({ session }, raw) => {
-        try {
-            if (!session)
-                return;
-            if (await isBlacklisted(session.userId))
-                return safeSend(session, cfg.messages.blacklisted);
-            const prompt = (raw || '').trim();
-            const imgs = koishi_1.h.select(session.elements, 'img');
-            const hasImage = imgs.length > 0;
-            if (hasImage && !cfg.enableImg2Video)
-                return safeSend(session, cfg.messages.img2videoDisabled);
-            if (!hasImage && !cfg.enableTxt2Video)
-                return safeSend(session, cfg.messages.txt2videoDisabled);
-            if (!prompt) {
-                if (hasImage)
-                    return safeSend(session, '图生视频请提供提示词');
-                return safeSend(session, cfg.messages.empty);
+        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));
             }
-            if (prompt.length > 6000)
-                return safeSend(session, '提示词过长，请限制在6000字符以内');
-            if (!hasImage) {
-                await safeSend(session, cfg.messages.generating);
-                await generateVideo(session, prompt);
-                return;
+            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(() => {
+            collectSessions.delete(key);
+            safeSend(session, cfg.messages.collectTimeout);
+        }, cfg.collectTimeout * 1000);
+    }
+    ctx.command('video [text]', 'AI视频生成')
+        .action(async ({ session }, text) => {
+        if (!session)
+            return;
+        if (await isBlacklisted(session.userId))
+            return safeSend(session, cfg.messages.blacklisted);
+        const key = `${session.guildId || 'private'}-${session.userId}`;
+        if (collectSessions.has(key))
+            return safeSend(session, '你已在收集模式中');
+        const hasImage = koishi_1.h.select(session.elements, 'img').length > 0;
+        if (!hasImage && !cfg.enableTxt2Video)
+            return safeSend(session, cfg.messages.txt2videoDisabled);
+        if (hasImage && !cfg.enableImg2Video)
+            return safeSend(session, cfg.messages.img2videoDisabled);
+        let imageUrl = '';
+        if (hasImage) {
             const assets = ctx.assets;
             if (!assets)
                 return safeSend(session, cfg.messages.needAssets);
-            const uploadResult = await assets.upload(imgs[0].attrs.src, 'ref_image.jpg');
-            if (!uploadResult || !/^https?:\/\//.test(uploadResult)) {
-                return safeSend(session, cfg.messages.needAssets);
+            const img = koishi_1.h.select(session.elements, 'img')[0];
+            try {
+                const up = await assets.upload(img.attrs.src, 'ref_image.jpg');
+                if (/^https?:\/\//.test(up))
+                    imageUrl = up;
             }
+            catch { }
+        }
+        const collect = { prompt: text || '', imageUrl, timer: null };
+        collect.timer = startTimer(session, key, collect);
+        collectSessions.set(key, collect);
+        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}`;
+        const collect = collectSessions.get(key);
+        if (!collect)
+            return;
+        const text = (session.content || '').trim();
+        const imgs = koishi_1.h.select(session.elements, 'img');
+        if (text === '取消' || text === 'cancel') {
+            clearTimeout(collect.timer);
+            collectSessions.delete(key);
+            return safeSend(session, cfg.messages.cancelCollect);
+        }
+        if (text === '开始' || text === 'start' || text === '生成') {
+            clearTimeout(collect.timer);
+            collectSessions.delete(key);
+            if (!collect.prompt && !collect.imageUrl)
+                return safeSend(session, cfg.messages.empty);
             await safeSend(session, cfg.messages.generating);
-            await generateVideo(session, prompt, uploadResult);
+            return generateVideos(session, collect.prompt || '默认', collect.imageUrl, cfg.maxVideos);
         }
-        catch (e) {
-            logger.error('video命令异常', e);
-            await safeSend(session, cfg.messages.fail);
+        if (imgs.length > 0 && !collect.imageUrl) {
+            const assets = ctx.assets;
+            if (!assets)
+                return safeSend(session, cfg.messages.needAssets);
+            try {
+                const up = await assets.upload(imgs[0].attrs.src, 'ref_image.jpg');
+                if (/^https?:\/\//.test(up)) {
+                    collect.imageUrl = up;
+                    clearTimeout(collect.timer);
+                    collect.timer = startTimer(session, key, collect);
+                    await safeSend(session, cfg.messages.collectUpdate.replace('{images}', '1'));
+                }
+            }
+            catch { }
+            return;
+        }
+        if (text) {
+            collect.prompt = collect.prompt ? collect.prompt + ' ' + text : text;
+            clearTimeout(collect.timer);
+            collect.timer = startTimer(session, key, collect);
+            await safeSend(session, cfg.messages.collectUpdate.replace('{images}', collect.imageUrl ? '1' : '0'));
         }
     });
     ctx.command('redraw', '重绘上一次文生视频')
         .action(async ({ session }) => {
-        try {
-            if (!session)
-                return;
-            if (await isBlacklisted(session.userId))
-                return safeSend(session, cfg.messages.blacklisted);
-            const userId = `${session.guildId || 'private'}-${session.userId}`;
-            const last = lastTaskMap.get(userId);
-            if (!last)
-                return safeSend(session, cfg.messages.noLastTask);
-            if (last.isImg2Video)
-                return safeSend(session, cfg.messages.redrawImg2Video);
-            await safeSend(session, cfg.messages.redrawing);
-            await generateVideo(session, last.prompt, '', last.model);
-        }
-        catch (e) {
-            logger.error('重绘命令异常', e);
-            await safeSend(session, cfg.messages.fail);
-        }
-    });
-    const blacklistCmd = ctx.command('blacklist', '黑名单管理');
-    blacklistCmd.subcommand('.list', '查看黑名单').action(async ({ session }) => {
         if (!session)
             return;
-        if (!cfg.blacklistAdmins.includes(session.userId))
-            return safeSend(session, cfg.messages.noPermission);
+        if (await isBlacklisted(session.userId))
+            return safeSend(session, cfg.messages.blacklisted);
+        const last = lastTaskMap.get(`${session.guildId || 'private'}-${session.userId}`);
+        if (!last)
+            return safeSend(session, cfg.messages.noLastTask);
+        if (last.isImg2Video)
+            return safeSend(session, cfg.messages.redrawImg2Video);
+        await safeSend(session, cfg.messages.redrawing);
+        return generateVideos(session, last.prompt, '', cfg.maxVideos);
+    });
+    function isValidQQ(id) { return /^\d{5,11}$/.test(id); }
+    async function isBlacklisted(userId) {
         try {
-            const entries = await ctx.database.get('ai_video_blacklist', {});
-            if (entries.length === 0)
-                return safeSend(session, cfg.messages.blacklistListEmpty);
-            const list = entries.map(e => e.id).join('\n');
-            return safeSend(session, cfg.messages.blacklistListTitle + '\n' + list);
+            return (await ctx.database.get('ai_video_blacklist', { id: userId })).length > 0;
         }
-        catch (e) {
-            return safeSend(session, cfg.messages.fail);
+        catch {
+            return false;
         }
+    }
+    const blacklistCmd = ctx.command('blacklist', '黑名单管理');
+    blacklistCmd.subcommand('.list').action(async ({ session }) => {
+        if (!session || !cfg.blacklistAdmins.includes(session.userId))
+            return safeSend(session, cfg.messages.noPermission);
+        const entries = await ctx.database.get('ai_video_blacklist', {});
+        if (!entries.length)
+            return safeSend(session, cfg.messages.blacklistListEmpty);
+        return safeSend(session, cfg.messages.blacklistListTitle + '\n' + entries.map(e => e.id).join('\n'));
     });
-    blacklistCmd.subcommand('.add <...targets:string>', '添加黑名单').action(async ({ session }, ...targets) => {
-        if (!session)
-            return;
-        if (!cfg.blacklistAdmins.includes(session.userId))
+    blacklistCmd.subcommand('.add <...targets:string>').action(async ({ session }, ...targets) => {
+        if (!session || !cfg.blacklistAdmins.includes(session.userId))
             return safeSend(session, cfg.messages.noPermission);
-        const ids = targets.map(t => t.trim()).filter(id => id.length > 0);
+        const ids = targets.map(t => t.trim()).filter(Boolean);
         const invalid = ids.filter(id => !isValidQQ(id));
         if (invalid.length)
             return safeSend(session, cfg.messages.invalidUserId.replace('{targets}', invalid.join(', ')));
-        const { success, fail } = await addToBlacklist(ids);
+        const success = [], fail = [];
+        for (const id of ids) {
+            try {
+                const exist = await ctx.database.get('ai_video_blacklist', { id });
+                if (exist.length)
+                    fail.push(id);
+                else {
+                    await ctx.database.create('ai_video_blacklist', { id, createdAt: new Date() });
+                    success.push(id);
+                }
+            }
+            catch {
+                fail.push(id);
+            }
+        }
         if (success.length)
             await safeSend(session, cfg.messages.blacklistAddSuccess.replace('{targets}', success.join(', ')));
         if (fail.length)
             await safeSend(session, cfg.messages.blacklistAddFail.replace('{targets}', fail.join(', ')));
     });
-    blacklistCmd.subcommand('.remove <...targets:string>', '移除黑名单').action(async ({ session }, ...targets) => {
-        if (!session)
-            return;
-        if (!cfg.blacklistAdmins.includes(session.userId))
+    blacklistCmd.subcommand('.remove <...targets:string>').action(async ({ session }, ...targets) => {
+        if (!session || !cfg.blacklistAdmins.includes(session.userId))
             return safeSend(session, cfg.messages.noPermission);
-        const ids = targets.map(t => t.trim()).filter(id => id.length > 0);
+        const ids = targets.map(t => t.trim()).filter(Boolean);
         const invalid = ids.filter(id => !isValidQQ(id));
         if (invalid.length)
             return safeSend(session, cfg.messages.invalidUserId.replace('{targets}', invalid.join(', ')));
-        const { success, fail } = await removeFromBlacklist(ids);
+        const success = [], fail = [];
+        for (const id of ids) {
+            try {
+                const exist = await ctx.database.get('ai_video_blacklist', { id });
+                if (exist.length) {
+                    await ctx.database.remove('ai_video_blacklist', { id });
+                    success.push(id);
+                }
+                else
+                    fail.push(id);
+            }
+            catch {
+                fail.push(id);
+            }
+        }
         if (success.length)
             await safeSend(session, cfg.messages.blacklistRemoveSuccess.replace('{targets}', success.join(', ')));
         if (fail.length)

package/package.json CHANGED Viewed

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