feat: add enhanceApp API

Author: condorheroblog

packages/vuepress-plugin-export-pdf-v2/README.md

@@ -109,6 +109,7 @@ config options:
 - `routePatterns` - Specify the patterns of files you want to be exported. The patterns are relative to the source directory (default `["/**", "!/404.html"]`).Patterns to match Route path using [multimatch](https://github.com/sindresorhus/multimatch)
 - `puppeteerLaunchOptions` - [Puppeteer launch options object](https://github.com/puppeteer/puppeteer/blob/main/docs/api/puppeteer.puppeteerlaunchoptions.md)
 - `pdfOptions` - [Valid options to configure PDF generation via Page.pdf()](https://github.com/puppeteer/puppeteer/blob/main/docs/api/puppeteer.pdfoptions.md) (default `{ format: 'A4 }`)
+- `enhanceApp` - Enhanceapp is a function that is executed before generating PDF. It receives two parameters: the created [browser](https://github.com/puppeteer/puppeteer/blob/main/docs/api/puppeteer.browser.md) instance and the [page](https://github.com/puppeteer/puppeteer/blob/main/docs/api/puppeteer.page.md) instance
 
 ## PDF print style
 

packages/vuepress-plugin-export-pdf-v2/src/configs/userConfigTypes.ts

@@ -1,6 +1,7 @@
-import type { PDFOptions, PuppeteerLaunchOptions } from "puppeteer";
+import type { Browser, Page as BrowserPage, PDFOptions, PuppeteerLaunchOptions } from "puppeteer";
 import type { Bundler, Page, Theme } from "vuepress";
 
+export type EnhanceApp = (browser: Browser, browserPage: BrowserPage) => void;
 export type UserSorter = (a: Page, b: Page) => number;
 
 /**
@@ -15,4 +16,5 @@ export interface UserConfig {
   pdfOptions?: PDFOptions
   outFile?: string
   outDir?: string
+  enhanceApp?: EnhanceApp
 }

packages/vuepress-plugin-export-pdf-v2/src/dev-server/vuePressServer.ts

@@ -57,6 +57,7 @@ export const vuePressServer = async(dir = "docs", commandOptions: CommandOptions
     outFile = vuepressOutFile,
     outDir = vuepressOutDir,
     routePatterns,
+    enhanceApp,
   } = userConfig;
 
   const devApp = createDevApp({ source: sourceDir, bundler, theme, host: "localhost", port: 8714 });
@@ -81,6 +82,7 @@ export const vuePressServer = async(dir = "docs", commandOptions: CommandOptions
       puppeteerLaunchOptions,
       pdfOptions,
       routePatterns,
+      enhanceApp,
     });
   }
   catch (error) {

packages/vuepress-plugin-export-pdf-v2/src/utils/generatePdf.ts

@@ -2,7 +2,7 @@ import type { DevApp } from "vuepress";
 import type { PDFOptions, PuppeteerLaunchOptions } from "puppeteer";
 import { chalk, fs, path } from "@vuepress/utils";
 import puppeteer from "puppeteer";
-import type { UserSorter } from "../configs";
+import type { EnhanceApp, UserSorter } from "../configs";
 
 import { filterRoute, singleProgressBar } from "../utils";
 import { mergePDF } from "./mergePDF";
@@ -16,6 +16,7 @@ interface IGeneratePdfOptions {
   sorter?: UserSorter
   puppeteerLaunchOptions?: PuppeteerLaunchOptions
   pdfOptions?: PDFOptions
+  enhanceApp?: EnhanceApp
 }
 
 const { yellow, gray } = chalk;
@@ -35,6 +36,7 @@ export const generatePdf = async(ctx: DevApp, {
   puppeteerLaunchOptions,
   pdfOptions,
   routePatterns,
+  enhanceApp,
 }: IGeneratePdfOptions) => {
   const { pages, options: { temp: tempPath } } = ctx;
   const tempPdfDir = join(tempPath, "pdf");
@@ -61,6 +63,7 @@ export const generatePdf = async(ctx: DevApp, {
 
   for (const { location, pagePath, url, title } of normalizePages) {
     const browserPage = await browser.newPage();
+    typeof enhanceApp === "function" && enhanceApp(browser, browserPage);
     browserPage.setDefaultNavigationTimeout(0);
 
     await browserPage.goto(

packages/vuepress-plugin-export-pdf/README.md

@@ -129,6 +129,7 @@ config options:
 - `routePatterns` - Specify the patterns of files you want to be exported. The patterns are relative to the source directory (default `["/**", "!/404.html"]`).Patterns to match Route path using [multimatch](https://github.com/sindresorhus/multimatch)
 - `puppeteerLaunchOptions` - [Puppeteer launch options object](https://github.com/puppeteer/puppeteer/blob/main/docs/api/puppeteer.puppeteerlaunchoptions.md)
 - `pdfOptions` - [Valid options to configure PDF generation via Page.pdf()](https://github.com/puppeteer/puppeteer/blob/main/docs/api/puppeteer.pdfoptions.md) (default `{ format: 'A4 }`)
+- `enhanceApp` - Enhanceapp is a function that is executed before generating PDF. It receives two parameters: the created [browser](https://github.com/puppeteer/puppeteer/blob/main/docs/api/puppeteer.browser.md) instance and the [page](https://github.com/puppeteer/puppeteer/blob/main/docs/api/puppeteer.page.md) instance
 
 ## PDF print style
 

packages/vuepress-plugin-export-pdf/src/configs/userConfigTypes.ts

@@ -1,6 +1,7 @@
-import type { PDFOptions, PuppeteerLaunchOptions } from "puppeteer";
+import type { Browser, Page as BrowserPage, PDFOptions, PuppeteerLaunchOptions } from "puppeteer";
 import type { Page } from "vuepress";
 
+export type EnhanceApp = (browser: Browser, browserPage: BrowserPage) => void;
 export type UserSorter = (a: Page, b: Page) => number;
 
 /**
@@ -14,4 +15,5 @@ export interface UserConfig {
   pdfOptions?: PDFOptions
   outFile?: string
   outDir?: string
+  enhanceApp?: EnhanceApp
 }

packages/vuepress-plugin-export-pdf/src/dev-server/vuePressServer.ts

@@ -62,6 +62,7 @@ export const vuePressServer = async(dir = "docs", commandOptions: CommandOptions
     outDir = vuepressOutDir,
     theme = vuepressTheme,
     routePatterns,
+    enhanceApp,
   } = userConfig;
 
   const devContext = await dev({
@@ -94,6 +95,7 @@ export const vuePressServer = async(dir = "docs", commandOptions: CommandOptions
       puppeteerLaunchOptions,
       pdfOptions,
       routePatterns,
+      enhanceApp,
     });
   }
   catch (error) {

packages/vuepress-plugin-export-pdf/src/utils/generatePdf.ts

@@ -2,7 +2,7 @@ import type { Context } from "vuepress";
 import type { PDFOptions, PuppeteerLaunchOptions } from "puppeteer";
 import { chalk, fs, path } from "@vuepress/utils";
 import puppeteer from "puppeteer";
-import type { UserSorter } from "../configs";
+import type { EnhanceApp, UserSorter } from "../configs";
 
 import { filterRoute, singleProgressBar } from "../utils";
 import { mergePDF } from "./mergePDF";
@@ -16,6 +16,7 @@ interface IGeneratePdfOptions {
   sorter?: UserSorter
   puppeteerLaunchOptions?: PuppeteerLaunchOptions
   pdfOptions?: PDFOptions
+  enhanceApp?: EnhanceApp
 }
 
 const { yellow, gray } = chalk;
@@ -35,6 +36,7 @@ export const generatePdf = async(ctx: Context, {
   puppeteerLaunchOptions,
   pdfOptions,
   routePatterns,
+  enhanceApp,
 }: IGeneratePdfOptions) => {
   const { pages, tempPath } = ctx;
   const tempPdfDir = join(tempPath, "pdf");
@@ -61,6 +63,7 @@ export const generatePdf = async(ctx: Context, {
 
   for (const { location, pagePath, url, title } of normalizePages) {
     const browserPage = await browser.newPage();
+    typeof enhanceApp === "function" && enhanceApp(browser, browserPage);
     browserPage.setDefaultNavigationTimeout(0);
 
     await browserPage.goto(