|
| 1 | +import { promises as fs } from 'fs' |
| 2 | +import { chromium } from 'playwright' |
| 3 | +import { InlineConfig } from 'vite' |
| 4 | +import { PDFDocument } from 'pdf-lib' |
| 5 | +import { green } from 'chalk' |
| 6 | +import { createServer } from './server' |
| 7 | +import { parseSlidesMarkdown } from './parser' |
| 8 | + |
| 9 | +export async function genratePDF(entry = 'slides.md', config: InlineConfig = {}) { |
| 10 | + const pages = parseSlidesMarkdown(await fs.readFile(entry, 'utf-8')) |
| 11 | + const server = await createServer(entry, { |
| 12 | + ...config, |
| 13 | + logLevel: 'silent', |
| 14 | + clearScreen: false, |
| 15 | + }) |
| 16 | + const port = 18724 |
| 17 | + await server.listen(port) |
| 18 | + |
| 19 | + const browser = await chromium.launch() |
| 20 | + const context = await browser.newContext({ |
| 21 | + viewport: { |
| 22 | + width: 1920, |
| 23 | + height: 1080, |
| 24 | + }, |
| 25 | + deviceScaleFactor: 1, |
| 26 | + }) |
| 27 | + const page = await context.newPage() |
| 28 | + |
| 29 | + const buffers: Buffer[] = [] |
| 30 | + const pagesCount = pages.length - 1 |
| 31 | + for (let i = 0; i < pagesCount; i++) { |
| 32 | + console.log(`Exporting: ${i + 1} / ${pagesCount}`) |
| 33 | + await page.goto(`http://localhost:${port}/${i}?print`, { |
| 34 | + waitUntil: 'networkidle', |
| 35 | + }) |
| 36 | + await page.emulateMedia({ media: 'screen' }) |
| 37 | + const pdf = await page.pdf({ |
| 38 | + width: 1920, |
| 39 | + height: 1080, |
| 40 | + margin: { |
| 41 | + left: 0, |
| 42 | + top: 0, |
| 43 | + right: 0, |
| 44 | + bottom: 0, |
| 45 | + }, |
| 46 | + pageRanges: '1', |
| 47 | + printBackground: true, |
| 48 | + preferCSSPageSize: true, |
| 49 | + }) |
| 50 | + buffers.push(pdf) |
| 51 | + } |
| 52 | + |
| 53 | + const mergedPdf = await PDFDocument.create({}) |
| 54 | + for (const pdfBytes of buffers) { |
| 55 | + const pdf = await PDFDocument.load(pdfBytes) |
| 56 | + const copiedPages = await mergedPdf.copyPages(pdf, pdf.getPageIndices()) |
| 57 | + copiedPages.forEach((page) => { |
| 58 | + mergedPdf.addPage(page) |
| 59 | + }) |
| 60 | + } |
| 61 | + |
| 62 | + const buffer = await mergedPdf.save() |
| 63 | + await fs.writeFile('slides.pdf', buffer) |
| 64 | + console.log(green`Exporting finished: ./slides.pdf`) |
| 65 | + |
| 66 | + browser.close() |
| 67 | + server.close() |
| 68 | + process.exit(0) |
| 69 | +} |
0 commit comments