videosdkagent-cli 0.0.1__py2.py3-none-any.whl

videosdk_cli/utils/apis/videosdk_auth_api_client.py

@@ -0,0 +1,58 @@
+import os
+import aiohttp
+from videosdk_cli.utils.config_manager import get_config_value
+from videosdk_cli.utils.apis.error import AuthenticationError, APIRequestError
+
+class VideoSDKAsyncClient:
+    def __init__(self):
+        self.auth_token = get_config_value("VIDEOSDK_AUTH_TOKEN") or ""
+        self.base_url = os.environ.get(
+            "API_BASE_URL", "https://api.videosdk.live"
+        )
+
+        self.headers = {
+            "Authorization": self.auth_token,
+            "Content-Type": "application/json",
+        }
+
+    async def _request(
+        self,
+        method: str,
+        endpoint: str,
+        *,
+        session: aiohttp.ClientSession | None = None,
+        **kwargs,
+    ):
+        """
+        Common aiohttp request handler
+        """
+        url = endpoint if endpoint.startswith("http") else f"{self.base_url}{endpoint}"
+        close_session = False
+
+        if session is None:
+            session = aiohttp.ClientSession(headers=self.headers)
+            close_session = True
+
+        try:
+            async with session.request(method, url, **kwargs) as resp:
+                # 🔐 Common auth handling
+                if resp.status in (401, 403):
+                    raise AuthenticationError(
+                        "Authentication failed. Please check your VIDEOSDK_AUTH_TOKEN."
+                    )
+
+                if resp.status >= 400:
+                    text = await resp.text()
+                    raise APIRequestError(
+                        f"Request failed [{resp.status}]: {text}"
+                    )
+
+                # auto-detect JSON vs text
+                content_type = resp.headers.get("Content-Type", "")
+                if "application/json" in content_type:
+                    return await resp.json()
+                return await resp.text()
+
+        finally:
+            if close_session:
+                await session.close()

videosdk_cli/utils/auth_api_client.py

@@ -0,0 +1,49 @@
+import aiohttp
+
+
+class AuthAPIClient:
+    """
+    Client for VideoSDK CLI authentication flow.
+    Responsible ONLY for talking to your backend auth APIs.
+    """
+
+    def __init__(self, base_url: str = "http://localhost:8000"):
+        self.base_url = base_url.rstrip("/")
+
+    async def start_auth(self) -> dict:
+        """
+        Starts the auth flow.
+
+        Expected response:
+        {
+            "verify_url": "...",
+            "request_id": "...",
+            "expires_in": 600
+        }
+        """
+        async with aiohttp.ClientSession() as session:
+            async with session.post(f"{self.base_url}/v2/cli/request/start") as resp:
+                resp.raise_for_status()
+                return await resp.json()
+
+    async def check_status(self, requestId: str) -> dict:
+        """
+        Polls auth status.
+
+        Expected approved response:
+        {
+            "status": "approved",
+            "project_token": "VIDEOSDK_AUTH_TOKEN"
+        }
+
+        Other responses:
+        { "status": "pending" }
+        { "status": "expired" }
+        """
+        async with aiohttp.ClientSession() as session:
+            async with session.get(
+                f"{self.base_url}/v2/cli/request/status",
+                params={"requestId": requestId},
+            ) as resp:
+                resp.raise_for_status()
+                return await resp.json()

videosdk_cli/utils/config_manager.py

