socaity-cli 0.1.1__py3-none-any.whl

socaity_cli/registry_push.py

@@ -0,0 +1,122 @@
+"""Docker push with ephemeral registry credentials.
+
+Credentials live only in a throwaway DOCKER_CONFIG directory that is deleted
+after the push, so nothing persists in the user's ~/.docker/config.json.
+"""
+from __future__ import annotations
+
+import json
+import os
+import re
+import subprocess
+import tempfile
+from typing import Optional, Tuple
+
+_DIGEST_RE = re.compile(r"digest:\s*(sha256:[0-9a-f]{64})")
+
+
+class DockerNotAvailable(RuntimeError):
+    """Docker CLI missing or the daemon is not running."""
+
+
+class CredentialsExpired(RuntimeError):
+    """Registry rejected the credentials (expired mid-push); refresh and retry."""
+
+
+class PushFailed(RuntimeError):
+    """Any other push failure; message is safe to print."""
+
+
+def _docker(args: list, *, env: Optional[dict] = None, input_bytes: Optional[bytes] = None) -> subprocess.CompletedProcess:
+    try:
+        return subprocess.run(["docker", *args], capture_output=True, env=env, input=input_bytes)
+    except FileNotFoundError as exc:
+        raise DockerNotAvailable("The docker CLI is not installed or not on PATH.") from exc
+
+
+def image_size_bytes(local_tag: str) -> int:
+    """Uncompressed size of a local image (upper bound of the pushed size)."""
+    result = _docker(["image", "inspect", local_tag, "--format", "{{.Size}}"])
+    if result.returncode != 0:
+        return 0
+    try:
+        return int(result.stdout.decode().strip())
+    except ValueError:
+        return 0
+
+
+def push_image(
+    registry: str,
+    repository: str,
+    tag: str,
+    username: str,
+    password: str,
+    local_tag: str,
+) -> Tuple[str, int]:
+    """Tag ``local_tag`` and push it as ``registry/repository:tag``.
+
+    Returns:
+        (manifest digest reported by the registry, local image size in bytes).
+
+    Raises:
+        DockerNotAvailable, CredentialsExpired, PushFailed.
+    """
+    ref = f"{registry}/{repository}:{tag}"
+    size = image_size_bytes(local_tag)
+    if size == 0:
+        raise PushFailed(f"Local image '{local_tag}' not found. Build it first (apipod build).")
+
+    result = _docker(["tag", local_tag, ref])
+    if result.returncode != 0:
+        raise PushFailed(f"docker tag failed: {result.stderr.decode()[:300]}")
+
+    with tempfile.TemporaryDirectory(prefix="socaity-push-") as docker_config:
+        env = {**os.environ, "DOCKER_CONFIG": docker_config}
+        try:
+            login = _docker(["login", registry, "-u", username, "--password-stdin"],
+                            env=env, input_bytes=password.encode())
+            if login.returncode != 0:
+                stderr = login.stderr.decode()
+                if "unauthorized" in stderr.lower() or "401" in stderr:
+                    raise CredentialsExpired("Registry login rejected; the push credentials expired.")
+                raise PushFailed(f"docker login failed: {stderr[:300]}")
+
+            push = subprocess.Popen(
+                ["docker", "push", ref],
+                stdout=subprocess.PIPE, stderr=subprocess.STDOUT, env=env, text=True,
+            )
+            digest, last_lines = None, []
+            for line in push.stdout:
+                line = line.rstrip()
+                last_lines = (last_lines + [line])[-5:]
+                match = _DIGEST_RE.search(line)
+                if match:
+                    digest = match.group(1)
+                if line and not line.startswith(("The push refers", "Preparing", "Waiting")):
+                    print(f"  {line}")
+            push.wait()
+            if push.returncode != 0:
+                tail = " | ".join(last_lines)
+                if "unauthorized" in tail.lower() or "authentication required" in tail.lower():
+                    raise CredentialsExpired("Push rejected mid-transfer; the credentials expired.")
+                raise PushFailed(f"docker push failed: {tail[:400]}")
+            if not digest:
+                digest = _digest_from_local_metadata(ref, env)
+            if not digest:
+                raise PushFailed("Push succeeded but no digest was reported by the registry.")
+            return digest, size
+        finally:
+            _docker(["rmi", ref], env=env)  # remove the extra tag, keep the local image
+
+
+def _digest_from_local_metadata(ref: str, env: dict) -> Optional[str]:
+    """Fallback: read the RepoDigest docker recorded for the pushed reference."""
+    result = _docker(["image", "inspect", ref, "--format", "{{json .RepoDigests}}"], env=env)
+    if result.returncode != 0:
+        return None
+    repo = ref.rsplit(":", 1)[0]
+    for entry in json.loads(result.stdout.decode() or "[]"):
+        name, _, digest = entry.partition("@")
+        if name == repo and digest.startswith("sha256:"):
+            return digest
+    return None

socaity_cli/requires.py

