90 lines
2.3 KiB
TypeScript
90 lines
2.3 KiB
TypeScript
import puppeteer from 'puppeteer';
|
|
import fs from 'node:fs';
|
|
import path from 'node:path';
|
|
import { PDFDocument } from 'pdf-lib';
|
|
|
|
async function exportPDF(): Promise<void> {
|
|
const browser = await puppeteer.launch();
|
|
const page = await browser.newPage();
|
|
|
|
// 设置页面样式适应 PDF
|
|
await page.setViewport({ width: 1200, height: 800 });
|
|
|
|
// const baseUrl = 'http://localhost:5173'; // VitePress 开发服务器地址
|
|
const baseUrl = 'https://look-good.xiongxiao.me'; // VitePress 开发服务器生产地址
|
|
const outputPath = path.resolve(process.cwd(), 'pdf');
|
|
|
|
if (!fs.existsSync(outputPath)) {
|
|
fs.mkdirSync(outputPath);
|
|
}
|
|
|
|
// 根据 sidebar 配置，导出的路径列表
|
|
const routes: string[] = [
|
|
'/',
|
|
'/preface',
|
|
'/self-introduction',
|
|
'/book/0-1',
|
|
'/book/01',
|
|
'/book/02',
|
|
'/book/03',
|
|
'/book/0-2',
|
|
'/book/04',
|
|
'/book/05',
|
|
'/book/06',
|
|
'/book/07',
|
|
'/book/08',
|
|
'/book/09',
|
|
'/book/0-3',
|
|
'/book/10',
|
|
'/book/11',
|
|
'/book/12',
|
|
'/appendix/a',
|
|
'/appendix/b',
|
|
'/appendix/c',
|
|
'/appendix/d',
|
|
'/appendix/e'
|
|
];
|
|
|
|
const pdfFiles: string[] = [];
|
|
|
|
for (const route of routes) {
|
|
const url = baseUrl + route;
|
|
const fileName = route === '/' ? 'index' : route.slice(1).replace(/\//g, '_');
|
|
const pdfPath = path.join(outputPath, `${fileName}.pdf`);
|
|
console.log(`⏳ 正在生成: ${fileName}.pdf from ${url}`);
|
|
await page.goto(url, { waitUntil: 'networkidle0' });
|
|
|
|
await page.pdf({
|
|
path: pdfPath,
|
|
format: 'A4',
|
|
printBackground: true,
|
|
margin: {
|
|
top: '20mm',
|
|
right: '20mm',
|
|
bottom: '20mm',
|
|
left: '20mm',
|
|
},
|
|
});
|
|
|
|
pdfFiles.push(pdfPath);
|
|
console.log(`✅ 已生成: ${fileName}.pdf`);
|
|
}
|
|
|
|
await browser.close();
|
|
|
|
// 合并所有 PDF
|
|
const mergedPdf = await PDFDocument.create();
|
|
for (const pdfFile of pdfFiles) {
|
|
const pdfBytes = fs.readFileSync(pdfFile);
|
|
const pdf = await PDFDocument.load(pdfBytes);
|
|
const copiedPages = await mergedPdf.copyPages(pdf, pdf.getPageIndices());
|
|
copiedPages.forEach((page) => mergedPdf.addPage(page));
|
|
}
|
|
const mergedPdfBytes = await mergedPdf.save();
|
|
const bookPdfPath = path.join(outputPath, 'book.pdf');
|
|
fs.writeFileSync(bookPdfPath, mergedPdfBytes);
|
|
console.log(`🎉 已合并为: book.pdf`);
|
|
}
|
|
|
|
exportPDF();
|