@@ -0,0 +1,66 @@
+import json
+import click
+from pathlib import Path
+
+CONFIG_DIR = Path(click.get_app_dir("videosdk_config"))
+CONFIG_FILE = CONFIG_DIR / "config.json"
+
+
+
+def load_config():
+    """
+    Load the configuration from the standard user config directory.
+    For Windows : C:/Users/User_name/AppData/Roaming/videosdk_config/config.json
+    For mac/linux :  ~/.config/videosdk_config/config.json
+
+    Returns:
+        dict or None: Configuration dictionary, or None if file is empty or missing.
+    """
+    if not CONFIG_FILE.is_file():
+        return None
+
+    try:
+        with open(CONFIG_FILE, "r") as f:
+            content = f.read().strip()
+            if not content:
+                return None
+            return json.loads(content)
+    
+    except json.JSONDecodeError:
+        print(f"[WARNING] Config file at {CONFIG_FILE} is invalid. A new one will be created on next save.")
+        return None
+    
+    except Exception as e:
+        print(f"[ERROR] An unexpected error occurred while loading the config file: {e}")
+        return None
+
+
+
+
+def save_config(config: dict):
+    """Save the configuration to the standard user config directory."""
+    try:
+        CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+        with open(CONFIG_FILE, "w") as f:
+            json.dump(config, f, indent=2)
+    
+    except Exception as e:
+        print(f"[ERROR] Unable to save config to {CONFIG_FILE}: {e}")
+
+
+
+def get_config_value(key: str, default=None):
+    """A helper function to safely get a single value from the config."""
+    config = load_config()
+    
+    if not config:
+        return default
+    return config.get(key, default)
+
+
+
+def set_config_value(key: str, value):
+    """A helper function to safely set a single value in the config."""
+    config = load_config() or {}
+    config[key] = value
+    save_config(config)

videosdk_cli/utils/exceptions.py

@@ -0,0 +1,3 @@
+class AuthenticationError(Exception):
+    """Custom exception for API authentication failures (e.g., 401 Unauthorized)."""
+    pass

videosdk_cli/utils/manager/agent_manager.py

