cli.py

"""CLI 入口：将 Simprint 浏览器构建产物打包为 ZIP。"""

import argparse
from importlib.metadata import PackageNotFoundError, version
from pathlib import Path

from config import (
    DEFAULT_BUILD_DIR,
    DEFAULT_CHROMIUM_ROOT,
    DEFAULT_EXECUTABLE_NAME,
    DEFAULT_FLATTEN_ZIP,
    DEFAULT_OUTPUT_PREFIX,
    DEFAULT_PACKAGE_DIR,
    DEFAULT_RELEASE_FILE,
    DEFAULT_SECTIONS,
    resolve_chromium_root,
    resolve_path_from_root,
)
from pack import create_zip, get_version

PACKAGE_NAME = "simprint-browser-packager"


def main() -> None:
    parser = argparse.ArgumentParser(
        description="将 Simprint 浏览器构建产物打包为 ZIP（基于 chrome.release 文件列表）"
    )
    parser.add_argument(
        "--version",
        action="version",
        version=f"%(prog)s {get_cli_version()}",
    )
    parser.add_argument(
        "--chromium-root",
        type=Path,
        default=Path(DEFAULT_CHROMIUM_ROOT),
        metavar="PATH",
        help="Chromium / Simprint 源码根目录，默认 ../simprint-browser/src",
    )
    parser.add_argument(
        "--release-file",
        type=Path,
        default=Path(DEFAULT_RELEASE_FILE),
        help="release 文件路径；相对路径按 chromium-root 解析",
    )
    parser.add_argument(
        "--build-dir",
        type=Path,
        default=Path(DEFAULT_BUILD_DIR),
        help="构建目录；相对路径按 chromium-root 解析",
    )
    parser.add_argument(
        "--output",
        "-o",
        type=Path,
        default=None,
        help="输出 ZIP 完整路径（覆盖时不再自动加版本后缀）",
    )
    parser.add_argument(
        "--executable-name",
        default=DEFAULT_EXECUTABLE_NAME,
        help="可执行文件名，如 simprint",
    )
    parser.add_argument(
        "--sections",
        default=DEFAULT_SECTIONS,
        help="release 段落，逗号分隔，如 GENERAL,HIDPI,FFMPEG",
    )
    parser.add_argument(
        "--package-dir",
        default=DEFAULT_PACKAGE_DIR,
        help="ZIP 顶层目录名；仅在禁用扁平化时影响最终输出结构",
    )
    parser.add_argument(
        "--output-prefix",
        default=DEFAULT_OUTPUT_PREFIX,
        help="默认输出文件名前缀，最终文件名为 {prefix}-{version}.zip",
    )
    parser.add_argument(
        "--flatten-zip",
        action=argparse.BooleanOptionalAction,
        default=DEFAULT_FLATTEN_ZIP,
        help="是否扁平化 ZIP 根目录；默认开启，可用 --no-flatten-zip 关闭",
    )
    parser.add_argument(
        "--verbose",
        "-v",
        action="store_true",
        help="打印每个打包进 ZIP 的文件",
    )
    args = parser.parse_args()

    chromium_root = resolve_chromium_root(args.chromium_root)
    if not chromium_root.is_dir():
        parser.error(f"SIMPRINT_CHROMIUM_ROOT 目录不存在: {chromium_root}")

    build_dir = resolve_path_from_root(chromium_root, args.build_dir)
    if not build_dir.is_dir():
        parser.error(f"构建目录不存在: {build_dir}")

    release_file = resolve_path_from_root(chromium_root, args.release_file)
    if not release_file.is_file():
        parser.error(f"release 文件不存在: {release_file}")

    executable_name = args.executable_name
    package_dir_name = args.package_dir.strip()
    if not package_dir_name:
        parser.error("PACKAGE_DIR 不能为空。")
    sections = [s.strip() for s in args.sections.split(",") if s.strip()]

    version = get_version(chromium_root)
    if args.output is not None:
        output = args.output
    else:
        prefix = args.output_prefix.strip()
        # 最终文件名：{前缀}-{版本}.zip
        output = Path(prefix).parent / f"{Path(prefix).name}-{version}.zip"
    if not output.is_absolute():
        output = Path.cwd() / output

    create_zip(
        build_dir=build_dir,
        release_file=release_file,
        chromium_root=chromium_root,
        executable_name=executable_name,
        sections=sections,
        output_path=output,
        package_dir_name=package_dir_name,
        flatten_top_dir=args.flatten_zip,
        verbose=args.verbose,
    )


def get_cli_version() -> str:
    try:
        return version(PACKAGE_NAME)
    except PackageNotFoundError:
        return "0.1.0"