ima2-gen 1.1.7 → 1.1.9
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/README.md +56 -27
- package/bin/commands/annotate.js +137 -0
- package/bin/commands/annotate.ts +118 -0
- package/bin/commands/cancel.js +37 -33
- package/bin/commands/cancel.ts +45 -0
- package/bin/commands/canvas-versions.js +91 -0
- package/bin/commands/canvas-versions.ts +80 -0
- package/bin/commands/cardnews.js +293 -0
- package/bin/commands/cardnews.ts +248 -0
- package/bin/commands/comfy.js +63 -0
- package/bin/commands/comfy.ts +54 -0
- package/bin/commands/config.js +270 -0
- package/bin/commands/config.ts +265 -0
- package/bin/commands/edit.js +97 -72
- package/bin/commands/edit.ts +116 -0
- package/bin/commands/gen.js +140 -118
- package/bin/commands/gen.ts +176 -0
- package/bin/commands/history.js +164 -0
- package/bin/commands/history.ts +145 -0
- package/bin/commands/ls.js +60 -42
- package/bin/commands/ls.ts +60 -0
- package/bin/commands/metadata.js +45 -0
- package/bin/commands/metadata.ts +36 -0
- package/bin/commands/multimode.js +159 -0
- package/bin/commands/multimode.ts +146 -0
- package/bin/commands/node.js +176 -0
- package/bin/commands/node.ts +157 -0
- package/bin/commands/observability.js +201 -0
- package/bin/commands/observability.ts +176 -0
- package/bin/commands/ping.js +26 -20
- package/bin/commands/ping.ts +29 -0
- package/bin/commands/prompt.js +506 -0
- package/bin/commands/prompt.ts +421 -0
- package/bin/commands/ps.js +78 -71
- package/bin/commands/ps.ts +78 -0
- package/bin/commands/session.js +308 -0
- package/bin/commands/session.ts +265 -0
- package/bin/commands/show.js +75 -40
- package/bin/commands/show.ts +69 -0
- package/bin/ima2.js +324 -310
- package/bin/ima2.ts +444 -0
- package/bin/lib/args.js +75 -66
- package/bin/lib/args.ts +73 -0
- package/bin/lib/browser-id.js +15 -0
- package/bin/lib/browser-id.ts +16 -0
- package/bin/lib/client.js +91 -83
- package/bin/lib/client.ts +109 -0
- package/bin/lib/error-hints.js +14 -17
- package/bin/lib/error-hints.ts +23 -0
- package/bin/lib/files.js +26 -28
- package/bin/lib/files.ts +39 -0
- package/bin/lib/output.js +44 -42
- package/bin/lib/output.ts +58 -0
- package/bin/lib/platform.js +60 -56
- package/bin/lib/platform.ts +97 -0
- package/bin/lib/sse.js +73 -0
- package/bin/lib/sse.ts +73 -0
- package/bin/lib/star-prompt.js +69 -76
- package/bin/lib/star-prompt.ts +97 -0
- package/bin/lib/storage-doctor.js +34 -35
- package/bin/lib/storage-doctor.ts +38 -0
- package/config.js +147 -190
- package/config.ts +331 -0
- package/docs/API.md +48 -8
- package/docs/CLI.md +190 -0
- package/docs/FAQ.ko.md +5 -5
- package/docs/FAQ.md +5 -5
- package/docs/README.ja.md +71 -25
- package/docs/README.ko.md +61 -24
- package/docs/README.zh-CN.md +73 -27
- package/lib/assetLifecycle.js +130 -130
- package/lib/assetLifecycle.ts +142 -0
- package/lib/canvasVersionStore.js +135 -153
- package/lib/canvasVersionStore.ts +181 -0
- package/lib/cardNewsGenerator.js +127 -142
- package/lib/cardNewsGenerator.ts +162 -0
- package/lib/cardNewsJobStore.js +78 -84
- package/lib/cardNewsJobStore.ts +107 -0
- package/lib/cardNewsManifestStore.js +88 -93
- package/lib/cardNewsManifestStore.ts +112 -0
- package/lib/cardNewsPlanner.js +157 -152
- package/lib/cardNewsPlanner.ts +180 -0
- package/lib/cardNewsPlannerClient.js +101 -98
- package/lib/cardNewsPlannerClient.ts +114 -0
- package/lib/cardNewsPlannerPrompt.js +56 -56
- package/lib/cardNewsPlannerPrompt.ts +60 -0
- package/lib/cardNewsPlannerSchema.js +231 -223
- package/lib/cardNewsPlannerSchema.ts +259 -0
- package/lib/cardNewsRoleTemplateStore.js +39 -41
- package/lib/cardNewsRoleTemplateStore.ts +47 -0
- package/lib/cardNewsTemplateStore.js +171 -175
- package/lib/cardNewsTemplateStore.ts +210 -0
- package/lib/codexDetect.js +44 -47
- package/lib/codexDetect.ts +69 -0
- package/lib/comfyBridge.js +164 -184
- package/lib/comfyBridge.ts +214 -0
- package/lib/db.js +41 -51
- package/lib/db.ts +166 -0
- package/lib/errorClassify.js +62 -78
- package/lib/errorClassify.ts +100 -0
- package/lib/generationErrors.js +140 -103
- package/lib/generationErrors.ts +125 -0
- package/lib/historyList.js +149 -147
- package/lib/historyList.ts +164 -0
- package/lib/imageMetadata.js +86 -89
- package/lib/imageMetadata.ts +111 -0
- package/lib/imageMetadataStore.js +46 -51
- package/lib/imageMetadataStore.ts +67 -0
- package/lib/imageModels.js +38 -45
- package/lib/imageModels.ts +52 -0
- package/lib/inflight.js +131 -150
- package/lib/inflight.ts +204 -0
- package/lib/localImportStore.js +105 -0
- package/lib/localImportStore.ts +111 -0
- package/lib/logger.js +105 -112
- package/lib/logger.ts +150 -0
- package/lib/nodeStore.js +65 -64
- package/lib/nodeStore.ts +81 -0
- package/lib/oauthLauncher.js +61 -59
- package/lib/oauthLauncher.ts +64 -0
- package/lib/oauthNormalize.js +15 -19
- package/lib/oauthNormalize.ts +30 -0
- package/lib/oauthProxy.js +834 -832
- package/lib/oauthProxy.ts +995 -0
- package/lib/openDirectory.js +41 -40
- package/lib/openDirectory.ts +45 -0
- package/lib/pngInfo.js +18 -20
- package/lib/pngInfo.ts +26 -0
- package/lib/promptImport/curatedSources.js +135 -0
- package/lib/promptImport/curatedSources.ts +139 -0
- package/lib/promptImport/discoveryRegistry.js +218 -0
- package/lib/promptImport/discoveryRegistry.ts +236 -0
- package/lib/promptImport/errors.js +10 -10
- package/lib/promptImport/errors.ts +18 -0
- package/lib/promptImport/githubDiscovery.js +238 -0
- package/lib/promptImport/githubDiscovery.ts +248 -0
- package/lib/promptImport/githubFolder.js +302 -0
- package/lib/promptImport/githubFolder.ts +308 -0
- package/lib/promptImport/githubSource.js +194 -171
- package/lib/promptImport/githubSource.ts +239 -0
- package/lib/promptImport/gptImageHints.js +61 -0
- package/lib/promptImport/gptImageHints.ts +68 -0
- package/lib/promptImport/parsePromptCandidates.js +110 -112
- package/lib/promptImport/parsePromptCandidates.ts +153 -0
- package/lib/promptImport/promptIndex.js +230 -0
- package/lib/promptImport/promptIndex.ts +248 -0
- package/lib/promptImport/rankPromptCandidates.js +52 -0
- package/lib/promptImport/rankPromptCandidates.ts +49 -0
- package/lib/providerOptions.js +31 -0
- package/lib/providerOptions.ts +41 -0
- package/lib/referenceImageCompress.js +51 -62
- package/lib/referenceImageCompress.ts +75 -0
- package/lib/refs.js +93 -81
- package/lib/refs.ts +117 -0
- package/lib/requestLogger.js +32 -38
- package/lib/requestLogger.ts +48 -0
- package/lib/responsesImageAdapter.js +351 -0
- package/lib/responsesImageAdapter.ts +352 -0
- package/lib/runtimePorts.js +71 -73
- package/lib/runtimePorts.ts +93 -0
- package/lib/sessionStore.js +179 -230
- package/lib/sessionStore.ts +272 -0
- package/lib/storageMigration.js +247 -245
- package/lib/storageMigration.ts +284 -0
- package/lib/styleSheet.js +86 -90
- package/lib/styleSheet.ts +128 -0
- package/lib/systemTrash.js +18 -0
- package/lib/systemTrash.ts +20 -0
- package/package.json +26 -10
- package/routes/annotations.js +76 -79
- package/routes/annotations.ts +95 -0
- package/routes/canvasVersions.js +50 -54
- package/routes/canvasVersions.ts +64 -0
- package/routes/cardNews.js +158 -171
- package/routes/cardNews.ts +183 -0
- package/routes/comfy.js +23 -31
- package/routes/comfy.ts +39 -0
- package/routes/edit.js +183 -214
- package/routes/edit.ts +230 -0
- package/routes/generate.js +269 -291
- package/routes/generate.ts +309 -0
- package/routes/health.js +102 -107
- package/routes/health.ts +114 -0
- package/routes/history.js +136 -144
- package/routes/history.ts +153 -0
- package/routes/imageImport.js +33 -0
- package/routes/imageImport.ts +33 -0
- package/routes/index.js +18 -16
- package/routes/index.ts +35 -0
- package/routes/metadata.js +60 -64
- package/routes/metadata.ts +71 -0
- package/routes/multimode.js +228 -263
- package/routes/multimode.ts +280 -0
- package/routes/nodes.js +378 -424
- package/routes/nodes.ts +455 -0
- package/routes/promptImport.js +291 -152
- package/routes/promptImport.ts +354 -0
- package/routes/prompts.js +333 -360
- package/routes/prompts.ts +379 -0
- package/routes/sessions.js +277 -285
- package/routes/sessions.ts +292 -0
- package/routes/storage.js +29 -31
- package/routes/storage.ts +39 -0
- package/server.js +189 -196
- package/server.ts +235 -0
- package/ui/dist/.vite/manifest.json +101 -0
- package/ui/dist/assets/CardNewsWorkspace-BJOCey7Z.js +2 -0
- package/ui/dist/assets/NodeCanvas-BZV40eAE.css +1 -0
- package/ui/dist/assets/NodeCanvas-C3dzYNsk.js +7 -0
- package/ui/dist/assets/PromptImportDialog-Dqu1VpUh.js +2 -0
- package/ui/dist/assets/PromptImportDiscoverySection-Dg8T9X0L.js +1 -0
- package/ui/dist/assets/PromptImportFolderSection-DBaqsFO4.js +1 -0
- package/ui/dist/assets/PromptLibraryPanel-p5QqR97M.js +2 -0
- package/ui/dist/assets/SettingsWorkspace-B5bSAZ6u.js +1 -0
- package/ui/dist/assets/index-C9cXwiWE.js +25 -0
- package/ui/dist/assets/index-CGMIkZXn.css +1 -0
- package/ui/dist/assets/index-Cvld7dUZ.js +1 -0
- package/ui/dist/index.html +6 -3
- package/assets/screenshot.png +0 -0
- package/assets/screenshots/classic-generate-light.png +0 -0
- package/assets/screenshots/node-graph-branching.png +0 -0
- package/assets/screenshots/settings-oauth-generation.png +0 -0
- package/assets/screenshots/settings-workspace.png +0 -0
- package/assets/screenshots/style-sheet-editor.png +0 -0
- package/integrations/comfyui/ima2_gen_bridge/__pycache__/__init__.cpython-313.pyc +0 -0
- package/integrations/comfyui/ima2_gen_bridge/__pycache__/nodes.cpython-313.pyc +0 -0
- package/ui/dist/assets/index-DARPdT4Q.css +0 -1
- package/ui/dist/assets/index-ht80GMq4.js +0 -31
- package/ui/dist/assets/index-ht80GMq4.js.map +0 -1
|
@@ -0,0 +1,80 @@
|
|
|
1
|
+
import { readFile } from "fs/promises";
|
|
2
|
+
import { parseArgs } from "../lib/args.js";
|
|
3
|
+
import { resolveServer, request } from "../lib/client.js";
|
|
4
|
+
import { out, die, color, json, exitCodeForError } from "../lib/output.js";
|
|
5
|
+
|
|
6
|
+
const HELP = `
|
|
7
|
+
ima2 canvas-versions <subcommand> [options]
|
|
8
|
+
|
|
9
|
+
Subcommands:
|
|
10
|
+
save <imagefile> [--source <filename>] [--prompt <text>]
|
|
11
|
+
update <filename> <imagefile> [--source <filename>] [--prompt <text>]
|
|
12
|
+
|
|
13
|
+
Notes: server only exposes POST /api/canvas-versions (collection) and
|
|
14
|
+
PUT /api/canvas-versions/:filename. No GET, no DELETE.
|
|
15
|
+
`;
|
|
16
|
+
|
|
17
|
+
const FLAGS = {
|
|
18
|
+
json: { type: "boolean" },
|
|
19
|
+
server: { type: "string" },
|
|
20
|
+
source: { type: "string" },
|
|
21
|
+
prompt: { type: "string" },
|
|
22
|
+
help: { short: "h", type: "boolean" },
|
|
23
|
+
};
|
|
24
|
+
|
|
25
|
+
async function getServer(args) {
|
|
26
|
+
try { return await resolveServer({ serverFlag: args.server }); }
|
|
27
|
+
catch (e: any) { die(exitCodeForError(e), e.message); throw e; }
|
|
28
|
+
}
|
|
29
|
+
|
|
30
|
+
function buildHeaders(args) {
|
|
31
|
+
const h: Record<string, string> = { "Content-Type": "image/png" };
|
|
32
|
+
if (args.source) h["X-Ima2-Canvas-Source-Filename"] = args.source;
|
|
33
|
+
if (args.prompt) h["X-Ima2-Canvas-Prompt"] = args.prompt;
|
|
34
|
+
return h;
|
|
35
|
+
}
|
|
36
|
+
|
|
37
|
+
async function saveSub(argv) {
|
|
38
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
39
|
+
const file = args.positional[0];
|
|
40
|
+
if (!file) die(2, "image file required");
|
|
41
|
+
const buf = await readFile(file);
|
|
42
|
+
const server = await getServer(args);
|
|
43
|
+
const resp = await request(server.base, "/api/canvas-versions", {
|
|
44
|
+
method: "POST",
|
|
45
|
+
body: buf,
|
|
46
|
+
raw: true,
|
|
47
|
+
headers: buildHeaders(args),
|
|
48
|
+
}).catch((e) => die(exitCodeForError(e), `${e.message}${e.code ? ` (${e.code})` : ""}`));
|
|
49
|
+
if (args.json) { json(resp); return; }
|
|
50
|
+
out(color.green("✓ saved canvas version"));
|
|
51
|
+
}
|
|
52
|
+
|
|
53
|
+
async function updateSub(argv) {
|
|
54
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
55
|
+
const [filename, file] = args.positional;
|
|
56
|
+
if (!filename || !file) die(2, "usage: canvas-versions update <filename> <imagefile>");
|
|
57
|
+
const buf = await readFile(file);
|
|
58
|
+
const server = await getServer(args);
|
|
59
|
+
const resp = await request(server.base, `/api/canvas-versions/${encodeURIComponent(filename)}`, {
|
|
60
|
+
method: "PUT",
|
|
61
|
+
body: buf,
|
|
62
|
+
raw: true,
|
|
63
|
+
headers: buildHeaders(args),
|
|
64
|
+
}).catch((e) => die(exitCodeForError(e), `${e.message}${e.code ? ` (${e.code})` : ""}`));
|
|
65
|
+
if (args.json) { json(resp); return; }
|
|
66
|
+
out(color.green("✓ updated"));
|
|
67
|
+
}
|
|
68
|
+
|
|
69
|
+
const SUB: Record<string, (argv: any[]) => Promise<void>> = {
|
|
70
|
+
save: saveSub,
|
|
71
|
+
update: updateSub,
|
|
72
|
+
};
|
|
73
|
+
|
|
74
|
+
export default async function canvasVersionsCmd(argv) {
|
|
75
|
+
const sub = argv[0];
|
|
76
|
+
if (!sub || sub === "--help" || sub === "-h") { out(HELP); return; }
|
|
77
|
+
const handler = SUB[sub];
|
|
78
|
+
if (!handler) die(2, `unknown subcommand: ${sub}\n${HELP}`);
|
|
79
|
+
return handler(argv.slice(1));
|
|
80
|
+
}
|
|
@@ -0,0 +1,293 @@
|
|
|
1
|
+
import { parseArgs } from "../lib/args.js";
|
|
2
|
+
import { resolveServer, request } from "../lib/client.js";
|
|
3
|
+
import { out, die, color, json, exitCodeForError } from "../lib/output.js";
|
|
4
|
+
import { config as runtimeConfig } from "../../config.js";
|
|
5
|
+
const HELP = `
|
|
6
|
+
ima2 cardnews <subcommand> [options]
|
|
7
|
+
|
|
8
|
+
Subcommands:
|
|
9
|
+
templates List image-templates + role-templates
|
|
10
|
+
template preview <templateId> Preview an image template
|
|
11
|
+
sets List card news sets
|
|
12
|
+
set show <setId> Show a set
|
|
13
|
+
set manifest <setId> Show a set manifest
|
|
14
|
+
draft [--data <json>] Create a draft
|
|
15
|
+
generate [--data <json>] Run generation
|
|
16
|
+
job create [--data <json>] Create and start a job
|
|
17
|
+
job show <jobId> Show a job
|
|
18
|
+
job retry <jobId> [--cards <id,...>] Retry a job (optionally specific cards)
|
|
19
|
+
card regenerate <cardId> [--data <json>] Regenerate a single card
|
|
20
|
+
export [--data <json>] Export a bundle
|
|
21
|
+
|
|
22
|
+
Note: Card News must be enabled (IMA2_CARD_NEWS=1 or features.cardNews: true in config.json).
|
|
23
|
+
|
|
24
|
+
Options:
|
|
25
|
+
--data <json> JSON body to send (for draft/generate/job create/card regenerate/export)
|
|
26
|
+
--cards <ids> Comma-separated card IDs for job retry
|
|
27
|
+
--json Output raw JSON
|
|
28
|
+
--server <url> Override server URL
|
|
29
|
+
`;
|
|
30
|
+
const FLAGS = {
|
|
31
|
+
json: { type: "boolean" },
|
|
32
|
+
server: { type: "string" },
|
|
33
|
+
data: { type: "string" },
|
|
34
|
+
cards: { type: "string" },
|
|
35
|
+
help: { short: "h", type: "boolean" },
|
|
36
|
+
};
|
|
37
|
+
function ensureCardNewsEnabled() {
|
|
38
|
+
if (!runtimeConfig.features?.cardNews) {
|
|
39
|
+
die(2, "Card News feature is disabled. Set IMA2_CARD_NEWS=1 (or features.cardNews: true in config.json) and restart the server.");
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
async function getServer(args) {
|
|
43
|
+
try {
|
|
44
|
+
return await resolveServer({ serverFlag: args.server });
|
|
45
|
+
}
|
|
46
|
+
catch (e) {
|
|
47
|
+
die(exitCodeForError(e), e.message);
|
|
48
|
+
throw e;
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
function parseData(raw) {
|
|
52
|
+
if (!raw)
|
|
53
|
+
return {};
|
|
54
|
+
try {
|
|
55
|
+
return JSON.parse(raw);
|
|
56
|
+
}
|
|
57
|
+
catch {
|
|
58
|
+
die(2, `--data must be valid JSON`);
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
async function templatesSub(argv) {
|
|
62
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
63
|
+
if (args.help) {
|
|
64
|
+
out(HELP);
|
|
65
|
+
return;
|
|
66
|
+
}
|
|
67
|
+
ensureCardNewsEnabled();
|
|
68
|
+
const server = await getServer(args);
|
|
69
|
+
const [imgTemplates, roleTemplates] = await Promise.all([
|
|
70
|
+
request(server.base, "/api/cardnews/image-templates")
|
|
71
|
+
.catch((e) => die(exitCodeForError(e), e.message)),
|
|
72
|
+
request(server.base, "/api/cardnews/role-templates")
|
|
73
|
+
.catch((e) => die(exitCodeForError(e), e.message)),
|
|
74
|
+
]);
|
|
75
|
+
if (args.json) {
|
|
76
|
+
json({ imageTemplates: imgTemplates, roleTemplates });
|
|
77
|
+
return;
|
|
78
|
+
}
|
|
79
|
+
out(color.bold("Image templates:"));
|
|
80
|
+
out(JSON.stringify(imgTemplates, null, 2));
|
|
81
|
+
out(color.bold("Role templates:"));
|
|
82
|
+
out(JSON.stringify(roleTemplates, null, 2));
|
|
83
|
+
}
|
|
84
|
+
async function templatePreviewSub(argv) {
|
|
85
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
86
|
+
const templateId = args.positional[0];
|
|
87
|
+
if (!templateId)
|
|
88
|
+
die(2, "templateId required");
|
|
89
|
+
ensureCardNewsEnabled();
|
|
90
|
+
const server = await getServer(args);
|
|
91
|
+
const resp = await request(server.base, `/api/cardnews/image-templates/${encodeURIComponent(templateId)}/preview`)
|
|
92
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
93
|
+
if (args.json) {
|
|
94
|
+
json(resp);
|
|
95
|
+
return;
|
|
96
|
+
}
|
|
97
|
+
out(JSON.stringify(resp, null, 2));
|
|
98
|
+
}
|
|
99
|
+
async function setsSub(argv) {
|
|
100
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
101
|
+
ensureCardNewsEnabled();
|
|
102
|
+
const server = await getServer(args);
|
|
103
|
+
const resp = await request(server.base, "/api/cardnews/sets")
|
|
104
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
105
|
+
if (args.json) {
|
|
106
|
+
json(resp);
|
|
107
|
+
return;
|
|
108
|
+
}
|
|
109
|
+
out(JSON.stringify(resp, null, 2));
|
|
110
|
+
}
|
|
111
|
+
async function setShowSub(argv) {
|
|
112
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
113
|
+
const setId = args.positional[0];
|
|
114
|
+
if (!setId)
|
|
115
|
+
die(2, "setId required");
|
|
116
|
+
ensureCardNewsEnabled();
|
|
117
|
+
const server = await getServer(args);
|
|
118
|
+
const resp = await request(server.base, `/api/cardnews/sets/${encodeURIComponent(setId)}`)
|
|
119
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
120
|
+
if (args.json) {
|
|
121
|
+
json(resp);
|
|
122
|
+
return;
|
|
123
|
+
}
|
|
124
|
+
out(JSON.stringify(resp, null, 2));
|
|
125
|
+
}
|
|
126
|
+
async function setManifestSub(argv) {
|
|
127
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
128
|
+
const setId = args.positional[0];
|
|
129
|
+
if (!setId)
|
|
130
|
+
die(2, "setId required");
|
|
131
|
+
ensureCardNewsEnabled();
|
|
132
|
+
const server = await getServer(args);
|
|
133
|
+
const resp = await request(server.base, `/api/cardnews/sets/${encodeURIComponent(setId)}/manifest`)
|
|
134
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
135
|
+
if (args.json) {
|
|
136
|
+
json(resp);
|
|
137
|
+
return;
|
|
138
|
+
}
|
|
139
|
+
out(JSON.stringify(resp, null, 2));
|
|
140
|
+
}
|
|
141
|
+
async function draftSub(argv) {
|
|
142
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
143
|
+
ensureCardNewsEnabled();
|
|
144
|
+
const server = await getServer(args);
|
|
145
|
+
const body = parseData(args.data);
|
|
146
|
+
const resp = await request(server.base, "/api/cardnews/draft", { method: "POST", body })
|
|
147
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
148
|
+
if (args.json) {
|
|
149
|
+
json(resp);
|
|
150
|
+
return;
|
|
151
|
+
}
|
|
152
|
+
out(JSON.stringify(resp, null, 2));
|
|
153
|
+
}
|
|
154
|
+
async function generateSub(argv) {
|
|
155
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
156
|
+
ensureCardNewsEnabled();
|
|
157
|
+
const server = await getServer(args);
|
|
158
|
+
const body = parseData(args.data);
|
|
159
|
+
const resp = await request(server.base, "/api/cardnews/generate", { method: "POST", body })
|
|
160
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
161
|
+
if (args.json) {
|
|
162
|
+
json(resp);
|
|
163
|
+
return;
|
|
164
|
+
}
|
|
165
|
+
out(JSON.stringify(resp, null, 2));
|
|
166
|
+
}
|
|
167
|
+
async function jobCreateSub(argv) {
|
|
168
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
169
|
+
ensureCardNewsEnabled();
|
|
170
|
+
const server = await getServer(args);
|
|
171
|
+
const body = parseData(args.data);
|
|
172
|
+
const resp = await request(server.base, "/api/cardnews/jobs", { method: "POST", body })
|
|
173
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
174
|
+
if (args.json) {
|
|
175
|
+
json(resp);
|
|
176
|
+
return;
|
|
177
|
+
}
|
|
178
|
+
out(JSON.stringify(resp, null, 2));
|
|
179
|
+
}
|
|
180
|
+
async function jobShowSub(argv) {
|
|
181
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
182
|
+
const jobId = args.positional[0];
|
|
183
|
+
if (!jobId)
|
|
184
|
+
die(2, "jobId required");
|
|
185
|
+
ensureCardNewsEnabled();
|
|
186
|
+
const server = await getServer(args);
|
|
187
|
+
const resp = await request(server.base, `/api/cardnews/jobs/${encodeURIComponent(jobId)}`)
|
|
188
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
189
|
+
if (args.json) {
|
|
190
|
+
json(resp);
|
|
191
|
+
return;
|
|
192
|
+
}
|
|
193
|
+
out(JSON.stringify(resp, null, 2));
|
|
194
|
+
}
|
|
195
|
+
async function jobRetrySub(argv) {
|
|
196
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
197
|
+
const jobId = args.positional[0];
|
|
198
|
+
if (!jobId)
|
|
199
|
+
die(2, "jobId required");
|
|
200
|
+
ensureCardNewsEnabled();
|
|
201
|
+
const server = await getServer(args);
|
|
202
|
+
const body = {};
|
|
203
|
+
if (args.cards)
|
|
204
|
+
body.cardIds = args.cards.split(",").map((s) => s.trim()).filter(Boolean);
|
|
205
|
+
const resp = await request(server.base, `/api/cardnews/jobs/${encodeURIComponent(jobId)}/retry`, { method: "POST", body })
|
|
206
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
207
|
+
if (args.json) {
|
|
208
|
+
json(resp);
|
|
209
|
+
return;
|
|
210
|
+
}
|
|
211
|
+
out(color.green("✓ ") + "retry requested");
|
|
212
|
+
if (resp && typeof resp === "object" && Object.keys(resp).length > 0)
|
|
213
|
+
out(JSON.stringify(resp, null, 2));
|
|
214
|
+
}
|
|
215
|
+
async function cardRegenerateSub(argv) {
|
|
216
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
217
|
+
const cardId = args.positional[0];
|
|
218
|
+
if (!cardId)
|
|
219
|
+
die(2, "cardId required");
|
|
220
|
+
ensureCardNewsEnabled();
|
|
221
|
+
const server = await getServer(args);
|
|
222
|
+
const body = parseData(args.data);
|
|
223
|
+
const resp = await request(server.base, `/api/cardnews/cards/${encodeURIComponent(cardId)}/regenerate`, { method: "POST", body })
|
|
224
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
225
|
+
if (args.json) {
|
|
226
|
+
json(resp);
|
|
227
|
+
return;
|
|
228
|
+
}
|
|
229
|
+
out(color.green("✓ ") + "regeneration started");
|
|
230
|
+
if (resp && typeof resp === "object" && Object.keys(resp).length > 0)
|
|
231
|
+
out(JSON.stringify(resp, null, 2));
|
|
232
|
+
}
|
|
233
|
+
async function exportSub(argv) {
|
|
234
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
235
|
+
ensureCardNewsEnabled();
|
|
236
|
+
const server = await getServer(args);
|
|
237
|
+
const body = parseData(args.data);
|
|
238
|
+
const resp = await request(server.base, "/api/cardnews/export", { method: "POST", body })
|
|
239
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
240
|
+
if (args.json) {
|
|
241
|
+
json(resp);
|
|
242
|
+
return;
|
|
243
|
+
}
|
|
244
|
+
out(color.green("✓ ") + "export requested");
|
|
245
|
+
if (resp && typeof resp === "object" && Object.keys(resp).length > 0)
|
|
246
|
+
out(JSON.stringify(resp, null, 2));
|
|
247
|
+
}
|
|
248
|
+
export default async function cardnewsCmd(argv) {
|
|
249
|
+
const sub = argv[0];
|
|
250
|
+
if (!sub || sub === "--help" || sub === "-h") {
|
|
251
|
+
out(HELP);
|
|
252
|
+
return;
|
|
253
|
+
}
|
|
254
|
+
if (sub === "template") {
|
|
255
|
+
const s2 = argv[1];
|
|
256
|
+
if (s2 === "preview")
|
|
257
|
+
return templatePreviewSub(argv.slice(2));
|
|
258
|
+
die(2, `unknown subcommand: template ${s2 || ""}\n${HELP}`);
|
|
259
|
+
}
|
|
260
|
+
if (sub === "set") {
|
|
261
|
+
const s2 = argv[1];
|
|
262
|
+
const SET_SUB = { show: setShowSub, manifest: setManifestSub };
|
|
263
|
+
const handler = SET_SUB[s2];
|
|
264
|
+
if (!handler)
|
|
265
|
+
die(2, `unknown subcommand: set ${s2 || ""}\n${HELP}`);
|
|
266
|
+
return handler(argv.slice(2));
|
|
267
|
+
}
|
|
268
|
+
if (sub === "job") {
|
|
269
|
+
const s2 = argv[1];
|
|
270
|
+
const JOB_SUB = { create: jobCreateSub, show: jobShowSub, retry: jobRetrySub };
|
|
271
|
+
const handler = JOB_SUB[s2];
|
|
272
|
+
if (!handler)
|
|
273
|
+
die(2, `unknown subcommand: job ${s2 || ""}\n${HELP}`);
|
|
274
|
+
return handler(argv.slice(2));
|
|
275
|
+
}
|
|
276
|
+
if (sub === "card") {
|
|
277
|
+
const s2 = argv[1];
|
|
278
|
+
if (s2 === "regenerate")
|
|
279
|
+
return cardRegenerateSub(argv.slice(2));
|
|
280
|
+
die(2, `unknown subcommand: card ${s2 || ""}\n${HELP}`);
|
|
281
|
+
}
|
|
282
|
+
const FLAT = {
|
|
283
|
+
templates: templatesSub,
|
|
284
|
+
sets: setsSub,
|
|
285
|
+
draft: draftSub,
|
|
286
|
+
generate: generateSub,
|
|
287
|
+
export: exportSub,
|
|
288
|
+
};
|
|
289
|
+
const handler = FLAT[sub];
|
|
290
|
+
if (!handler)
|
|
291
|
+
die(2, `unknown subcommand: ${sub}\n${HELP}`);
|
|
292
|
+
return handler(argv.slice(1));
|
|
293
|
+
}
|
|
@@ -0,0 +1,248 @@
|
|
|
1
|
+
import { parseArgs } from "../lib/args.js";
|
|
2
|
+
import { resolveServer, request } from "../lib/client.js";
|
|
3
|
+
import { out, die, color, json, exitCodeForError } from "../lib/output.js";
|
|
4
|
+
import { config as runtimeConfig } from "../../config.js";
|
|
5
|
+
|
|
6
|
+
const HELP = `
|
|
7
|
+
ima2 cardnews <subcommand> [options]
|
|
8
|
+
|
|
9
|
+
Subcommands:
|
|
10
|
+
templates List image-templates + role-templates
|
|
11
|
+
template preview <templateId> Preview an image template
|
|
12
|
+
sets List card news sets
|
|
13
|
+
set show <setId> Show a set
|
|
14
|
+
set manifest <setId> Show a set manifest
|
|
15
|
+
draft [--data <json>] Create a draft
|
|
16
|
+
generate [--data <json>] Run generation
|
|
17
|
+
job create [--data <json>] Create and start a job
|
|
18
|
+
job show <jobId> Show a job
|
|
19
|
+
job retry <jobId> [--cards <id,...>] Retry a job (optionally specific cards)
|
|
20
|
+
card regenerate <cardId> [--data <json>] Regenerate a single card
|
|
21
|
+
export [--data <json>] Export a bundle
|
|
22
|
+
|
|
23
|
+
Note: Card News must be enabled (IMA2_CARD_NEWS=1 or features.cardNews: true in config.json).
|
|
24
|
+
|
|
25
|
+
Options:
|
|
26
|
+
--data <json> JSON body to send (for draft/generate/job create/card regenerate/export)
|
|
27
|
+
--cards <ids> Comma-separated card IDs for job retry
|
|
28
|
+
--json Output raw JSON
|
|
29
|
+
--server <url> Override server URL
|
|
30
|
+
`;
|
|
31
|
+
|
|
32
|
+
const FLAGS = {
|
|
33
|
+
json: { type: "boolean" },
|
|
34
|
+
server: { type: "string" },
|
|
35
|
+
data: { type: "string" },
|
|
36
|
+
cards: { type: "string" },
|
|
37
|
+
help: { short: "h", type: "boolean" },
|
|
38
|
+
};
|
|
39
|
+
|
|
40
|
+
function ensureCardNewsEnabled() {
|
|
41
|
+
if (!runtimeConfig.features?.cardNews) {
|
|
42
|
+
die(2, "Card News feature is disabled. Set IMA2_CARD_NEWS=1 (or features.cardNews: true in config.json) and restart the server.");
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
|
|
46
|
+
async function getServer(args) {
|
|
47
|
+
try { return await resolveServer({ serverFlag: args.server }); }
|
|
48
|
+
catch (e: any) { die(exitCodeForError(e), e.message); throw e; }
|
|
49
|
+
}
|
|
50
|
+
|
|
51
|
+
function parseData(raw: string | undefined): any {
|
|
52
|
+
if (!raw) return {};
|
|
53
|
+
try { return JSON.parse(raw); }
|
|
54
|
+
catch { die(2, `--data must be valid JSON`); }
|
|
55
|
+
}
|
|
56
|
+
|
|
57
|
+
async function templatesSub(argv) {
|
|
58
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
59
|
+
if (args.help) { out(HELP); return; }
|
|
60
|
+
ensureCardNewsEnabled();
|
|
61
|
+
const server = await getServer(args);
|
|
62
|
+
const [imgTemplates, roleTemplates] = await Promise.all([
|
|
63
|
+
request(server.base, "/api/cardnews/image-templates")
|
|
64
|
+
.catch((e) => die(exitCodeForError(e), e.message)),
|
|
65
|
+
request(server.base, "/api/cardnews/role-templates")
|
|
66
|
+
.catch((e) => die(exitCodeForError(e), e.message)),
|
|
67
|
+
]);
|
|
68
|
+
if (args.json) { json({ imageTemplates: imgTemplates, roleTemplates }); return; }
|
|
69
|
+
out(color.bold("Image templates:"));
|
|
70
|
+
out(JSON.stringify(imgTemplates, null, 2));
|
|
71
|
+
out(color.bold("Role templates:"));
|
|
72
|
+
out(JSON.stringify(roleTemplates, null, 2));
|
|
73
|
+
}
|
|
74
|
+
|
|
75
|
+
async function templatePreviewSub(argv) {
|
|
76
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
77
|
+
const templateId = args.positional[0];
|
|
78
|
+
if (!templateId) die(2, "templateId required");
|
|
79
|
+
ensureCardNewsEnabled();
|
|
80
|
+
const server = await getServer(args);
|
|
81
|
+
const resp = await request(server.base, `/api/cardnews/image-templates/${encodeURIComponent(templateId)}/preview`)
|
|
82
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
83
|
+
if (args.json) { json(resp); return; }
|
|
84
|
+
out(JSON.stringify(resp, null, 2));
|
|
85
|
+
}
|
|
86
|
+
|
|
87
|
+
async function setsSub(argv) {
|
|
88
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
89
|
+
ensureCardNewsEnabled();
|
|
90
|
+
const server = await getServer(args);
|
|
91
|
+
const resp = await request(server.base, "/api/cardnews/sets")
|
|
92
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
93
|
+
if (args.json) { json(resp); return; }
|
|
94
|
+
out(JSON.stringify(resp, null, 2));
|
|
95
|
+
}
|
|
96
|
+
|
|
97
|
+
async function setShowSub(argv) {
|
|
98
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
99
|
+
const setId = args.positional[0];
|
|
100
|
+
if (!setId) die(2, "setId required");
|
|
101
|
+
ensureCardNewsEnabled();
|
|
102
|
+
const server = await getServer(args);
|
|
103
|
+
const resp = await request(server.base, `/api/cardnews/sets/${encodeURIComponent(setId)}`)
|
|
104
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
105
|
+
if (args.json) { json(resp); return; }
|
|
106
|
+
out(JSON.stringify(resp, null, 2));
|
|
107
|
+
}
|
|
108
|
+
|
|
109
|
+
async function setManifestSub(argv) {
|
|
110
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
111
|
+
const setId = args.positional[0];
|
|
112
|
+
if (!setId) die(2, "setId required");
|
|
113
|
+
ensureCardNewsEnabled();
|
|
114
|
+
const server = await getServer(args);
|
|
115
|
+
const resp = await request(server.base, `/api/cardnews/sets/${encodeURIComponent(setId)}/manifest`)
|
|
116
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
117
|
+
if (args.json) { json(resp); return; }
|
|
118
|
+
out(JSON.stringify(resp, null, 2));
|
|
119
|
+
}
|
|
120
|
+
|
|
121
|
+
async function draftSub(argv) {
|
|
122
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
123
|
+
ensureCardNewsEnabled();
|
|
124
|
+
const server = await getServer(args);
|
|
125
|
+
const body = parseData(args.data);
|
|
126
|
+
const resp = await request(server.base, "/api/cardnews/draft", { method: "POST", body })
|
|
127
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
128
|
+
if (args.json) { json(resp); return; }
|
|
129
|
+
out(JSON.stringify(resp, null, 2));
|
|
130
|
+
}
|
|
131
|
+
|
|
132
|
+
async function generateSub(argv) {
|
|
133
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
134
|
+
ensureCardNewsEnabled();
|
|
135
|
+
const server = await getServer(args);
|
|
136
|
+
const body = parseData(args.data);
|
|
137
|
+
const resp = await request(server.base, "/api/cardnews/generate", { method: "POST", body })
|
|
138
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
139
|
+
if (args.json) { json(resp); return; }
|
|
140
|
+
out(JSON.stringify(resp, null, 2));
|
|
141
|
+
}
|
|
142
|
+
|
|
143
|
+
async function jobCreateSub(argv) {
|
|
144
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
145
|
+
ensureCardNewsEnabled();
|
|
146
|
+
const server = await getServer(args);
|
|
147
|
+
const body = parseData(args.data);
|
|
148
|
+
const resp = await request(server.base, "/api/cardnews/jobs", { method: "POST", body })
|
|
149
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
150
|
+
if (args.json) { json(resp); return; }
|
|
151
|
+
out(JSON.stringify(resp, null, 2));
|
|
152
|
+
}
|
|
153
|
+
|
|
154
|
+
async function jobShowSub(argv) {
|
|
155
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
156
|
+
const jobId = args.positional[0];
|
|
157
|
+
if (!jobId) die(2, "jobId required");
|
|
158
|
+
ensureCardNewsEnabled();
|
|
159
|
+
const server = await getServer(args);
|
|
160
|
+
const resp = await request(server.base, `/api/cardnews/jobs/${encodeURIComponent(jobId)}`)
|
|
161
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
162
|
+
if (args.json) { json(resp); return; }
|
|
163
|
+
out(JSON.stringify(resp, null, 2));
|
|
164
|
+
}
|
|
165
|
+
|
|
166
|
+
async function jobRetrySub(argv) {
|
|
167
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
168
|
+
const jobId = args.positional[0];
|
|
169
|
+
if (!jobId) die(2, "jobId required");
|
|
170
|
+
ensureCardNewsEnabled();
|
|
171
|
+
const server = await getServer(args);
|
|
172
|
+
const body: any = {};
|
|
173
|
+
if (args.cards) body.cardIds = args.cards.split(",").map((s) => s.trim()).filter(Boolean);
|
|
174
|
+
const resp = await request(server.base, `/api/cardnews/jobs/${encodeURIComponent(jobId)}/retry`, { method: "POST", body })
|
|
175
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
176
|
+
if (args.json) { json(resp); return; }
|
|
177
|
+
out(color.green("✓ ") + "retry requested");
|
|
178
|
+
if (resp && typeof resp === "object" && Object.keys(resp).length > 0) out(JSON.stringify(resp, null, 2));
|
|
179
|
+
}
|
|
180
|
+
|
|
181
|
+
async function cardRegenerateSub(argv) {
|
|
182
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
183
|
+
const cardId = args.positional[0];
|
|
184
|
+
if (!cardId) die(2, "cardId required");
|
|
185
|
+
ensureCardNewsEnabled();
|
|
186
|
+
const server = await getServer(args);
|
|
187
|
+
const body = parseData(args.data);
|
|
188
|
+
const resp = await request(server.base, `/api/cardnews/cards/${encodeURIComponent(cardId)}/regenerate`, { method: "POST", body })
|
|
189
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
190
|
+
if (args.json) { json(resp); return; }
|
|
191
|
+
out(color.green("✓ ") + "regeneration started");
|
|
192
|
+
if (resp && typeof resp === "object" && Object.keys(resp).length > 0) out(JSON.stringify(resp, null, 2));
|
|
193
|
+
}
|
|
194
|
+
|
|
195
|
+
async function exportSub(argv) {
|
|
196
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
197
|
+
ensureCardNewsEnabled();
|
|
198
|
+
const server = await getServer(args);
|
|
199
|
+
const body = parseData(args.data);
|
|
200
|
+
const resp = await request(server.base, "/api/cardnews/export", { method: "POST", body })
|
|
201
|
+
.catch((e) => die(exitCodeForError(e), e.message));
|
|
202
|
+
if (args.json) { json(resp); return; }
|
|
203
|
+
out(color.green("✓ ") + "export requested");
|
|
204
|
+
if (resp && typeof resp === "object" && Object.keys(resp).length > 0) out(JSON.stringify(resp, null, 2));
|
|
205
|
+
}
|
|
206
|
+
|
|
207
|
+
type Sub = (argv: any[]) => Promise<void>;
|
|
208
|
+
|
|
209
|
+
export default async function cardnewsCmd(argv) {
|
|
210
|
+
const sub = argv[0];
|
|
211
|
+
if (!sub || sub === "--help" || sub === "-h") { out(HELP); return; }
|
|
212
|
+
|
|
213
|
+
if (sub === "template") {
|
|
214
|
+
const s2 = argv[1];
|
|
215
|
+
if (s2 === "preview") return templatePreviewSub(argv.slice(2));
|
|
216
|
+
die(2, `unknown subcommand: template ${s2 || ""}\n${HELP}`);
|
|
217
|
+
}
|
|
218
|
+
if (sub === "set") {
|
|
219
|
+
const s2 = argv[1];
|
|
220
|
+
const SET_SUB: Record<string, Sub> = { show: setShowSub, manifest: setManifestSub };
|
|
221
|
+
const handler = SET_SUB[s2];
|
|
222
|
+
if (!handler) die(2, `unknown subcommand: set ${s2 || ""}\n${HELP}`);
|
|
223
|
+
return handler(argv.slice(2));
|
|
224
|
+
}
|
|
225
|
+
if (sub === "job") {
|
|
226
|
+
const s2 = argv[1];
|
|
227
|
+
const JOB_SUB: Record<string, Sub> = { create: jobCreateSub, show: jobShowSub, retry: jobRetrySub };
|
|
228
|
+
const handler = JOB_SUB[s2];
|
|
229
|
+
if (!handler) die(2, `unknown subcommand: job ${s2 || ""}\n${HELP}`);
|
|
230
|
+
return handler(argv.slice(2));
|
|
231
|
+
}
|
|
232
|
+
if (sub === "card") {
|
|
233
|
+
const s2 = argv[1];
|
|
234
|
+
if (s2 === "regenerate") return cardRegenerateSub(argv.slice(2));
|
|
235
|
+
die(2, `unknown subcommand: card ${s2 || ""}\n${HELP}`);
|
|
236
|
+
}
|
|
237
|
+
|
|
238
|
+
const FLAT: Record<string, Sub> = {
|
|
239
|
+
templates: templatesSub,
|
|
240
|
+
sets: setsSub,
|
|
241
|
+
draft: draftSub,
|
|
242
|
+
generate: generateSub,
|
|
243
|
+
export: exportSub,
|
|
244
|
+
};
|
|
245
|
+
const handler = FLAT[sub];
|
|
246
|
+
if (!handler) die(2, `unknown subcommand: ${sub}\n${HELP}`);
|
|
247
|
+
return handler(argv.slice(1));
|
|
248
|
+
}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
import { writeFile, access } from "fs/promises";
|
|
2
|
+
import { parseArgs } from "../lib/args.js";
|
|
3
|
+
import { resolveServer, request } from "../lib/client.js";
|
|
4
|
+
import { out, die, color, json, exitCodeForError } from "../lib/output.js";
|
|
5
|
+
const HELP = `
|
|
6
|
+
ima2 comfy <subcommand> [options]
|
|
7
|
+
|
|
8
|
+
Subcommands:
|
|
9
|
+
export <filename> [-o <out>] [--force]
|
|
10
|
+
`;
|
|
11
|
+
const FLAGS = {
|
|
12
|
+
json: { type: "boolean" },
|
|
13
|
+
server: { type: "string" },
|
|
14
|
+
out: { short: "o", type: "string" },
|
|
15
|
+
force: { type: "boolean" },
|
|
16
|
+
help: { short: "h", type: "boolean" },
|
|
17
|
+
};
|
|
18
|
+
async function exportSub(argv) {
|
|
19
|
+
const args = parseArgs(argv, { flags: FLAGS });
|
|
20
|
+
const filename = args.positional[0];
|
|
21
|
+
if (!filename)
|
|
22
|
+
die(2, "filename required");
|
|
23
|
+
let server;
|
|
24
|
+
try {
|
|
25
|
+
server = await resolveServer({ serverFlag: args.server });
|
|
26
|
+
}
|
|
27
|
+
catch (e) {
|
|
28
|
+
die(exitCodeForError(e), e.message);
|
|
29
|
+
throw e;
|
|
30
|
+
}
|
|
31
|
+
const resp = await request(server.base, "/api/comfy/export-image", {
|
|
32
|
+
method: "POST",
|
|
33
|
+
body: { filename },
|
|
34
|
+
}).catch((e) => die(exitCodeForError(e), `${e.message}${e.code ? ` (${e.code})` : ""}`));
|
|
35
|
+
const target = args.out || `${filename}.workflow.json`;
|
|
36
|
+
if (!args.force) {
|
|
37
|
+
try {
|
|
38
|
+
await access(target);
|
|
39
|
+
die(2, `${target} already exists. Pass --force to overwrite.`);
|
|
40
|
+
}
|
|
41
|
+
catch { /* file does not exist — proceed */ }
|
|
42
|
+
}
|
|
43
|
+
await writeFile(target, JSON.stringify(resp, null, 2));
|
|
44
|
+
if (args.json) {
|
|
45
|
+
json({ path: target });
|
|
46
|
+
return;
|
|
47
|
+
}
|
|
48
|
+
out(color.green("✓ ") + target);
|
|
49
|
+
}
|
|
50
|
+
const SUB = {
|
|
51
|
+
export: exportSub,
|
|
52
|
+
};
|
|
53
|
+
export default async function comfyCmd(argv) {
|
|
54
|
+
const sub = argv[0];
|
|
55
|
+
if (!sub || sub === "--help" || sub === "-h") {
|
|
56
|
+
out(HELP);
|
|
57
|
+
return;
|
|
58
|
+
}
|
|
59
|
+
const handler = SUB[sub];
|
|
60
|
+
if (!handler)
|
|
61
|
+
die(2, `unknown subcommand: ${sub}\n${HELP}`);
|
|
62
|
+
return handler(argv.slice(1));
|
|
63
|
+
}
|