feat: print to PDF
This commit is contained in:
parent
1f1e99e690
commit
d0ea5d1b1b
25
cli.py
25
cli.py
@ -40,7 +40,7 @@ def serve(args):
|
||||
"""serve the documentation and reload on changes"""
|
||||
parser = argparse.ArgumentParser(description='Serve the documentation and reload changes', prog="cli.py serve")
|
||||
parser.add_argument('-v', '--verbose', action='store_true', help='verbose mode')
|
||||
parser.add_argument('-p', '--port', default=8000, help='port')
|
||||
parser.add_argument('-p', '--port', default=3000, help='port')
|
||||
parser.add_argument('--update_issues', action='store_true', help='update issues')
|
||||
outDir = "build"
|
||||
issuePath = os.path.join(outDir,"issues.json")
|
||||
@ -68,11 +68,32 @@ def issueUpdate(args):
|
||||
issuePath = os.path.join(args.outDir,"issues.json")
|
||||
updateIssue(issuePath, args.verbose)
|
||||
|
||||
def buildPdf(args):
|
||||
parser = argparse.ArgumentParser(description='Print to pdf', prog="cli.py buildPdf")
|
||||
parser.add_argument('-v', '--verbose', action='store_true', help='verbose mode')
|
||||
parser.add_argument('--outDir', default="build/doc.pdf", help='output directory')
|
||||
parser.add_argument('--browser-path', help='path to the browser')
|
||||
args = parser.parse_args(args)
|
||||
absPath = os.path.normpath(os.path.join( os.getcwd(),'book' , 'print.html' )).replace('\\', '/')
|
||||
url = f"file://{absPath}"
|
||||
if args.verbose:
|
||||
print("build start")
|
||||
print("print url", url)
|
||||
cmd = ["deno", "run","--unstable" ,"-A","tools/printPdf.ts", "--outDir", args.outDir, "--url", url]
|
||||
if args.browser_path:
|
||||
cmd.append("--chromeDir")
|
||||
cmd.append(args.browser_path)
|
||||
if args.verbose:
|
||||
print("cmd: ", " ".join(cmd))
|
||||
p = subprocess.run(cmd)
|
||||
p.check_returncode()
|
||||
|
||||
commandList = {
|
||||
'build': build,
|
||||
'help': help,
|
||||
'issueUpdate': issueUpdate,
|
||||
'serve': serve
|
||||
'serve': serve,
|
||||
'buildPdf': buildPdf
|
||||
}
|
||||
|
||||
def main():
|
||||
|
69
tools/printPdf.ts
Normal file
69
tools/printPdf.ts
Normal file
@ -0,0 +1,69 @@
|
||||
import puppeteer from "https://deno.land/x/puppeteer@9.0.2/mod.ts";
|
||||
import { parse as argParse } from "https://deno.land/std@0.136.0/flags/mod.ts";
|
||||
import {join as pathJoin} from "https://deno.land/std@0.136.0/path/mod.ts";
|
||||
|
||||
const executablePathMap = {
|
||||
windows: "C:\\Program Files (x86)\\Microsoft\\Edge\\Application\\msedge.exe",
|
||||
//"C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe",
|
||||
darwin: "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome",
|
||||
linux: "/usr/bin/chromium-browser",
|
||||
};
|
||||
|
||||
function findChromeDir(){
|
||||
const dir = executablePathMap[Deno.build.os];
|
||||
if (Deno.build.os === "linux"){
|
||||
const pathes = Deno.env.get("PATH")?.split(";");
|
||||
if(!pathes){
|
||||
return dir;
|
||||
}
|
||||
for (const path of pathes){
|
||||
const entries = [...Deno.readDirSync(path)];
|
||||
const chrome = entries.find((entry) => entry.name === "chrome");
|
||||
if(chrome) {
|
||||
return pathJoin(path, chrome.name);
|
||||
}
|
||||
}
|
||||
}
|
||||
return dir;
|
||||
}
|
||||
|
||||
async function main() {
|
||||
const args = argParse(Deno.args);
|
||||
let { url, outDir, chromeDir } = args;
|
||||
if (!url) {
|
||||
console.error("No url provided.");
|
||||
Deno.exit(1);
|
||||
}
|
||||
outDir = outDir ?? "out.pdf";
|
||||
if(!chromeDir) {
|
||||
console.log("No chromeDir provided. Try to find chrome.");
|
||||
chromeDir = findChromeDir();
|
||||
console.log(`chromeDir: ${chromeDir}`);
|
||||
}
|
||||
|
||||
const browser = await puppeteer.launch({
|
||||
product: "chrome",
|
||||
headless: true,
|
||||
executablePath: chromeDir,
|
||||
});
|
||||
|
||||
const page = await browser.newPage();
|
||||
console.log(`goto url ${url}`);
|
||||
await page.goto(url, {
|
||||
waitUntil: "networkidle2",
|
||||
});
|
||||
await page.pdf({
|
||||
path: outDir,
|
||||
margin: {
|
||||
bottom: "1cm",
|
||||
left: "1cm",
|
||||
right: "1cm",
|
||||
top: "1cm",
|
||||
}
|
||||
});
|
||||
await browser.close();
|
||||
}
|
||||
|
||||
if (import.meta.main) {
|
||||
main();
|
||||
}
|
Loading…
Reference in New Issue
Block a user