@@ -0,0 +1,381 @@
+from genericpath import exists
+import click
+from typing import Optional
+from pathlib import Path
+from dotenv import dotenv_values
+import os
+import re
+from videosdk_cli.utils.apis.deployment_client import DeploymentClient
+from videosdk_cli.utils.videosdk_yaml_helper import update_agent_config
+from videosdk_cli.utils.videosdk_yaml_helper import AgentConfig
+from rich.console import Console
+from InquirerPy import inquirer
+from rich.table import Table
+from videosdk_cli.utils.ui.kv_blocks import print_kv_blocks
+console = Console()
+
+async def init_agent(app_dir: Path, name: Optional[str] = None):
+    client = DeploymentClient()
+
+    # Call backend
+    response = await client.agent_init(name)
+
+    agent_id = response.get("agentId")
+    agent_name = response.get("name") or "agent-test"
+
+    if not agent_id or not agent_name:
+        raise RuntimeError("Invalid response from agent_init API")
+
+    # Build AgentConfig dataclass
+    agent_config = AgentConfig(
+        id=agent_id,
+        name=agent_name,
+    )
+
+    # Update videosdk.yaml
+    update_agent_config(
+        app_dir=app_dir,
+        agent_config=agent_config,
+    )
+
+    return response
+
+
+async def list_deployment(deployment_id: Optional[str] = None,head: Optional[int] = None,tail: Optional[int] = None):
+    client = DeploymentClient()
+    try:
+        response = await client.agent_list(deployment_id,head,tail)
+        deployments = response.get("deployments", [])
+
+        if not deployments:
+            console.print("[bold red]No deployments found.[/bold red]")
+            return
+
+        table = Table(title="Deployments", show_lines=True)
+
+        table.add_column("Name", style="cyan", no_wrap=True)
+        table.add_column("Region", style="green")
+        table.add_column("Profile", style="magenta")
+        table.add_column("Deployment ID", style="yellow",no_wrap=True)
+        table.add_column("Created At", style="white")
+
+
+        for d in deployments:
+            created = d["createdAt"].replace("T", " ").replace("Z", "")
+
+            table.add_row(
+                d["name"],
+                d["region"],
+                d["profile"],
+                d["deploymentId"],
+                created
+            )
+        
+        console.print(table)
+        return response
+    except Exception as e:
+        raise e
+
+async def list_agents_manager(agent_id: Optional[str] = None,head: Optional[int] = None,tail: Optional[int] = None):
+    client = DeploymentClient()
+    try:
+        print(agent_id)
+        response = await client.agent_list(agent_id,head,tail)
+        deployments = response.get("deployments", [])
+
+        if not deployments:
+            console.print("[bold red]No deployments found.[/bold red]")
+            return
+
+        table = Table(title="Deployments", show_lines=True)
+
+        table.add_column("Name", style="cyan", no_wrap=True)
+        table.add_column("Region", style="green")
+        table.add_column("Profile", style="magenta")
+        table.add_column("Deployment ID", style="yellow",no_wrap=True)
+        table.add_column("Created At", style="white")
+
+
+        for d in deployments:
+            created = d["createdAt"].replace("T", " ").replace("Z", "")
+
+            table.add_row(
+                d["name"],
+                d["region"],
+                d["profile"],
+                d["deploymentId"],
+                created
+            )
+        
+        console.print(table)
+        return response
+    except Exception as e:
+        raise e
+
+async def describe_agent_manager(agent_id:str,deployment_id:Optional[str] = None):
+    client = DeploymentClient()
+    try:
+        response = await client.agent_describe(agent_id,deployment_id)
+        data_to_print = {}
+        if response:
+            data_to_print = {
+                "Agent ID": response.get("agentId",'N/A'),
+                "Name": response.get("name",'N/A'),
+                "Region": response.get("region",'N/A'),
+                "Profile": response.get("profile",'N/A'),
+                "Deployment ID": response.get("deploymentId",'N/A'),
+                "Created At": response.get("createdAt",'N/A').replace("T", " ").replace("Z", ""),
+                "minReplica": response.get("minReplica",'N/A'),
+                "maxReplica": response.get("maxReplica",'N/A'),
+                "imageCR": response.get("image",{}).get("imageCR",'N/A'),
+                "imageUrl": response.get("image",{}).get("imageUrl",'N/A'),
+                "Active": response.get("active",'N/A'),
+            }
+
+        print_kv_blocks([data_to_print])
+        
+    except Exception as e:
+        raise e
+
+def parse_env_file(env_file: Path) -> dict:
+    if not env_file.exists():
+        raise click.ClickException("Env file does not exist")
+    
+    env_vars = dotenv_values(env_file)
+
+    if not env_vars:
+        raise click.ClickException("Env file is empty or invalid")
+    
+
+    # Convert OrderedDict -> dict
+    return dict(env_vars)
+
+async def secret_set_manager(name:str,file:Optional[str] = None):
+    client = DeploymentClient()
+    try:
+        console.print(f"Secret Name: [green]{name}[/green]")
+
+        # 🔒 Validate secret name
+        if not re.fullmatch(r"[a-z0-9_-]+", name):
+            console.print(
+                "❌ Invalid secret name.\n"
+                "Rules:\n"
+                "  • must be lowercase\n"
+                "  • no spaces\n"
+                "  • allowed characters: a-z, 0-9, -, _",
+            )
+            raise click.Abort()
+
+        secrets = {}
+
+        if file:
+            console.print(f"File: [green]{file}[/green]")
+            secrets = parse_env_file(Path(os.getcwd()) / file)
+
+        else:
+            while True:
+                key = click.prompt("Enter key", type=str)
+                if not re.fullmatch(r"^[A-Za-z0-9_-]+$", key):
+                    console.print(
+                        "❌ Invalid key.\n"
+                        "Rules:\n"
+                        "  • must be lowercase\n"
+                        "  • no spaces\n"
+                        "  • allowed characters: a-z, 0-9, -, _",
+                    )
+                    continue
+
+                value = click.prompt("Enter value", type=str, hide_input=True)
+
+                secrets[key] = value
+
+                choice = inquirer.select(
+                    message="Add another secret?",
+                    choices=["Yes", "No"],
+                    default="No",
+                ).execute()
+
+                if choice == "No":
+                    break
+
+        if not secrets:
+            console.print("No secrets provided. Aborting.")
+            return
+
+        console.print("\nSecrets to be saved:")
+        for k in secrets:
+            console.print(f"- {k}: ******")
+
+        action = inquirer.select(
+            message="Confirm action",
+            choices=["Save secrets", "Cancel"],
+            default="Save secrets",
+        ).execute()
+
+        if action == "Cancel":
+            console.print("[yellow]Cancelled. No secrets were saved.[/yellow]")
+            return
+    
+        console.print("\nSaving secrets...")
+        response = await client.secret_set(name, secrets)
+        console.print("Secrets saved successfully.")
+    except Exception as e:
+        raise e
+
+
+async def list_secret_manager():
+    client = DeploymentClient()
+    try:
+        console.print("[bold blue]Listing secrets...[/bold blue]")
+        response = await client.secret_list()
+        data = response.get("data", [])
+
+        if not data:
+            console.print("[bold red]No secrets found.[/bold red]")
+            return
+
+        table = Table(title="Secrets", show_lines=True)
+
+        table.add_column("name", style="cyan", no_wrap=True)
+        table.add_column("secretId", style="green")
+        table.add_column("type", style="magenta")
+
+        for d in data:
+            table.add_row(  
+                d.get('name','N/A'),
+                d.get('secretId','N/A'),
+                d.get('type','N/A'),
+            )
+
+        console.print(table)
+        console.print("Secrets listed successfully.")
+    except Exception as e:
+        raise e
+
+async def remove_secret_set(name: str):
+    client = DeploymentClient()
+    try:
+        console.print("[bold blue]Removing secret...[/bold blue]")
+        response = await client.secret_remove(name)
+        console.print("Secret removed successfully.")
+    except Exception as e:
+        raise e
+
+async def describe_secret_set(name: str):
+    client = DeploymentClient()
+    try:
+        console.print("[bold blue]Getting secret...[/bold blue]")
+        response = await client.secret_get(name)
+        data_to_print = {}
+        data=response.get("data", {})
+        if data:
+            data_to_print = {
+                "Name": data.get("name",'N/A'),
+                "Secret ID": data.get("secretId",'N/A'),
+                "type": data.get("type",'N/A'),
+            }
+        print_kv_blocks([data_to_print])
+            # 2️⃣ Keys table
+        keys = data.get("keys", {})
+
+        if not keys:
+            console.print("\n[dim]No keys found[/dim]")
+            return
+
+        table = Table(title="Secret Keys", show_header=True, header_style="bold")
+
+        table.add_column("Key", style="cyan", no_wrap=True)
+        table.add_column("Value", style="green")
+
+        for k in keys:
+            table.add_row(
+                k,
+                keys.get(k,'N/A')
+            )
+
+        console.print(table)
+    except Exception as e:
+        raise e
+
+async def add_secret_set(name: str):
+    client = DeploymentClient()
+    try:
+        console.print("[bold blue]Adding secret...[/bold blue]")
+        secrets = {}
+
+        while True:
+            key = click.prompt("Enter key", type=str)
+            value = click.prompt("Enter value", type=str, hide_input=True)
+
+            secrets[key] = value
+
+            choice = inquirer.select(
+                message="Add another secret?",
+                choices=["Yes", "No"],
+                default="No",
+            ).execute()
+
+            if choice == "No":
+                break
+
+        if not secrets:
+            console.print("No secrets provided. Aborting.")
+            return
+
+        console.print("\nSecrets to be saved:")
+        for k in secrets:
+            console.print(f"- {k}: ******")
+
+        action = inquirer.select(
+            message="Confirm action",
+            choices=["Save secrets", "Cancel"],
+            default="Save secrets",
+        ).execute()
+
+        if action == "Cancel":
+            console.print("Cancelled. No secrets were saved.", fg="yellow")
+            return
+        
+        response = await client.secret_add_key(name, secrets)
+    
+        console.print("Secret added successfully.")
+    except Exception as e:
+        raise e
+
+async def remove_secret_set_key(name: str):
+    client = DeploymentClient()
+    try:
+        console.print("[bold blue]Removing secret...[/bold blue]")
+        keys=[]
+        while True:
+            key = click.prompt("Enter key", type=str)
+            keys.append(key)
+
+            choice = inquirer.select(
+                message="Remove another key?",
+                choices=["Yes", "No"],
+                default="No",
+            ).execute()
+
+            if choice == "No":
+                break
+        if not keys:
+            console.print("No keys provided.")
+            return
+        response = await client.secret_remove_key(name,keys)
+        console.print("Secret removed successfully.")
+    except Exception as e:
+        raise e
+
+async def image_pull_secret_manager(name: str):
+    client = DeploymentClient()
+    try:
+        console.print("[bold blue]Setting image pull secret...[/bold blue]")
+        console.print(f"Secret Name: [green]{name}[/green]")
+        server =click.prompt("Enter server name", type=str)
+        username =click.prompt("Enter username", type=str)
+        password =click.prompt("Enter password", type=str,hide_input=True)
+        response = await client.secret_set(name,{"serverName":server,"USERNAME":username,"PASSWORD":password},type="IMAGE_PULL")
+        console.print("Image pull secret set successfully.")
+    except Exception as e:
+        raise e

