unitysvc-services 0.1.0__py3-none-any.whl

unitysvc_services/__init__.py

@@ -0,0 +1,4 @@
+"""Top-level package for UnitySvc Provider."""
+
+__author__ = """Bo Peng"""
+__email__ = "bo.peng@unitysvc.com"

unitysvc_services/cli.py

@@ -0,0 +1,21 @@
+"""Console script for unitysvc_services."""
+
+import typer
+
+from . import format_data, populate, publisher, query, scaffold, update, validator
+from . import list as list_cmd
+
+app = typer.Typer()
+
+# Register command groups
+# Init commands are defined in scaffold.py alongside their implementation
+app.add_typer(scaffold.app, name="init")
+app.add_typer(list_cmd.app, name="list")
+app.add_typer(query.app, name="query")
+app.add_typer(publisher.app, name="publish")
+app.add_typer(update.app, name="update")
+
+# Register standalone commands at root level
+app.command("format")(format_data.format_data)
+app.command("validate")(validator.validate)
+app.command("populate")(populate.populate)

unitysvc_services/format_data.py

@@ -0,0 +1,145 @@
+"""Format command - format data files."""
+
+import os
+from pathlib import Path
+
+import typer
+from rich.console import Console
+
+app = typer.Typer(help="Format data files")
+console = Console()
+
+
+@app.command()
+def format_data(
+    data_dir: Path | None = typer.Argument(
+        None,
+        help="Directory containing data files to format (default: ./data or UNITYSVC_DATA_DIR env var)",
+    ),
+    check_only: bool = typer.Option(
+        False,
+        "--check",
+        help="Check if files are formatted without modifying them",
+    ),
+):
+    """
+    Format data files (JSON, TOML, MD) to match pre-commit requirements.
+
+    This command:
+    - Formats JSON files with 2-space indentation
+    - Removes trailing whitespace
+    - Ensures files end with a newline
+    - Validates TOML syntax
+    """
+    import json as json_lib
+
+    # Set data directory
+    if data_dir is None:
+        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
+        if data_dir_str:
+            data_dir = Path(data_dir_str)
+        else:
+            data_dir = Path.cwd() / "data"
+
+    if not data_dir.is_absolute():
+        data_dir = Path.cwd() / data_dir
+
+    if not data_dir.exists():
+        console.print(f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red")
+        raise typer.Exit(code=1)
+
+    console.print(f"[blue]{'Checking' if check_only else 'Formatting'} files in:[/blue] {data_dir}\n")
+
+    # Find all JSON, TOML, and MD files
+    all_files: list[Path] = []
+    for ext in ["json", "toml", "md"]:
+        all_files.extend(data_dir.rglob(f"*.{ext}"))
+
+    if not all_files:
+        console.print("[yellow]No files found to format.[/yellow]")
+        return
+
+    console.print(f"[cyan]Found {len(all_files)} file(s) to process[/cyan]\n")
+
+    files_formatted = 0
+    files_with_issues = []
+    files_failed = []
+
+    for file_path in sorted(all_files):
+        try:
+            # Read file content
+            with open(file_path, encoding="utf-8") as f:
+                original_content = f.read()
+
+            modified_content = original_content
+            changes = []
+
+            # Format based on file type
+            if file_path.suffix == ".json":
+                # Parse and reformat JSON
+                try:
+                    data = json_lib.loads(original_content)
+                    formatted_json = json_lib.dumps(data, indent=2, sort_keys=True, separators=(",", ": "))
+                    modified_content = formatted_json
+                    if modified_content != original_content.rstrip("\n"):
+                        changes.append("reformatted JSON")
+                except json_lib.JSONDecodeError as e:
+                    console.print(f"[red]✗[/red] Invalid JSON in {file_path}: {e}")
+                    files_failed.append(str(file_path.relative_to(data_dir)))
+                    continue
+
+            # Remove trailing whitespace from each line
+            lines = modified_content.split("\n")
+            stripped_lines = [line.rstrip() for line in lines]
+            modified_content = "\n".join(stripped_lines)
+            if "\n".join([line.rstrip() for line in original_content.split("\n")]) != modified_content:
+                if "reformatted JSON" not in changes:
+                    changes.append("removed trailing whitespace")
+
+            # Ensure file ends with single newline
+            if not modified_content.endswith("\n"):
+                modified_content += "\n"
+                changes.append("added end-of-file newline")
+            elif modified_content.endswith("\n\n"):
+                # Remove extra newlines at end
+                modified_content = modified_content.rstrip("\n") + "\n"
+                changes.append("fixed multiple end-of-file newlines")
+
+            # Check if file was modified
+            if modified_content != original_content:
+                files_with_issues.append(str(file_path.relative_to(data_dir)))
+
+                if check_only:
+                    console.print(f"[yellow]✗ Would format:[/yellow] {file_path.relative_to(data_dir)}")
+                    if changes:
+                        console.print(f"  [dim]Changes: {', '.join(changes)}[/dim]")
+                else:
+                    # Write formatted content
+                    with open(file_path, "w", encoding="utf-8") as f:
+                        f.write(modified_content)
+                    console.print(f"[green]✓ Formatted:[/green] {file_path.relative_to(data_dir)}")
+                    if changes:
+                        console.print(f"  [dim]Changes: {', '.join(changes)}[/dim]")
+                    files_formatted += 1
+            else:
+                if not check_only:
+                    console.print(f"[dim]✓ Already formatted:[/dim] {file_path.relative_to(data_dir)}")
+
+        except Exception as e:
+            console.print(f"[red]✗ Error processing {file_path.relative_to(data_dir)}: {e}[/red]")
+            files_failed.append(str(file_path.relative_to(data_dir)))
+
+    # Print summary
+    console.print("\n" + "=" * 50)
+    console.print("[bold]Format Summary:[/bold]")
+    console.print(f"  Total files: {len(all_files)}")
+    if check_only:
+        console.print(f"  [yellow]Files needing formatting: {len(files_with_issues)}[/yellow]")
+    else:
+        console.print(f"  [green]✓ Files formatted: {files_formatted}[/green]")
+        console.print(f"  [dim]Already formatted: {len(all_files) - files_formatted - len(files_failed)}[/dim]")
+    if files_failed:
+        console.print(f"  [red]✗ Failed: {len(files_failed)}[/red]")
+
+    if files_failed or (check_only and files_with_issues):
+        raise typer.Exit(code=1)

unitysvc_services/list.py

@@ -0,0 +1,245 @@
+"""List command group - list local data files."""
+
+import os
+from pathlib import Path
+
+import typer
+from rich.console import Console
+from rich.table import Table
+
+from .utils import (
+    find_files_by_schema,
+    resolve_provider_name,
+    resolve_service_name_for_listing,
+)
+
+app = typer.Typer(help="List data files in directory")
+console = Console()
+
+
+@app.command("providers")
+def list_providers(
+    data_dir: Path | None = typer.Argument(
+        None,
+        help="Directory containing provider files (default: ./data or UNITYSVC_DATA_DIR env var)",
+    ),
+):
+    """List all provider files found in the data directory."""
+    # Set data directory
+    if data_dir is None:
+        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
+        if data_dir_str:
+            data_dir = Path(data_dir_str)
+        else:
+            data_dir = Path.cwd() / "data"
+
+    if not data_dir.is_absolute():
+        data_dir = Path.cwd() / data_dir
+
+    if not data_dir.exists():
+        console.print(
+            f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red"
+        )
+        raise typer.Exit(code=1)
+
+    console.print(f"[blue]Searching for providers in:[/blue] {data_dir}\n")
+
+    # Find provider files by schema
+    provider_files = find_files_by_schema(data_dir, "provider_v1")
+
+    if not provider_files:
+        console.print("[yellow]No provider files found.[/yellow]")
+        return
+
+    # Create table
+    table = Table(title="Provider Files", show_lines=True)
+    table.add_column("File", style="cyan")
+    table.add_column("Name", style="green")
+    table.add_column("Display Name", style="blue")
+
+    for file_path, _file_format, data in sorted(provider_files, key=lambda x: x[0]):
+        table.add_row(
+            str(file_path.relative_to(data_dir)),
+            data.get("name", "N/A"),
+            data.get("display_name", "N/A"),
+        )
+
+    console.print(table)
+    console.print(f"\n[green]Total:[/green] {len(provider_files)} provider file(s)")
+
+
+@app.command("sellers")
+def list_sellers(
+    data_dir: Path | None = typer.Argument(
+        None,
+        help="Directory containing seller files (default: ./data or UNITYSVC_DATA_DIR env var)",
+    ),
+):
+    """List all seller files found in the data directory."""
+    # Set data directory
+    if data_dir is None:
+        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
+        if data_dir_str:
+            data_dir = Path(data_dir_str)
+        else:
+            data_dir = Path.cwd() / "data"
+
+    if not data_dir.is_absolute():
+        data_dir = Path.cwd() / data_dir
+
+    if not data_dir.exists():
+        console.print(
+            f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red"
+        )
+        raise typer.Exit(code=1)
+
+    console.print(f"[blue]Searching for sellers in:[/blue] {data_dir}\n")
+
+    # Find seller files by schema
+    seller_files = find_files_by_schema(data_dir, "seller_v1")
+
+    if not seller_files:
+        console.print("[yellow]No seller files found.[/yellow]")
+        return
+
+    # Create table
+    table = Table(title="Seller Files", show_lines=True)
+    table.add_column("File", style="cyan")
+    table.add_column("Name", style="green")
+    table.add_column("Display Name", style="blue")
+
+    for file_path, _file_format, data in sorted(seller_files, key=lambda x: x[0]):
+        table.add_row(
+            str(file_path.relative_to(data_dir)),
+            data.get("name", "N/A"),
+            data.get("display_name", "N/A"),
+        )
+
+    console.print(table)
+    console.print(f"\n[green]Total:[/green] {len(seller_files)} seller file(s)")
+
+
+@app.command("offerings")
+def list_offerings(
+    data_dir: Path | None = typer.Argument(
+        None,
+        help="Directory containing service files (default: ./data or UNITYSVC_DATA_DIR env var)",
+    ),
+):
+    """List all service offering files found in the data directory."""
+    # Set data directory
+    if data_dir is None:
+        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
+        if data_dir_str:
+            data_dir = Path(data_dir_str)
+        else:
+            data_dir = Path.cwd() / "data"
+
+    if not data_dir.is_absolute():
+        data_dir = Path.cwd() / data_dir
+
+    if not data_dir.exists():
+        console.print(
+            f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red"
+        )
+        raise typer.Exit(code=1)
+
+    console.print(f"[blue]Searching for service offerings in:[/blue] {data_dir}\n")
+
+    # Find service files by schema
+    service_files = find_files_by_schema(data_dir, "service_v1")
+
+    if not service_files:
+        console.print("[yellow]No service offering files found.[/yellow]")
+        return
+
+    # Create table
+    table = Table(title="Service Offering Files", show_lines=True)
+    table.add_column("File", style="cyan")
+    table.add_column("Provider", style="yellow")
+    table.add_column("Name", style="green")
+    table.add_column("Display Name", style="blue")
+    table.add_column("Status", style="magenta")
+
+    for file_path, _file_format, data in sorted(service_files, key=lambda x: x[0]):
+        provider_name = resolve_provider_name(file_path) or "N/A"
+        table.add_row(
+            str(file_path.relative_to(data_dir)),
+            provider_name,
+            data.get("name", "N/A"),
+            data.get("display_name", "N/A"),
+            data.get("upstream_status", "N/A"),
+        )
+
+    console.print(table)
+    console.print(
+        f"\n[green]Total:[/green] {len(service_files)} service offering file(s)"
+    )
+
+
+@app.command("listings")
+def list_listings(
+    data_dir: Path | None = typer.Argument(
+        None,
+        help="Directory containing listing files (default: ./data or UNITYSVC_DATA_DIR env var)",
+    ),
+):
+    """List all service listing files found in the data directory."""
+    # Set data directory
+    if data_dir is None:
+        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
+        if data_dir_str:
+            data_dir = Path(data_dir_str)
+        else:
+            data_dir = Path.cwd() / "data"
+
+    if not data_dir.is_absolute():
+        data_dir = Path.cwd() / data_dir
+
+    if not data_dir.exists():
+        console.print(
+            f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red"
+        )
+        raise typer.Exit(code=1)
+
+    console.print(f"[blue]Searching for service listings in:[/blue] {data_dir}\n")
+
+    # Find seller definition to get display names
+    seller_files = find_files_by_schema(data_dir, "seller_v1")
+    seller_info = {}
+    if seller_files:
+        # Use the first (and should be only) seller file
+        seller_info = seller_files[0][-1]
+
+    # Find listing files by schema
+    listing_files = find_files_by_schema(data_dir, "listing_v1")
+
+    if not listing_files:
+        console.print("[yellow]No service listing files found.[/yellow]")
+        return
+
+    # Create table
+    table = Table(title="Service Listing Files", show_lines=True)
+    table.add_column("File", style="cyan")
+    table.add_column("Provider", style="yellow")
+    table.add_column("Service", style="blue")
+    table.add_column("Seller", style="green")
+    table.add_column("Status", style="magenta")
+
+    for file_path, _file_format, data in sorted(listing_files, key=lambda x: x[0]):
+        # Resolve provider and service names using the utility functions
+        provider_name = resolve_provider_name(file_path) or "N/A"
+        service_name = resolve_service_name_for_listing(file_path, data) or "N/A"
+
+        table.add_row(
+            str(file_path.relative_to(data_dir)),
+            provider_name,
+            service_name,
+            seller_info.get("name", "N/A"),
+            data.get("listing_status", "N/A"),
+        )
+
+    console.print(table)
+    console.print(
+        f"\n[green]Total:[/green] {len(listing_files)} service listing file(s)"
+    )

unitysvc_services/models/__init__.py

@@ -0,0 +1,6 @@
+from .listing_v1 import ListingV1
+from .provider_v1 import ProviderV1
+from .seller_v1 import SellerV1
+from .service_v1 import ServiceV1
+
+__all__ = ["ProviderV1", "ServiceV1", "ListingV1", "SellerV1"]