checkdk-cli 0.2.0__tar.gz

checkdk_cli-0.2.0/PKG-INFO

@@ -0,0 +1,131 @@
+Metadata-Version: 2.4
+Name: checkdk-cli
+Version: 0.2.0
+Summary: checkDK CLI – AI-powered Docker/Kubernetes issue detector and pod failure predictor
+Author-email: checkDK Team <team@checkdk.app>
+License: MIT
+Project-URL: Homepage, https://checkdk.app
+Project-URL: Documentation, https://checkdk.app/docs
+Project-URL: Repository, https://github.com/your-org/checkDK_AWS
+Project-URL: Bug Tracker, https://github.com/your-org/checkDK_AWS/issues
+Project-URL: Changelog, https://github.com/your-org/checkDK_AWS/releases
+Keywords: docker,kubernetes,devops,cli,ai,chaos-engineering,monitoring
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: System Administrators
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Build Tools
+Classifier: Topic :: System :: Systems Administration
+Classifier: Topic :: Utilities
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: click>=8.1.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: requests>=2.31.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: websocket-client>=1.6.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+
+# checkDK CLI
+
+Thin CLI tool that wraps `docker` and `kubectl` commands with pre-execution
+analysis, and provides a `predict` command for pod failure detection.
+
+All analysis is delegated to the **checkDK backend API** running in Docker.
+The CLI itself has no ML or LLM dependencies — it is just an HTTP client with
+a rich terminal UI.
+
+---
+
+## Quick start
+
+```bash
+# 1. Clone and enter the cli folder
+cd cli
+
+# 2. Run the setup script (creates .venv + installs the package)
+bash setup.sh
+
+# 3. Activate the virtual environment
+source .venv/bin/activate
+
+# 4. Point at the backend (start it first with docker compose)
+export CHECKDK_API_URL=http://localhost:8000
+# — or save it permanently:
+checkdk init
+```
+
+---
+
+## Start the backend first
+
+```bash
+# From the repo root
+cp .env.example .env          # add GROQ_API_KEY / GEMINI_API_KEY
+docker compose up --build     # starts backend on :8000, frontend on :3000
+```
+
+---
+
+## Commands
+
+| Command | Description |
+|---|---|
+| `checkdk init` | Save API URL + keys to `~/.checkdk/.env` |
+| `checkdk docker compose up -d` | Analyse `docker-compose.yml`, then run the command |
+| `checkdk docker compose up --dry-run` | Analyse only, do not execute |
+| `checkdk kubectl apply -f deploy.yaml` | Analyse Kubernetes manifest, then apply |
+| `checkdk predict --cpu 93 --memory 91` | Predict pod failure risk |
+| `checkdk predict --cpu 93 --memory 91 --no-ai` | ML prediction only, skip LLM |
+
+### Predict options
+
+```
+  --cpu            CPU usage % (required)
+  --memory         Memory usage % (required)
+  --disk           Disk usage % (default 50)
+  --latency        Network latency ms (default 10)
+  --restarts       Restart count (default 0)
+  --probe-failures Probe failure count (default 0)
+  --cpu-pressure   Node CPU pressure 0/1 (default 0)
+  --mem-pressure   Node memory pressure 0/1 (default 0)
+  --age            Pod age in minutes (default 60)
+  --service        Service/pod name (optional label)
+  --platform       docker | kubernetes (default docker)
+  --no-ai          Skip LLM analysis
+```
+
+---
+
+## Environment variables
+
+| Variable | Default | Description |
+|---|---|---|
+| `CHECKDK_API_URL` | `http://localhost:8000` | Backend API base URL |
+
+The CLI auto-loads `~/.checkdk/.env` and `./.env` on startup.
+
+---
+
+## Development
+
+```bash
+# Install with dev extras
+pip install -e ".[dev]"
+
+# Run tests
+pytest
+
+# Lint
+ruff check .
+```

checkdk_cli-0.2.0/README.md

