77 lines
3.2 KiB
Bash
Executable File
77 lines
3.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# 设计稿构建：把整页 HTML 按顶层 .page 拆成单页 → html2poster 逐页渲染 → 合并为同名 PDF
|
||
# 用法：bash build.sh [stem ...] # stem 为 design-refs/mockup 下的 html 文件名（不含扩展名）
|
||
# 默认构建 course-shelf 与 components 两个目标
|
||
set -euo pipefail
|
||
cd "$(dirname "$0")"
|
||
|
||
PDF_SKILL_DIR="${PDF_SKILL_DIR:-$HOME/.zcode/cli/plugins/cache/zcode-plugins-official/pdf/0.1.7/skills/pdf}"
|
||
export PDF_SKILL_DIR
|
||
|
||
# 本地渲染器（固定 headless-shell 规避系统 Chrome 占用；TMPDIR 重定向规避 /tmp 磁盘配额）
|
||
HTML2POSTER="$PWD/tools/html2poster.js"
|
||
H2P_CHROMIUM="${H2P_CHROMIUM:-$HOME/.cache/ms-playwright/chromium_headless_shell-1234/chrome-headless-shell-linux64/chrome-headless-shell}"
|
||
export HTML2POSTER H2P_CHROMIUM
|
||
mkdir -p "$HOME/.cache/h2p-tmp"
|
||
export TMPDIR="$HOME/.cache/h2p-tmp"
|
||
|
||
STEMS=("$@")
|
||
if [ ${#STEMS[@]} -eq 0 ]; then
|
||
STEMS=(course-shelf components color-system)
|
||
fi
|
||
|
||
for STEM in "${STEMS[@]}"; do
|
||
SRC="${STEM}.html"
|
||
[ -f "$SRC" ] || { echo "!! 缺少 $SRC，跳过"; continue; }
|
||
OUT_DIR="single-${STEM}"
|
||
mkdir -p "$OUT_DIR"
|
||
|
||
# 1. 按顶层 .page 拆单页（固定布局必须走 html2poster，不能用 html2pdf-next）
|
||
python3 - "$SRC" "$OUT_DIR" <<'EOF'
|
||
import re, sys
|
||
src, out_dir = sys.argv[1], sys.argv[2]
|
||
html = open(src, encoding="utf-8").read()
|
||
head, rest = html.split("<body>", 1)
|
||
body_inner = rest.rsplit("</body>", 1)[0]
|
||
pat = re.compile(r'(?=<div class="page(?: cover)?">)')
|
||
chunks = [c for c in pat.split(body_inner) if c.lstrip().startswith('<div class="page')]
|
||
for i, page_html in enumerate(chunks, 1):
|
||
page_html = re.sub(r'<!-- =+[^>]+-->\s*$', '', page_html.rstrip()) + "\n"
|
||
page_html = page_html.replace("page-break-after: always;", "page-break-after: auto;")
|
||
open(f"{out_dir}/page-{i}.html", "w", encoding="utf-8").write(head + "<body>\n" + page_html + "</body>\n</html>\n")
|
||
print(f"[{src}] split {len(chunks)} pages")
|
||
EOF
|
||
|
||
# 2. 单页素材（img 相对路径以单页目录为基准）
|
||
[ -d assets ] && cp -r assets "$OUT_DIR/assets"
|
||
|
||
# 3. 逐页渲染
|
||
python3 - "$OUT_DIR" <<'EOF'
|
||
import glob, os, re, subprocess, sys
|
||
out_dir = sys.argv[1]
|
||
pages = sorted(glob.glob(f"{out_dir}/page-*.html"), key=lambda p: int(re.search(r"(\d+)", os.path.basename(p)).group(1)))
|
||
for p in pages:
|
||
out = p.replace(".html", ".pdf")
|
||
subprocess.run(["node", os.environ['HTML2POSTER'], p, "--output", out, "--width", "1280px"], check=True, capture_output=True)
|
||
print(f"[{out_dir}] rendered {len(pages)} pages")
|
||
EOF
|
||
|
||
# 4. 合并 + 元数据
|
||
python3 - "$OUT_DIR" "${STEM}" <<'EOF'
|
||
import glob, os, re, sys
|
||
from pypdf import PdfWriter
|
||
out_dir, stem = sys.argv[1], sys.argv[2]
|
||
pdfs = sorted(glob.glob(f"{out_dir}/page-*.pdf"), key=lambda p: int(re.search(r"(\d+)", os.path.basename(p)).group(1)))
|
||
w = PdfWriter()
|
||
for p in pdfs:
|
||
# 只取第 1 页：Chromium 打印偶发产生一个空白尾页（几何已验证无溢出）
|
||
w.append(p, pages=(0, 1))
|
||
w.add_metadata({"/Title": f"课程书架设计稿 · {stem} · stack idealjs", "/Author": "Z.ai", "/Creator": "Z.ai"})
|
||
with open(f"{stem}.pdf", "wb") as f:
|
||
w.write(f)
|
||
print(f"merged {len(pdfs)} pages -> {stem}.pdf")
|
||
EOF
|
||
|
||
echo "done: ${STEM}.pdf"
|
||
done
|