@@ -0,0 +1,47 @@
+"""Optional-dependency guard for CLI commands and library addons.
+
+Mirrors media_toolkit's ``dependency_requirements`` decorator. In CLI mode it
+exits with a friendly install hint instead of raising: CLI users should see
+one line, not a traceback. In library mode (``cli=False``) it raises
+``ImportError`` so calling code can handle the missing dependency.
+"""
+import functools
+from importlib.util import find_spec
+from typing import Optional
+
+# Cache of packages already confirmed importable (find_spec is not free).
+_available: set = set()
+
+
+def requires(package: str, pip_name: Optional[str] = None, cli: bool = True):
+    """Run the decorated function only when ``package`` is installed.
+
+    Args:
+        package: Importable module name to check (e.g. ``apipod``).
+        pip_name: PyPI name for the install hint when it differs from ``package``.
+        cli: When True (default), print an install hint and ``SystemExit(1)``.
+            When False, raise ``ImportError`` for programmatic use.
+    """
+    install_name = pip_name or package
+
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            if package not in _available:
+                try:
+                    spec = find_spec(package)
+                except (ImportError, ValueError):
+                    spec = None
+                if spec is None:
+                    message = (
+                        f"This {'command' if cli else 'feature'} needs the '{package}' package. "
+                        f"Please install it with: pip install {install_name}"
+                    )
+                    if cli:
+                        print(message)
+                        raise SystemExit(1)
+                    raise ImportError(message)
+                _available.add(package)
+            return func(*args, **kwargs)
+        return wrapper
+    return decorator

socaity_cli/sdk_commands.py

@@ -0,0 +1,28 @@
+"""Commands that need the socaity SDK package (stub generation, local model zoo).
+
+Imports stay inside the functions: the whole point of this module is that
+``socaity`` is optional for the CLI. The ``requires`` decorator guards each
+entry with a friendly install hint.
+"""
+from socaity_cli.requires import requires
+
+
+@requires("socaity")
+def run_install(service_name: str) -> None:
+    from socaity.core.socaity_service_registry import SocaityServiceRegistry
+
+    registry = SocaityServiceRegistry()
+    if service_name.lower() == "all":
+        print("Installing all available services...")
+        registry.install_all()
+    else:
+        print(f"Installing service: {service_name}...")
+        registry.install_service(service_name)
+
+
+@requires("socaity")
+def run_update() -> None:
+    from socaity.core.socaity_service_registry import SocaityServiceRegistry
+
+    print("Checking for service updates...")
+    SocaityServiceRegistry().update_package()

socaity_cli-0.1.1.dist-info/METADATA

