nat-engine 1__py3-none-any.whl

mannf/product/cli_admin.py

@@ -0,0 +1,408 @@
+# Copyright (C) 2026 Brad Guider
+# This file is part of NAT (Neural Agent Testing Framework).
+# Licensed under the AGPL-3.0. See LICENSE for details.
+
+"""NAT Admin CLI — manage tenants, API keys, and plans.
+
+Usage::
+
+    nat-admin tenant create --name "Acme Corp" --email admin@acme.com [--plan free]
+    nat-admin tenant list [--plan free|pro|team|enterprise] [--limit 50]
+    nat-admin tenant show <tenant-id>
+    nat-admin tenant update <tenant-id> --plan pro [--quota 500] [--security-quota 100]
+    nat-admin tenant deactivate <tenant-id>
+    nat-admin tenant rotate-key <tenant-id>
+    nat-admin tenant usage <tenant-id>
+
+Each subcommand supports ``--json`` for machine-readable output.
+
+Exit codes:
+    0 — success
+    1 — error (tenant not found, validation error, etc.)
+"""
+
+from __future__ import annotations
+
+import argparse
+import asyncio
+import json
+import os
+import sys
+from typing import Any, Optional
+
+from mannf.product.billing.tenant_manager import TenantManager
+
+
+# ---------------------------------------------------------------------------
+# Output helpers
+# ---------------------------------------------------------------------------
+
+
+def _print_json(data: Any) -> None:
+    print(json.dumps(data, indent=2, default=str))
+
+
+def _print_table(rows: list[dict], columns: Optional[list[str]] = None) -> None:
+    """Print a list of dicts as a simple ASCII table."""
+    if not rows:
+        print("(no results)")
+        return
+
+    if columns is None:
+        columns = list(rows[0].keys())
+
+    # Calculate column widths
+    widths: dict[str, int] = {col: len(col) for col in columns}
+    for row in rows:
+        for col in columns:
+            val = str(row.get(col, ""))
+            if len(val) > widths[col]:
+                widths[col] = min(len(val), 60)  # cap at 60 chars
+
+    header = "  ".join(col.upper().ljust(widths[col]) for col in columns)
+    separator = "  ".join("-" * widths[col] for col in columns)
+    print(header)
+    print(separator)
+    for row in rows:
+        line = "  ".join(
+            str(row.get(col, ""))[:widths[col]].ljust(widths[col])
+            for col in columns
+        )
+        print(line)
+
+
+def _print_record(data: dict) -> None:
+    """Print a single record as aligned key-value pairs."""
+    max_key = max((len(k) for k in data), default=0)
+    for key, value in data.items():
+        print(f"  {key.ljust(max_key)}  {value}")
+
+
+# ---------------------------------------------------------------------------
+# DB initialisation
+# ---------------------------------------------------------------------------
+
+
+def _init_db() -> None:
+    """Ensure the database engine is initialised from DATABASE_URL."""
+    db_url = os.environ.get("DATABASE_URL")
+    if not db_url:
+        print(
+            "Error: DATABASE_URL is not set. "
+            "Set it to a PostgreSQL connection string and retry.",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    try:
+        from mannf.product.database import _build_engine  # noqa: PLC0415
+
+        if _build_engine() is None:
+            print(
+                "Error: Failed to initialise the database engine. "
+                "Check DATABASE_URL and ensure the database is reachable.",
+                file=sys.stderr,
+            )
+            sys.exit(1)
+    except ImportError as exc:
+        print(f"Error: Database module not available: {exc}", file=sys.stderr)
+        sys.exit(1)
+
+
+# ---------------------------------------------------------------------------
+# Command handlers
+# ---------------------------------------------------------------------------
+
+
+async def _cmd_create(args: argparse.Namespace) -> int:
+    manager = TenantManager()
+    try:
+        result = await manager.create_tenant(
+            name=args.name,
+            email=args.email,
+            plan_tier=args.plan,
+        )
+    except (RuntimeError, ValueError) as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+
+    if args.json:
+        _print_json(result)
+    else:
+        print("Tenant created successfully.")
+        print()
+        _print_record(result)
+        print()
+        print("⚠️  Save the api_key now — it will not be shown again.")
+    return 0
+
+
+async def _cmd_list(args: argparse.Namespace) -> int:
+    manager = TenantManager()
+    try:
+        tenants = await manager.list_tenants(
+            plan_filter=args.plan,
+            active_only=not args.all,
+            limit=args.limit,
+        )
+    except RuntimeError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+
+    if args.json:
+        _print_json(tenants)
+    else:
+        _print_table(
+            tenants,
+            columns=["tenant_id", "name", "email", "plan", "is_active", "created_at"],
+        )
+        print(f"\n{len(tenants)} tenant(s) listed.")
+    return 0
+
+
+async def _cmd_show(args: argparse.Namespace) -> int:
+    manager = TenantManager()
+    try:
+        result = await manager.get_tenant(args.tenant_id)
+    except RuntimeError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+
+    if result is None:
+        print(f"Error: Tenant {args.tenant_id!r} not found.", file=sys.stderr)
+        return 1
+
+    if args.json:
+        _print_json(result)
+    else:
+        _print_record(result)
+    return 0
+
+
+async def _cmd_update(args: argparse.Namespace) -> int:
+    manager = TenantManager()
+    try:
+        result = await manager.update_tenant(
+            tenant_id=args.tenant_id,
+            plan_tier=args.plan,
+            monthly_scan_quota=args.quota,
+            monthly_security_scan_quota=args.security_quota,
+        )
+    except KeyError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+    except (RuntimeError, ValueError) as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+
+    if args.json:
+        _print_json(result)
+    else:
+        print("Tenant updated.")
+        print()
+        _print_record(result)
+    return 0
+
+
+async def _cmd_deactivate(args: argparse.Namespace) -> int:
+    manager = TenantManager()
+    try:
+        result = await manager.deactivate_tenant(args.tenant_id)
+    except KeyError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+    except RuntimeError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+
+    if args.json:
+        _print_json(result)
+    else:
+        print(f"Tenant {args.tenant_id} has been deactivated.")
+        print()
+        _print_record(result)
+    return 0
+
+
+async def _cmd_rotate_key(args: argparse.Namespace) -> int:
+    manager = TenantManager()
+    try:
+        result = await manager.rotate_api_key(args.tenant_id)
+    except KeyError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+    except RuntimeError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+
+    if args.json:
+        _print_json(result)
+    else:
+        print("API key rotated.")
+        print()
+        _print_record(result)
+        print()
+        print("⚠️  Save the new api_key now — it will not be shown again.")
+    return 0
+
+
+async def _cmd_usage(args: argparse.Namespace) -> int:
+    manager = TenantManager()
+    try:
+        result = await manager.get_tenant_usage(args.tenant_id)
+    except KeyError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+    except RuntimeError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+
+    if args.json:
+        _print_json(result)
+    else:
+        _print_record(result)
+    return 0
+
+
+# ---------------------------------------------------------------------------
+# Argument parser
+# ---------------------------------------------------------------------------
+
+
+def _build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        prog="nat-admin",
+        description="NAT Admin CLI — manage tenants, API keys, and plans.",
+    )
+    parser.add_argument(
+        "--json",
+        action="store_true",
+        default=False,
+        help="Output results as JSON instead of a formatted table.",
+    )
+
+    subparsers = parser.add_subparsers(dest="resource", metavar="<resource>")
+    subparsers.required = True
+
+    # ---- tenant ----
+    tenant_parser = subparsers.add_parser("tenant", help="Tenant operations.")
+    tenant_subs = tenant_parser.add_subparsers(dest="action", metavar="<action>")
+    tenant_subs.required = True
+
+    # tenant create
+    create_p = tenant_subs.add_parser("create", help="Create a new tenant.")
+    create_p.add_argument("--name", required=True, help="Tenant display name.")
+    create_p.add_argument("--email", required=True, help="Tenant billing email.")
+    create_p.add_argument(
+        "--plan",
+        default="free",
+        choices=["free", "pro", "team", "enterprise"],
+        help="Plan tier (default: free).",
+    )
+
+    # tenant list
+    list_p = tenant_subs.add_parser("list", help="List tenants.")
+    list_p.add_argument(
+        "--plan",
+        choices=["free", "pro", "team", "enterprise"],
+        default=None,
+        help="Filter by plan tier.",
+    )
+    list_p.add_argument(
+        "--limit",
+        type=int,
+        default=100,
+        metavar="N",
+        help="Maximum number of results (default: 100).",
+    )
+    list_p.add_argument(
+        "--all",
+        action="store_true",
+        default=False,
+        help="Include inactive tenants.",
+    )
+
+    # tenant show
+    show_p = tenant_subs.add_parser("show", help="Show tenant details.")
+    show_p.add_argument("tenant_id", metavar="<tenant-id>")
+
+    # tenant update
+    update_p = tenant_subs.add_parser("update", help="Update tenant settings.")
+    update_p.add_argument("tenant_id", metavar="<tenant-id>")
+    update_p.add_argument(
+        "--plan",
+        choices=["free", "pro", "team", "enterprise"],
+        default=None,
+        help="New plan tier.",
+    )
+    update_p.add_argument(
+        "--quota",
+        type=int,
+        default=None,
+        metavar="N",
+        dest="quota",
+        help="Override monthly scan quota.",
+    )
+    update_p.add_argument(
+        "--security-quota",
+        type=int,
+        default=None,
+        metavar="N",
+        dest="security_quota",
+        help="Override monthly security scan quota.",
+    )
+
+    # tenant deactivate
+    deactivate_p = tenant_subs.add_parser("deactivate", help="Deactivate a tenant.")
+    deactivate_p.add_argument("tenant_id", metavar="<tenant-id>")
+
+    # tenant rotate-key
+    rotate_p = tenant_subs.add_parser("rotate-key", help="Rotate the tenant's API key.")
+    rotate_p.add_argument("tenant_id", metavar="<tenant-id>")
+
+    # tenant usage
+    usage_p = tenant_subs.add_parser("usage", help="Show current month's usage.")
+    usage_p.add_argument("tenant_id", metavar="<tenant-id>")
+
+    return parser
+
+
+# ---------------------------------------------------------------------------
+# Dispatch
+# ---------------------------------------------------------------------------
+
+
+_ACTION_MAP = {
+    "create": _cmd_create,
+    "list": _cmd_list,
+    "show": _cmd_show,
+    "update": _cmd_update,
+    "deactivate": _cmd_deactivate,
+    "rotate-key": _cmd_rotate_key,
+    "usage": _cmd_usage,
+}
+
+
+def main(argv: Optional[list[str]] = None) -> None:
+    """Entry point for the ``nat-admin`` CLI command."""
+    parser = _build_parser()
+    args = parser.parse_args(argv)
+
+    # Propagate the top-level --json flag to the parsed namespace so
+    # sub-command handlers can access it.
+    if not hasattr(args, "json"):
+        args.json = False
+
+    _init_db()
+
+    handler = _ACTION_MAP.get(args.action)
+    if handler is None:
+        parser.print_help()
+        sys.exit(1)
+
+    exit_code = asyncio.run(handler(args))
+    sys.exit(exit_code)
+
+
+if __name__ == "__main__":
+    main()

