validio-cli 10.0.0__tar.gz → 10.0.3__tar.gz

{validio_cli-10.0.0 → validio_cli-10.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: validio-cli
-Version: 10.0.0
+Version: 10.0.3
 Summary: CLI tool to interact with the Validio platform
 Author: Validio
 Author-email: Validio <support@validio.io>

{validio_cli-10.0.0 → validio_cli-10.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "validio-cli"
-version = "10.0.0"
+version = "10.0.3"
 description = "CLI tool to interact with the Validio platform"
 authors = [{ name = "Validio", email = "support@validio.io" }]
 requires-python = ">=3.10,<4"

{validio_cli-10.0.0 → validio_cli-10.0.3}/validio_cli/__init__.py

@@ -11,6 +11,7 @@ from pathlib import Path
 from types import EllipsisType
 from typing import Any, TypeVar
 
+import click
 import typer
 from camel_converter import to_snake
 from tabulate import tabulate
@@ -29,9 +30,12 @@ T = TypeVar("T", bound="OutputSettings")
 DEFAULT_NAMESPACE: str = "default"
 
 # User-Agent sent for all CLI requests. We report both the CLI version and the
-# SDK version it builds on. The CLI is only used for IaC, so this lets Surface
-# tell IaC traffic apart from direct SDK usage and see which versions are in use.
+# SDK version it builds on.
 CLI_USER_AGENT = f"validio-cli@{cli_version()} {SDK_USER_AGENT}"
+# Header reporting the CLI command (e.g. `channels get`)
+CLI_COMMAND_HEADER = "x-validio-cli-command"
+# Header flagging IaC operations (`validio code` plan/apply/import).
+IAC_HEADER = "x-validio-iac"
 
 
 class OutputFormat(str, Enum):
@@ -97,12 +101,35 @@ class AsyncTyper(typer.Typer):
         return decorator
 
 
+def _cli_headers(iac: bool = False) -> dict[str, str]:
+    """Build the CLI headers for an outgoing request.
+
+    :param iac: Whether the request originates from an IaC operation.
+    """
+    headers: dict[str, str] = {}
+
+    ctx = click.get_current_context(silent=True)
+    if ctx is not None:
+        # `command_path` is e.g. "validio channels get",
+        # drop the program name to get the command itself.
+        command = " ".join(ctx.command_path.split()[1:])
+        if command:
+            headers[CLI_COMMAND_HEADER] = command
+
+    if iac:
+        headers[IAC_HEADER] = "true"
+
+    return headers
+
+
 def get_client(
     config_dir: str = ConfigDir,
     user_agent: str = CLI_USER_AGENT,
+    iac: bool = False,
 ) -> tuple[APIClient, ValidioConfig]:
     cfg = Config(Path(config_dir)).read()
-    return APIClient(config=cfg, user_agent=user_agent), cfg
+    headers = _cli_headers(iac=iac)
+    return APIClient(config=cfg, user_agent=user_agent, headers=headers), cfg
 
 
 @dataclass

{validio_cli-10.0.0 → validio_cli-10.0.3}/validio_cli/bin/entities/code.py

@@ -188,7 +188,7 @@ async def plan(
     progress: bool = show_progress_option,
 ) -> None:
     dir_path = directory_or_default(directory)
-    client, _ = get_client(config_dir)
+    client, _ = get_client(config_dir, iac=True)
 
     schema_reinference = create_source_schema_reinference(
         update_schema, update_all_schemas
@@ -258,7 +258,7 @@ async def apply(
     ),
 ) -> None:
     dir_path = directory_or_default(directory)
-    client, _ = get_client(config_dir)
+    client, _ = get_client(config_dir, iac=True)
 
     schema_reinference = create_source_schema_reinference(
         update_schema, update_all_schemas
@@ -374,7 +374,7 @@ async def _import(
     )
 
     dir_path = directory_or_default(directory)
-    client, _ = get_client(config_dir)
+    client, _ = get_client(config_dir, iac=True)
     namespace = get_namespace(dir_path)
 
     if import_namespace: