Workspace Affected

#!/usr/bin/env python3
"""
Analyze workspace dependency graph and show affected projects.
"""

import sys
import re
import os
from pathlib import Path

def parse_relationships(config_path: str) -> list[tuple[str, str, str]]:
    """
    Parse relationships from WORKSPACE.md.
    Returns list of (consumer, provider, description) tuples.
    """
    relationships = []

    try:
        with open(config_path, 'r') as f:
            content = f.read()
    except FileNotFoundError:
        return relationships

    # Find the Relationships section
    in_relationships = False
    for line in content.split('\n'):
        if line.strip().startswith('## Relationships'):
            in_relationships = True
            continue
        if in_relationships and line.strip().startswith('## '):
            break
        if in_relationships and line.strip().startswith('- '):
            # Parse: - consumer -> provider (description)
            match = re.match(r'-\s*(\w+)\s*->\s*(\w+)\s*(?:\(([^)]+)\))?', line.strip())
            if match:
                consumer = match.group(1)
                provider = match.group(2)
                description = match.group(3) or ""
                relationships.append((consumer, provider, description))

    return relationships


def get_dependents(project: str, relationships: list[tuple[str, str, str]]) -> list[tuple[str, str]]:
    """Get projects that depend on the given project."""
    return [(consumer, desc) for consumer, provider, desc in relationships if provider == project]


def get_dependencies(project: str, relationships: list[tuple[str, str, str]]) -> list[tuple[str, str]]:
    """Get projects that the given project depends on."""
    return [(provider, desc) for consumer, provider, desc in relationships if consumer == project]


def print_tree(items: list[tuple[str, str]], empty_message: str = "(none)"):
    """Print items as a tree structure."""
    if not items:
        print(f"  └── {empty_message}")
        return

    for i, (name, desc) in enumerate(items):
        is_last = i == len(items) - 1
        prefix = "└──" if is_last else "├──"
        if desc:
            print(f"  {prefix} {name} ({desc})")
        else:
            print(f"  {prefix} {name}")


def main():
    if len(sys.argv) < 3:
        print("Usage: affected.py <project> <config_file>")
        sys.exit(1)

    project = sys.argv[1]
    config_file = sys.argv[2]

    if not os.path.exists(config_file):
        print(f"Error: Config file not found: {config_file}")
        sys.exit(1)

    relationships = parse_relationships(config_file)

    dependents = get_dependents(project, relationships)
    dependencies = get_dependencies(project, relationships)

    print(f"\n🔗 Dependency graph for: {project}\n")

    print(f"Projects that depend on {project} (will be affected by changes):")
    print_tree(dependents)

    print(f"\nProjects that {project} depends on:")
    print_tree(dependencies)

    # Impact summary
    print("\nImpact summary:")
    if dependents:
        dependent_names = [name for name, _ in dependents]
        print(f"  • Changes to {project} may affect {len(dependents)} project(s)")
        print(f"  • Consider running tests in: {', '.join(dependent_names)}")
    else:
        print(f"  • Changes to {project} affect only itself")

    if dependencies:
        dependency_names = [name for name, _ in dependencies]
        print(f"  • Ensure these are running when testing: {', '.join(dependency_names)}")

    print()


if __name__ == "__main__":
    main()