import subprocess
import os
import shutil

def generate_ebook(source_dir="documentation/docs", build_dir="documentation/docs/_build", output_format="epub"):
    """
    使用 Sphinx 生成电子书 (EPUB 或 HTML)。
    """
    print(f"开始生成 {output_format} 格式的电子书...")

    # 确保构建目录存在
    if not os.path.exists(build_dir):
        os.makedirs(build_dir)

    # 清理之前的构建
    if os.path.exists(build_dir):
        shutil.rmtree(build_dir)
        print(f"清理目录: {build_dir}")
    else:
        print(f"构建目录不存在，无需清理: {build_dir}")

    # 构建文档
    build_command = [
        f"./documentation/analysis/phallic-worship-analysis/venv/bin/sphinx-build",
        "-b", output_format,
        source_dir,
        os.path.join(build_dir, output_format)
    ]
    print(f"执行构建命令: {' '.join(build_command)}")
    try:
        subprocess.run(build_command, check=True, cwd='.')
        print(f"{output_format} 电子书生成成功！")
        return True
    except subprocess.CalledProcessError as e:
        print(f"生成 {output_format} 电子书失败: {e}")
        return False

if __name__ == "__main__":
    # 示例用法：生成 EPUB
    if generate_ebook(output_format="epub"):
        print("EPUB 电子书已生成。")
    else:
        print("EPUB 电子书生成失败。")

    # 示例用法：生成 HTML
    if generate_ebook(output_format="html"):
        print("HTML 网站已生成。")
    else:
        print("HTML 网站生成失败。")