rio-cli 0.1.0__tar.gz → 0.2.1__tar.gz

{rio_cli-0.1.0 → rio_cli-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: rio-cli
-Version: 0.1.0
+Version: 0.2.1
 Summary: Command-line client for Rio, an automated code review tool.
 Project-URL: Homepage, https://github.com/JayTheCoder77/rio
 Project-URL: Repository, https://github.com/JayTheCoder77/rio
@@ -29,6 +29,18 @@ pip install rio-cli
 
 This installs the `rio` command.
 
+## Authentication
+
+Before using `rio review`, authenticate with a Rio API key (create one on your Rio
+dashboard's API keys page):
+
+```sh
+rio auth
+```
+
+This prompts for your API key, validates it against your Rio instance, and saves it to
+`~/.config/rio/config.toml` with restricted (`0600`) file permissions.
+
 ## Usage
 
 ```sh
@@ -50,7 +62,9 @@ rio review --staged --include-untracked
 
 ## Configuration
 
-Rio reads `~/.config/rio/config.toml` for the AI engine URL and API key:
+`rio auth` manages `~/.config/rio/config.toml` for you. To point the CLI at a
+self-hosted `ai-engine` instance instead of the default (`http://localhost:8000`), edit
+the `[api] url` value in that file, or set it before running `rio auth`:
 
 ```toml
 [api]
@@ -58,9 +72,6 @@ url = "https://your-rio-instance.example.com"
 api_key = "your-api-key"
 ```
 
-If no config file is found, the CLI defaults to `http://localhost:8000` for local
-development against a self-hosted `ai-engine`.
-
 ## License
 
 [ISC](LICENSE) © 2026 JayTheCoder77

{rio_cli-0.1.0 → rio_cli-0.2.1}/README.md

@@ -12,6 +12,18 @@ pip install rio-cli
 
 This installs the `rio` command.
 
+## Authentication
+
+Before using `rio review`, authenticate with a Rio API key (create one on your Rio
+dashboard's API keys page):
+
+```sh
+rio auth
+```
+
+This prompts for your API key, validates it against your Rio instance, and saves it to
+`~/.config/rio/config.toml` with restricted (`0600`) file permissions.
+
 ## Usage
 
 ```sh
@@ -33,7 +45,9 @@ rio review --staged --include-untracked
 
 ## Configuration
 
-Rio reads `~/.config/rio/config.toml` for the AI engine URL and API key:
+`rio auth` manages `~/.config/rio/config.toml` for you. To point the CLI at a
+self-hosted `ai-engine` instance instead of the default (`http://localhost:8000`), edit
+the `[api] url` value in that file, or set it before running `rio auth`:
 
 ```toml
 [api]
@@ -41,9 +55,6 @@ url = "https://your-rio-instance.example.com"
 api_key = "your-api-key"
 ```
 
-If no config file is found, the CLI defaults to `http://localhost:8000` for local
-development against a self-hosted `ai-engine`.
-
 ## License
 
 [ISC](LICENSE) © 2026 JayTheCoder77

{rio_cli-0.1.0 → rio_cli-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rio-cli"
-version = "0.1.0"
+version = "0.2.1"
 description = "Command-line client for Rio, an automated code review tool."
 readme = "README.md"
 license = "ISC"

rio_cli-0.2.1/rio_cli/auth.py

@@ -0,0 +1,72 @@
+import os
+import stat
+
+import httpx
+import tomllib
+import typer
+
+from rio_cli.config import CONFIG_PATH, get_ai_engine_url
+from rio_cli.utils import _fail
+
+
+def _load_existing_config() -> dict:
+    if not CONFIG_PATH.exists():
+        return {}
+    try:
+        with CONFIG_PATH.open("rb") as f:
+            return tomllib.load(f)
+    except tomllib.TOMLDecodeError:
+        return {}
+
+
+def _write_config(data: dict) -> None:
+    CONFIG_PATH.parent.mkdir(parents=True, exist_ok=True)
+
+    api_section = data.get("api", {})
+    lines = ["[api]"]
+    if "url" in api_section:
+        lines.append(f'url = "{api_section["url"]}"')
+    lines.append(f'api_key = "{api_section["api_key"]}"')
+
+    CONFIG_PATH.write_text("\n".join(lines) + "\n")
+    # This file holds a real credential — don't rely on the default umask.
+    os.chmod(CONFIG_PATH, stat.S_IRUSR | stat.S_IWUSR)  # 0o600
+
+
+def _validate_api_key(api_key: str, ai_engine_url: str) -> None:
+    try:
+        response = httpx.get(
+            f"{ai_engine_url}/v1/me",
+            headers={"Authorization": f"Bearer {api_key}"},
+            timeout=15.0,
+        )
+    except httpx.ConnectError:
+        _fail(f"Error: could not connect to ai-engine at {ai_engine_url}.")
+    except httpx.TimeoutException:
+        _fail("Error: ai-engine request timed out.")
+
+    if response.status_code == 401:
+        _fail("Error: invalid API key. Create one at your Rio dashboard's API keys page.")
+    if response.status_code != 200:
+        _fail(f"Error: ai-engine returned {response.status_code}: {response.text}")
+
+
+def auth() -> None:
+    """Authenticate the CLI with a Rio API key, created via the Rio dashboard."""
+    existing = _load_existing_config()
+    ai_engine_url = existing.get("api", {}).get("url") or get_ai_engine_url()
+
+    api_key = typer.prompt(
+        "Paste your Rio API key (create one at your Rio dashboard's API keys page)",
+        hide_input=True,
+    ).strip()
+
+    if not api_key:
+        _fail("Error: no API key provided.")
+
+    _validate_api_key(api_key, ai_engine_url)
+
+    existing["api"] = {**existing.get("api", {}), "api_key": api_key, "url": ai_engine_url}
+    _write_config(existing)
+
+    typer.echo(f"Authenticated. Config saved to {CONFIG_PATH}")

{rio_cli-0.1.0 → rio_cli-0.2.1}/rio_cli/config.py

@@ -1,10 +1,14 @@
+import os
 from pathlib import Path
 
 import tomllib
 
 from rio_cli.utils import _fail
 
-DEFAULT_AI_ENGINE_URL = "http://localhost:8000"
+DEFAULT_AI_ENGINE_URL = os.environ.get(
+    "RIO_API_URL", "https://rio-ai-engine.onrender.com"
+)
+# DEFAULT_AI_ENGINE_URL = "http://localhost:8000"
 CONFIG_PATH = Path.home() / ".config" / "rio" / "config.toml"
 
 def get_ai_engine_url() -> str:

{rio_cli-0.1.0 → rio_cli-0.2.1}/rio_cli/main.py

@@ -1,5 +1,6 @@
 import typer
 
+from rio_cli.auth import auth as _auth
 from rio_cli.git import (
     _get_combined_diff,
     _get_committed_diff,
@@ -22,6 +23,11 @@ app = typer.Typer()
 def callback():
     pass
 
+@app.command()
+def auth():
+    """Authenticate the CLI with a Rio API key."""
+    _auth()
+
 def _resolve_diff(
     staged: bool,
     uncommitted: bool,

{rio_cli-0.1.0 → rio_cli-0.2.1}/rio_cli/post.py

@@ -7,11 +7,12 @@ from rio_cli.utils import _fail
 
 
 def _post_review(diff_text : str) -> list[Finding]:
-    headers : dict = {}
     api_key = get_api_key()
-    if api_key:
-        headers["Authorization"] = f"Bearer {api_key}"
-    
+    if not api_key:
+        _fail("Error: not authenticated. Run `rio auth` first.")
+
+    headers = {"Authorization": f"Bearer {api_key}"}
+
     rio_config = load_rio_config()
     ai_engine_url = get_ai_engine_url()
     try:
@@ -26,7 +27,13 @@ def _post_review(diff_text : str) -> list[Finding]:
     
     except httpx.TimeoutException:
         _fail("Error: ai-engine request timed out.")
-    
+
+    if response.status_code == 401:
+        _fail("Error: not authenticated. Run `rio auth` first.")
+    if response.status_code == 412:
+        _fail(f"Error: {response.json().get('detail', response.text)}")
+    if response.status_code == 422:
+        _fail(f"Error: {response.json().get('detail', response.text)}")
     if response.status_code != 200:
         _fail(f"Error: ai-engine returned {response.status_code}: {response.text}")