feat(setup): add puppeteerLaunchOptions param

Author: condorheroblog

src/commands/htmlExportPdf/htmlExportPdf.ts

@@ -134,7 +134,7 @@ export async function htmlExportPdf(args: undefined | string[], options: HtmlExp
 			output = replaceExt(output, ".html");
 		}
 		else if (options.debug === true) {
-			await printer.preview(inputPath);
+			await printer.render(inputPath);
 		}
 		else {
 			const format = options.pageSize;
@@ -175,7 +175,7 @@ export async function htmlExportPdf(args: undefined | string[], options: HtmlExp
 	});
 
 	await Promise.all(promises);
-	await printer.close();
+	await printer.closeBrowser();
 	progress.stop();
 	!isSingleFile && process.stdout.write(`\n\n ${green("  ✓ ")}${dim("Saved to ")} ${path.join(dir, options.outDir ?? "")}\n\n`);
 	process.exit(0);

src/core/printer.ts

@@ -76,13 +76,14 @@ export class Printer extends EventEmitter {
 		}
 	}
 
-	async setup() {
+	async setup(puppeteerLaunchOptions?: LaunchOptions) {
 		const puppeteerOptions = {
 			// https://github.com/puppeteer/puppeteer/issues/2735#issuecomment-470309033
 			// pipe: true,
 			headless: this.headless,
 			args: ["--disable-dev-shm-usage", "--export-tagged-pdf"],
 			ignoreHTTPSErrors: this.ignoreHTTPSErrors,
+			...puppeteerLaunchOptions,
 		};
 
 		if (this.allowLocal)
@@ -168,7 +169,7 @@ export class Printer extends EventEmitter {
 			return page;
 		}
 		catch (error) {
-			this.closeAfter && this.close();
+			this.closeAfter && await this.closeBrowser();
 			throw error;
 		}
 	}
@@ -232,7 +233,7 @@ export class Printer extends EventEmitter {
 					throw e;
 				});
 
-			this.closeAfter && page.close();
+			this.closeAfter && await page.close();
 			this.pages.delete(input);
 
 			this.emit("postprocessing");
@@ -244,32 +245,35 @@ export class Printer extends EventEmitter {
 			return await pdfDoc.save();
 		}
 		catch (error) {
-			this.closeAfter && this.close();
+			this.closeAfter && await this.closeBrowser();
 			throw error;
 		}
 	}
 
 	async html(input: string) {
-		const page = await this.render(input);
+		let page = this.pages.get(input);
+		if (!page) {
+			page = await this.render(input)
+				.catch((e) => {
+					throw e;
+				});
+		}
 
 		const content = await page.content();
 
 		if (this.closeAfter) {
-			page.close();
-			this.close();
+			await page.close();
+			await this.closeBrowser();
 		}
 
 		return content;
 	}
 
-	async preview(input: string) {
-		const page = await this.render(input);
-		this.closeAfter && this.close();
-		return page;
-	}
-
-	async close() {
-		return this.browser && this.browser.close();
+	async closeBrowser() {
+		if (this.browser) {
+			await this.browser.close();
+			this.browser = undefined;
+		}
 	}
 
 	needsAllowedRules() {