#!/usr/bin/env bash
set -euo pipefail
# =============================================================
# run_test.sh — 生成少量预览页面 → Hugo静态部署 → 线上预览
# 用法:
# ./run_test.sh # 默认从两个数据源各取5条，共10条
# ./run_test.sh 20 # 自定义总数(SQL取10, JSON取10)
# ./run_test.sh 20 deploy # 生成+自动部署
# =============================================================
SITE_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
REPO_ROOT="$(cd "$SITE_ROOT/../.." && pwd)"
PIPELINE_DIR="$REPO_ROOT/cloudflare-pipeline"
TOTAL="${1:-10}"
ACTION="${2:-generate}" # generate | deploy
HALF=$(( TOTAL / 2 ))
REMAINDER=$(( TOTAL - HALF ))
PREVIEW_DIR="$SITE_ROOT/hugo/static/_preview"
echo "╔══════════════════════════════════════════════════════╗"
echo "║ 🧪 Test Mode — 预览页面生成 ($TOTAL 页) ║"
echo "╚══════════════════════════════════════════════════════╝"
# 清理旧的预览文件
rm -rf "$PREVIEW_DIR"
mkdir -p "$PREVIEW_DIR"
# ---- Step 1: 从 SQL 数据源生成 ----
SQL_FILE="${DIGIKEY_SQL_FILE:-/Volumes/win10/（Jada）spider_digikey_data.sql}"
if [ -f "$SQL_FILE" ]; then
echo ""
echo "===== Step 1: SQL 数据源 → 取 $HALF 条 ====="
TARGET_COUNT="$HALF" \
GENERATE_HTML=1 \
GENERATE_SQL=0 \
OUTPUT_HTML_DIR="$PREVIEW_DIR" \
python3 "$PIPELINE_DIR/generate_r2_product_pages.py" --clean-output --skip-qa
# ---- Step 2: 从 JSON 数据源生成 ----
echo ""
echo "===== Step 2: JSON 数据源 → 取 $REMAINDER 条 ====="
TARGET_COUNT="$REMAINDER" \
GENERATE_HTML=1 \
GENERATE_SQL=0 \
OUTPUT_HTML_DIR="$PREVIEW_DIR" \
python3 "$PIPELINE_DIR/page_pipeline.py" generate --source json --skip-qa
else
echo ""
echo "⚠️ SQL 数据源文件 $SQL_FILE 不存在，将自动降级为 JSON 数据源生成全部 $TOTAL 页..."
echo ""
echo "===== Step 1: JSON 数据源 → 取 $TOTAL 条 ====="
TARGET_COUNT="$TOTAL" \
GENERATE_HTML=1 \
GENERATE_SQL=0 \
OUTPUT_HTML_DIR="$PREVIEW_DIR" \
python3 "$PIPELINE_DIR/page_pipeline.py" generate --source json --clean-output --skip-qa
fi
# ---- 汇总 ----
FILE_COUNT=$(find "$PREVIEW_DIR" -name '*.html' | wc -l | tr -d ' ')
echo ""
echo "✅ 共生成 $FILE_COUNT 个预览页面 → $PREVIEW_DIR/"
echo ""
# 生成预览索引页
INDEX_FILE="$PREVIEW_DIR/index.html"
cat > "$INDEX_FILE" << 'HEADER'
Template Preview Index
🧪 Template Preview
Generated at: __TIMESTAMP__ · noindex
| # | File | Size |
|---|
HEADER
# 替换时间戳
sed -i '' "s|__TIMESTAMP__|$(date '+%Y-%m-%d %H:%M:%S')|g" "$INDEX_FILE"
# 填充文件列表
COUNT=0
for f in "$PREVIEW_DIR"/*.html; do
fname=$(basename "$f")
[ "$fname" = "index.html" ] && continue
COUNT=$((COUNT + 1))
fsize=$(stat -f%z "$f" 2>/dev/null || stat -c%s "$f" 2>/dev/null)
fsize_kb=$(echo "scale=1; $fsize / 1024" | bc)
echo "| $COUNT | $fname | ${fsize_kb} KB |
" >> "$INDEX_FILE"
done
cat >> "$INDEX_FILE" << 'FOOTER'
确认无误后，执行 ./run_all.sh 进行全量生成与部署。
FOOTER
echo "📄 预览索引页 → $PREVIEW_DIR/index.html"
echo ""
# ---- Step 3: 可选部署 ----
if [[ "$ACTION" == "deploy" ]]; then
echo "===== Step 3: Hugo Build + Deploy ====="
cd hugo
npx wrangler pages deploy public --project-name=jp-inotoday-parts
cd ..
echo ""
echo "🌐 预览地址: https://parts.jp.inotoday.com/_preview/"
else
echo "📌 预览文件已就绪，手动部署步骤:"
echo " cd hugo && npx wrangler pages deploy public --project-name=jp-inotoday-parts"
echo ""
echo "🌐 部署后预览: https://parts.jp.inotoday.com/_preview/"
fi