gh-danielscholl-agent-skills-plugins-osdu/skills/osdu/scripts/status.py

#!/usr/bin/env python3
# /// script
# dependencies = [
#     "click",
# ]
# ///

"""
OSDU Quality Status Script

Get latest test status by stage (unit/integration/acceptance) for OSDU projects.
Wraps the osdu-quality CLI tool for GitLab CI/CD pipeline analysis.

Usage:
    uv run status.py --json --pipelines 10
    uv run status.py --json --project "partition"
    uv run status.py --json --venus --no-release
"""

import json
import os
import subprocess
import sys
from typing import Any

import click


def check_prerequisites() -> None:
    """
    Check that osdu-quality CLI is installed and GitLab authentication is configured.

    Authentication can be either:
    - GITLAB_TOKEN environment variable
    - glab CLI authentication (glab auth login)

    Raises:
        Exception: If prerequisites are not met
    """
    # Check GitLab authentication (GITLAB_TOKEN or glab auth)
    has_token = bool(os.getenv("GITLAB_TOKEN"))
    has_glab_auth = False

    if not has_token:
        # Check if glab is authenticated
        try:
            result = subprocess.run(
                ["glab", "auth", "status"],
                capture_output=True,
                check=True,
                text=True,
            )
            has_glab_auth = True
        except (FileNotFoundError, subprocess.CalledProcessError):
            has_glab_auth = False

    if not has_token and not has_glab_auth:
        raise Exception(
            "GitLab authentication not configured.\n"
            "Use one of:\n"
            "  - export GITLAB_TOKEN='your-token-here'\n"
            "  - glab auth login"
        )

    # Check osdu-quality CLI
    try:
        subprocess.run(
            ["osdu-quality", "--version"],
            capture_output=True,
            check=True,
            text=True,
        )
    except FileNotFoundError:
        raise Exception(
            "osdu-quality CLI not found.\n"
            "Install with: uv tool install git+https://community.opengroup.org/danielscholl/osdu-quality.git"
        )
    except subprocess.CalledProcessError as e:
        raise Exception(f"osdu-quality CLI check failed: {e.stderr}")


def run_osdu_status(
    pipelines: int,
    project: str | None,
    venus: bool,
    no_release: bool,
    output_format: str,
) -> tuple[str, int]:
    """
    Execute osdu-quality status command.

    Args:
        pipelines: Number of pipelines to analyze
        project: Optional project name filter
        venus: Filter Venus provider only
        no_release: Exclude release pipelines
        output_format: Output format (json, markdown, or terminal)

    Returns:
        Tuple of (output_string, exit_code)

    Raises:
        Exception: If command execution fails
    """
    # Build command
    cmd = ["osdu-quality", "status", "--pipelines", str(pipelines)]

    if project:
        cmd.extend(["--project", project])
    if venus:
        cmd.append("--venus")
    if no_release:
        cmd.append("--no-release")

    # Add output format if not default terminal (tty)
    if output_format != "terminal":
        cmd.extend(["--output", output_format])

    try:
        # Execute command
        result = subprocess.run(
            cmd,
            capture_output=True,
            check=True,
            text=True,
        )
        return result.stdout, 0

    except subprocess.CalledProcessError as e:
        # Command failed - return stderr
        error_msg = e.stderr if e.stderr else e.stdout
        raise Exception(f"osdu-quality command failed:\n{error_msg}")
    except Exception as e:
        raise Exception(f"Unexpected error: {str(e)}")


@click.command()
@click.option(
    "--pipelines",
    default=10,
    type=int,
    help="Number of pipelines to analyze (default: 10)",
)
@click.option(
    "--project",
    type=str,
    help="Filter by project name (e.g., 'partition')",
)
@click.option(
    "--venus",
    is_flag=True,
    help="Filter Venus provider pipelines only",
)
@click.option(
    "--no-release",
    is_flag=True,
    help="Exclude release pipelines from analysis",
)
@click.option(
    "--format",
    "output_format",
    type=click.Choice(["json", "markdown", "terminal"], case_sensitive=False),
    default="terminal",
    help="Output format: json (structured data), markdown (reports/docs), or terminal (default, colored tables)",
)
def main(
    pipelines: int,
    project: str | None,
    venus: bool,
    no_release: bool,
    output_format: str,
):
    """
    Get latest test status by stage for OSDU projects.

    Returns pipeline status with test results organized by stage
    (unit, integration, acceptance).

    Output Formats:
    - json: Structured data for AI parsing and automation
    - markdown: Human-readable reports, great for documentation and sharing
    - terminal: Interactive display with colors and tables (default)

    Requires:
    - osdu-quality CLI installed (uv tool install git+https://...)
    - GitLab authentication (GITLAB_TOKEN env var OR glab auth login)
    """
    try:
        # Check prerequisites
        check_prerequisites()

        # Run osdu-quality status command
        output, exit_code = run_osdu_status(
            pipelines=pipelines,
            project=project,
            venus=venus,
            no_release=no_release,
            output_format=output_format,
        )

        # Output results
        click.echo(output)
        sys.exit(exit_code)

    except Exception as e:
        if output_format == "json":
            # JSON error format
            error_data = {"error": str(e)}
            click.echo(json.dumps(error_data, indent=2))
        else:
            # Human-readable error (markdown and terminal)
            click.echo(f"Error: {e}", err=True)
        sys.exit(1)


if __name__ == "__main__":
    main()