#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
从小程序/正式 API 下载单章正文，保存为书稿目录下的 md 文件。

用法：
  SOUL_TEST_ENV=soulapi python3 scripts/content_download.py 128
  SOUL_TEST_ENV=soulapi python3 scripts/content_download.py --id 10.27
  python3 scripts/content_download.py 128 --out-dir /path/to/2026每日派对干货

2026 场次（第102场起）对应 id 10.01、10.02、…、10.27（第128场）…
"""
import argparse
import os
import sys
from pathlib import Path

# 项目根
PROJECT_ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(PROJECT_ROOT / "scripts" / "test"))
try:
    from config import API_BASE, ENV_LABEL
except Exception:
    API_BASE = os.environ.get("SOUL_API_BASE", "https://soulapi.quwanzhi.com").rstrip("/")
    ENV_LABEL = "env"

# 书稿 2026 目录默认路径（与上传 README 一致）
DEFAULT_BOOK_2026 = Path(
    os.environ.get(
        "SOUL_BOOK_2026",
        "/Users/karuo/Documents/个人/2、我写的书/《一场soul的创业实验》/2026每日派对干货",
    )
)


def field_to_id(field: int) -> str:
    """第 N 场（≥102）→ 10.xx；第101场及以前在第九章为 9.xx。API 上第128场可能为 9.28。"""
    if field >= 102:
        n = field - 102 + 1
        return f"10.{n:02d}"
    if field >= 1:
        return f"9.{field:02d}"  # 第9章 9.01～9.99，按场次
    raise ValueError("场次请用 1～999")


def main():
    parser = argparse.ArgumentParser(description="从小程序 API 下载单章为 md")
    parser.add_argument("field", nargs="?", type=int, help="场次号，如 128 表示第128场")
    parser.add_argument("--id", type=str, help="章节 id，如 10.27（与 field 二选一）")
    parser.add_argument("--out-dir", type=Path, default=DEFAULT_BOOK_2026, help="输出目录，默认 2026每日派对干货")
    parser.add_argument("--base", type=str, default=API_BASE, help="API 根地址")
    args = parser.parse_args()

    if args.id:
        chapter_id = args.id
    elif args.field is not None:
        chapter_id = field_to_id(args.field)
    else:
        parser.error("请指定 field（如 128）或 --id（如 10.27）")

    base = args.base.rstrip("/")
    try:
        import requests
    except ImportError:
        print("请安装: pip install requests", file=sys.stderr)
        sys.exit(1)

    url = f"{base}/api/miniprogram/book/chapter/by-id/{chapter_id}"
    print(f"环境: {ENV_LABEL} | GET {url}")
    r = requests.get(url, timeout=30)
    # 2026 场次可能仍挂在第9章（9.28=第128场），404 时用 9.xx 再试
    if r.status_code == 404 and args.field is not None and not args.id and args.field >= 1 and args.field <= 101:
        fallback_id = f"9.{args.field:02d}"
        url = f"{base}/api/miniprogram/book/chapter/by-id/{fallback_id}"
        print(f"404，尝试第9章 id: {fallback_id} | GET {url}")
        r = requests.get(url, timeout=30)
        if r.status_code == 200:
            chapter_id = fallback_id
    if r.status_code == 404 and args.field is not None and not args.id and args.field >= 102:
        fallback_id = f"9.{(args.field - 100):02d}"  # 128 → 9.28
        url = f"{base}/api/miniprogram/book/chapter/by-id/{fallback_id}"
        print(f"404，尝试第9章 id: {fallback_id} | GET {url}")
        r = requests.get(url, timeout=30)
        if r.status_code == 200:
            chapter_id = fallback_id
    r.raise_for_status()
    data = r.json()
    if not data.get("success"):
        print("API 返回失败:", data.get("error", data), file=sys.stderr)
        sys.exit(2)

    content = data.get("content") or (data.get("data") or {}).get("content") or ""
    section_title = data.get("sectionTitle") or (data.get("data") or {}).get("sectionTitle") or f"第{args.field or '?'}场"
    if not content:
        print("未获取到正文 content", file=sys.stderr)
        sys.exit(3)

    # 若标题里没有「第X场」，用场次补上（sectionTitle 可能是「赚最多那个月…」）
    if args.field and "第" not in section_title and "场" not in section_title:
        display_title = f"第{args.field}场｜{section_title}"
    else:
        display_title = section_title

    out_dir = args.out_dir
    out_dir.mkdir(parents=True, exist_ok=True)
    out_file = out_dir / f"{display_title}.md"
    body = f"# {display_title}\n\n{content.strip()}\n"
    out_file.write_text(body, encoding="utf-8")
    print(f"已写入: {out_file}")
    print(f"字数: {len(content)}")


if __name__ == "__main__":
    main()