1import assert from "node:assert/strict";
2import fs from "node:fs/promises";
3import os from "node:os";
4import path from "node:path";
5import { execFile } from "node:child_process";
6import { promisify } from "node:util";
7import test from "node:test";
8 
9const execFileAsync = promisify(execFile);
10const repoRoot = path.resolve(import.meta.dirname, "..", "..", "..");
11const scriptPath = path.join(repoRoot, "skills", "baoyu-image-gen", "scripts", "build-batch.ts");
12 
13async function makeFixture(): Promise<{
14  root: string;
15  outlinePath: string;
16  promptsDir: string;
17  outputPath: string;
18}> {
19  const root = await fs.mkdtemp(path.join(os.tmpdir(), "baoyu-image-gen-build-batch-"));
20  const outlinePath = path.join(root, "outline.md");
21  const promptsDir = path.join(root, "prompts");
22  const outputPath = path.join(root, "batch.json");
23 
24  await fs.mkdir(promptsDir, { recursive: true });
25  await fs.writeFile(
26    outlinePath,
27    `## Illustration 1
28**Position**: demo
29**Purpose**: demo
30**Visual Content**: demo
31**Filename**: 01-demo.png
32`,
33  );
34  await fs.writeFile(path.join(promptsDir, "01-demo.md"), "A demo prompt\n");
35 
36  return { root, outlinePath, promptsDir, outputPath };
37}
38 
39async function runBuildBatch(args: string[]): Promise<void> {
40  await execFileAsync(process.execPath, ["--import", "tsx", scriptPath, ...args], {
41    cwd: repoRoot,
42  });
43}
44 
45test("build-batch omits default model so baoyu-image-gen can resolve env or EXTEND defaults", async () => {
46  const fixture = await makeFixture();
47 
48  await runBuildBatch([
49    "--outline",
50    fixture.outlinePath,
51    "--prompts",
52    fixture.promptsDir,
53    "--output",
54    fixture.outputPath,
55  ]);
56 
57  const batch = JSON.parse(await fs.readFile(fixture.outputPath, "utf8")) as {
58    tasks: Array<Record<string, unknown>>;
59  };
60 
61  assert.equal(batch.tasks.length, 1);
62  assert.equal(batch.tasks[0]?.provider, "replicate");
63  assert.equal(Object.hasOwn(batch.tasks[0]!, "model"), false);
64});
65 
66test("build-batch preserves explicit model overrides", async () => {
67  const fixture = await makeFixture();
68 
69  await runBuildBatch([
70    "--outline",
71    fixture.outlinePath,
72    "--prompts",
73    fixture.promptsDir,
74    "--output",
75    fixture.outputPath,
76    "--model",
77    "acme/custom-model",
78  ]);
79 
80  const batch = JSON.parse(await fs.readFile(fixture.outputPath, "utf8")) as {
81    tasks: Array<Record<string, unknown>>;
82  };
83 
84  assert.equal(batch.tasks[0]?.model, "acme/custom-model");
85});
86 
87test("build-batch propagates direct-usage references from prompt frontmatter", async () => {
88  const fixture = await makeFixture();
89 
90  await fs.writeFile(
91    path.join(fixture.promptsDir, "01-demo.md"),
92    `---
93illustration_id: 01
94type: infographic
95references:
96  - ref_id: 01
97    filename: 01-ref-brand.png
98    usage: direct
99  - ref_id: 02
100    filename: 02-ref-style.png
101    usage: style
102---
103 
104A demo prompt
105`,
106  );
107 
108  await runBuildBatch([
109    "--outline",
110    fixture.outlinePath,
111    "--prompts",
112    fixture.promptsDir,
113    "--output",
114    fixture.outputPath,
115  ]);
116 
117  const batch = JSON.parse(await fs.readFile(fixture.outputPath, "utf8")) as {
118    tasks: Array<Record<string, unknown>>;
119  };
120 
121  assert.deepEqual(batch.tasks[0]?.ref, ["references/01-ref-brand.png"]);
122});
123 
124test("build-batch omits ref field when no direct references exist", async () => {
125  const fixture = await makeFixture();
126 
127  await fs.writeFile(
128    path.join(fixture.promptsDir, "01-demo.md"),
129    `---
130illustration_id: 01
131references:
132  - ref_id: 01
133    filename: 01-ref-palette.png
134    usage: palette
135---
136 
137A demo prompt
138`,
139  );
140 
141  await runBuildBatch([
142    "--outline",
143    fixture.outlinePath,
144    "--prompts",
145    fixture.promptsDir,
146    "--output",
147    fixture.outputPath,
148  ]);
149 
150  const batch = JSON.parse(await fs.readFile(fixture.outputPath, "utf8")) as {
151    tasks: Array<Record<string, unknown>>;
152  };
153 
154  assert.equal(Object.hasOwn(batch.tasks[0]!, "ref"), false);
155});
156 
157test("build-batch honors --refs-dir override", async () => {
158  const fixture = await makeFixture();
159 
160  await fs.writeFile(
161    path.join(fixture.promptsDir, "01-demo.md"),
162    `---
163illustration_id: 01
164references:
165  - ref_id: 01
166    filename: brand.png
167    usage: direct
168---
169 
170A demo prompt
171`,
172  );
173 
174  await runBuildBatch([
175    "--outline",
176    fixture.outlinePath,
177    "--prompts",
178    fixture.promptsDir,
179    "--output",
180    fixture.outputPath,
181    "--refs-dir",
182    "refs",
183  ]);
184 
185  const batch = JSON.parse(await fs.readFile(fixture.outputPath, "utf8")) as {
186    tasks: Array<Record<string, unknown>>;
187  };
188 
189  assert.deepEqual(batch.tasks[0]?.ref, ["refs/brand.png"]);
190});
191