@@ -0,0 +1,73 @@
+Metadata-Version: 2.4
+Name: socaity-cli
+Version: 0.1.1
+Summary: Socaity platform CLI: login, catalog browsing, SDK installs and APIPod deployment commands.
+Author: SocAIty
+License: MIT
+Project-URL: Repository, https://github.com/SocAIty/socaity-cli
+Project-URL: Homepage, https://www.socaity.ai
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: httpx
+Requires-Dist: socaity-schemas>=0.0.1
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+
+# socaity-cli
+
+The command line for the [socaity.ai](https://www.socaity.ai) platform: login, catalog browsing,
+SDK installs, and APIPod deployment tooling. One `socaity` command, shared by the
+[socaity SDK](https://github.com/SocAIty/socaity) and [APIPod](https://github.com/SocAIty/apipod).
+
+The package is deliberately small: `httpx` and `socaity-schemas` are the only dependencies.
+Commands that need more (stub generation, container builds) delegate to the `socaity` or
+`apipod` package and tell you what to install when it is missing.
+
+## Install
+
+```bash
+pip install socaity-cli
+```
+
+You get `socaity-cli` as part of `pip install socaity` or `pip install apipod` as well.
+
+## Commands
+
+| Command | Needs | What it does |
+|---------|-------|--------------|
+| `socaity login` / `logout` / `whoami` | - | Browser login, credentials in `~/.config/socaity/` |
+| `socaity list services\|models` | - | Browse the AI catalog |
+| `socaity search "flux schnel"` | - | Typo-tolerant search (`--collections services,models,jobs`) |
+| `socaity jobs [job_id]` | login | List/search jobs; `--q`, `--expand data,files,billing,social`, `--visibility` |
+| `socaity projects [project_id]` | login | Your projects; `--expand jobs` embeds member jobs |
+| `socaity install <service>` | `socaity` | Generate an SDK stub for a service |
+| `socaity update` | `socaity` | Sync installed services with the platform |
+| `socaity scan` | `apipod` | Scan a project and generate `apipod.json` |
+| `socaity build [file]` | `apipod` | Build the deployment container |
+| `socaity start [entrypoint]` | `apipod` | Run an APIPod service locally |
+| `socaity deploy [target]` | `apipod` | Deploy to Socaity (wraps `apipod deploy`) |
+
+Use `-C DIR` with scan/build/start/deploy to point at a service project directory
+(e.g. `socaity -C simple_test_service deploy serverless-runpod --yes`).
+
+## Configuration
+
+| Env var | Default | Purpose |
+|---------|---------|---------|
+| `SOCAITY_API_KEY` | from login | API key (wins over stored credentials) |
+| `SOCAITY_BACKEND_URL` | `https://webapi.socaity.ai` | Platform API |
+| `SOCAITY_FRONTEND_URL` | `https://socaity.ai` | Web UI for browser login |
+
+## Using it as a library
+
+`socaity` and `apipod` build on the same primitives this package exposes:
+
+```python
+from socaity_cli import SocaityBackendClient, get_api_key, requires
+```
+
+`SocaityBackendClient` covers catalog reads, search, and SDK install/update calls.
+`requires("apipod")` guards a function behind an optional package with a friendly
+install hint instead of an ImportError traceback.

socaity_cli-0.1.1.dist-info/RECORD

@@ -0,0 +1,27 @@
+socaity_cli/__init__.py,sha256=_XBpsipDc33SmZMUhAXfdTb94cFNmQEmsHbZvzytaK4,811
+socaity_cli/__main__.py,sha256=_LzdSFvyHkmOXVQip9vcACz_pkkahnjeKpq4RVQsM2A,72
+socaity_cli/apipod_commands.py,sha256=eYDEJOi6WGxXXHKyvjS7bLEZcGt66U6zQGcOlJgwElo,2105
+socaity_cli/auth.py,sha256=6bj0f6IesojmeCsrrYu_NHaiV8SZu3gEMeQFcJ9zk6U,5959
+socaity_cli/backend_client.py,sha256=QVkQ_M2PW7eWdIqtJH43ODxhV4GUapl7SY-0PG2pFjs,827
+socaity_cli/cli.py,sha256=eEUEHyO5aSQ6U1zdYZz9jfxe6xhkAH4Ux3NftgdfgZw,13223
+socaity_cli/credentials.py,sha256=BciEbzqA0kerY4FN4pUmmd_-9NdDZimcNAhNuKH2Bp8,3999
+socaity_cli/deployment.py,sha256=a2cKM_hLLx8MPnZVJx9aN6WnpO-5RSmvPeNYh9z7b2o,13171
+socaity_cli/errors.py,sha256=QdsrNsngd8Jvca33Hw_w5mbTMJxUDwS8yrPHjGRtuJU,1094
+socaity_cli/http.py,sha256=shQy_vBX3oUt0eh9BaBhA_aWcmxLDiP0DXdfys05jvE,4280
+socaity_cli/prompts.py,sha256=kWMn1CbzyDGZegY8u5xMezXQL4ZCASCx4UOZkC6HUc4,639
+socaity_cli/registry_push.py,sha256=Fwrdn07l1xrvEo8q6bKiKQ3aJkQwCn0FwqjsJXnJ1ss,4697
+socaity_cli/requires.py,sha256=UYPTx4VIAu4kTgdQ0L4iI-jqCjvwLPxc7wQ4CvxGze4,1880
+socaity_cli/sdk_commands.py,sha256=WLFbXP0aXDiTMOvXnLdyRNKU1DTY_NNBxM3LqIv_ycc,954
+socaity_cli/clients/__init__.py,sha256=iSvjkoBsbZGMmoYvYdMcYHtGNw981VefEyQFhMgOPdY,608
+socaity_cli/clients/analytics.py,sha256=ufIsrfpv6OuNK7AREASGPZdr0zIuyCFkVaJhyURZlwY,2146
+socaity_cli/clients/catalog.py,sha256=Q-JCd8eOLXfUrSsya1BG-GjiaITMxoz3h7AcCcue4FY,5476
+socaity_cli/clients/deployment.py,sha256=PEf9BCTRWReqZFdr9VryIkOE_vN7ijL1MzSSuxWtBQ4,4430
+socaity_cli/clients/files.py,sha256=gieJtK886gQv4QgFBx8y7z3sMCZX-Updm4FHk0wEINY,3070
+socaity_cli/clients/jobs.py,sha256=2vOBWTMi65Kh_hjhvMspRSxOKdC3CYSyuYTUv5A7UiQ,2679
+socaity_cli/clients/projects.py,sha256=Tdd1N2gTNCKvRvnmv4l3hkUETDWtr9m6U6d1CKfd0ng,2171
+socaity_cli/clients/sdk_install.py,sha256=2wepEZ9Xv1EF4nnRAISuuJHtvkKF4-OB--32wrbVri4,690
+socaity_cli-0.1.1.dist-info/METADATA,sha256=j0zh-JTgFjzJKv8R4W365-cLsK5kwXtd400bPBm0FS4,3113
+socaity_cli-0.1.1.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+socaity_cli-0.1.1.dist-info/entry_points.txt,sha256=mA_DYjFU2lbAQ6PVTq_onISa7yp85hP2BSynX_HgJUA,49
+socaity_cli-0.1.1.dist-info/top_level.txt,sha256=G0kM_WkhvCrj-vW_f9xPP9Ae6ujdgiePN11_j0n7aN8,12
+socaity_cli-0.1.1.dist-info/RECORD,,

socaity_cli-0.1.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (83.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

socaity_cli-0.1.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+socaity = socaity_cli.cli:main

socaity_cli-0.1.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+socaity_cli