#!/usr/bin/env python
"""
代码格式化脚本 - 自动格式化代码
"""
import subprocess
import sys
from pathlib import Path


def run_command(cmd: list, description: str) -> bool:
    """运行命令"""
    print(f"\n{'='*60}")
    print(f"执行: {description}")
    print(f"命令: {' '.join(cmd)}")
    print('='*60)

    try:
        result = subprocess.run(cmd, check=True, capture_output=True, text=True)
        print(result.stdout)
        if result.stderr:
            print("STDERR:", result.stderr)
        return True
    except subprocess.CalledProcessError as e:
        print(f"错误: {e}")
        print(f"STDOUT: {e.stdout}")
        print(f"STDERR: {e.stderr}")
        return False


def main():
    """主函数"""
    project_root = Path(__file__).parent

    print(f"项目根目录: {project_root}")
    print("开始代码格式化...")

    success = True

    # 1. 格式化代码 (black)
    success &= run_command(
        [sys.executable, "-m", "black", str(project_root)],
        "格式化代码 (black)"
    )

    # 2. 排序导入 (isort)
    success &= run_command(
        [sys.executable, "-m", "isort", str(project_root)],
        "排序导入 (isort)"
    )

    # 3. 类型检查 (mypy)
    success &= run_command(
        [sys.executable, "-m", "mypy", str(project_root), "--ignore-missing-imports"],
        "类型检查 (mypy)"
    )

    # 4. 代码检查 (flake8)
    success &= run_command(
        [sys.executable, "-m", "flake8", str(project_root), "--max-line-length=100"],
        "代码检查 (flake8)"
    )

    # 5. 代码审查 (pylint)
    success &= run_command(
        [sys.executable, "-m", "pylint", str(project_root), "--max-line-length=100"],
        "代码审查 (pylint)"
    )

    print("\n" + "="*60)
    if success:
        print("✅ 代码格式化完成")
    else:
        print("❌ 部分命令失败，请检查输出")
    print("="*60)

    return 0 if success else 1


if __name__ == "__main__":
    sys.exit(main())