@@ -0,0 +1,93 @@
+# checkDK CLI
+
+Thin CLI tool that wraps `docker` and `kubectl` commands with pre-execution
+analysis, and provides a `predict` command for pod failure detection.
+
+All analysis is delegated to the **checkDK backend API** running in Docker.
+The CLI itself has no ML or LLM dependencies — it is just an HTTP client with
+a rich terminal UI.
+
+---
+
+## Quick start
+
+```bash
+# 1. Clone and enter the cli folder
+cd cli
+
+# 2. Run the setup script (creates .venv + installs the package)
+bash setup.sh
+
+# 3. Activate the virtual environment
+source .venv/bin/activate
+
+# 4. Point at the backend (start it first with docker compose)
+export CHECKDK_API_URL=http://localhost:8000
+# — or save it permanently:
+checkdk init
+```
+
+---
+
+## Start the backend first
+
+```bash
+# From the repo root
+cp .env.example .env          # add GROQ_API_KEY / GEMINI_API_KEY
+docker compose up --build     # starts backend on :8000, frontend on :3000
+```
+
+---
+
+## Commands
+
+| Command | Description |
+|---|---|
+| `checkdk init` | Save API URL + keys to `~/.checkdk/.env` |
+| `checkdk docker compose up -d` | Analyse `docker-compose.yml`, then run the command |
+| `checkdk docker compose up --dry-run` | Analyse only, do not execute |
+| `checkdk kubectl apply -f deploy.yaml` | Analyse Kubernetes manifest, then apply |
+| `checkdk predict --cpu 93 --memory 91` | Predict pod failure risk |
+| `checkdk predict --cpu 93 --memory 91 --no-ai` | ML prediction only, skip LLM |
+
+### Predict options
+
+```
+  --cpu            CPU usage % (required)
+  --memory         Memory usage % (required)
+  --disk           Disk usage % (default 50)
+  --latency        Network latency ms (default 10)
+  --restarts       Restart count (default 0)
+  --probe-failures Probe failure count (default 0)
+  --cpu-pressure   Node CPU pressure 0/1 (default 0)
+  --mem-pressure   Node memory pressure 0/1 (default 0)
+  --age            Pod age in minutes (default 60)
+  --service        Service/pod name (optional label)
+  --platform       docker | kubernetes (default docker)
+  --no-ai          Skip LLM analysis
+```
+
+---
+
+## Environment variables
+
+| Variable | Default | Description |
+|---|---|---|
+| `CHECKDK_API_URL` | `http://localhost:8000` | Backend API base URL |
+
+The CLI auto-loads `~/.checkdk/.env` and `./.env` on startup.
+
+---
+
+## Development
+
+```bash
+# Install with dev extras
+pip install -e ".[dev]"
+
+# Run tests
+pytest
+
+# Lint
+ruff check .
+```

checkdk_cli-0.2.0/checkdk_cli.egg-info/PKG-INFO

@@ -0,0 +1,131 @@
+Metadata-Version: 2.4
+Name: checkdk-cli
+Version: 0.2.0
+Summary: checkDK CLI – AI-powered Docker/Kubernetes issue detector and pod failure predictor
+Author-email: checkDK Team <team@checkdk.app>
+License: MIT
+Project-URL: Homepage, https://checkdk.app
+Project-URL: Documentation, https://checkdk.app/docs
+Project-URL: Repository, https://github.com/your-org/checkDK_AWS
+Project-URL: Bug Tracker, https://github.com/your-org/checkDK_AWS/issues
+Project-URL: Changelog, https://github.com/your-org/checkDK_AWS/releases
+Keywords: docker,kubernetes,devops,cli,ai,chaos-engineering,monitoring
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: System Administrators
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Build Tools
+Classifier: Topic :: System :: Systems Administration
+Classifier: Topic :: Utilities
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: click>=8.1.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: requests>=2.31.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: websocket-client>=1.6.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+
+# checkDK CLI
+
+Thin CLI tool that wraps `docker` and `kubectl` commands with pre-execution
+analysis, and provides a `predict` command for pod failure detection.
+
+All analysis is delegated to the **checkDK backend API** running in Docker.
+The CLI itself has no ML or LLM dependencies — it is just an HTTP client with
+a rich terminal UI.
+
+---
+
+## Quick start
+
+```bash
+# 1. Clone and enter the cli folder
+cd cli
+
+# 2. Run the setup script (creates .venv + installs the package)
+bash setup.sh
+
+# 3. Activate the virtual environment
+source .venv/bin/activate
+
+# 4. Point at the backend (start it first with docker compose)
+export CHECKDK_API_URL=http://localhost:8000
+# — or save it permanently:
+checkdk init
+```
+
+---
+
+## Start the backend first
+
+```bash
+# From the repo root
+cp .env.example .env          # add GROQ_API_KEY / GEMINI_API_KEY
+docker compose up --build     # starts backend on :8000, frontend on :3000
+```
+
+---
+
+## Commands
+
+| Command | Description |
+|---|---|
+| `checkdk init` | Save API URL + keys to `~/.checkdk/.env` |
+| `checkdk docker compose up -d` | Analyse `docker-compose.yml`, then run the command |
+| `checkdk docker compose up --dry-run` | Analyse only, do not execute |
+| `checkdk kubectl apply -f deploy.yaml` | Analyse Kubernetes manifest, then apply |
+| `checkdk predict --cpu 93 --memory 91` | Predict pod failure risk |
+| `checkdk predict --cpu 93 --memory 91 --no-ai` | ML prediction only, skip LLM |
+
+### Predict options
+
+```
+  --cpu            CPU usage % (required)
+  --memory         Memory usage % (required)
+  --disk           Disk usage % (default 50)
+  --latency        Network latency ms (default 10)
+  --restarts       Restart count (default 0)
+  --probe-failures Probe failure count (default 0)
+  --cpu-pressure   Node CPU pressure 0/1 (default 0)
+  --mem-pressure   Node memory pressure 0/1 (default 0)
+  --age            Pod age in minutes (default 60)
+  --service        Service/pod name (optional label)
+  --platform       docker | kubernetes (default docker)
+  --no-ai          Skip LLM analysis
+```
+
+---
+
+## Environment variables
+
+| Variable | Default | Description |
+|---|---|---|
+| `CHECKDK_API_URL` | `http://localhost:8000` | Backend API base URL |
+
+The CLI auto-loads `~/.checkdk/.env` and `./.env` on startup.
+
+---
+
+## Development
+
+```bash
+# Install with dev extras
+pip install -e ".[dev]"
+
+# Run tests
+pytest
+
+# Lint
+ruff check .
+```