mannf/product/dashboard/__init__.py

@@ -0,0 +1,61 @@
+# Copyright (C) 2026 Brad Guider
+# This file is part of NAT (Neural Agent Testing Framework).
+# Licensed under the AGPL-3.0. See LICENSE for details.
+# Commercial licensing available — see COMMERCIAL_LICENSE.md.
+
+"""NAT Web Dashboard package.
+
+Provides a single-page web dashboard for visualizing BDI agent belief
+states, endpoint risk heatmaps, anomaly alerts, and scan history.
+"""
+
+from mannf.product.dashboard.app import router as dashboard_router
+from mannf.product.dashboard.models import (
+    A11yViolationSummary,
+    AnomalyTimelineEntry,
+    BeliefStateSnapshot,
+    BeliefTrend,
+    DashboardSummary,
+    DiscoveryRequest,
+    DiscoveryResult,
+    DiscoveryScenario,
+    EndpointRiskCard,
+    FailureCluster,
+    FlakinessSummary,
+    FunctionalTestItem,
+    FunctionalTestResult,
+    InvoiceEntry,
+    OnboardingStatus,
+    OnboardingStep,
+    PerfMetrics,
+    PredictiveRiskScore,
+    RootCauseSuggestion,
+    UsageSummary,
+    VisualDiffEntry,
+)
+
+__all__ = [
+    "dashboard_router",
+    "DashboardSummary",
+    "EndpointRiskCard",
+    "BeliefStateSnapshot",
+    "AnomalyTimelineEntry",
+    "FunctionalTestResult",
+    "FunctionalTestItem",
+    "VisualDiffEntry",
+    "A11yViolationSummary",
+    "PerfMetrics",
+    "UsageSummary",
+    "InvoiceEntry",
+    "OnboardingStatus",
+    "OnboardingStep",
+    "DiscoveryRequest",
+    "DiscoveryScenario",
+    "DiscoveryResult",
+    # Phase 6.5
+    "BeliefTrend",
+    "FlakinessSummary",
+    "FailureCluster",
+    "RootCauseSuggestion",
+    "PredictiveRiskScore",
+]