ima2-gen 1.1.8 → 1.1.10

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 (255) hide show
  1. package/README.md +60 -27
  2. package/bin/commands/annotate.js +139 -0
  3. package/bin/commands/annotate.ts +119 -0
  4. package/bin/commands/cancel.js +40 -33
  5. package/bin/commands/cancel.ts +48 -0
  6. package/bin/commands/canvas-versions.js +91 -0
  7. package/bin/commands/canvas-versions.ts +80 -0
  8. package/bin/commands/cardnews.js +295 -0
  9. package/bin/commands/cardnews.ts +249 -0
  10. package/bin/commands/comfy.js +63 -0
  11. package/bin/commands/comfy.ts +54 -0
  12. package/bin/commands/config.js +270 -0
  13. package/bin/commands/config.ts +265 -0
  14. package/bin/commands/edit.js +100 -72
  15. package/bin/commands/edit.ts +119 -0
  16. package/bin/commands/gen.js +143 -118
  17. package/bin/commands/gen.ts +179 -0
  18. package/bin/commands/history.js +165 -0
  19. package/bin/commands/history.ts +146 -0
  20. package/bin/commands/ls.js +63 -42
  21. package/bin/commands/ls.ts +63 -0
  22. package/bin/commands/metadata.js +48 -0
  23. package/bin/commands/metadata.ts +39 -0
  24. package/bin/commands/multimode.js +159 -0
  25. package/bin/commands/multimode.ts +145 -0
  26. package/bin/commands/node.js +176 -0
  27. package/bin/commands/node.ts +157 -0
  28. package/bin/commands/observability.js +201 -0
  29. package/bin/commands/observability.ts +176 -0
  30. package/bin/commands/ping.js +28 -20
  31. package/bin/commands/ping.ts +31 -0
  32. package/bin/commands/prompt.js +509 -0
  33. package/bin/commands/prompt.ts +423 -0
  34. package/bin/commands/ps.js +81 -71
  35. package/bin/commands/ps.ts +81 -0
  36. package/bin/commands/session.js +309 -0
  37. package/bin/commands/session.ts +266 -0
  38. package/bin/commands/show.js +78 -40
  39. package/bin/commands/show.ts +72 -0
  40. package/bin/ima2.js +325 -311
  41. package/bin/ima2.ts +444 -0
  42. package/bin/lib/args.js +75 -66
  43. package/bin/lib/args.ts +92 -0
  44. package/bin/lib/browser-id.js +15 -0
  45. package/bin/lib/browser-id.ts +16 -0
  46. package/bin/lib/client.js +91 -83
  47. package/bin/lib/client.ts +122 -0
  48. package/bin/lib/error-hints.js +14 -17
  49. package/bin/lib/error-hints.ts +23 -0
  50. package/bin/lib/files.js +26 -28
  51. package/bin/lib/files.ts +39 -0
  52. package/bin/lib/output.js +46 -42
  53. package/bin/lib/output.ts +73 -0
  54. package/bin/lib/platform.js +62 -56
  55. package/bin/lib/platform.ts +99 -0
  56. package/bin/lib/sse.js +73 -0
  57. package/bin/lib/sse.ts +73 -0
  58. package/bin/lib/star-prompt.js +69 -76
  59. package/bin/lib/star-prompt.ts +97 -0
  60. package/bin/lib/storage-doctor.js +34 -35
  61. package/bin/lib/storage-doctor.ts +39 -0
  62. package/config.js +147 -243
  63. package/config.ts +333 -0
  64. package/docs/API.md +48 -8
  65. package/docs/CLI.md +190 -0
  66. package/docs/FAQ.ko.md +5 -5
  67. package/docs/FAQ.md +5 -5
  68. package/docs/README.ja.md +71 -25
  69. package/docs/README.ko.md +61 -24
  70. package/docs/README.zh-CN.md +73 -27
  71. package/docs/migration/runtime-test-inventory.md +135 -0
  72. package/lib/assetLifecycle.js +130 -130
  73. package/lib/assetLifecycle.ts +142 -0
  74. package/lib/canvasVersionStore.js +135 -153
  75. package/lib/canvasVersionStore.ts +221 -0
  76. package/lib/cardNewsGenerator.js +132 -143
  77. package/lib/cardNewsGenerator.ts +265 -0
  78. package/lib/cardNewsJobStore.js +78 -84
  79. package/lib/cardNewsJobStore.ts +142 -0
  80. package/lib/cardNewsManifestStore.js +98 -96
  81. package/lib/cardNewsManifestStore.ts +154 -0
  82. package/lib/cardNewsPlanner.js +159 -152
  83. package/lib/cardNewsPlanner.ts +236 -0
  84. package/lib/cardNewsPlannerClient.js +114 -98
  85. package/lib/cardNewsPlannerClient.ts +155 -0
  86. package/lib/cardNewsPlannerPrompt.js +56 -56
  87. package/lib/cardNewsPlannerPrompt.ts +60 -0
  88. package/lib/cardNewsPlannerSchema.js +238 -223
  89. package/lib/cardNewsPlannerSchema.ts +321 -0
  90. package/lib/cardNewsRoleTemplateStore.js +39 -41
  91. package/lib/cardNewsRoleTemplateStore.ts +47 -0
  92. package/lib/cardNewsTemplateStore.js +156 -175
  93. package/lib/cardNewsTemplateStore.ts +252 -0
  94. package/lib/codexDetect.js +46 -47
  95. package/lib/codexDetect.ts +71 -0
  96. package/lib/comfyBridge.js +166 -184
  97. package/lib/comfyBridge.ts +235 -0
  98. package/lib/db.js +41 -51
  99. package/lib/db.ts +166 -0
  100. package/lib/errInfo.js +32 -0
  101. package/lib/errInfo.ts +43 -0
  102. package/lib/errorClassify.js +62 -78
  103. package/lib/errorClassify.ts +100 -0
  104. package/lib/generationErrors.js +140 -103
  105. package/lib/generationErrors.ts +151 -0
  106. package/lib/historyList.js +153 -147
  107. package/lib/historyList.ts +168 -0
  108. package/lib/imageMetadata.js +86 -89
  109. package/lib/imageMetadata.ts +111 -0
  110. package/lib/imageMetadataStore.js +46 -51
  111. package/lib/imageMetadataStore.ts +67 -0
  112. package/lib/imageModels.js +38 -45
  113. package/lib/imageModels.ts +54 -0
  114. package/lib/inflight.js +131 -157
  115. package/lib/inflight.ts +247 -0
  116. package/lib/localImportStore.js +87 -93
  117. package/lib/localImportStore.ts +112 -0
  118. package/lib/logger.js +107 -112
  119. package/lib/logger.ts +161 -0
  120. package/lib/nodeStore.js +65 -64
  121. package/lib/nodeStore.ts +89 -0
  122. package/lib/oauthLauncher.js +61 -59
  123. package/lib/oauthLauncher.ts +65 -0
  124. package/lib/oauthNormalize.js +15 -19
  125. package/lib/oauthNormalize.ts +30 -0
  126. package/lib/oauthProxy/errors.js +93 -0
  127. package/lib/oauthProxy/errors.ts +128 -0
  128. package/lib/oauthProxy/generators.js +426 -0
  129. package/lib/oauthProxy/generators.ts +494 -0
  130. package/lib/oauthProxy/index.js +8 -0
  131. package/lib/oauthProxy/index.ts +28 -0
  132. package/lib/oauthProxy/prompts.js +84 -0
  133. package/lib/oauthProxy/prompts.ts +108 -0
  134. package/lib/oauthProxy/references.js +32 -0
  135. package/lib/oauthProxy/references.ts +45 -0
  136. package/lib/oauthProxy/runtime.js +101 -0
  137. package/lib/oauthProxy/runtime.ts +115 -0
  138. package/lib/oauthProxy/streams.js +211 -0
  139. package/lib/oauthProxy/streams.ts +232 -0
  140. package/lib/oauthProxy/types.js +6 -0
  141. package/lib/oauthProxy/types.ts +9 -0
  142. package/lib/oauthProxy.js +3 -909
  143. package/lib/oauthProxy.ts +3 -0
  144. package/lib/openDirectory.js +43 -40
  145. package/lib/openDirectory.ts +47 -0
  146. package/lib/pngInfo.js +18 -20
  147. package/lib/pngInfo.ts +26 -0
  148. package/lib/promptImport/curatedSources.js +125 -129
  149. package/lib/promptImport/curatedSources.ts +141 -0
  150. package/lib/promptImport/discoveryRegistry.js +198 -208
  151. package/lib/promptImport/discoveryRegistry.ts +329 -0
  152. package/lib/promptImport/errors.js +10 -10
  153. package/lib/promptImport/errors.ts +18 -0
  154. package/lib/promptImport/githubDiscovery.js +216 -220
  155. package/lib/promptImport/githubDiscovery.ts +309 -0
  156. package/lib/promptImport/githubFolder.js +261 -259
  157. package/lib/promptImport/githubFolder.ts +397 -0
  158. package/lib/promptImport/githubSource.js +191 -200
  159. package/lib/promptImport/githubSource.ts +257 -0
  160. package/lib/promptImport/gptImageHints.js +49 -56
  161. package/lib/promptImport/gptImageHints.ts +70 -0
  162. package/lib/promptImport/parsePromptCandidates.js +110 -123
  163. package/lib/promptImport/parsePromptCandidates.ts +179 -0
  164. package/lib/promptImport/promptIndex.js +203 -211
  165. package/lib/promptImport/promptIndex.ts +326 -0
  166. package/lib/promptImport/rankPromptCandidates.js +46 -43
  167. package/lib/promptImport/rankPromptCandidates.ts +65 -0
  168. package/lib/promptImport/types.js +1 -0
  169. package/lib/promptImport/types.ts +103 -0
  170. package/lib/promptSafetyPolicy.js +5 -0
  171. package/lib/promptSafetyPolicy.ts +5 -0
  172. package/lib/providerOptions.js +31 -0
  173. package/lib/providerOptions.ts +42 -0
  174. package/lib/referenceImageCompress.js +51 -62
  175. package/lib/referenceImageCompress.ts +84 -0
  176. package/lib/refs.js +95 -81
  177. package/lib/refs.ts +133 -0
  178. package/lib/requestLogger.js +32 -38
  179. package/lib/requestLogger.ts +49 -0
  180. package/lib/responsesImageAdapter.js +357 -0
  181. package/lib/responsesImageAdapter.ts +430 -0
  182. package/lib/runtimeContext.js +100 -0
  183. package/lib/runtimeContext.ts +131 -0
  184. package/lib/runtimePorts.js +72 -73
  185. package/lib/runtimePorts.ts +105 -0
  186. package/lib/sessionStore.js +183 -232
  187. package/lib/sessionStore.ts +308 -0
  188. package/lib/storageMigration.js +254 -245
  189. package/lib/storageMigration.ts +310 -0
  190. package/lib/styleSheet.js +87 -90
  191. package/lib/styleSheet.ts +139 -0
  192. package/lib/systemTrash.js +18 -0
  193. package/lib/systemTrash.ts +20 -0
  194. package/package.json +28 -10
  195. package/routes/annotations.js +81 -79
  196. package/routes/annotations.ts +118 -0
  197. package/routes/canvasVersions.js +56 -55
  198. package/routes/canvasVersions.ts +69 -0
  199. package/routes/cardNews.js +172 -172
  200. package/routes/cardNews.ts +211 -0
  201. package/routes/comfy.js +27 -32
  202. package/routes/comfy.ts +43 -0
  203. package/routes/edit.js +188 -215
  204. package/routes/edit.ts +250 -0
  205. package/routes/generate.js +282 -292
  206. package/routes/generate.ts +323 -0
  207. package/routes/health.js +107 -108
  208. package/routes/health.ts +119 -0
  209. package/routes/history.js +152 -145
  210. package/routes/history.ts +171 -0
  211. package/routes/imageImport.js +31 -27
  212. package/routes/imageImport.ts +37 -0
  213. package/routes/index.js +20 -18
  214. package/routes/index.ts +38 -0
  215. package/routes/metadata.js +66 -65
  216. package/routes/metadata.ts +77 -0
  217. package/routes/multimode.js +235 -264
  218. package/routes/multimode.ts +301 -0
  219. package/routes/nodes.js +389 -425
  220. package/routes/nodes.ts +494 -0
  221. package/routes/promptImport.js +291 -325
  222. package/routes/promptImport.ts +379 -0
  223. package/routes/prompts.js +357 -361
  224. package/routes/prompts.ts +428 -0
  225. package/routes/sessions.js +299 -285
  226. package/routes/sessions.ts +317 -0
  227. package/routes/storage.js +32 -32
  228. package/routes/storage.ts +47 -0
  229. package/server.js +193 -196
  230. package/server.ts +255 -0
  231. package/ui/dist/.vite/manifest.json +102 -0
  232. package/ui/dist/assets/CardNewsWorkspace-C9Cpxuxc.js +2 -0
  233. package/ui/dist/assets/NodeCanvas-BZV40eAE.css +1 -0
  234. package/ui/dist/assets/NodeCanvas-BllpfcQW.js +7 -0
  235. package/ui/dist/assets/PromptImportDialog-D8EMO--u.js +2 -0
  236. package/ui/dist/assets/PromptImportDiscoverySection-BB2FrKuq.js +1 -0
  237. package/ui/dist/assets/PromptImportFolderSection-aVteBUcb.js +1 -0
  238. package/ui/dist/assets/PromptLibraryPanel-Z-4B8RSs.js +2 -0
  239. package/ui/dist/assets/SettingsWorkspace-DBYdgpPI.js +1 -0
  240. package/ui/dist/assets/index-B2TDuGqy.css +1 -0
  241. package/ui/dist/assets/index-DFlbOIxI.js +25 -0
  242. package/ui/dist/assets/index-Deo5wBiA.js +1 -0
  243. package/ui/dist/index.html +2 -2
  244. package/assets/phase-a-bg-cleanup-test.png +0 -0
  245. package/assets/screenshot.png +0 -0
  246. package/assets/screenshots/classic-generate-light.png +0 -0
  247. package/assets/screenshots/node-graph-branching.png +0 -0
  248. package/assets/screenshots/settings-oauth-generation.png +0 -0
  249. package/assets/screenshots/settings-workspace.png +0 -0
  250. package/assets/screenshots/style-sheet-editor.png +0 -0
  251. package/integrations/comfyui/ima2_gen_bridge/__pycache__/__init__.cpython-313.pyc +0 -0
  252. package/integrations/comfyui/ima2_gen_bridge/__pycache__/nodes.cpython-313.pyc +0 -0
  253. package/ui/dist/assets/index-BDffwmLs.css +0 -1
  254. package/ui/dist/assets/index-D0fdHLkJ.js +0 -31
  255. package/ui/dist/assets/index-D0fdHLkJ.js.map +0 -1
