feat: print to PDF

This commit is contained in:
monoid 2022-04-29 01:00:41 +09:00
parent 1f1e99e690
commit d0ea5d1b1b
2 changed files with 92 additions and 2 deletions

25
cli.py
View File

@ -40,7 +40,7 @@ def serve(args):
"""serve the documentation and reload on changes""" """serve the documentation and reload on changes"""
parser = argparse.ArgumentParser(description='Serve the documentation and reload changes', prog="cli.py serve") 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('-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') parser.add_argument('--update_issues', action='store_true', help='update issues')
outDir = "build" outDir = "build"
issuePath = os.path.join(outDir,"issues.json") issuePath = os.path.join(outDir,"issues.json")
@ -68,11 +68,32 @@ def issueUpdate(args):
issuePath = os.path.join(args.outDir,"issues.json") issuePath = os.path.join(args.outDir,"issues.json")
updateIssue(issuePath, args.verbose) 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 = { commandList = {
'build': build, 'build': build,
'help': help, 'help': help,
'issueUpdate': issueUpdate, 'issueUpdate': issueUpdate,
'serve': serve 'serve': serve,
'buildPdf': buildPdf
} }
def main(): def main():

69
tools/printPdf.ts Normal file
View 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();
}