feat: support change the origin of the print url #5

Author: condorheroblog

.gitignore

@@ -87,3 +87,4 @@ dist
 **/.vuepress/.temp/
 **/.vuepress/.cache/
 **/.vuepress/dist/
+*.pdf

packages/vuepress-plugin-export-pdf-core/package.json

@@ -55,7 +55,7 @@
 		"cac": "^6.7.14",
 		"envinfo": "^7.8.1",
 		"fs-extra": "^11.1.1",
-		"html-export-pdf-cli": "^1.0.2",
+		"html-export-pdf-cli": "^1.0.3",
 		"multimatch": "^6.0.0",
 		"ora": "^6.3.0",
 		"pdfjs": "^2.4.7",

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

@@ -1,12 +1,12 @@
 import { join } from "node:path";
 import fse from "fs-extra";
 import pc from "picocolors";
-import { Printer, createProgress, writeFileSafe } from "html-export-pdf-cli";
+import { Printer, createProgress, isValidUrl, writeFileSafe } from "html-export-pdf-cli";
 import type { LaunchOptions, PDFOptions } from "html-export-pdf-cli";
 
-import { filterRoute } from "../utils";
-import type { Page } from "../";
-import { mergePDF } from "./mergePDF";
+import { filterRoute } from "./utils";
+import type { Page } from ".";
+import { mergePDF } from ".";
 
 export type UserSorter = (a: Page, b: Page) => number;
 
@@ -22,6 +22,7 @@ export interface IGeneratePdfOptions {
 	puppeteerLaunchOptions?: LaunchOptions
 	pdfOptions?: PDFOptions
 	pdfOutlines?: boolean
+	urlOrigin?: string
 }
 
 /**
@@ -36,10 +37,12 @@ export const generatePdf = async ({
 	sorter,
 	outFile,
 	outDir,
+	urlOrigin,
 	pdfOptions,
 	pdfOutlines,
 	routePatterns,
 	puppeteerLaunchOptions,
+// eslint-disable-next-line sonarjs/cognitive-complexity
 }: IGeneratePdfOptions) => {
 	const tempPdfDir = join(tempDir, "pdf");
 	fse.ensureDirSync(tempPdfDir);
@@ -49,21 +52,62 @@ export const generatePdf = async ({
 	if (typeof sorter === "function")
 		exportPages = exportPages.sort(sorter);
 
+	const isValidUrlOrigin = isValidUrl(urlOrigin ?? "");
+	if (urlOrigin && !isValidUrlOrigin) {
+		process.stdout.write(pc.red(`${urlOrigin} is not a valid URL`));
+		process.exit(1);
+	}
+
 	const normalizePages = exportPages.map((page) => {
 		return {
 			url: page.path,
 			title: page.title,
-			location: `http://${host}:${port}${page.path}`,
+			location: urlOrigin ? `${new URL(urlOrigin).origin}${page.path}` : `http://${host}:${port}${page.path}`,
 			pagePath: `${tempPdfDir}/${page.key}.pdf`,
 		};
 	});
 
 	const singleBar = createProgress();
 	singleBar.start(normalizePages.length);
 
-	const printer = new Printer(puppeteerLaunchOptions);
+	const printer = new Printer();
+	await printer.setup(puppeteerLaunchOptions);
 
 	for (const { location, pagePath, title } of normalizePages) {
+		const page = await printer.createNewPage(location);
+
+		if (urlOrigin && isValidUrlOrigin) {
+			await page.setRequestInterception(true);
+			page.on("request", (request) => {
+				const reqUrl = request.url();
+				// http or https
+				if (isValidUrl(reqUrl)) {
+					const parsedUrl = new URL(reqUrl);
+					parsedUrl.host = host;
+					parsedUrl.protocol = "http:";
+					parsedUrl.port = `${port}`;
+					const parsedUrlString = parsedUrl.toString();
+					request.continue({
+						url: parsedUrlString,
+						headers: Object.assign(
+							{},
+							request.headers(),
+							{
+								refer: parsedUrlString,
+								// Same origin
+								// origin: parsedUrl.origin,
+								// CORS
+								// host: parsedUrl.host,
+							}),
+					});
+				}
+				else {
+					request.continue();
+				}
+			});
+		}
+
+		await printer.render(location);
 		const data = await printer.pdf(location, {
 			format: "A4",
 			...pdfOptions,
@@ -74,7 +118,7 @@ export const generatePdf = async ({
 	}
 
 	singleBar.stop();
-	await printer.close();
+	await printer.closeBrowser();
 
 	const exportedPath = await mergePDF(normalizePages, outFile, outDir, pdfOutlines);
 	const message = `\nExported to ${pc.yellow(exportedPath)}\n`;

packages/vuepress-plugin-export-pdf-core/src/index.ts

@@ -2,4 +2,5 @@ export * from "./types";
 export * from "./runner";
 export * from "./utils";
 export * from "./configs";
+export * from "./generatePdf";
 export * from "./systemInfo";

packages/vuepress-plugin-export-pdf-core/src/types.ts

@@ -3,6 +3,7 @@ export interface CommandOptions {
 	outDir?: string
 	outFile?: string
 	debug?: boolean
+	urlOrigin?: string
 	pdfOutlines?: boolean
 }
 

packages/vuepress-plugin-export-pdf-core/src/utils/index.ts

@@ -1,5 +1,4 @@
 export * from "./timeTransformer";
-export * from "./generatePdf";
 export * from "./filterRoute";
 export * from "./checkEnv";
 export * from "./mergePDF";

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

@@ -54,6 +54,7 @@ The package provides the `press-export-pdf` command with the following command l
   - `--outFile <outFile>`: Name of output file
   - `--outDir <outDir>`: Directory of output files
   - `--pdfOutlines <pdfOutlines>`: Keep PDF outlines/bookmarks
+  - `--urlOrigin <urlOrigin>`: Change the origin of the print url
   - `--debug`: Enable debug mode
 - `info`: Display environment information
 - `--help`: Display help information
@@ -111,7 +112,7 @@ config options:
 - `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 }`)
 - `pdfOutlines` - Keep PDF outlines/bookmarks(default `true`)
-
+- `urlOrigin`: Change the origin of the print url([How do I change the URL point to the localhost](https://github.com/condorheroblog/vuepress-plugin/issues/5))
 ## PDF print style
 
 By default, `A4` paper is used for printing, The size of A4 paper is (8.27in x 11.7in), One inch is equal to ninety-six pixels: `1 in = 96 pixel (X)` ,the inch unit of A4 is converted to (793.92px x 1123.2px).

packages/vuepress-plugin-export-pdf-v2/assets/vuepress-plugin-export-pdf-v2-outline.tldr

@@ -8,6 +8,7 @@
 		"help": "press-export-pdf --help",
 		"version": "press-export-pdf --version",
 		"systemInfo": "press-export-pdf info",
+		"docs:urlOrigin": "press-export-pdf export docs --urlOrigin https://www.google.com/",
 		"docs:outFile": "press-export-pdf export docs --outFile docs.pdf",
 		"docs:outDir": "press-export-pdf export docs --outDir /tmp/outDir",
 		"docs:debug": "press-export-pdf export docs --debug",