checkdk_cli-0.2.0/checkdk_cli.egg-info/SOURCES.txt

@@ -0,0 +1,21 @@
+README.md
+pyproject.toml
+checkdk_cli.egg-info/PKG-INFO
+checkdk_cli.egg-info/SOURCES.txt
+checkdk_cli.egg-info/dependency_links.txt
+checkdk_cli.egg-info/entry_points.txt
+checkdk_cli.egg-info/requires.txt
+checkdk_cli.egg-info/top_level.txt
+checkdkcli/__init__.py
+checkdkcli/client.py
+checkdkcli/display.py
+checkdkcli/main.py
+checkdkcli/commands/__init__.py
+checkdkcli/commands/auth.py
+checkdkcli/commands/chaos.py
+checkdkcli/commands/docker.py
+checkdkcli/commands/init.py
+checkdkcli/commands/kubectl.py
+checkdkcli/commands/monitor.py
+checkdkcli/commands/playground.py
+checkdkcli/commands/predict.py

checkdk_cli-0.2.0/checkdk_cli.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

checkdk_cli-0.2.0/checkdk_cli.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+checkdk = checkdkcli.main:main

checkdk_cli-0.2.0/checkdk_cli.egg-info/requires.txt

@@ -0,0 +1,11 @@
+click>=8.1.0
+rich>=13.0.0
+requests>=2.31.0
+pyyaml>=6.0
+python-dotenv>=1.0.0
+websocket-client>=1.6.0
+
+[dev]
+pytest>=7.4.0
+black>=23.0.0
+ruff>=0.1.0

checkdk_cli-0.2.0/checkdk_cli.egg-info/top_level.txt

@@ -0,0 +1 @@
+checkdkcli

checkdk_cli-0.2.0/checkdkcli/__init__.py

@@ -0,0 +1,3 @@
+"""checkDK CLI package."""
+
+__version__ = "0.2.0"

checkdk_cli-0.2.0/checkdkcli/client.py

