@optima-chat/gen-cli 1.0.0
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.
- package/dist/backends/bfl.d.ts +18 -0
- package/dist/backends/bfl.d.ts.map +1 -0
- package/dist/backends/bfl.js +130 -0
- package/dist/backends/bfl.js.map +1 -0
- package/dist/backends/comfyui.d.ts +10 -0
- package/dist/backends/comfyui.d.ts.map +1 -0
- package/dist/backends/comfyui.js +121 -0
- package/dist/backends/comfyui.js.map +1 -0
- package/dist/backends/index.d.ts +24 -0
- package/dist/backends/index.d.ts.map +1 -0
- package/dist/backends/index.js +59 -0
- package/dist/backends/index.js.map +1 -0
- package/dist/backends/types.d.ts +99 -0
- package/dist/backends/types.d.ts.map +1 -0
- package/dist/backends/types.js +2 -0
- package/dist/backends/types.js.map +1 -0
- package/dist/backends/video-dashscope.d.ts +10 -0
- package/dist/backends/video-dashscope.d.ts.map +1 -0
- package/dist/backends/video-dashscope.js +108 -0
- package/dist/backends/video-dashscope.js.map +1 -0
- package/dist/backends/video-index.d.ts +23 -0
- package/dist/backends/video-index.d.ts.map +1 -0
- package/dist/backends/video-index.js +57 -0
- package/dist/backends/video-index.js.map +1 -0
- package/dist/commands/asr.d.ts +3 -0
- package/dist/commands/asr.d.ts.map +1 -0
- package/dist/commands/asr.js +133 -0
- package/dist/commands/asr.js.map +1 -0
- package/dist/commands/config.d.ts +3 -0
- package/dist/commands/config.d.ts.map +1 -0
- package/dist/commands/config.js +105 -0
- package/dist/commands/config.js.map +1 -0
- package/dist/commands/download.d.ts +3 -0
- package/dist/commands/download.d.ts.map +1 -0
- package/dist/commands/download.js +169 -0
- package/dist/commands/download.js.map +1 -0
- package/dist/commands/edit.d.ts +3 -0
- package/dist/commands/edit.d.ts.map +1 -0
- package/dist/commands/edit.js +155 -0
- package/dist/commands/edit.js.map +1 -0
- package/dist/commands/generate.d.ts +3 -0
- package/dist/commands/generate.d.ts.map +1 -0
- package/dist/commands/generate.js +135 -0
- package/dist/commands/generate.js.map +1 -0
- package/dist/commands/image.d.ts +3 -0
- package/dist/commands/image.d.ts.map +1 -0
- package/dist/commands/image.js +139 -0
- package/dist/commands/image.js.map +1 -0
- package/dist/commands/init.d.ts +3 -0
- package/dist/commands/init.d.ts.map +1 -0
- package/dist/commands/init.js +144 -0
- package/dist/commands/init.js.map +1 -0
- package/dist/commands/interrupt.d.ts +3 -0
- package/dist/commands/interrupt.d.ts.map +1 -0
- package/dist/commands/interrupt.js +54 -0
- package/dist/commands/interrupt.js.map +1 -0
- package/dist/commands/model.d.ts +3 -0
- package/dist/commands/model.d.ts.map +1 -0
- package/dist/commands/model.js +82 -0
- package/dist/commands/model.js.map +1 -0
- package/dist/commands/node.d.ts +3 -0
- package/dist/commands/node.d.ts.map +1 -0
- package/dist/commands/node.js +148 -0
- package/dist/commands/node.js.map +1 -0
- package/dist/commands/queue.d.ts +3 -0
- package/dist/commands/queue.d.ts.map +1 -0
- package/dist/commands/queue.js +199 -0
- package/dist/commands/queue.js.map +1 -0
- package/dist/commands/system.d.ts +3 -0
- package/dist/commands/system.d.ts.map +1 -0
- package/dist/commands/system.js +85 -0
- package/dist/commands/system.js.map +1 -0
- package/dist/commands/task.d.ts +3 -0
- package/dist/commands/task.d.ts.map +1 -0
- package/dist/commands/task.js +176 -0
- package/dist/commands/task.js.map +1 -0
- package/dist/commands/tryon.d.ts +3 -0
- package/dist/commands/tryon.d.ts.map +1 -0
- package/dist/commands/tryon.js +178 -0
- package/dist/commands/tryon.js.map +1 -0
- package/dist/commands/tts.d.ts +3 -0
- package/dist/commands/tts.d.ts.map +1 -0
- package/dist/commands/tts.js +157 -0
- package/dist/commands/tts.js.map +1 -0
- package/dist/commands/video.d.ts +3 -0
- package/dist/commands/video.d.ts.map +1 -0
- package/dist/commands/video.js +137 -0
- package/dist/commands/video.js.map +1 -0
- package/dist/commands/workflow.d.ts +3 -0
- package/dist/commands/workflow.d.ts.map +1 -0
- package/dist/commands/workflow.js +244 -0
- package/dist/commands/workflow.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +58 -0
- package/dist/index.js.map +1 -0
- package/dist/services/api.d.ts +23 -0
- package/dist/services/api.d.ts.map +1 -0
- package/dist/services/api.js +108 -0
- package/dist/services/api.js.map +1 -0
- package/dist/services/auth.d.ts +10 -0
- package/dist/services/auth.d.ts.map +1 -0
- package/dist/services/auth.js +34 -0
- package/dist/services/auth.js.map +1 -0
- package/dist/services/bfl-api.d.ts +67 -0
- package/dist/services/bfl-api.d.ts.map +1 -0
- package/dist/services/bfl-api.js +108 -0
- package/dist/services/bfl-api.js.map +1 -0
- package/dist/services/config.d.ts +12 -0
- package/dist/services/config.d.ts.map +1 -0
- package/dist/services/config.js +30 -0
- package/dist/services/config.js.map +1 -0
- package/dist/services/dashscope-api.d.ts +69 -0
- package/dist/services/dashscope-api.d.ts.map +1 -0
- package/dist/services/dashscope-api.js +134 -0
- package/dist/services/dashscope-api.js.map +1 -0
- package/dist/services/dashscope-tts-api.d.ts +23 -0
- package/dist/services/dashscope-tts-api.d.ts.map +1 -0
- package/dist/services/dashscope-tts-api.js +56 -0
- package/dist/services/dashscope-tts-api.js.map +1 -0
- package/dist/services/fashn-api.d.ts +57 -0
- package/dist/services/fashn-api.d.ts.map +1 -0
- package/dist/services/fashn-api.js +84 -0
- package/dist/services/fashn-api.js.map +1 -0
- package/dist/services/generation-api.d.ts +139 -0
- package/dist/services/generation-api.d.ts.map +1 -0
- package/dist/services/generation-api.js +236 -0
- package/dist/services/generation-api.js.map +1 -0
- package/dist/services/groq-api.d.ts +18 -0
- package/dist/services/groq-api.d.ts.map +1 -0
- package/dist/services/groq-api.js +30 -0
- package/dist/services/groq-api.js.map +1 -0
- package/dist/services/websocket.d.ts +33 -0
- package/dist/services/websocket.d.ts.map +1 -0
- package/dist/services/websocket.js +217 -0
- package/dist/services/websocket.js.map +1 -0
- package/dist/utils/helpText.d.ts +15 -0
- package/dist/utils/helpText.d.ts.map +1 -0
- package/dist/utils/helpText.js +37 -0
- package/dist/utils/helpText.js.map +1 -0
- package/dist/utils/logger.d.ts +5 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +14 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/output.d.ts +54 -0
- package/dist/utils/output.d.ts.map +1 -0
- package/dist/utils/output.js +76 -0
- package/dist/utils/output.js.map +1 -0
- package/dist/utils/workflow.d.ts +17 -0
- package/dist/utils/workflow.d.ts.map +1 -0
- package/dist/utils/workflow.js +81 -0
- package/dist/utils/workflow.js.map +1 -0
- package/package.json +61 -0
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import type { VideoBackend } from './types.js';
|
|
2
|
+
export type VideoBackendType = 'auto' | 'dashscope' | 'comfyui';
|
|
3
|
+
export { VideoBackend, VideoParams, VideoResult } from './types.js';
|
|
4
|
+
export { DashScopeVideoBackend } from './video-dashscope.js';
|
|
5
|
+
/**
|
|
6
|
+
* 获取 DashScope API Key(优先环境变量,其次配置文件)
|
|
7
|
+
*/
|
|
8
|
+
export declare function getDashScopeApiKey(): string | undefined;
|
|
9
|
+
/**
|
|
10
|
+
* 获取视频生成后端
|
|
11
|
+
*
|
|
12
|
+
* - auto: 有 DashScope API Key 则用 DashScope,否则用 ComfyUI
|
|
13
|
+
* - dashscope: 强制使用 DashScope(需要 API Key)
|
|
14
|
+
* - comfyui: 强制使用 ComfyUI
|
|
15
|
+
*
|
|
16
|
+
* API Key 优先级: 环境变量 DASHSCOPE_API_KEY > 配置文件 dashscope.apiKey
|
|
17
|
+
*/
|
|
18
|
+
export declare function getVideoBackend(type?: VideoBackendType): VideoBackend;
|
|
19
|
+
/**
|
|
20
|
+
* 获取当前会使用的视频后端名称
|
|
21
|
+
*/
|
|
22
|
+
export declare function getCurrentVideoBackendName(type?: VideoBackendType): 'dashscope' | 'comfyui';
|
|
23
|
+
//# sourceMappingURL=video-index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"video-index.d.ts","sourceRoot":"","sources":["../../src/backends/video-index.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE/C,MAAM,MAAM,gBAAgB,GAAG,MAAM,GAAG,WAAW,GAAG,SAAS,CAAC;AAEhE,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAE7D;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,MAAM,GAAG,SAAS,CAEvD;AAED;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAAC,IAAI,CAAC,EAAE,gBAAgB,GAAG,YAAY,CAyBrE;AAED;;GAEG;AACH,wBAAgB,0BAA0B,CAAC,IAAI,CAAC,EAAE,gBAAgB,GAAG,WAAW,GAAG,SAAS,CAe3F"}
|
|
@@ -0,0 +1,57 @@
|
|
|
1
|
+
import { getConfig } from '../services/config.js';
|
|
2
|
+
import { DashScopeVideoBackend } from './video-dashscope.js';
|
|
3
|
+
export { DashScopeVideoBackend } from './video-dashscope.js';
|
|
4
|
+
/**
|
|
5
|
+
* 获取 DashScope API Key(优先环境变量,其次配置文件)
|
|
6
|
+
*/
|
|
7
|
+
export function getDashScopeApiKey() {
|
|
8
|
+
return process.env.DASHSCOPE_API_KEY || getConfig().dashscope?.apiKey;
|
|
9
|
+
}
|
|
10
|
+
/**
|
|
11
|
+
* 获取视频生成后端
|
|
12
|
+
*
|
|
13
|
+
* - auto: 有 DashScope API Key 则用 DashScope,否则用 ComfyUI
|
|
14
|
+
* - dashscope: 强制使用 DashScope(需要 API Key)
|
|
15
|
+
* - comfyui: 强制使用 ComfyUI
|
|
16
|
+
*
|
|
17
|
+
* API Key 优先级: 环境变量 DASHSCOPE_API_KEY > 配置文件 dashscope.apiKey
|
|
18
|
+
*/
|
|
19
|
+
export function getVideoBackend(type) {
|
|
20
|
+
const cfg = getConfig();
|
|
21
|
+
const backendType = type || cfg.preferVideoBackend || 'auto';
|
|
22
|
+
// 强制使用 DashScope
|
|
23
|
+
if (backendType === 'dashscope') {
|
|
24
|
+
const apiKey = getDashScopeApiKey();
|
|
25
|
+
if (!apiKey) {
|
|
26
|
+
throw new Error('DashScope API key not configured. Set DASHSCOPE_API_KEY env or run: comfy config set dashscope.apiKey <key>');
|
|
27
|
+
}
|
|
28
|
+
return new DashScopeVideoBackend(apiKey);
|
|
29
|
+
}
|
|
30
|
+
// 强制使用 ComfyUI(返回 null,由调用方处理)
|
|
31
|
+
if (backendType === 'comfyui') {
|
|
32
|
+
return null; // ComfyUI 视频后端由 video.ts 原有逻辑处理
|
|
33
|
+
}
|
|
34
|
+
// 自动选择:优先 DashScope
|
|
35
|
+
const apiKey = getDashScopeApiKey();
|
|
36
|
+
if (apiKey) {
|
|
37
|
+
return new DashScopeVideoBackend(apiKey);
|
|
38
|
+
}
|
|
39
|
+
return null; // 回退到 ComfyUI
|
|
40
|
+
}
|
|
41
|
+
/**
|
|
42
|
+
* 获取当前会使用的视频后端名称
|
|
43
|
+
*/
|
|
44
|
+
export function getCurrentVideoBackendName(type) {
|
|
45
|
+
const cfg = getConfig();
|
|
46
|
+
const backendType = type || cfg.preferVideoBackend || 'auto';
|
|
47
|
+
if (backendType === 'dashscope') {
|
|
48
|
+
return 'dashscope';
|
|
49
|
+
}
|
|
50
|
+
if (backendType === 'comfyui') {
|
|
51
|
+
return 'comfyui';
|
|
52
|
+
}
|
|
53
|
+
// auto
|
|
54
|
+
const apiKey = getDashScopeApiKey();
|
|
55
|
+
return apiKey ? 'dashscope' : 'comfyui';
|
|
56
|
+
}
|
|
57
|
+
//# sourceMappingURL=video-index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"video-index.js","sourceRoot":"","sources":["../../src/backends/video-index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAClD,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAM7D,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAE7D;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,OAAO,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,SAAS,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC;AACxE,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,eAAe,CAAC,IAAuB;IACrD,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;IACxB,MAAM,WAAW,GAAG,IAAI,IAAI,GAAG,CAAC,kBAAkB,IAAI,MAAM,CAAC;IAE7D,iBAAiB;IACjB,IAAI,WAAW,KAAK,WAAW,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAC;QACpC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,6GAA6G,CAAC,CAAC;QACjI,CAAC;QACD,OAAO,IAAI,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC;IAED,+BAA+B;IAC/B,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;QAC9B,OAAO,IAAW,CAAC,CAAC,gCAAgC;IACtD,CAAC;IAED,oBAAoB;IACpB,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAC;IACpC,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,IAAI,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC;IAED,OAAO,IAAW,CAAC,CAAC,cAAc;AACpC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,0BAA0B,CAAC,IAAuB;IAChE,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;IACxB,MAAM,WAAW,GAAG,IAAI,IAAI,GAAG,CAAC,kBAAkB,IAAI,MAAM,CAAC;IAE7D,IAAI,WAAW,KAAK,WAAW,EAAE,CAAC;QAChC,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;QAC9B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO;IACP,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAC;IACpC,OAAO,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;AAC1C,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"asr.d.ts","sourceRoot":"","sources":["../../src/commands/asr.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAepC,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,OAAO,QA+HlD"}
|
|
@@ -0,0 +1,133 @@
|
|
|
1
|
+
import * as path from 'path';
|
|
2
|
+
import * as fs from 'fs';
|
|
3
|
+
import ora from 'ora';
|
|
4
|
+
import { success, info } from '../utils/logger.js';
|
|
5
|
+
import { outputSuccess, outputError, isPrettyMode } from '../utils/output.js';
|
|
6
|
+
import { addEnhancedHelp } from '../utils/helpText.js';
|
|
7
|
+
import { GenerationApiClient, GenerationApiError, validateASRParams, fileToBase64, } from '../services/generation-api.js';
|
|
8
|
+
export function registerAsrCommand(program) {
|
|
9
|
+
const cmd = program
|
|
10
|
+
.command('asr')
|
|
11
|
+
.description('语音识别 (Speech-to-Text) - 使用 Groq Whisper')
|
|
12
|
+
.argument('<file>', '音频文件路径 (支持 mp3, wav, m4a, ogg, webm)')
|
|
13
|
+
.option('--language <lang>', '语言提示: zh | en | ja | ko 等')
|
|
14
|
+
.option('--model <name>', 'Whisper 模型', 'whisper-large-v3-turbo')
|
|
15
|
+
.action(async (file, options) => {
|
|
16
|
+
const pretty = isPrettyMode(options);
|
|
17
|
+
try {
|
|
18
|
+
// 检查文件是否存在
|
|
19
|
+
const filePath = path.resolve(file);
|
|
20
|
+
if (!fs.existsSync(filePath)) {
|
|
21
|
+
outputError('FILE_NOT_FOUND', `音频文件不存在: ${filePath}`, options);
|
|
22
|
+
return;
|
|
23
|
+
}
|
|
24
|
+
// CLI 层快速校验
|
|
25
|
+
const stat = fs.statSync(filePath);
|
|
26
|
+
validateASRParams({ fileSize: stat.size });
|
|
27
|
+
// 音频文件转 base64
|
|
28
|
+
const fileBase64 = await fileToBase64(filePath);
|
|
29
|
+
const params = {
|
|
30
|
+
file: fileBase64,
|
|
31
|
+
filename: path.basename(filePath),
|
|
32
|
+
language: options.language,
|
|
33
|
+
model: options.model,
|
|
34
|
+
};
|
|
35
|
+
const client = new GenerationApiClient();
|
|
36
|
+
const startTime = Date.now();
|
|
37
|
+
if (pretty) {
|
|
38
|
+
const spinner = ora('正在识别语音...').start();
|
|
39
|
+
try {
|
|
40
|
+
const result = await client.submitAndPoll('asr', params, {
|
|
41
|
+
onProgress: (status) => {
|
|
42
|
+
spinner.text = `语音识别中 (${status})...`;
|
|
43
|
+
},
|
|
44
|
+
});
|
|
45
|
+
const durationMs = Date.now() - startTime;
|
|
46
|
+
if (result.status === 'completed') {
|
|
47
|
+
spinner.succeed('语音识别完成!');
|
|
48
|
+
if (result.result_meta?.text) {
|
|
49
|
+
success(`识别结果: ${result.result_meta.text}`);
|
|
50
|
+
}
|
|
51
|
+
else {
|
|
52
|
+
info(`任务完成但未返回文本,使用 'gen task get ${result.task_id}' 查看详情`);
|
|
53
|
+
}
|
|
54
|
+
info(`耗时: ${(durationMs / 1000).toFixed(1)}s`);
|
|
55
|
+
}
|
|
56
|
+
else {
|
|
57
|
+
spinner.info(`任务状态: ${result.status}`);
|
|
58
|
+
if (result.error_message)
|
|
59
|
+
info(`错误: ${result.error_message}`);
|
|
60
|
+
info(`使用 'gen task get ${result.task_id}' 查看最新状态`);
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
catch (err) {
|
|
64
|
+
spinner.fail(`识别失败: ${err?.message || err}`);
|
|
65
|
+
throw err;
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
else {
|
|
69
|
+
const result = await client.submitAndPoll('asr', params);
|
|
70
|
+
const durationMs = Date.now() - startTime;
|
|
71
|
+
if (result.status === 'completed') {
|
|
72
|
+
outputSuccess({
|
|
73
|
+
task_id: result.task_id,
|
|
74
|
+
text: result.result_meta?.text || '',
|
|
75
|
+
file: filePath,
|
|
76
|
+
language: options.language || 'auto',
|
|
77
|
+
model: options.model,
|
|
78
|
+
durationMs,
|
|
79
|
+
});
|
|
80
|
+
}
|
|
81
|
+
else {
|
|
82
|
+
outputSuccess({
|
|
83
|
+
task_id: result.task_id,
|
|
84
|
+
status: result.status,
|
|
85
|
+
error_message: result.error_message,
|
|
86
|
+
});
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
catch (err) {
|
|
91
|
+
if (err instanceof GenerationApiError) {
|
|
92
|
+
outputError(err.code, err.message, options);
|
|
93
|
+
}
|
|
94
|
+
else {
|
|
95
|
+
outputError('ASR_FAILED', `语音识别失败: ${err}`, options);
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
});
|
|
99
|
+
addEnhancedHelp(cmd, {
|
|
100
|
+
examples: [
|
|
101
|
+
'# 基本用法 - 自动检测语言',
|
|
102
|
+
'$ gen asr recording.mp3',
|
|
103
|
+
'',
|
|
104
|
+
'# 指定语言提示(提高准确率)',
|
|
105
|
+
'$ gen asr recording.wav --language zh',
|
|
106
|
+
'',
|
|
107
|
+
'# 识别英文音频',
|
|
108
|
+
'$ gen asr speech.m4a --language en',
|
|
109
|
+
],
|
|
110
|
+
outputJson: `{
|
|
111
|
+
"success": true,
|
|
112
|
+
"data": {
|
|
113
|
+
"task_id": "abc-123",
|
|
114
|
+
"text": "识别出的文字内容",
|
|
115
|
+
"file": "/path/to/audio.mp3",
|
|
116
|
+
"language": "zh",
|
|
117
|
+
"model": "whisper-large-v3-turbo",
|
|
118
|
+
"durationMs": 850
|
|
119
|
+
}
|
|
120
|
+
}`,
|
|
121
|
+
relatedCommands: [
|
|
122
|
+
{ command: 'tts <text>', description: '文本转语音' },
|
|
123
|
+
{ command: 'task get <id>', description: '查看任务状态' },
|
|
124
|
+
],
|
|
125
|
+
notes: [
|
|
126
|
+
'通过 optima-generation 调用 Groq Whisper 模型',
|
|
127
|
+
'需要 Optima 账号(OPTIMA_TOKEN 或 ~/.optima/token.json)',
|
|
128
|
+
'支持 mp3, wav, m4a, ogg, webm 等常见音频格式',
|
|
129
|
+
'文件大小限制 25MB',
|
|
130
|
+
],
|
|
131
|
+
});
|
|
132
|
+
}
|
|
133
|
+
//# sourceMappingURL=asr.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"asr.js","sourceRoot":"","sources":["../../src/commands/asr.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,GAAG,MAAM,KAAK,CAAC;AACtB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAC9E,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAElB,iBAAiB,EACjB,YAAY,GACb,MAAM,+BAA+B,CAAC;AAEvC,MAAM,UAAU,kBAAkB,CAAC,OAAgB;IACjD,MAAM,GAAG,GAAG,OAAO;SAChB,OAAO,CAAC,KAAK,CAAC;SACd,WAAW,CAAC,yCAAyC,CAAC;SACtD,QAAQ,CAAC,QAAQ,EAAE,sCAAsC,CAAC;SAC1D,MAAM,CAAC,mBAAmB,EAAE,2BAA2B,CAAC;SACxD,MAAM,CAAC,gBAAgB,EAAE,YAAY,EAAE,wBAAwB,CAAC;SAChE,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,OAAO,EAAE,EAAE;QACtC,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;QAErC,IAAI,CAAC;YACH,WAAW;YACX,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC7B,WAAW,CAAC,gBAAgB,EAAE,YAAY,QAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;gBAC/D,OAAO;YACT,CAAC;YAED,YAAY;YACZ,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACnC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAE3C,eAAe;YACf,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC,CAAC;YAEhD,MAAM,MAAM,GAAiB;gBAC3B,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACjC,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;aACrB,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,mBAAmB,EAAE,CAAC;YACzC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,OAAO,GAAG,GAAG,CAAC,WAAW,CAAC,CAAC,KAAK,EAAE,CAAC;gBAEzC,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE;wBACvD,UAAU,EAAE,CAAC,MAAM,EAAE,EAAE;4BACrB,OAAO,CAAC,IAAI,GAAG,UAAU,MAAM,MAAM,CAAC;wBACxC,CAAC;qBACF,CAAC,CAAC;oBAEH,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;oBAE1C,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;wBAClC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;wBAC3B,IAAI,MAAM,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC;4BAC7B,OAAO,CAAC,SAAS,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC;wBAC9C,CAAC;6BAAM,CAAC;4BACN,IAAI,CAAC,+BAA+B,MAAM,CAAC,OAAO,QAAQ,CAAC,CAAC;wBAC9D,CAAC;wBACD,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;oBACjD,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,IAAI,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;wBACvC,IAAI,MAAM,CAAC,aAAa;4BAAE,IAAI,CAAC,OAAO,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC;wBAC9D,IAAI,CAAC,oBAAoB,MAAM,CAAC,OAAO,UAAU,CAAC,CAAC;oBACrD,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAQ,EAAE,CAAC;oBAClB,OAAO,CAAC,IAAI,CAAC,SAAS,GAAG,EAAE,OAAO,IAAI,GAAG,EAAE,CAAC,CAAC;oBAC7C,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBACzD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBAE1C,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;oBAClC,aAAa,CAAC;wBACZ,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,IAAI,EAAE,MAAM,CAAC,WAAW,EAAE,IAAI,IAAI,EAAE;wBACpC,IAAI,EAAE,QAAQ;wBACd,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,MAAM;wBACpC,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,UAAU;qBACX,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC;wBACZ,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,MAAM,EAAE,MAAM,CAAC,MAAM;wBACrB,aAAa,EAAE,MAAM,CAAC,aAAa;qBACpC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,kBAAkB,EAAE,CAAC;gBACtC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,YAAY,EAAE,WAAW,GAAG,EAAE,EAAE,OAAO,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,eAAe,CAAC,GAAG,EAAE;QACnB,QAAQ,EAAE;YACR,iBAAiB;YACjB,yBAAyB;YACzB,EAAE;YACF,iBAAiB;YACjB,uCAAuC;YACvC,EAAE;YACF,UAAU;YACV,oCAAoC;SACrC;QACD,UAAU,EAAE;;;;;;;;;;EAUd;QACE,eAAe,EAAE;YACf,EAAE,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,OAAO,EAAE;YAC/C,EAAE,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,QAAQ,EAAE;SACpD;QACD,KAAK,EAAE;YACL,yCAAyC;YACzC,mDAAmD;YACnD,qCAAqC;YACrC,aAAa;SACd;KACF,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/commands/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAMpC,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,OAAO,QA0GrD"}
|
|
@@ -0,0 +1,105 @@
|
|
|
1
|
+
import { config, getConfig, setConfig, resetConfig, isValidConfigKey } from '../services/config.js';
|
|
2
|
+
import { success, info } from '../utils/logger.js';
|
|
3
|
+
import { outputSuccess, outputError, isPrettyMode } from '../utils/output.js';
|
|
4
|
+
import { addEnhancedHelp } from '../utils/helpText.js';
|
|
5
|
+
export function registerConfigCommand(program) {
|
|
6
|
+
const configCmd = program
|
|
7
|
+
.command('config')
|
|
8
|
+
.description('配置管理');
|
|
9
|
+
// gen config set
|
|
10
|
+
const setCmd = configCmd
|
|
11
|
+
.command('set')
|
|
12
|
+
.description('设置配置项')
|
|
13
|
+
.argument('<key>', '配置键')
|
|
14
|
+
.argument('<value>', '配置值')
|
|
15
|
+
.action((key, value, options) => {
|
|
16
|
+
const pretty = isPrettyMode(options);
|
|
17
|
+
try {
|
|
18
|
+
if (!isValidConfigKey(key)) {
|
|
19
|
+
outputError('INVALID_CONFIG_KEY', `无效的配置键: ${key}。可用键: outputDir`, options);
|
|
20
|
+
return;
|
|
21
|
+
}
|
|
22
|
+
setConfig(key, value);
|
|
23
|
+
if (pretty) {
|
|
24
|
+
success(`已设置 ${key} = ${value}`);
|
|
25
|
+
}
|
|
26
|
+
else {
|
|
27
|
+
outputSuccess({ message: '配置已更新', key, value });
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
catch (err) {
|
|
31
|
+
outputError('CONFIG_SET_FAILED', `设置失败: ${err}`, options);
|
|
32
|
+
}
|
|
33
|
+
});
|
|
34
|
+
addEnhancedHelp(setCmd, {
|
|
35
|
+
examples: [
|
|
36
|
+
'# 设置输出目录',
|
|
37
|
+
'$ gen config set outputDir ./my-outputs',
|
|
38
|
+
],
|
|
39
|
+
outputJson: `{
|
|
40
|
+
"success": true,
|
|
41
|
+
"data": { "message": "配置已更新", "key": "outputDir", "value": "./my-outputs" }
|
|
42
|
+
}`,
|
|
43
|
+
relatedCommands: [
|
|
44
|
+
{ command: 'config list', description: '查看当前配置' },
|
|
45
|
+
{ command: 'config reset --confirm', description: '重置为默认配置' },
|
|
46
|
+
],
|
|
47
|
+
notes: ['可用配置项: outputDir'],
|
|
48
|
+
});
|
|
49
|
+
// gen config list
|
|
50
|
+
const listCmd = configCmd
|
|
51
|
+
.command('list')
|
|
52
|
+
.description('查看当前配置')
|
|
53
|
+
.action((options) => {
|
|
54
|
+
const pretty = isPrettyMode(options);
|
|
55
|
+
try {
|
|
56
|
+
const cfg = getConfig();
|
|
57
|
+
if (pretty) {
|
|
58
|
+
info(`Output Dir: ${cfg.outputDir}`);
|
|
59
|
+
info(`配置文件: ${config.path}`);
|
|
60
|
+
}
|
|
61
|
+
else {
|
|
62
|
+
outputSuccess({ config: cfg, config_path: config.path });
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
catch (err) {
|
|
66
|
+
outputError('CONFIG_LIST_FAILED', `获取配置失败: ${err}`, options);
|
|
67
|
+
}
|
|
68
|
+
});
|
|
69
|
+
addEnhancedHelp(listCmd, {
|
|
70
|
+
examples: ['$ gen config list'],
|
|
71
|
+
outputJson: `{
|
|
72
|
+
"success": true,
|
|
73
|
+
"data": { "config": { "outputDir": "./gen-output" }, "config_path": "..." }
|
|
74
|
+
}`,
|
|
75
|
+
relatedCommands: [
|
|
76
|
+
{ command: 'config set <key> <value>', description: '设置配置项' },
|
|
77
|
+
],
|
|
78
|
+
notes: ['显示所有配置项的当前值和配置文件路径'],
|
|
79
|
+
});
|
|
80
|
+
// gen config reset
|
|
81
|
+
configCmd
|
|
82
|
+
.command('reset')
|
|
83
|
+
.description('重置为默认配置(需要 --confirm)')
|
|
84
|
+
.option('--confirm', '确认重置')
|
|
85
|
+
.action((options) => {
|
|
86
|
+
const pretty = isPrettyMode(options);
|
|
87
|
+
try {
|
|
88
|
+
if (!options.confirm) {
|
|
89
|
+
outputError('CONFIRM_REQUIRED', '请使用 --confirm 确认重置操作', options);
|
|
90
|
+
return;
|
|
91
|
+
}
|
|
92
|
+
resetConfig();
|
|
93
|
+
if (pretty) {
|
|
94
|
+
success('配置已重置为默认值');
|
|
95
|
+
}
|
|
96
|
+
else {
|
|
97
|
+
outputSuccess({ message: '配置已重置为默认值' });
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
catch (err) {
|
|
101
|
+
outputError('CONFIG_RESET_FAILED', `重置失败: ${err}`, options);
|
|
102
|
+
}
|
|
103
|
+
});
|
|
104
|
+
}
|
|
105
|
+
//# sourceMappingURL=config.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/commands/config.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACpG,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAC9E,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,MAAM,UAAU,qBAAqB,CAAC,OAAgB;IACpD,MAAM,SAAS,GAAG,OAAO;SACtB,OAAO,CAAC,QAAQ,CAAC;SACjB,WAAW,CAAC,MAAM,CAAC,CAAC;IAEvB,iBAAiB;IACjB,MAAM,MAAM,GAAG,SAAS;SACrB,OAAO,CAAC,KAAK,CAAC;SACd,WAAW,CAAC,OAAO,CAAC;SACpB,QAAQ,CAAC,OAAO,EAAE,KAAK,CAAC;SACxB,QAAQ,CAAC,SAAS,EAAE,KAAK,CAAC;SAC1B,MAAM,CAAC,CAAC,GAAW,EAAE,KAAa,EAAE,OAAO,EAAE,EAAE;QAC9C,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;QAErC,IAAI,CAAC;YACH,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC3B,WAAW,CAAC,oBAAoB,EAAE,WAAW,GAAG,iBAAiB,EAAE,OAAO,CAAC,CAAC;gBAC5E,OAAO;YACT,CAAC;YAED,SAAS,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YAEtB,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,CAAC,OAAO,GAAG,MAAM,KAAK,EAAE,CAAC,CAAC;YACnC,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,WAAW,CAAC,mBAAmB,EAAE,SAAS,GAAG,EAAE,EAAE,OAAO,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,eAAe,CAAC,MAAM,EAAE;QACtB,QAAQ,EAAE;YACR,UAAU;YACV,yCAAyC;SAC1C;QACD,UAAU,EAAE;;;EAGd;QACE,eAAe,EAAE;YACf,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,QAAQ,EAAE;YACjD,EAAE,OAAO,EAAE,wBAAwB,EAAE,WAAW,EAAE,SAAS,EAAE;SAC9D;QACD,KAAK,EAAE,CAAC,kBAAkB,CAAC;KAC5B,CAAC,CAAC;IAEH,kBAAkB;IAClB,MAAM,OAAO,GAAG,SAAS;SACtB,OAAO,CAAC,MAAM,CAAC;SACf,WAAW,CAAC,QAAQ,CAAC;SACrB,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;QAClB,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;QAErC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;YAExB,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,CAAC,eAAe,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;gBACrC,IAAI,CAAC,SAAS,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,WAAW,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,WAAW,CAAC,oBAAoB,EAAE,WAAW,GAAG,EAAE,EAAE,OAAO,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,eAAe,CAAC,OAAO,EAAE;QACvB,QAAQ,EAAE,CAAC,mBAAmB,CAAC;QAC/B,UAAU,EAAE;;;EAGd;QACE,eAAe,EAAE;YACf,EAAE,OAAO,EAAE,0BAA0B,EAAE,WAAW,EAAE,OAAO,EAAE;SAC9D;QACD,KAAK,EAAE,CAAC,oBAAoB,CAAC;KAC9B,CAAC,CAAC;IAEH,mBAAmB;IACnB,SAAS;SACN,OAAO,CAAC,OAAO,CAAC;SAChB,WAAW,CAAC,uBAAuB,CAAC;SACpC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC;SAC3B,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;QAClB,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;QAErC,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBACrB,WAAW,CAAC,kBAAkB,EAAE,sBAAsB,EAAE,OAAO,CAAC,CAAC;gBACjE,OAAO;YACT,CAAC;YAED,WAAW,EAAE,CAAC;YAEd,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,CAAC,WAAW,CAAC,CAAC;YACvB,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,WAAW,CAAC,qBAAqB,EAAE,SAAS,GAAG,EAAE,EAAE,OAAO,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"download.d.ts","sourceRoot":"","sources":["../../src/commands/download.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AASpC,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,OAAO,QAyKvD"}
|
|
@@ -0,0 +1,169 @@
|
|
|
1
|
+
import * as path from 'path';
|
|
2
|
+
import * as fs from 'fs';
|
|
3
|
+
import ora from 'ora';
|
|
4
|
+
import { ComfyAPIClient } from '../services/api.js';
|
|
5
|
+
import { success, error } from '../utils/logger.js';
|
|
6
|
+
import { outputSuccess, outputError, isPrettyMode } from '../utils/output.js';
|
|
7
|
+
import { addEnhancedHelp } from '../utils/helpText.js';
|
|
8
|
+
export function registerDownloadCommand(program) {
|
|
9
|
+
const cmd = program
|
|
10
|
+
.command('download')
|
|
11
|
+
.description('下载生成的图像或视频 - 根据 Prompt ID 下载 ComfyUI 生成的文件')
|
|
12
|
+
.argument('<prompt_id>', 'Prompt ID(从 generate/edit/video 命令获取)')
|
|
13
|
+
.option('-o, --output <path>', '输出文件路径或目录')
|
|
14
|
+
.action(async (promptId, options) => {
|
|
15
|
+
const pretty = isPrettyMode(options);
|
|
16
|
+
try {
|
|
17
|
+
const client = new ComfyAPIClient();
|
|
18
|
+
// 获取输出文件列表
|
|
19
|
+
let files;
|
|
20
|
+
if (pretty) {
|
|
21
|
+
const spinner = ora('正在获取输出文件...').start();
|
|
22
|
+
files = await client.getOutputFiles(promptId);
|
|
23
|
+
if (files.length === 0) {
|
|
24
|
+
spinner.fail('未找到输出文件');
|
|
25
|
+
outputError('NO_OUTPUT_FILES', `Prompt ID: ${promptId} 没有生成任何文件`, options);
|
|
26
|
+
}
|
|
27
|
+
spinner.succeed(`找到 ${files.length} 个文件`);
|
|
28
|
+
}
|
|
29
|
+
else {
|
|
30
|
+
files = await client.getOutputFiles(promptId);
|
|
31
|
+
if (files.length === 0) {
|
|
32
|
+
outputError('NO_OUTPUT_FILES', `Prompt ID: ${promptId} 没有生成任何文件`, options);
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
const downloadedFiles = [];
|
|
36
|
+
// 下载所有文件
|
|
37
|
+
for (let i = 0; i < files.length; i++) {
|
|
38
|
+
const file = files[i];
|
|
39
|
+
if (pretty) {
|
|
40
|
+
const downloadSpinner = ora(`正在下载 (${i + 1}/${files.length}): ${file.filename}`).start();
|
|
41
|
+
try {
|
|
42
|
+
// 下载文件
|
|
43
|
+
const buffer = await client.downloadImage(file.filename, file.subfolder, file.type);
|
|
44
|
+
// 确定输出路径
|
|
45
|
+
let outputPath;
|
|
46
|
+
if (options.output) {
|
|
47
|
+
// 如果指定了输出路径
|
|
48
|
+
if (files.length === 1) {
|
|
49
|
+
// 只有一个文件,直接使用指定路径
|
|
50
|
+
outputPath = options.output;
|
|
51
|
+
}
|
|
52
|
+
else {
|
|
53
|
+
// 多个文件,将指定路径作为目录
|
|
54
|
+
const dir = options.output;
|
|
55
|
+
if (!fs.existsSync(dir)) {
|
|
56
|
+
fs.mkdirSync(dir, { recursive: true });
|
|
57
|
+
}
|
|
58
|
+
outputPath = path.join(dir, file.filename);
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
else {
|
|
62
|
+
// 默认保存到 ./output 目录
|
|
63
|
+
const outputDir = './output';
|
|
64
|
+
if (!fs.existsSync(outputDir)) {
|
|
65
|
+
fs.mkdirSync(outputDir, { recursive: true });
|
|
66
|
+
}
|
|
67
|
+
outputPath = path.join(outputDir, file.filename);
|
|
68
|
+
}
|
|
69
|
+
// 确保输出目录存在
|
|
70
|
+
const outputDir = path.dirname(outputPath);
|
|
71
|
+
if (!fs.existsSync(outputDir)) {
|
|
72
|
+
fs.mkdirSync(outputDir, { recursive: true });
|
|
73
|
+
}
|
|
74
|
+
// 保存文件
|
|
75
|
+
fs.writeFileSync(outputPath, buffer);
|
|
76
|
+
downloadSpinner.succeed(`已下载: ${outputPath}`);
|
|
77
|
+
downloadedFiles.push(outputPath);
|
|
78
|
+
}
|
|
79
|
+
catch (err) {
|
|
80
|
+
downloadSpinner.fail(`下载失败: ${file.filename}`);
|
|
81
|
+
error(`错误: ${err}`);
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
else {
|
|
85
|
+
// JSON 模式:直接下载,不显示进度
|
|
86
|
+
const buffer = await client.downloadImage(file.filename, file.subfolder, file.type);
|
|
87
|
+
let outputPath;
|
|
88
|
+
if (options.output) {
|
|
89
|
+
if (files.length === 1) {
|
|
90
|
+
outputPath = options.output;
|
|
91
|
+
}
|
|
92
|
+
else {
|
|
93
|
+
const dir = options.output;
|
|
94
|
+
if (!fs.existsSync(dir)) {
|
|
95
|
+
fs.mkdirSync(dir, { recursive: true });
|
|
96
|
+
}
|
|
97
|
+
outputPath = path.join(dir, file.filename);
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
else {
|
|
101
|
+
const outputDir = './output';
|
|
102
|
+
if (!fs.existsSync(outputDir)) {
|
|
103
|
+
fs.mkdirSync(outputDir, { recursive: true });
|
|
104
|
+
}
|
|
105
|
+
outputPath = path.join(outputDir, file.filename);
|
|
106
|
+
}
|
|
107
|
+
const outputDir = path.dirname(outputPath);
|
|
108
|
+
if (!fs.existsSync(outputDir)) {
|
|
109
|
+
fs.mkdirSync(outputDir, { recursive: true });
|
|
110
|
+
}
|
|
111
|
+
fs.writeFileSync(outputPath, buffer);
|
|
112
|
+
downloadedFiles.push(outputPath);
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
if (pretty) {
|
|
116
|
+
success(`完成!共下载 ${files.length} 个文件`);
|
|
117
|
+
}
|
|
118
|
+
else {
|
|
119
|
+
outputSuccess({
|
|
120
|
+
prompt_id: promptId,
|
|
121
|
+
files: downloadedFiles,
|
|
122
|
+
total: downloadedFiles.length,
|
|
123
|
+
});
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
catch (err) {
|
|
127
|
+
outputError('DOWNLOAD_FAILED', `下载失败: ${err}`, options);
|
|
128
|
+
}
|
|
129
|
+
});
|
|
130
|
+
// Add enhanced help text
|
|
131
|
+
addEnhancedHelp(cmd, {
|
|
132
|
+
examples: [
|
|
133
|
+
'# 下载生成的图像(默认 JSON 输出)',
|
|
134
|
+
'$ comfy download abc-123-def',
|
|
135
|
+
'',
|
|
136
|
+
'# 指定输出路径',
|
|
137
|
+
'$ comfy download abc-123-def -o my-image.png',
|
|
138
|
+
'',
|
|
139
|
+
'# 指定输出目录(多文件时)',
|
|
140
|
+
'$ comfy download abc-123-def -o ./images/',
|
|
141
|
+
'',
|
|
142
|
+
'# Pretty 模式(显示下载进度)',
|
|
143
|
+
'$ comfy download abc-123-def --pretty',
|
|
144
|
+
],
|
|
145
|
+
outputJson: `{
|
|
146
|
+
"success": true,
|
|
147
|
+
"data": {
|
|
148
|
+
"prompt_id": "abc-123-def",
|
|
149
|
+
"files": [
|
|
150
|
+
"output/comfy-cli_00001_.png"
|
|
151
|
+
],
|
|
152
|
+
"total": 1
|
|
153
|
+
}
|
|
154
|
+
}`,
|
|
155
|
+
relatedCommands: [
|
|
156
|
+
{ command: 'generate <prompt>', description: '生成图像获取 prompt_id' },
|
|
157
|
+
{ command: 'workflow list', description: '查看历史 prompt_id' },
|
|
158
|
+
{ command: 'workflow get <id>', description: '查看工作流详情' },
|
|
159
|
+
],
|
|
160
|
+
notes: [
|
|
161
|
+
'Prompt ID 从 generate/edit/video 命令的输出中获取',
|
|
162
|
+
'默认保存到 ./output 目录',
|
|
163
|
+
'单文件时 -o 指定文件路径,多文件时 -o 指定目录',
|
|
164
|
+
'文件名保持 ComfyUI 原始命名',
|
|
165
|
+
'如果找不到文件,可能任务还在执行中',
|
|
166
|
+
],
|
|
167
|
+
});
|
|
168
|
+
}
|
|
169
|
+
//# sourceMappingURL=download.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"download.js","sourceRoot":"","sources":["../../src/commands/download.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,GAAG,MAAM,KAAK,CAAC;AACtB,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAQ,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAC9E,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,MAAM,UAAU,uBAAuB,CAAC,OAAgB;IACtD,MAAM,GAAG,GAAG,OAAO;SAChB,OAAO,CAAC,UAAU,CAAC;SACnB,WAAW,CAAC,4CAA4C,CAAC;SACzD,QAAQ,CAAC,aAAa,EAAE,uCAAuC,CAAC;SAChE,MAAM,CAAC,qBAAqB,EAAE,WAAW,CAAC;SAC1C,MAAM,CAAC,KAAK,EAAE,QAAgB,EAAE,OAAO,EAAE,EAAE;QAC1C,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;QAErC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;YAEpC,WAAW;YACX,IAAI,KAAK,CAAC;YACV,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,OAAO,GAAG,GAAG,CAAC,aAAa,CAAC,CAAC,KAAK,EAAE,CAAC;gBAC3C,KAAK,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;gBAE9C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACvB,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBACxB,WAAW,CAAC,iBAAiB,EAAE,cAAc,QAAQ,WAAW,EAAE,OAAO,CAAC,CAAC;gBAC7E,CAAC;gBAED,OAAO,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,MAAM,MAAM,CAAC,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACN,KAAK,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;gBAE9C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACvB,WAAW,CAAC,iBAAiB,EAAE,cAAc,QAAQ,WAAW,EAAE,OAAO,CAAC,CAAC;gBAC7E,CAAC;YACH,CAAC;YAED,MAAM,eAAe,GAAa,EAAE,CAAC;YAErC,SAAS;YACT,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;gBAEtB,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,eAAe,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,MAAM,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;oBAEzF,IAAI,CAAC;wBACH,OAAO;wBACP,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;wBAEpF,SAAS;wBACT,IAAI,UAAkB,CAAC;wBACvB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;4BACnB,YAAY;4BACZ,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gCACvB,kBAAkB;gCAClB,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC;4BAC9B,CAAC;iCAAM,CAAC;gCACN,iBAAiB;gCACjB,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;gCAC3B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oCACxB,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gCACzC,CAAC;gCACD,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;4BAC7C,CAAC;wBACH,CAAC;6BAAM,CAAC;4BACN,oBAAoB;4BACpB,MAAM,SAAS,GAAG,UAAU,CAAC;4BAC7B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gCAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;4BAC/C,CAAC;4BACD,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;wBACnD,CAAC;wBAED,WAAW;wBACX,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;wBAC3C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;4BAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;wBAC/C,CAAC;wBAED,OAAO;wBACP,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;wBAErC,eAAe,CAAC,OAAO,CAAC,QAAQ,UAAU,EAAE,CAAC,CAAC;wBAC9C,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;oBACnC,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,eAAe,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;wBAC/C,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;oBACtB,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,qBAAqB;oBACrB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;oBAEpF,IAAI,UAAkB,CAAC;oBACvB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;wBACnB,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;4BACvB,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC;wBAC9B,CAAC;6BAAM,CAAC;4BACN,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;4BAC3B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gCACxB,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;4BACzC,CAAC;4BACD,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;wBAC7C,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,MAAM,SAAS,GAAG,UAAU,CAAC;wBAC7B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;4BAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;wBAC/C,CAAC;wBACD,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACnD,CAAC;oBAED,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;oBAC3C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;wBAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC/C,CAAC;oBAED,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;oBACrC,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;YAED,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,CAAC,UAAU,KAAK,CAAC,MAAM,MAAM,CAAC,CAAC;YACxC,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC;oBACZ,SAAS,EAAE,QAAQ;oBACnB,KAAK,EAAE,eAAe;oBACtB,KAAK,EAAE,eAAe,CAAC,MAAM;iBAC9B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,WAAW,CAAC,iBAAiB,EAAE,SAAS,GAAG,EAAE,EAAE,OAAO,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,yBAAyB;IACzB,eAAe,CAAC,GAAG,EAAE;QACnB,QAAQ,EAAE;YACR,uBAAuB;YACvB,8BAA8B;YAC9B,EAAE;YACF,UAAU;YACV,8CAA8C;YAC9C,EAAE;YACF,gBAAgB;YAChB,2CAA2C;YAC3C,EAAE;YACF,qBAAqB;YACrB,uCAAuC;SACxC;QACD,UAAU,EAAE;;;;;;;;;EASd;QACE,eAAe,EAAE;YACf,EAAE,OAAO,EAAE,mBAAmB,EAAE,WAAW,EAAE,kBAAkB,EAAE;YACjE,EAAE,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,gBAAgB,EAAE;YAC3D,EAAE,OAAO,EAAE,mBAAmB,EAAE,WAAW,EAAE,SAAS,EAAE;SACzD;QACD,KAAK,EAAE;YACL,0CAA0C;YAC1C,mBAAmB;YACnB,6BAA6B;YAC7B,oBAAoB;YACpB,mBAAmB;SACpB;KACF,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"edit.d.ts","sourceRoot":"","sources":["../../src/commands/edit.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAWpC,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,QAyJnD"}
|