cli.py

import os
import sys

import typer
from diffsync.diff import Diff
from rich import print
from rich.console import Console
from rich.table import Table

from diff_nautobot_understack.device.main import ironic_nodes_diff_from_nautobot_devices
from diff_nautobot_understack.network.main import (
    openstack_network_diff_from_ucvni_network,
)
from diff_nautobot_understack.project.main import (
    openstack_all_projects_diff_from_nautobot_tenant,
)
from diff_nautobot_understack.project.main import (
    openstack_project_diff_from_nautobot_tenant,
)
from diff_nautobot_understack.settings import app_settings as settings
from diff_nautobot_understack.subnet.main import (
    openstack_subnets_diff_from_nautobot_prefixes,
)

required_env_vars = ["NAUTOBOT_TOKEN", "NAUTOBOT_URL", "OS_CLOUD"]


app = typer.Typer(
    name="diff",
    add_completion=False,
    help="compare data between Openstack and Nautobot.",
)
diff_outputs = {
    "project": {"title": "Project Diff", "id_column_name": "Project ID"},
    "network": {"title": "Network Diff", "id_column_name": "Network ID"},
    "subnet": {"title": "Subnet Diff", "id_column_name": "Subnet ID"},
    "device": {"title": "Device Diff", "id_column_name": "Device ID"},
}


def display_output(
    diff_result: Diff, diff_output: str, output_format: str | None = None
):
    print(diff_result.summary())
    __output_format = (
        output_format if output_format is not None else settings.output_format
    )
    if __output_format == "table":
        diff_output_props = diff_outputs.get(diff_output)
        tabular_output(
            diff_result.dict().get(diff_output, {}),
            diff_output_props.get("title"),
            diff_output_props.get("id_column_name"),
        )
    elif __output_format == "human":
        print("Summary:\n")
        print(diff_result.str())
    else:
        print(diff_result.dict())


def tabular_output(diffs, title, id_column_name):
    table = Table(title=title, show_lines=True)

    table.add_column(id_column_name, style="cyan", no_wrap=True)
    table.add_column("Change Type", style="magenta")
    table.add_column("Details", style="yellow")

    for diff_id, changes in diffs.items():
        for change_type, details in changes.items():
            table.add_row(diff_id, change_type, str(details))

    console = Console()
    console.print(table)


@app.command()
def projects(
    debug: bool = typer.Option(False, "--debug", "-v", help="Enable debug mode"),
    output_format: str = typer.Option(
        "json", "--format", help="Available formats json, table"
    ),
):
    """OpenStack projects ⟹ Nautobot tenants"""
    settings.debug = debug
    diff_result = openstack_all_projects_diff_from_nautobot_tenant()
    display_output(diff_result, "project", output_format)


@app.command()
def project(
    name: str,
    debug: bool = typer.Option(False, "--debug", "-v", help="Enable debug mode"),
    output_format: str = typer.Option(
        "json", "--format", help="Available formats json, table"
    ),
):
    """OpenStack projects ⟹ Nautobot tenants"""
    settings.debug = debug
    diff_result = openstack_project_diff_from_nautobot_tenant(os_project=name)
    display_output(diff_result, "project", output_format)


@app.command()
def network(
    debug: bool = typer.Option(False, "--debug", "-v", help="Enable debug mode"),
    output_format: str = typer.Option(
        "json", "--format", help="Available formats: json, table, human"
    ),
):
    """OpenStack networks ⟹ Nautobot UCVNIs"""
    settings.debug = debug
    diff_result = openstack_network_diff_from_ucvni_network()
    display_output(diff_result, "network", output_format)


@app.command()
def subnets(
    debug: bool = typer.Option(False, "--debug", "-v", help="Enable debug mode"),
    output_format: str = typer.Option(
        "json", "--format", help="Available formats: json, table, human"
    ),
):
    """OpenStack subnets ⟹ Nautobot prefixes"""
    settings.debug = debug
    diff_result = openstack_subnets_diff_from_nautobot_prefixes()
    display_output(diff_result, "subnet", output_format)


@app.command()
def devices(
    debug: bool = typer.Option(False, "--debug", "-v", help="Enable debug mode"),
    output_format: str = typer.Option(
        "json", "--format", help="Available formats: json, table, human"
    ),
):
    """Ironic nodes ⟹ Nautobot devices"""
    settings.debug = debug
    diff_result = ironic_nodes_diff_from_nautobot_devices()
    display_output(diff_result, "device", output_format)


def check_env_vars(required_vars):
    missing_vars = [var for var in required_vars if var not in os.environ]

    if missing_vars:
        print(f"Error: Missing environment variables: {', '.join(missing_vars)}")
        sys.exit(1)
    else:
        print("All required environment variables are set.")


check_env_vars(required_env_vars)