1import { existsSync, readdirSync, readFileSync } from "fs";
2import { join, basename } from "path";
3import { PDFDocument, rgb } from "pdf-lib";
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-pdf.ts <slide-deck-dir> [--output filename.pdf]");
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 
69async function createPdf(slides: SlideInfo[], outputPath: string) {
70  const pdfDoc = await PDFDocument.create();
71  pdfDoc.setAuthor("baoyu-slide-deck");
72  pdfDoc.setSubject("Generated Slide Deck");
73 
74  for (const slide of slides) {
75    const imageData = readFileSync(slide.path);
76    const isPng = imageData[0] === 0x89 && imageData[1] === 0x50 && imageData[2] === 0x4e && imageData[3] === 0x47;
77    const image = isPng
78      ? await pdfDoc.embedPng(imageData)
79      : await pdfDoc.embedJpg(imageData);
80 
81    const { width, height } = image;
82    const page = pdfDoc.addPage([width, height]);
83 
84    page.drawImage(image, {
85      x: 0,
86      y: 0,
87      width,
88      height,
89    });
90 
91    console.log(`Added: ${slide.filename}${slide.promptPath ? " (prompt available)" : ""}`);
92  }
93 
94  const pdfBytes = await pdfDoc.save();
95  await Bun.write(outputPath, pdfBytes);
96 
97  console.log(`\nCreated: ${outputPath}`);
98  console.log(`Total pages: ${slides.length}`);
99}
100 
101async function main() {
102  const { dir, output } = parseArgs();
103  const slides = findSlideImages(dir);
104 
105  const dirName = basename(dir) === "slide-deck" ? basename(join(dir, "..")) : basename(dir);
106  const outputPath = output || join(dir, `${dirName}.pdf`);
107 
108  console.log(`Found ${slides.length} slides in: ${dir}\n`);
109 
110  await createPdf(slides, outputPath);
111}
112 
113main().catch((err) => {
114  console.error("Error:", err.message);
115  process.exit(1);
116});
117