add cover and some tweaks

3 years ago · 40adfb49c9
parent 42485da43b
commit 40adfb49c9
3 changed files with 73 additions and 5 deletions
--- a/cover.html
+++ b/cover.html
@ -0,0 +1,30 @@
 <!DOCTYPE html>
 <html lang="en">
 <head>
    <meta charset="UTF-8">
    <title>Cover</title>
    <style>
        html, body{
            height: 100%;
            width: 100%;
            margin: 0;
            font-family: Helvetica, Arial, sans-serif;
        }
        body{
            display: flex;
            flex-flow: column;
            justify-content: center;
            align-items: center;
        }
        .subtitle{
            font-weight: bold;
            font-size: 1.2rem ;
        }
    </style>
 </head>
 <body>
    <h1>{{title}}</h1>
    <p class="subtitle">{{subtitle}}</p>
    <p class="date">{{date}}</p>
 </body>
 </html>
--- a/index.js
+++ b/index.js
@ -1,9 +1,14 @@
 import puppeteer from 'puppeteer';
 import fs from 'fs';
 import moment from 'moment';
 import PDFMerger from 'pdf-merger-js';
-const [url, title = '', subtitle = ''] = process.argv.slice(2);
+let [
-const date = moment().format('YYYY-MM-DD');
+  url,
  title = '',
  subtitle = '',
  date = moment().format('YYYY-MM-DD')
 ] = process.argv.slice(2);
 const path = './pdf';
@ -29,8 +34,12 @@ try{
    new Promise(resolve => setTimeout(resolve,1000))
  ]);
-  await page.pdf({
+  if (!title) title = await (await page.waitForSelector('h1')).evaluate(el => el.textContent);
-    path: `${path}/${date}_${title}.pdf`,
+  if (!subtitle) subtitle = await (await page.waitForSelector('h2')).evaluate(el => el.textContent);
  console.log(title,subtitle)
  const pages = await page.pdf({
    //path: `${path}/${date}_${title}_pages.pdf`,
    format: 'A4',
    margin: {
      top: '20mm',
@ -48,7 +57,7 @@ try{
        width: 100%;
        padding: 2mm;
        border-bottom: 1px #222 solid;
-      ">${[title, subtitle].join(' - ')}</div>
+      ">${[title, subtitle].map(el=>el.trim()).filter(el=>!!el).join(' - ')}</div>
    `,
    footerTemplate: `
      <div style="
@ -63,7 +72,35 @@ try{
    printBackground: true
  });
  let coverHTML = fs.readFileSync('./cover.html', {encoding: 'utf-8', flag: 'r'});
  Object.entries({
    title, subtitle, date
  }).forEach(([key, value]) => {
    coverHTML = coverHTML.replaceAll(`{{${key}}}`, value);
  });
  const page2 = await browser.newPage();
  await page2.setContent(coverHTML);
  const cover = await page2.pdf({
    //path: `${path}/${date}_${title}_cover.pdf`,
    format: 'A4',
    margin: {
      top: '20mm',
      bottom: '20mm',
      left: '12mm',
      right: '12mm'
    }
  });
  await browser.close();
  const merger = new PDFMerger();
  merger.add(cover);
  merger.add(pages);
  await merger.save(`${path}/${date}_${title.trim().replaceAll(' ', '_')}.pdf`)
 } catch (e) {
  if (browser) await browser.close();
  throw e;
--- a/package.json
+++ b/package.json
@ -12,6 +12,7 @@
  "license": "",
  "dependencies": {
    "moment": "^2.29.2",
    "pdf-merger-js": "^3.4.0",
    "puppeteer": "^13.5.2"
  }
 }