"""
Centralized final draft file operations.

Change the naming convention by editing ONLY these constants:
"""

# Configuration - change these to rename all final drafts
FINAL_DRAFT_PREFIX = "6"
LEGACY_FILENAME = "4-final-draft.md"  # For backwards compatibility


def get_final_draft_filename(output_dir: Path) -> str:
    """Get the final draft filename based on output directory name."""
    dir_name = output_dir.name  # e.g., "Aito-v0.0.2"
    return f"{FINAL_DRAFT_PREFIX}-{dir_name}.md"  # "6-Aito-v0.0.2.md"


def get_final_draft_path(output_dir: Path) -> Path:
    """Get the full path to the final draft file."""
    return output_dir / get_final_draft_filename(output_dir)


def find_final_draft(output_dir: Path) -> Optional[Path]:
    """
    Find the final draft file with automatic legacy fallback.

    Tries new naming pattern first (6-*.md), falls back to legacy (4-final-draft.md).
    """
    # Try new naming pattern first
    new_pattern_files = list(output_dir.glob(f"{FINAL_DRAFT_PREFIX}-*.md"))
    if new_pattern_files:
        return new_pattern_files[0]

    # Fall back to legacy naming
    legacy_path = output_dir / LEGACY_FILENAME
    if legacy_path.exists():
        return legacy_path

    return None


def read_final_draft(output_dir: Path) -> Optional[str]:
    """Read final draft content with automatic file detection."""
    draft_path = find_final_draft(output_dir)
    if draft_path and draft_path.exists():
        return draft_path.read_text(encoding="utf-8")
    return None


def write_final_draft(output_dir: Path, content: str) -> Path:
    """Write final draft with new naming pattern."""
    draft_path = get_final_draft_path(output_dir)
    draft_path.write_text(content, encoding="utf-8")
    return draft_path


def find_all_final_drafts(root_dir: Path) -> list[Path]:
    """Find all final draft files recursively (both new and legacy patterns)."""
    drafts = []
    # New pattern
    drafts.extend(root_dir.rglob(f"{FINAL_DRAFT_PREFIX}-*.md"))
    # Legacy pattern
    drafts.extend(root_dir.rglob(LEGACY_FILENAME))
    return drafts


def is_final_draft_file(filepath: Path) -> bool:
    """Check if a file matches the final draft naming pattern."""
    name = filepath.name
    return (
        name.startswith(f"{FINAL_DRAFT_PREFIX}-") and name.endswith(".md")
    ) or name == LEGACY_FILENAME

# New naming pattern
output/Aito-v0.0.2/
├── 0-deck-analysis.json
├── 1-research.json
├── 2-sections/
├── 3-validation.json
├── 4-fact-check.json
├── 4-fact-check.md
├── 5-scorecard/
├── 6-Aito-v0.0.2.md          # Final draft with descriptive name
└── state.json

# Edit src/final_draft.py ONLY
FINAL_DRAFT_PREFIX = "7"  # Change this one line
# All 15+ consumer files automatically use new naming

# Verify module imports correctly
python -c "from src.final_draft import write_final_draft, find_final_draft; print('✓ Module imports')"

# Verify workflow compiles
python -c "from src.workflow import build_workflow; w = build_workflow(); print('✓ Workflow compiles')"

# Test legacy detection
python -c "
from pathlib import Path
from src.final_draft import find_final_draft
# Should find either 6-*.md or 4-final-draft.md
print('✓ Legacy detection works')
"