1import { existsSync, readdirSync, readFileSync } from "fs";
2import { join, basename, extname } from "path";
3import PptxGenJS from "pptxgenjs";
4 
5interface SlideInfo {
6  filename: string;
7  path: string;
8  index: number;
9  promptPath?: string;
10}
11 
12function parseArgs(): { dir: string; output?: string } {
13  const args = process.argv.slice(2);
14  let dir = "";
15  let output: string | undefined;
16 
17  for (let i = 0; i < args.length; i++) {
18    if (args[i] === "--output" || args[i] === "-o") {
19      output = args[++i];
20    } else if (!args[i].startsWith("-")) {
21      dir = args[i];
22    }
23  }
24 
25  if (!dir) {
26    console.error("Usage: bun merge-to-pptx.ts <slide-deck-dir> [--output filename.pptx]");
27    process.exit(1);
28  }
29 
30  return { dir, output };
31}
32 
33function findSlideImages(dir: string): SlideInfo[] {
34  if (!existsSync(dir)) {
35    console.error(`Directory not found: ${dir}`);
36    process.exit(1);
37  }
38 
39  const files = readdirSync(dir);
40  const slidePattern = /^(\d+)-slide-.*\.(png|jpg|jpeg)$/i;
41  const promptsDir = join(dir, "prompts");
42  const hasPrompts = existsSync(promptsDir);
43 
44  const slides: SlideInfo[] = files
45    .filter((f) => slidePattern.test(f))
46    .map((f) => {
47      const match = f.match(slidePattern);
48      const baseName = f.replace(/\.(png|jpg|jpeg)$/i, "");
49      const promptPath = hasPrompts ? join(promptsDir, `${baseName}.md`) : undefined;
50 
51      return {
52        filename: f,
53        path: join(dir, f),
54        index: parseInt(match![1], 10),
55        promptPath: promptPath && existsSync(promptPath) ? promptPath : undefined,
56      };
57    })
58    .sort((a, b) => a.index - b.index);
59 
60  if (slides.length === 0) {
61    console.error(`No slide images found in: ${dir}`);
62    console.error("Expected format: 01-slide-*.png, 02-slide-*.png, etc.");
63    process.exit(1);
64  }
65 
66  return slides;
67}
68 
69function findBasePrompt(): string | undefined {
70  const scriptDir = import.meta.dir;
71  const basePromptPath = join(scriptDir, "..", "references", "base-prompt.md");
72  if (existsSync(basePromptPath)) {
73    return readFileSync(basePromptPath, "utf-8");
74  }
75  return undefined;
76}
77 
78async function createPptx(slides: SlideInfo[], outputPath: string) {
79  const pptx = new PptxGenJS();
80 
81  pptx.layout = "LAYOUT_16x9";
82  pptx.author = "baoyu-slide-deck";
83  pptx.subject = "Generated Slide Deck";
84 
85  const basePrompt = findBasePrompt();
86  let notesCount = 0;
87 
88  for (const slide of slides) {
89    const s = pptx.addSlide();
90    const imageData = readFileSync(slide.path);
91    const base64 = imageData.toString("base64");
92    const ext = extname(slide.filename).toLowerCase().replace(".", "");
93    const mimeType = ext === "png" ? "image/png" : "image/jpeg";
94 
95    s.addImage({
96      data: `data:${mimeType};base64,${base64}`,
97      x: 0,
98      y: 0,
99      w: "100%",
100      h: "100%",
101      sizing: { type: "cover", w: "100%", h: "100%" },
102    });
103 
104    if (slide.promptPath) {
105      const slidePrompt = readFileSync(slide.promptPath, "utf-8");
106      const fullNotes = basePrompt ? `${basePrompt}\n\n---\n\n${slidePrompt}` : slidePrompt;
107      s.addNotes(fullNotes);
108      notesCount++;
109    }
110 
111    console.log(`Added: ${slide.filename}${slide.promptPath ? " (with notes)" : ""}`);
112  }
113 
114  await pptx.writeFile({ fileName: outputPath });
115  console.log(`\nCreated: ${outputPath}`);
116  console.log(`Total slides: ${slides.length}`);
117  if (notesCount > 0) {
118    console.log(`Slides with notes: ${notesCount}${basePrompt ? " (includes base prompt)" : ""}`);
119  }
120}
121 
122async function main() {
123  const { dir, output } = parseArgs();
124  const slides = findSlideImages(dir);
125 
126  const dirName = basename(dir) === "slide-deck" ? basename(join(dir, "..")) : basename(dir);
127  const outputPath = output || join(dir, `${dirName}.pptx`);
128 
129  console.log(`Found ${slides.length} slides in: ${dir}\n`);
130 
131  await createPptx(slides, outputPath);
132}
133 
134main().catch((err) => {
135  console.error("Error:", err.message);
136  process.exit(1);
137});
138