@@ -0,0 +1,80 @@
1
+ import { readFile } from "fs/promises";
2
+ import { parseArgs, type ParsedArgs } 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: ParsedArgs) {
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: ParsedArgs) {
31
+ const h: Record<string, string> = { "Content-Type": "image/png" };
32
+ if (args.source) h["X-Ima2-Canvas-Source-Filename"] = String(args.source);
33
+ if (args.prompt) h["X-Ima2-Canvas-Prompt"] = String(args.prompt);
34
+ return h;
35
+ }
36
+
37
+ async function saveSub(argv: string[]) {
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: unknown) => { const err = e as { message?: string; code?: string }; die(exitCodeForError(e), `${err.message}${err.code ? ` (${err.code})` : ""}`); });
49
+ if (args.json) { json(resp); return; }
50
+ out(color.green("✓ saved canvas version"));
51
+ }
52
+
53
+ async function updateSub(argv: string[]) {
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: unknown) => { const err = e as { message?: string; code?: string }; die(exitCodeForError(e), `${err.message}${err.code ? ` (${err.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: string[]) {
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,295 @@
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
+ if (typeof raw !== "string")
55
+ return {};
56
+ try {
57
+ return JSON.parse(raw);
58
+ }
59
+ catch {
60
+ die(2, `--data must be valid JSON`);
61
+ }
62
+ }
63
+ async function templatesSub(argv) {
64
+ const args = parseArgs(argv, { flags: FLAGS });
65
+ if (args.help) {
66
+ out(HELP);
67
+ return;
68
+ }
69
+ ensureCardNewsEnabled();
70
+ const server = await getServer(args);
71
+ const [imgTemplates, roleTemplates] = await Promise.all([
72
+ request(server.base, "/api/cardnews/image-templates")
73
+ .catch((e) => die(exitCodeForError(e), e.message)),
74
+ request(server.base, "/api/cardnews/role-templates")
75
+ .catch((e) => die(exitCodeForError(e), e.message)),
76
+ ]);
77
+ if (args.json) {
78
+ json({ imageTemplates: imgTemplates, roleTemplates });
79
+ return;
80
+ }
81
+ out(color.bold("Image templates:"));
82
+ out(JSON.stringify(imgTemplates, null, 2));
83
+ out(color.bold("Role templates:"));
84
+ out(JSON.stringify(roleTemplates, null, 2));
85
+ }
86
+ async function templatePreviewSub(argv) {
87
+ const args = parseArgs(argv, { flags: FLAGS });
88
+ const templateId = args.positional[0];
89
+ if (!templateId)
90
+ die(2, "templateId required");
91
+ ensureCardNewsEnabled();
92
+ const server = await getServer(args);
93
+ const resp = await request(server.base, `/api/cardnews/image-templates/${encodeURIComponent(templateId)}/preview`)
94
+ .catch((e) => die(exitCodeForError(e), e.message));
95
+ if (args.json) {
96
+ json(resp);
97
+ return;
98
+ }
99
+ out(JSON.stringify(resp, null, 2));
100
+ }
101
+ async function setsSub(argv) {
102
+ const args = parseArgs(argv, { flags: FLAGS });
103
+ ensureCardNewsEnabled();
104
+ const server = await getServer(args);
105
+ const resp = await request(server.base, "/api/cardnews/sets")
106
+ .catch((e) => die(exitCodeForError(e), e.message));
107
+ if (args.json) {
108
+ json(resp);
109
+ return;
110
+ }
111
+ out(JSON.stringify(resp, null, 2));
112
+ }
113
+ async function setShowSub(argv) {
114
+ const args = parseArgs(argv, { flags: FLAGS });
115
+ const setId = args.positional[0];
116
+ if (!setId)
117
+ die(2, "setId required");
118
+ ensureCardNewsEnabled();
119
+ const server = await getServer(args);
120
+ const resp = await request(server.base, `/api/cardnews/sets/${encodeURIComponent(setId)}`)
121
+ .catch((e) => die(exitCodeForError(e), e.message));
122
+ if (args.json) {
123
+ json(resp);
124
+ return;
125
+ }
126
+ out(JSON.stringify(resp, null, 2));
127
+ }
128
+ async function setManifestSub(argv) {
129
+ const args = parseArgs(argv, { flags: FLAGS });
130
+ const setId = args.positional[0];
131
+ if (!setId)
132
+ die(2, "setId required");
133
+ ensureCardNewsEnabled();
134
+ const server = await getServer(args);
135
+ const resp = await request(server.base, `/api/cardnews/sets/${encodeURIComponent(setId)}/manifest`)
136
+ .catch((e) => die(exitCodeForError(e), e.message));
137
+ if (args.json) {
138
+ json(resp);
139
+ return;
140
+ }
141
+ out(JSON.stringify(resp, null, 2));
142
+ }
143
+ async function draftSub(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/draft", { method: "POST", body })
149
+ .catch((e) => die(exitCodeForError(e), e.message));
150
+ if (args.json) {
151
+ json(resp);
152
+ return;
153
+ }
154
+ out(JSON.stringify(resp, null, 2));
155
+ }
156
+ async function generateSub(argv) {
157
+ const args = parseArgs(argv, { flags: FLAGS });
158
+ ensureCardNewsEnabled();
159
+ const server = await getServer(args);
160
+ const body = parseData(args.data);
161
+ const resp = await request(server.base, "/api/cardnews/generate", { method: "POST", body })
162
+ .catch((e) => die(exitCodeForError(e), e.message));
163
+ if (args.json) {
164
+ json(resp);
165
+ return;
166
+ }
167
+ out(JSON.stringify(resp, null, 2));
168
+ }
169
+ async function jobCreateSub(argv) {
170
+ const args = parseArgs(argv, { flags: FLAGS });
171
+ ensureCardNewsEnabled();
172
+ const server = await getServer(args);
173
+ const body = parseData(args.data);
174
+ const resp = await request(server.base, "/api/cardnews/jobs", { method: "POST", body })
175
+ .catch((e) => die(exitCodeForError(e), e.message));
176
+ if (args.json) {
177
+ json(resp);
178
+ return;
179
+ }
180
+ out(JSON.stringify(resp, null, 2));
181
+ }
182
+ async function jobShowSub(argv) {
183
+ const args = parseArgs(argv, { flags: FLAGS });
184
+ const jobId = args.positional[0];
185
+ if (!jobId)
186
+ die(2, "jobId required");
187
+ ensureCardNewsEnabled();
188
+ const server = await getServer(args);
189
+ const resp = await request(server.base, `/api/cardnews/jobs/${encodeURIComponent(jobId)}`)
190
+ .catch((e) => die(exitCodeForError(e), e.message));
191
+ if (args.json) {
192
+ json(resp);
193
+ return;
194
+ }
195
+ out(JSON.stringify(resp, null, 2));
196
+ }
197
+ async function jobRetrySub(argv) {
198
+ const args = parseArgs(argv, { flags: FLAGS });
199
+ const jobId = args.positional[0];
200
+ if (!jobId)
201
+ die(2, "jobId required");
202
+ ensureCardNewsEnabled();
203
+ const server = await getServer(args);
204
+ const body = {};
205
+ if (args.cards)
206
+ body.cardIds = String(args.cards).split(",").map((s) => s.trim()).filter(Boolean);
207
+ const resp = await request(server.base, `/api/cardnews/jobs/${encodeURIComponent(jobId)}/retry`, { method: "POST", body })
208
+ .catch((e) => die(exitCodeForError(e), e.message));
209
+ if (args.json) {
210
+ json(resp);
211
+ return;
212
+ }
213
+ out(color.green("✓ ") + "retry requested");
214
+ if (resp && typeof resp === "object" && Object.keys(resp).length > 0)
215
+ out(JSON.stringify(resp, null, 2));
216
+ }
217
+ async function cardRegenerateSub(argv) {
218
+ const args = parseArgs(argv, { flags: FLAGS });
219
+ const cardId = args.positional[0];
220
+ if (!cardId)
221
+ die(2, "cardId required");
222
+ ensureCardNewsEnabled();
223
+ const server = await getServer(args);
224
+ const body = parseData(args.data);
225
+ const resp = await request(server.base, `/api/cardnews/cards/${encodeURIComponent(cardId)}/regenerate`, { method: "POST", body })
226
+ .catch((e) => die(exitCodeForError(e), e.message));
227
+ if (args.json) {
228
+ json(resp);
229
+ return;
230
+ }
231
+ out(color.green("✓ ") + "regeneration started");
232
+ if (resp && typeof resp === "object" && Object.keys(resp).length > 0)
233
+ out(JSON.stringify(resp, null, 2));
234
+ }
235
+ async function exportSub(argv) {
236
+ const args = parseArgs(argv, { flags: FLAGS });
237
+ ensureCardNewsEnabled();
238
+ const server = await getServer(args);
239
+ const body = parseData(args.data);
240
+ const resp = await request(server.base, "/api/cardnews/export", { method: "POST", body })
241
+ .catch((e) => die(exitCodeForError(e), e.message));
242
+ if (args.json) {
243
+ json(resp);
244
+ return;
245
+ }
246
+ out(color.green("✓ ") + "export requested");
247
+ if (resp && typeof resp === "object" && Object.keys(resp).length > 0)
248
+ out(JSON.stringify(resp, null, 2));
249
+ }
250
+ export default async function cardnewsCmd(argv) {
251
+ const sub = argv[0];
252
+ if (!sub || sub === "--help" || sub === "-h") {
253
+ out(HELP);
254
+ return;
255
+ }
256
+ if (sub === "template") {
257
+ const s2 = argv[1];
258
+ if (s2 === "preview")
259
+ return templatePreviewSub(argv.slice(2));
260
+ die(2, `unknown subcommand: template ${s2 || ""}\n${HELP}`);
261
+ }
262
+ if (sub === "set") {
263
+ const s2 = argv[1];
264
+ const SET_SUB = { show: setShowSub, manifest: setManifestSub };
265
+ const handler = SET_SUB[s2];
266
+ if (!handler)
267
+ die(2, `unknown subcommand: set ${s2 || ""}\n${HELP}`);
268
+ return handler(argv.slice(2));
269
+ }
270
+ if (sub === "job") {
271
+ const s2 = argv[1];
272
+ const JOB_SUB = { create: jobCreateSub, show: jobShowSub, retry: jobRetrySub };
273
+ const handler = JOB_SUB[s2];
274
+ if (!handler)
275
+ die(2, `unknown subcommand: job ${s2 || ""}\n${HELP}`);
276
+ return handler(argv.slice(2));
277
+ }
278
+ if (sub === "card") {
279
+ const s2 = argv[1];
280
+ if (s2 === "regenerate")
281
+ return cardRegenerateSub(argv.slice(2));
282
+ die(2, `unknown subcommand: card ${s2 || ""}\n${HELP}`);
283
+ }
284
+ const FLAT = {
285
+ templates: templatesSub,
286
+ sets: setsSub,
287
+ draft: draftSub,
288
+ generate: generateSub,
289
+ export: exportSub,
290
+ };
291
+ const handler = FLAT[sub];
292
+ if (!handler)
293
+ die(2, `unknown subcommand: ${sub}\n${HELP}`);
294
+ return handler(argv.slice(1));
295
+ }
@@ -0,0 +1,249 @@
1
+ import { parseArgs, type ParsedArgs } 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: ParsedArgs) {
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: unknown): any {
52
+ if (!raw) return {};
53
+ if (typeof raw !== "string") return {};
54
+ try { return JSON.parse(raw); }
55
+ catch { die(2, `--data must be valid JSON`); }
56
+ }
57
+
58
+ async function templatesSub(argv: string[]) {
59
+ const args = parseArgs(argv, { flags: FLAGS });
60
+ if (args.help) { out(HELP); return; }
61
+ ensureCardNewsEnabled();
62
+ const server = await getServer(args);
63
+ const [imgTemplates, roleTemplates] = await Promise.all([
64
+ request(server.base, "/api/cardnews/image-templates")
65
+ .catch((e) => die(exitCodeForError(e), e.message)),
66
+ request(server.base, "/api/cardnews/role-templates")
67
+ .catch((e) => die(exitCodeForError(e), e.message)),
68
+ ]);
69
+ if (args.json) { json({ imageTemplates: imgTemplates, roleTemplates }); return; }
70
+ out(color.bold("Image templates:"));
71
+ out(JSON.stringify(imgTemplates, null, 2));
72
+ out(color.bold("Role templates:"));
73
+ out(JSON.stringify(roleTemplates, null, 2));
74
+ }
75
+
76
+ async function templatePreviewSub(argv: string[]) {
77
+ const args = parseArgs(argv, { flags: FLAGS });
78
+ const templateId = args.positional[0];
79
+ if (!templateId) die(2, "templateId required");
80
+ ensureCardNewsEnabled();
81
+ const server = await getServer(args);
82
+ const resp = await request(server.base, `/api/cardnews/image-templates/${encodeURIComponent(templateId)}/preview`)
83
+ .catch((e) => die(exitCodeForError(e), e.message));
84
+ if (args.json) { json(resp); return; }
85
+ out(JSON.stringify(resp, null, 2));
86
+ }
87
+
88
+ async function setsSub(argv: string[]) {
89
+ const args = parseArgs(argv, { flags: FLAGS });
90
+ ensureCardNewsEnabled();
91
+ const server = await getServer(args);
92
+ const resp = await request(server.base, "/api/cardnews/sets")
93
+ .catch((e) => die(exitCodeForError(e), e.message));
94
+ if (args.json) { json(resp); return; }
95
+ out(JSON.stringify(resp, null, 2));
96
+ }
97
+
98
+ async function setShowSub(argv: string[]) {
99
+ const args = parseArgs(argv, { flags: FLAGS });
100
+ const setId = args.positional[0];
101
+ if (!setId) die(2, "setId required");
102
+ ensureCardNewsEnabled();
103
+ const server = await getServer(args);
104
+ const resp = await request(server.base, `/api/cardnews/sets/${encodeURIComponent(setId)}`)
105
+ .catch((e) => die(exitCodeForError(e), e.message));
106
+ if (args.json) { json(resp); return; }
107
+ out(JSON.stringify(resp, null, 2));
108
+ }
109
+
110
+ async function setManifestSub(argv: string[]) {
111
+ const args = parseArgs(argv, { flags: FLAGS });
112
+ const setId = args.positional[0];
113
+ if (!setId) die(2, "setId required");
114
+ ensureCardNewsEnabled();
115
+ const server = await getServer(args);
116
+ const resp = await request(server.base, `/api/cardnews/sets/${encodeURIComponent(setId)}/manifest`)
117
+ .catch((e) => die(exitCodeForError(e), e.message));
118
+ if (args.json) { json(resp); return; }
119
+ out(JSON.stringify(resp, null, 2));
120
+ }
121
+
122
+ async function draftSub(argv: string[]) {
123
+ const args = parseArgs(argv, { flags: FLAGS });
124
+ ensureCardNewsEnabled();
125
+ const server = await getServer(args);
126
+ const body = parseData(args.data);
127
+ const resp = await request(server.base, "/api/cardnews/draft", { method: "POST", body })
128
+ .catch((e) => die(exitCodeForError(e), e.message));
129
+ if (args.json) { json(resp); return; }
130
+ out(JSON.stringify(resp, null, 2));
131
+ }
132
+
133
+ async function generateSub(argv: string[]) {
134
+ const args = parseArgs(argv, { flags: FLAGS });
135
+ ensureCardNewsEnabled();
136
+ const server = await getServer(args);
137
+ const body = parseData(args.data);
138
+ const resp = await request(server.base, "/api/cardnews/generate", { method: "POST", body })
139
+ .catch((e) => die(exitCodeForError(e), e.message));
140
+ if (args.json) { json(resp); return; }
141
+ out(JSON.stringify(resp, null, 2));
142
+ }
143
+
144
+ async function jobCreateSub(argv: string[]) {
145
+ const args = parseArgs(argv, { flags: FLAGS });
146
+ ensureCardNewsEnabled();
147
+ const server = await getServer(args);
148
+ const body = parseData(args.data);
149
+ const resp = await request(server.base, "/api/cardnews/jobs", { method: "POST", body })
150
+ .catch((e) => die(exitCodeForError(e), e.message));
151
+ if (args.json) { json(resp); return; }
152
+ out(JSON.stringify(resp, null, 2));
153
+ }
154
+
155
+ async function jobShowSub(argv: string[]) {
156
+ const args = parseArgs(argv, { flags: FLAGS });
157
+ const jobId = args.positional[0];
158
+ if (!jobId) die(2, "jobId required");
159
+ ensureCardNewsEnabled();
160
+ const server = await getServer(args);
161
+ const resp = await request(server.base, `/api/cardnews/jobs/${encodeURIComponent(jobId)}`)
162
+ .catch((e) => die(exitCodeForError(e), e.message));
163
+ if (args.json) { json(resp); return; }
164
+ out(JSON.stringify(resp, null, 2));
165
+ }
166
+
167
+ async function jobRetrySub(argv: string[]) {
168
+ const args = parseArgs(argv, { flags: FLAGS });
169
+ const jobId = args.positional[0];
170
+ if (!jobId) die(2, "jobId required");
171
+ ensureCardNewsEnabled();
172
+ const server = await getServer(args);
173
+ const body: any = {};
174
+ if (args.cards) body.cardIds = String(args.cards).split(",").map((s: string) => s.trim()).filter(Boolean);
175
+ const resp = await request(server.base, `/api/cardnews/jobs/${encodeURIComponent(jobId)}/retry`, { method: "POST", body })
176
+ .catch((e) => die(exitCodeForError(e), e.message));
177
+ if (args.json) { json(resp); return; }
178
+ out(color.green("✓ ") + "retry requested");
179
+ if (resp && typeof resp === "object" && Object.keys(resp).length > 0) out(JSON.stringify(resp, null, 2));
180
+ }
181
+
182
+ async function cardRegenerateSub(argv: string[]) {
183
+ const args = parseArgs(argv, { flags: FLAGS });
184
+ const cardId = args.positional[0];
185
+ if (!cardId) die(2, "cardId required");
186
+ ensureCardNewsEnabled();
187
+ const server = await getServer(args);
188
+ const body = parseData(args.data);
189
+ const resp = await request(server.base, `/api/cardnews/cards/${encodeURIComponent(cardId)}/regenerate`, { method: "POST", body })
190
+ .catch((e) => die(exitCodeForError(e), e.message));
191
+ if (args.json) { json(resp); return; }
192
+ out(color.green("✓ ") + "regeneration started");
193
+ if (resp && typeof resp === "object" && Object.keys(resp).length > 0) out(JSON.stringify(resp, null, 2));
194
+ }
195
+
196
+ async function exportSub(argv: string[]) {
197
+ const args = parseArgs(argv, { flags: FLAGS });
198
+ ensureCardNewsEnabled();
199
+ const server = await getServer(args);
200
+ const body = parseData(args.data);
201
+ const resp = await request(server.base, "/api/cardnews/export", { method: "POST", body })
202
+ .catch((e) => die(exitCodeForError(e), e.message));
203
+ if (args.json) { json(resp); return; }
204
+ out(color.green("✓ ") + "export requested");
205
+ if (resp && typeof resp === "object" && Object.keys(resp).length > 0) out(JSON.stringify(resp, null, 2));
206
+ }
207
+
208
+ type Sub = (argv: any[]) => Promise<void>;
209
+
210
+ export default async function cardnewsCmd(argv: string[]) {
211
+ const sub = argv[0];
212
+ if (!sub || sub === "--help" || sub === "-h") { out(HELP); return; }
213
+
214
+ if (sub === "template") {
215
+ const s2 = argv[1];
216
+ if (s2 === "preview") return templatePreviewSub(argv.slice(2));
217
+ die(2, `unknown subcommand: template ${s2 || ""}\n${HELP}`);
218
+ }
219
+ if (sub === "set") {
220
+ const s2 = argv[1];
221
+ const SET_SUB: Record<string, Sub> = { show: setShowSub, manifest: setManifestSub };
222
+ const handler = SET_SUB[s2];
223
+ if (!handler) die(2, `unknown subcommand: set ${s2 || ""}\n${HELP}`);
224
+ return handler(argv.slice(2));
225
+ }
226
+ if (sub === "job") {
227
+ const s2 = argv[1];
228
+ const JOB_SUB: Record<string, Sub> = { create: jobCreateSub, show: jobShowSub, retry: jobRetrySub };
229
+ const handler = JOB_SUB[s2];
230
+ if (!handler) die(2, `unknown subcommand: job ${s2 || ""}\n${HELP}`);
231
+ return handler(argv.slice(2));
232
+ }
233
+ if (sub === "card") {
234
+ const s2 = argv[1];
235
+ if (s2 === "regenerate") return cardRegenerateSub(argv.slice(2));
236
+ die(2, `unknown subcommand: card ${s2 || ""}\n${HELP}`);
237
+ }
238
+
239
+ const FLAT: Record<string, Sub> = {
240
+ templates: templatesSub,
241
+ sets: setsSub,
242
+ draft: draftSub,
243
+ generate: generateSub,
244
+ export: exportSub,
245
+ };
246
+ const handler = FLAT[sub];
247
+ if (!handler) die(2, `unknown subcommand: ${sub}\n${HELP}`);
248
+ return handler(argv.slice(1));
249
+ }
@@ -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) => { const err = e; die(exitCodeForError(e), `${err.message}${err.code ? ` (${err.code})` : ""}`); });
35
+ const target = String(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
+ }