videosdk_cli/utils/project_config.py

@@ -0,0 +1,87 @@
+import yaml
+import click
+import os
+from pathlib import Path
+
+def load_project_config(search_path=None):
+    """
+    Load the project configuration from videosdk.yaml or videosdk.yml.
+    
+    Args:
+        search_path (Path, optional): Directory to search in. Defaults to current working directory.
+        
+    Returns:
+        dict or None: Configuration dictionary if found, None otherwise.
+    """
+    if search_path is None:
+        search_path = Path(os.getcwd())
+    else:
+        search_path = Path(search_path)
+        
+    config_path = search_path / "videosdk.yaml"
+    if not config_path.exists():
+        config_path = search_path / "videosdk.yml"
+        
+    if not config_path.exists():
+        return None
+        
+    try:
+        with open(config_path, 'r') as f:
+            return yaml.safe_load(f)
+    except Exception as e:
+        # We perform a lenient load here. Validation errors should generally be handled 
+        # by the caller or result in None/empty dict depending on strictness required.
+        # But per requirements we usually just want the config if valid.
+        # For now, if it fails to parse, we might return None or let the error bubble?
+        # The prompt said "if not there in yaml we will throw error" only if required config is missing.
+        # So if the file exists but is invalid, that's trickier. 
+        # Let's print a warning and return None for now to be safe, 
+        # preventing a crash if the file is just empty or garbage.
+        click.echo(f"[WARNING] Failed to load config from {config_path}: {e}", err=True)
+        return None
+
+def get_config_option(cli_value, config_keys, default=None, required=True, fail_message=None):
+    """
+    Get a configuration option resolving priority: CLI > videosdk.yaml > Error (if required) > Default.
+    
+    Args:
+        cli_value: The value provided via CLI (e.g. from click.option).
+        config_keys (list[str]): List of keys to navigate the config dict (e.g. ['agent', 'image']).
+        default: Default value if not found.
+        required (bool): If True, raise UsageError if value is not found in CLI or config.
+        fail_message (str, optional): Custom error message for UsageError.
+        
+    Returns:
+        The resolved value.
+    """
+    # 1. Check CLI value
+    if cli_value is not None:
+        return cli_value
+        
+    # 2. Check Project Config
+    config = load_project_config()
+    if config:
+        value = config
+        for key in config_keys:
+            if isinstance(value, dict):
+                value = value.get(key)
+            else:
+                value = None
+                break
+        
+        if value is not None:
+            return value
+            
+    # 3. Check Default
+    if default is not None:
+        return default
+        
+    # 4. Fail if required/no default
+    if required:
+        if fail_message:
+            raise click.UsageError(fail_message)
+        else:
+            key_path = ".".join(config_keys)
+            raise click.UsageError(f"Missing required configuration: '{key_path}'. Please provide it via CLI or videosdk.yaml.")
+            
+    return None