@@ -0,0 +1,160 @@
+"""HTTP client for the checkDK backend API.
+
+All communication between the CLI and the backend happens here.
+The base URL is read from $CHECKDK_API_URL (required).
+"""
+
+from __future__ import annotations
+
+import os
+from pathlib import Path
+from typing import Optional
+
+import requests
+
+_DEFAULT_TIMEOUT = 60  # seconds
+
+
+def get_api_url() -> str:
+    """Return the backend API base URL.
+
+    Priority:
+        1. $CHECKDK_API_URL environment variable
+        2. ~/.checkdk/.env  (loaded by python-dotenv at startup)
+        3. Fallback to http://localhost:8000 so local dev works without config
+    """
+    url = os.getenv("CHECKDK_API_URL", "http://localhost:8000").strip().rstrip("/")
+    return url
+
+
+def get_stored_token() -> Optional[str]:
+    """Read a stored JWT from ~/.checkdk/.env, if present."""
+    token = os.getenv("CHECKDK_TOKEN")
+    if token:
+        return token.strip()
+    env_file = Path.home() / ".checkdk" / ".env"
+    if env_file.exists():
+        for line in env_file.read_text().splitlines():
+            if line.startswith("CHECKDK_TOKEN="):
+                return line.split("=", 1)[1].strip().strip('"').strip("'")
+    return None
+
+
+def _auth_headers() -> dict:
+    """Return Authorization header dict if a token is stored, else empty dict."""
+    token = get_stored_token()
+    return {"Authorization": f"Bearer {token}"} if token else {}
+
+
+def _post(path: str, payload: dict, timeout: int = _DEFAULT_TIMEOUT) -> dict:
+    """POST to the API and return the parsed JSON body.
+
+    Raises requests.HTTPError on non-2xx responses.
+    """
+    url = f"{get_api_url()}{path}"
+    resp = requests.post(url, json=payload, headers=_auth_headers(), timeout=timeout)
+    resp.raise_for_status()
+    return resp.json()
+
+
+def _get(path: str, timeout: int = _DEFAULT_TIMEOUT) -> dict:
+    """GET from the API and return the parsed JSON body."""
+    url = f"{get_api_url()}{path}"
+    resp = requests.get(url, headers=_auth_headers(), timeout=timeout)
+    resp.raise_for_status()
+    return resp.json()
+
+
+def health_check() -> bool:
+    """Return True if the backend is reachable and healthy."""
+    try:
+        resp = requests.get(f"{get_api_url()}/health", timeout=5)
+        return resp.status_code == 200
+    except Exception:
+        return False
+
+
+# ── Analysis helpers ──────────────────────────────────────────────────────────
+
+def analyze_docker_compose(content: str, filename: Optional[str] = None,
+                           timeout: int = _DEFAULT_TIMEOUT) -> dict:
+    """POST docker-compose YAML content and return the analysis result dict."""
+    payload: dict = {"content": content}
+    if filename:
+        payload["filename"] = filename
+    return _post("/analyze/docker-compose", payload, timeout=timeout)
+
+
+def analyze_kubernetes(content: str, filename: Optional[str] = None,
+                       timeout: int = _DEFAULT_TIMEOUT) -> dict:
+    """POST Kubernetes manifest YAML content and return the analysis result dict."""
+    payload: dict = {"content": content}
+    if filename:
+        payload["filename"] = filename
+    return _post("/analyze/kubernetes", payload, timeout=timeout)
+
+
+def analyze_playground(content: str, filename: Optional[str] = None,
+                       timeout: int = _DEFAULT_TIMEOUT) -> dict:
+    """POST any config to the hybrid AI+rules playground endpoint."""
+    payload: dict = {"content": content}
+    if filename:
+        payload["filename"] = filename
+    return _post("/analyze/playground", payload, timeout=timeout)
+
+
+# ── Auth helpers ──────────────────────────────────────────────────────────────
+
+def validate_token(token: str) -> dict:
+    """POST a JWT to /auth/cli-token to validate it; returns user info dict."""
+    url = f"{get_api_url()}/auth/cli-token"
+    resp = requests.post(
+        url, json={"token": token},
+        headers={"Authorization": f"Bearer {token}"},
+        timeout=10,
+    )
+    resp.raise_for_status()
+    return resp.json()
+
+
+def get_current_user() -> dict:
+    """GET /auth/me — returns current user info (requires stored token)."""
+    return _get("/auth/me", timeout=10)
+
+
+# ── Pod health prediction ─────────────────────────────────────────────────────
+
+def predict_pod_health(
+    cpu: float,
+    memory: float,
+    disk: float = 50.0,
+    latency: float = 10.0,
+    restarts: int = 0,
+    probe_failures: int = 0,
+    cpu_pressure: int = 0,
+    mem_pressure: int = 0,
+    age: int = 60,
+    service: Optional[str] = None,
+    platform: str = "docker",
+    no_ai: bool = False,
+    timeout: int = 30,
+) -> dict:
+    """POST pod metrics to the /predict endpoint and return the result dict."""
+    return _post(
+        "/predict",
+        {
+            "cpu": cpu,
+            "memory": memory,
+            "disk": disk,
+            "latency": latency,
+            "restarts": restarts,
+            "probe_failures": probe_failures,
+            "cpu_pressure": cpu_pressure,
+            "mem_pressure": mem_pressure,
+            "age": age,
+            "service": service,
+            "platform": platform,
+            "no_ai": no_ai,
+        },
+        timeout=timeout,
+    )

