add endpoint from-html

Author: fabriciojs

package.json

@@ -3,6 +3,7 @@
     "start": "node pdf-service.js"
   },
   "dependencies": {
+    "body-parser": "^1.19.0",
     "express": "^4.17.1",
     "puppeteer": "^5.2.1"
   },

pdf-service.js

@@ -4,6 +4,10 @@ const path = require('path');
 const fs = require('fs');
 const crypto = require('crypto');
 const app = express();
+const bodyParser = require('body-parser');
+
+app.use(bodyParser.json()); // support json encoded bodies
+app.use(bodyParser.urlencoded({ extended: true })); // support encoded bodies
 
 let port = 3000;
 let calls = {
@@ -20,7 +24,25 @@ app.get('/health', (req, res) => {
     deliverJson(res, {isReady, calls}, isReady ? 200 : 503);
 });
 
-app.get('/from-html', async (req, res) => {
+app.post('/from-html', async (req, res) => {
+    const html = req.body.html;
+    const htmlFile = md5(html.substr(0, 100)) + '.html';
+    const fullHtmlPath = path.join(__dirname, storagePath, htmlFile);
+
+    fs.writeFileSync(fullHtmlPath, html);
+
+    let pdfFilePath;
+    try {
+        pdfFilePath = await generatePdf(`file://${fullHtmlPath}`);
+    } catch (err) {
+        console.log('/from-html: error generating PDF', e);
+        let msg = 'failure generating PDF';
+        deliverJson(res, {msg, err}, 500);
+        return;
+    }
+
+    deliverPdfFile(res, pdfFilePath);
+    fs.unlinkSync(fullHtmlPath);
 });
 
 app.get('/from-url', async (req, res) => {
@@ -75,7 +97,7 @@ async function generatePdf(url) {
         waitUntil: 'networkidle2',
     });
 
-    let filename = crypto.createHash('md5').update(url).digest('hex') + '.pdf';
+    let filename = md5(url) + '.pdf';
     const pdfFolder = path.join(__dirname, storagePath);
     if (!fs.existsSync(pdfFolder)) {
         fs.mkdirSync(pdfFolder);
@@ -95,6 +117,10 @@ async function generatePdf(url) {
     return pdfFilePath;
 }
 
+function md5(seed) {
+    return crypto.createHash('md5').update(seed).digest('hex');
+}
+
 app.listen(port, () => {
     console.log(`Kool PDF service running at port ${port}`);
     console.log(`Going to start puppeter`);

yarn.lock

@@ -51,7 +51,7 @@ bl@^4.0.1:
     inherits "^2.0.4"
     readable-stream "^3.4.0"
 
-body-parser@1.19.0:
+body-parser@1.19.0, body-parser@^1.19.0:
   version "1.19.0"
   resolved "https://registry.yarnpkg.com/body-parser/-/body-parser-1.19.0.tgz#96b2709e57c9c4e09a6fd66a8fd979844f69f08a"
   integrity sha512-dhEPs72UPbDnAQJ9ZKMNTP6ptJaionhP5cBb541nXPlW60Jepo9RV/a4fX4XWW9CuFNK22krhrj1+rgzifNCsw==