add userAgent and base64

Author: iamdual

README.md

@@ -64,6 +64,8 @@ html2pdf ./example.html --format A4 --pageRanges 1 -o example.pdf
 | scale      | float   | Set scale. The value must be greater than or equals to `0.1`, or lower than or equals to `2`. Default is `1`.                       |
 | landscape  | boolean | Enable landscape mode. To enable, the value should be true. Disabled by default.                                                    |
 | margin     | string  | Set margin(s) for the PDF document. It can be all four margin or specified by the values separated with space. Default is `0`.      |
+| userAgent  | string  | Set custom user-agent string.                                                                                                       |
+| base64     | boolean | Return the PDF as a base64-encoded string (REST API only).                                                                          |
 
 ## Author
 

src/config.ts

@@ -35,6 +35,8 @@ export default class Config {
   readonly scale: number = 1;
   readonly landscape: boolean = false;
   readonly margin: Margin = new Margin();
+  readonly userAgent?: string;
+  readonly base64: boolean = false;
 
   constructor(params?: { [key: string]: any }) {
     for (const param in params) {
@@ -142,6 +144,19 @@ export default class Config {
             );
           }
           break;
+        case "userAgent":
+          if (typeof params[param] === "string") {
+            this.userAgent = params[param];
+          }
+          break;
+        case "base64":
+          if (
+            typeof params[param] === "string" ||
+            typeof params[param] === "boolean"
+          ) {
+            this.base64 = params[param];
+          }
+          break;
       }
     }
   }

src/generator/index.ts

@@ -12,7 +12,7 @@ if (process.env.HTML2PDF_NO_SANDBOX) {
   args.push("--no-sandbox");
 }
 
-export default async (config: Config) => {
+export default async (config: Config): Promise<Buffer> => {
   // Create a browser instance
   const browser = await puppeteer.launch({
     headless: true,
@@ -31,6 +31,11 @@ export default async (config: Config) => {
   // Set CSS media type
   await page.emulateMediaType(config.mediaType);
 
+  if (config.userAgent) {
+    // Set user-agent
+    await page.setUserAgent(config.userAgent);
+  }
+
   if (config.isUrl) {
     // https://pptr.dev/api/puppeteer.page.goto
     await page.goto(config.source, { waitUntil: "networkidle0" });

src/rest-api/generate.ts

@@ -33,6 +33,13 @@ export default async (req: Request, res: Response) => {
 
   try {
     const pdf = await Generator(config);
+
+    if (config.base64) {
+      res.set("Content-Type", "text/plain");
+      res.status(200).send(Buffer.from(pdf).toString("base64"));
+      return;
+    }
+
     res.set("Content-Type", "application/pdf");
     res.status(200).send(pdf);
     return;