"""Example: Export UCC-style extended metrics to CSV.

This demonstrates how to use the extended counting features
to produce UCC-compatible reports.
"""

from pathlib import Path
import csv
from typing import List, Dict, Any

from pyucc.core.ucc_extended_counting import (
    analyze_file_ucc_style,
    format_ucc_table_header,
    format_ucc_table_line,
)


def export_ucc_extended_csv(files: List[Path], output_path: Path):
    """
    Export extended UCC-style metrics to CSV.

    Args:
        files: List of source files to analyze
        output_path: Output CSV file path
    """

    results = []

    print(f"Analyzing {len(files)} files...")
    for i, file_path in enumerate(files, 1):
        if i % 10 == 0:
            print(f"  Progress: {i}/{len(files)}")

        try:
            result = analyze_file_ucc_style(file_path)
            result["name"] = file_path.name
            result["path"] = str(file_path)
            results.append(result)
        except Exception as e:
            print(f"  Error analyzing {file_path}: {e}")

    # Write to CSV
    fieldnames = [
        "name",
        "path",
        "total_lines",
        "blank_lines",
        "comment_whole",
        "comment_embedded",
        "compiler_directives",
        "data_declarations",
        "exec_instructions",
        "logical_sloc",
        "physical_sloc",
        "language",
    ]

    with open(output_path, "w", newline="", encoding="utf-8") as f:
        writer = csv.DictWriter(f, fieldnames=fieldnames)
        writer.writeheader()
        writer.writerows(results)

    print(f"\n✅ Exported to: {output_path}")

    # Print summary in UCC format
    print("\n" + "=" * 100)
    print("UCC-Style Summary:")
    print(format_ucc_table_header())

    for result in results[:10]:  # Show first 10
        print(format_ucc_table_line(result))

    if len(results) > 10:
        print(f"... ({len(results) - 10} more files)")

    # Print totals
    totals = calculate_totals(results)
    print("-" * 100)
    print(format_ucc_table_line(totals, "TOTAL"))


def calculate_totals(results: List[Dict[str, Any]]) -> Dict[str, Any]:
    """Calculate aggregate totals across all files."""

    totals = {
        "total_lines": 0,
        "blank_lines": 0,
        "comment_whole": 0,
        "comment_embedded": 0,
        "compiler_directives": 0,
        "data_declarations": 0,
        "exec_instructions": 0,
        "logical_sloc": 0,
        "physical_sloc": 0,
        "file": "TOTAL",
    }

    for result in results:
        for key in totals:
            if key != "file":
                totals[key] += result.get(key, 0)

    return totals


def demo_extended_counting():
    """Demo the extended counting on test files."""

    # Find some C/C++ files to analyze
    test_dir = Path(r"C:\__temp\Metrics\attuale\REP\Projects\DSP")

    if not test_dir.exists():
        print("⚠️  Test directory not found. Using current directory...")
        test_dir = Path.cwd() / "pyucc"

    # Collect files
    files = []
    for pattern in ["**/*.c", "**/*.cpp", "**/*.h"]:
        files.extend(list(test_dir.glob(pattern))[:20])  # Limit to 20 files

    if not files:
        print("No C/C++ files found to analyze.")
        return

    print(f"Found {len(files)} files to analyze")

    # Export
    output_path = Path("ucc_extended_output.csv")
    export_ucc_extended_csv(files, output_path)


if __name__ == "__main__":
    demo_extended_counting()