stackbox 0.1.0__py3-none-any.whl

stackbox/catalog.py

@@ -0,0 +1,156 @@
+"""What the gateway knows about each provider out of the box.
+
+Defaults (list prices, plan limits, sizes) let a backend be declared with just its kind,
+or auto-detected from credentials in the environment. Everything here can be overridden
+per backend in the config file. Prices are list prices checked 2026-11; see the
+``pricing`` links.
+"""
+
+from __future__ import annotations
+
+import importlib.util
+import os
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Any
+
+
+@dataclass(frozen=True)
+class Provider:
+    kind: str
+    label: str
+    credentials: tuple[tuple[str, ...], ...]  # any one group of env vars, all set
+    extra: str | None  # `uv sync --extra <extra>` / pip install 'stackbox[<extra>]'
+    module: str | None  # import that proves the extra is installed
+    defaults: dict[str, Any] = field(default_factory=dict)
+    signup: str = ""
+    pricing: str = ""
+    notes: str = ""
+
+    def detected(self) -> bool:
+        if self.kind == "modal" and (Path.home() / ".modal.toml").is_file():
+            return True
+        return any(all(os.environ.get(v) for v in group) for group in self.credentials)
+
+    def missing_credentials(self) -> str:
+        return " + ".join(self.credentials[0]) if self.credentials else ""
+
+    def installed(self) -> bool:
+        return self.module is None or importlib.util.find_spec(self.module) is not None
+
+    def price_1vcpu_2gib(self) -> float:
+        d = self.defaults
+        return d.get("cost_per_vcpu_hour", 0.0) + 2 * d.get("cost_per_gib_hour", 0.0)
+
+
+PROVIDERS: dict[str, Provider] = {
+    p.kind: p
+    for p in (
+        Provider(
+            kind="e2b",
+            label="E2B",
+            credentials=(("E2B_API_KEY",),),
+            extra="e2b",
+            module="e2b",
+            defaults={
+                "max_active": 20,
+                "max_timeout_s": 3600,
+                "cost_per_vcpu_hour": 0.0504,
+                "cost_per_gib_hour": 0.0162,
+                "startup_prior_s": 1.5,
+            },
+            signup="https://e2b.dev/dashboard?tab=keys",
+            pricing="https://e2b.dev/docs/pricing",
+            notes="Firecracker microVMs; images become templates (built once, ~45 s)",
+        ),
+        Provider(
+            kind="modal",
+            label="Modal",
+            credentials=(("MODAL_TOKEN_ID", "MODAL_TOKEN_SECRET"),),
+            extra="modal",
+            module="modal",
+            defaults={
+                "max_active": 20,
+                "max_timeout_s": 86400,
+                "cost_per_vcpu_hour": 0.0710,
+                "cost_per_gib_hour": 0.0242,
+                "startup_prior_s": 1.0,
+                "cold_start_prior_s": 15.0,
+            },
+            signup="https://modal.com/settings/tokens",
+            pricing="https://modal.com/pricing",
+            notes="gVisor; any registry image; or `modal token set` (~/.modal.toml)",
+        ),
+        Provider(
+            kind="daytona",
+            label="Daytona",
+            credentials=(("DAYTONA_API_KEY",),),
+            extra="daytona",
+            module="daytona",
+            defaults={
+                "max_active": 20,
+                "max_timeout_s": 3600,
+                "cost_per_vcpu_hour": 0.0504,
+                "cost_per_gib_hour": 0.0162,
+                "startup_prior_s": 3.0,
+                "cold_start_prior_s": 20.0,
+            },
+            signup="https://app.daytona.io/dashboard/keys",
+            pricing="https://www.daytona.io/pricing",
+            notes="any registry image; account memory quota is discovered automatically",
+        ),
+        Provider(
+            kind="vercel",
+            label="Vercel Sandbox",
+            credentials=(
+                ("VERCEL_TOKEN", "VERCEL_TEAM_ID", "VERCEL_PROJECT_ID"),
+                ("VERCEL_OIDC_TOKEN",),
+            ),
+            extra="vercel",
+            module="vercel.sandbox",
+            defaults={
+                "max_active": 10,
+                "max_timeout_s": 2700,
+                "cost_per_vcpu_hour": 0.128,
+                "cost_per_gib_hour": 0.0212,
+                "cpu_billed_fraction": 0.5,
+                "startup_prior_s": 1.0,
+            },
+            signup="https://vercel.com/account/tokens",
+            pricing="https://vercel.com/docs/sandbox/pricing",
+            notes="bills active CPU only; images are pushed to Vercel's registry on prepare (needs Docker)",
+        ),
+        Provider(
+            kind="cloudflare",
+            label="Cloudflare Containers",
+            credentials=(
+                ("CLOUDFLARE_BRIDGE_URL", "CLOUDFLARE_BRIDGE_TOKEN"),
+                ("CLOUDFLARE_BRIDGE_URL", "BRIDGE_TOKEN"),
+            ),
+            extra=None,
+            module=None,
+            defaults={
+                "url": "env:CLOUDFLARE_BRIDGE_URL",
+                "api_key": "env:CLOUDFLARE_BRIDGE_TOKEN|BRIDGE_TOKEN",
+                "max_active": 20,
+                "max_timeout_s": 86400,
+                "cost_per_vcpu_hour": 0.072,
+                "cost_per_gib_hour": 0.009,
+                "cpu_billed_fraction": 0.5,
+                "startup_prior_s": 1.5,
+            },
+            signup="https://dash.cloudflare.com/profile/api-tokens",
+            pricing="https://developers.cloudflare.com/containers/pricing/",
+            notes="via a bridge Worker you deploy once (bridges/cloudflare); bills active CPU only",
+        ),
+    )
+}
+
+
+def defaults_for(kind: str) -> dict[str, Any]:
+    p = PROVIDERS.get(kind)
+    return dict(p.defaults) if p else {}
+
+
+def detected() -> list[Provider]:
+    return [p for p in PROVIDERS.values() if p.detected()]