Refactor

Author: hiresh-verma

index.js

@@ -4,22 +4,23 @@ const hb = require("handlebars");
 const fs = require("fs");
 const inlineCss = require("inline-css");
 module.exports;
+
 async function generatePdf(file, options, callback, font) {
-  // we are using headless mode
-  let args = ["--no-sandbox", "--disable-setuid-sandbox"];
-  if (options.args) {
-    args = options.args;
-    delete options.args;
-  }
+    // we are using headless mode
+    let args = ["--no-sandbox", "--disable-setuid-sandbox"];
+    if (options.args) {
+        args = options.args;
+        delete options.args;
+    }
 
-  const browser = await puppeteer.launch({
-    args: args,
-  });
-  const page = await browser.newPage();
-  let embeddedFontStyle = "";
-  if (file.content) {
-    if (font) {
-      embeddedFontStyle = `
+    const browser = await puppeteer.launch({
+        args: args,
+    });
+    const page = await browser.newPage();
+    let embeddedFontStyle = "";
+    if (file.content) {
+        if (font) {
+            embeddedFontStyle = `
     <style>
     @font-face {
         font-family: ${font.name};
@@ -29,76 +30,76 @@ async function generatePdf(file, options, callback, font) {
       }
     </style>
   `;
-    }
-    data = await inlineCss(file.content, { url: "/" });
-    data = embeddedFontStyle + data;
+        }
+        data = await inlineCss(file.content, {url: "/"});
+        data = embeddedFontStyle + data;
 
-    console.log("Compiling the template with handlebars");
-    // we have compile our code with handlebars
-    const template = hb.compile(data, { strict: true });
-    const result = template(data);
-    const html = result;
+        console.log("Compiling the template with handlebars");
+        // we have compile our code with handlebars
+        const template = hb.compile(data, {strict: true});
+        const result = template(data);
+        const html = result;
 
-    // We set the page content as the generated html by handlebars
-    await page.setContent(html, {
-      waitUntil: "networkidle0", // wait for page to load completely
-    });
-  } else {
-    await page.goto(file.url, {
-      waitUntil: ["load", "networkidle0"], // wait for page to load completely
-    });
-  }
+        // We set the page content as the generated html by handlebars
+        await page.setContent(html, {
+            waitUntil: "networkidle0", // wait for page to load completely
+        });
+    } else {
+        await page.goto(file.url, {
+            waitUntil: ["load", "networkidle0"], // wait for page to load completely
+        });
+    }
 
-  return Promise.props(page.pdf(options))
-    .then(async function (data) {
-      await browser.close();
+    return Promise.props(page.pdf(options))
+        .then(async function (data) {
+            await browser.close();
 
-      return Buffer.from(Object.values(data));
-    })
-    .asCallback(callback);
+            return Buffer.from(Object.values(data));
+        })
+        .asCallback(callback);
 }
 
 async function generatePdfs(files, options, callback) {
-  // we are using headless mode
-  let args = ["--no-sandbox", "--disable-setuid-sandbox"];
-  if (options.args) {
-    args = options.args;
-    delete options.args;
-  }
-  const browser = await puppeteer.launch({
-    args: args,
-  });
-  let pdfs = [];
-  const page = await browser.newPage();
-  for (let file of files) {
-    if (file.content) {
-      data = await inlineCss(file.content, { url: "/" });
-      console.log("Compiling the template with handlebars");
-      // we have compile our code with handlebars
-      const template = hb.compile(data, { strict: true });
-      const result = template(data);
-      const html = result;
-      // We set the page content as the generated html by handlebars
-      await page.setContent(html, {
-        waitUntil: "networkidle0", // wait for page to load completely
-      });
-    } else {
-      await page.goto(file.url, {
-        waitUntil: "networkidle0", // wait for page to load completely
-      });
+    // we are using headless mode
+    let args = ["--no-sandbox", "--disable-setuid-sandbox"];
+    if (options.args) {
+        args = options.args;
+        delete options.args;
+    }
+    const browser = await puppeteer.launch({
+        args: args,
+    });
+    let pdfs = [];
+    const page = await browser.newPage();
+    for (let file of files) {
+        if (file.content) {
+            data = await inlineCss(file.content, {url: "/"});
+            console.log("Compiling the template with handlebars");
+            // we have compile our code with handlebars
+            const template = hb.compile(data, {strict: true});
+            const result = template(data);
+            const html = result;
+            // We set the page content as the generated html by handlebars
+            await page.setContent(html, {
+                waitUntil: "networkidle0", // wait for page to load completely
+            });
+        } else {
+            await page.goto(file.url, {
+                waitUntil: "networkidle0", // wait for page to load completely
+            });
+        }
+        let pdfObj = JSON.parse(JSON.stringify(file));
+        delete pdfObj["content"];
+        pdfObj["buffer"] = Buffer.from(Object.values(await page.pdf(options)));
+        pdfs.push(pdfObj);
     }
-    let pdfObj = JSON.parse(JSON.stringify(file));
-    delete pdfObj["content"];
-    pdfObj["buffer"] = Buffer.from(Object.values(await page.pdf(options)));
-    pdfs.push(pdfObj);
-  }
 
-  return Promise.resolve(pdfs)
-    .then(async function (data) {
-      await browser.close();
-      return data;
-    })
-    .asCallback(callback);
+    return Promise.resolve(pdfs)
+        .then(async function (data) {
+            await browser.close();
+            return data;
+        })
+        .asCallback(callback);
 }
 
 module.exports.generatePdf = generatePdf;