cli.py

"""
MechCalc FEM CLI

python -m mechcalc run model.json
python -m mechcalc validate
python -m mechcalc benchmark
python -m mechcalc profile
"""
from __future__ import annotations

import argparse
import sys
from pathlib import Path

# 確保專案根目錄在 path
sys.path.insert(0, str(Path(__file__).resolve().parent))


def cmd_run(args: argparse.Namespace) -> int:
    """執行模型分析。"""
    input_path = args.input
    if not input_path:
        print("Error: --input required for 'run' command")
        return 1
    path = Path(input_path)
    if not path.exists():
        print(f"Error: file not found: {path}")
        return 1
    print(f"Running model: {input_path}")
    # TODO: 載入 JSON 並呼叫 api.run()
    print("  (JSON model loading not yet implemented)")
    return 0


def cmd_validate(args: argparse.Namespace) -> int:
    """執行驗證套件。"""
    print("Running validation suite...")
    try:
        from validation.run_all import run_full_validation
        run_full_validation()
        return 0
    except ImportError as e:
        print(f"Error: {e}")
        return 1


def cmd_benchmark(args: argparse.Namespace) -> int:
    """執行 benchmark 套件。"""
    print("Running benchmark suite...")
    try:
        from validation.buckling_column import buckling_column_benchmark
        from validation.plastic_tension_test import plastic_tension_benchmark
        from validation.contact_benchmark import run_contact_benchmark_suite
        from validation.benchmark_matrix import run_benchmark_matrix

        buckling_column_benchmark(verbose=True)
        plastic_tension_benchmark(verbose=True)
        run_contact_benchmark_suite(verbose=True)

        matrix = run_benchmark_matrix()
        try:
            print("\n" + matrix.to_markdown())
        except UnicodeEncodeError:
            print("\n[Benchmark matrix - see CSV for details]")

        if args.output:
            out = Path(args.output)
            out.parent.mkdir(parents=True, exist_ok=True)
            matrix.to_csv(str(out))
            yaml_path = out.with_suffix(".yaml")
            matrix.to_yaml(str(yaml_path))
            print(f"\nBenchmark matrix saved to: {args.output}, {yaml_path}")

        return 0
    except ImportError as e:
        print(f"Error: {e}")
        return 1


def cmd_profile(args: argparse.Namespace) -> int:
    """效能分析。"""
    print("Running profile...")
    try:
        from core.profiler import profile_example
        profile_example()
        return 0
    except ImportError:
        print("  (Profiler not available)")
        return 0


def main() -> int:
    parser = argparse.ArgumentParser(
        description="MechCalc FEM — 有限元素分析工具",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Examples:
  python -m mechcalc run model.json
  python -m mechcalc validate
  python -m mechcalc benchmark --output results/benchmark.csv
  python -m mechcalc profile
        """,
    )
    parser.add_argument(
        "command",
        choices=["run", "validate", "benchmark", "profile"],
        help="Command to execute",
    )
    parser.add_argument(
        "--input", "-i",
        type=str,
        default=None,
        help="Input model file (JSON) for 'run' command",
    )
    parser.add_argument(
        "--output", "-o",
        type=str,
        default=None,
        help="Output path for benchmark CSV",
    )
    args = parser.parse_args()

    handlers = {
        "run": cmd_run,
        "validate": cmd_validate,
        "benchmark": cmd_benchmark,
        "profile": cmd_profile,
    }
    return handlers[args.command](args)


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