checkdk_cli-0.2.0/checkdkcli/commands/__init__.py

@@ -0,0 +1 @@
+"""CLI commands package."""

checkdk_cli-0.2.0/checkdkcli/commands/auth.py

@@ -0,0 +1,110 @@
+"""checkdk auth commands - login, logout, whoami."""
+
+from __future__ import annotations
+
+import os
+import sys
+import webbrowser
+from pathlib import Path
+
+import click
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+
+from ..client import get_api_url, get_current_user, validate_token
+
+_console = Console()
+_ENV_FILE = Path.home() / ".checkdk" / ".env"
+
+
+def _save_token(token: str) -> None:
+    _ENV_FILE.parent.mkdir(parents=True, exist_ok=True)
+    lines = _ENV_FILE.read_text().splitlines() if _ENV_FILE.exists() else []
+    lines = [l for l in lines if not l.startswith("CHECKDK_TOKEN=")]
+    lines.append(f"CHECKDK_TOKEN={token}")
+    _ENV_FILE.write_text("\n".join(lines) + "\n")
+
+
+def _remove_token() -> None:
+    if not _ENV_FILE.exists():
+        return
+    lines = [l for l in _ENV_FILE.read_text().splitlines()
+             if not l.startswith("CHECKDK_TOKEN=")]
+    _ENV_FILE.write_text("\n".join(lines) + "\n")
+
+
+@click.group("auth")
+def auth_cmd() -> None:
+    """Authentication - log in, log out, or check who you are."""
+
+
+@auth_cmd.command("login")
+def login_cmd() -> None:
+    """Log in to checkDK via GitHub or Google OAuth.
+
+    Opens the sign-in page in your browser, then prompts you to paste
+    the JWT token shown after a successful login.
+    """
+    login_url = f"{get_api_url().replace('/api', '')}/login" if "/api" in get_api_url() else "https://checkdk.app/login"
+    _console.print(f"\n[bold]Opening sign-in page:[/] [cyan]{login_url}[/]")
+    _console.print("[dim]Sign in with GitHub or Google, then copy the token from the page.[/]\n")
+
+    try:
+        webbrowser.open(login_url)
+    except Exception:
+        _console.print("[yellow]Could not open browser automatically.[/]")
+        _console.print(f"Please visit: [cyan]{login_url}[/]\n")
+
+    token = _console.input("[bold]Paste your JWT token here:[/] ").strip()
+    if not token:
+        _console.print("[red]No token provided. Login cancelled.[/]")
+        sys.exit(1)
+
+    try:
+        user = validate_token(token)
+    except Exception as exc:
+        _console.print(f"[bold red]Token validation failed:[/] {exc}")
+        _console.print("[yellow]Make sure CHECKDK_API_URL points to a running backend.[/]")
+        sys.exit(1)
+
+    _save_token(token)
+    os.environ["CHECKDK_TOKEN"] = token
+
+    _console.print(Panel(
+        f"[bold green]Logged in successfully![/]\n\n"
+        f"  Name:     {user.get('name', '?')}\n"
+        f"  Email:    {user.get('email', '?')}\n"
+        f"  Provider: {user.get('provider', '?')}\n\n"
+        f"[dim]Token saved to {_ENV_FILE}[/]",
+        border_style="green",
+    ))
+
+
+@auth_cmd.command("logout")
+def logout_cmd() -> None:
+    """Remove the stored JWT token."""
+    _remove_token()
+    if "CHECKDK_TOKEN" in os.environ:
+        del os.environ["CHECKDK_TOKEN"]
+    _console.print("[bold green]Logged out.[/] Token removed from local config.")
+
+
+@auth_cmd.command("whoami")
+def whoami_cmd() -> None:
+    """Show the currently logged-in user."""
+    try:
+        user = get_current_user()
+    except Exception as exc:
+        _console.print(f"[bold red]Not logged in or API unreachable:[/] {exc}")
+        _console.print("[dim]Run [bold]checkdk auth login[/] first.[/]")
+        sys.exit(1)
+
+    t = Table.grid(padding=(0, 2))
+    t.add_column(style="bold cyan")
+    t.add_column()
+    t.add_row("Name:",     user.get("name", "?"))
+    t.add_row("Email:",    user.get("email", "?"))
+    t.add_row("Provider:", user.get("provider", "?"))
+    t.add_row("User ID:",  user.get("userId", "?"))
+    _console.print(Panel(t, title="Current User", border_style="cyan"))