botable-dev 0.3.1__py3-none-any.whl

botable/__init__.py

@@ -0,0 +1,3 @@
+"""Botable CLI — Generate Telegram bots with AI."""
+
+__version__ = "0.3.1"

botable/__main__.py

@@ -0,0 +1,5 @@
+"""Allow running as `python -m botable`."""
+
+from botable.cli import app
+
+app()

botable/api_client.py

@@ -0,0 +1,203 @@
+"""
+HTTP client for Botable Generator API.
+
+Handles authentication, project CRUD, and SSE event streaming.
+Implemented in Stage 024.
+"""
+from __future__ import annotations
+
+from typing import Any, AsyncIterator, Dict, Optional
+
+import httpx
+
+from botable.config import get_api_url, get_auth_token
+
+
+class BotableClient:
+    """HTTP client for the Botable Generator API."""
+
+    def __init__(
+        self,
+        base_url: Optional[str] = None,
+        api_key: Optional[str] = None,
+    ):
+        self.base_url = (base_url or get_api_url()).rstrip("/")
+        self.api_key = api_key or get_auth_token()
+        self._client: Optional[httpx.AsyncClient] = None
+
+    def _headers(self) -> Dict[str, str]:
+        headers = {"Content-Type": "application/json"}
+        if self.api_key:
+            headers["Authorization"] = f"Bearer {self.api_key}"
+        return headers
+
+    async def _get_client(self) -> httpx.AsyncClient:
+        if self._client is None:
+            # Retry on transient network errors (TCP reset, DNS timeout)
+            transport = httpx.AsyncHTTPTransport(retries=3)
+            self._client = httpx.AsyncClient(
+                base_url=self.base_url,
+                headers=self._headers(),
+                timeout=60.0,
+                transport=transport,
+            )
+        return self._client
+
+    async def close(self) -> None:
+        if self._client:
+            await self._client.aclose()
+            self._client = None
+
+    # ── Project operations (Stage 024) ───────────────────────────────────
+
+    async def create_project(self, name: str, description: str = "") -> Dict[str, Any]:
+        """Create a new project on the generator."""
+        client = await self._get_client()
+        resp = await client.post("/api/projects", json={
+            "name": name,
+            "description": description,
+        })
+        resp.raise_for_status()
+        return resp.json()
+
+    async def send_message(self, project_id: str, text: str) -> Dict[str, Any]:
+        """Send a message to a project's FSM."""
+        client = await self._get_client()
+        resp = await client.post(f"/api/projects/{project_id}/message", json={
+            "text": text,
+        })
+        resp.raise_for_status()
+        return resp.json()
+
+    async def get_project(self, project_id: str) -> Dict[str, Any]:
+        """Get project status."""
+        client = await self._get_client()
+        resp = await client.get(f"/api/projects/{project_id}")
+        resp.raise_for_status()
+        return resp.json()
+
+    # ── SSE streaming (Stage 024) ────────────────────────────────────────
+
+    async def stream_events(self, project_id: str) -> AsyncIterator[Dict[str, Any]]:
+        """Stream SSE events for a project."""
+        import json
+
+        client = await self._get_client()
+        async with client.stream(
+            "GET",
+            f"/api/v2/projects/{project_id}/events",
+            headers={"Accept": "text/event-stream"},
+        ) as resp:
+            async for line in resp.aiter_lines():
+                if line.startswith("data: "):
+                    try:
+                        yield json.loads(line[6:])
+                    except json.JSONDecodeError:
+                        continue
+
+    # ── Deploy (Stage 038) ────────────────────────────────────────────
+
+    async def deploy_project(self, tarball: bytes) -> Dict[str, Any]:
+        """Upload a tarball and trigger deployment.
+
+        Uses multipart/form-data (not JSON) so we build a one-shot client.
+        """
+        headers: Dict[str, str] = {}
+        if self.api_key:
+            headers["Authorization"] = f"Bearer {self.api_key}"
+
+        async with httpx.AsyncClient(
+            base_url=self.base_url, timeout=120.0,
+        ) as client:
+            files = {"file": ("project.tar.gz", tarball, "application/gzip")}
+            resp = await client.post(
+                "/api/v1/cli/deploy", files=files, headers=headers,
+            )
+            resp.raise_for_status()
+            return resp.json()
+
+    # ── Bots management (Stage 039) ───────────────────────────────────
+
+    async def list_bots(self) -> Dict[str, Any]:
+        """List all user's bots/projects."""
+        client = await self._get_client()
+        resp = await client.get("/api/v1/cli/bots")
+        resp.raise_for_status()
+        return resp.json()
+
+    async def stop_bot(self, project_id: str) -> Dict[str, Any]:
+        """Stop a running bot."""
+        client = await self._get_client()
+        resp = await client.post(f"/api/v1/cli/bots/{project_id}/stop")
+        resp.raise_for_status()
+        return resp.json()
+
+    async def get_bot_status(self, project_id: str) -> Dict[str, Any]:
+        """Get bot logs/status."""
+        client = await self._get_client()
+        resp = await client.get(f"/api/v1/cli/bots/{project_id}/logs")
+        resp.raise_for_status()
+        return resp.json()
+
+    async def start_bot(self, project_id: str) -> Dict[str, Any]:
+        """Start a stopped bot."""
+        client = await self._get_client()
+        resp = await client.post(f"/api/v1/cli/bots/{project_id}/start")
+        resp.raise_for_status()
+        return resp.json()
+
+    async def delete_bot(self, project_id: str) -> Dict[str, Any]:
+        """Delete a bot/project."""
+        client = await self._get_client()
+        resp = await client.delete(f"/api/v1/cli/bots/{project_id}")
+        resp.raise_for_status()
+        return resp.json()
+
+    async def edit_bot(self, project_id: str, instruction: str) -> Dict[str, Any]:
+        """Send an edit instruction to a bot."""
+        client = await self._get_client()
+        resp = await client.post(f"/api/v1/cli/bots/{project_id}/edit", json={
+            "instruction": instruction,
+        })
+        resp.raise_for_status()
+        return resp.json()
+
+    # ── API Keys (CLI expansion) ─────────────────────────────────────────
+
+    async def list_api_keys(self) -> Dict[str, Any]:
+        """List all API keys."""
+        client = await self._get_client()
+        resp = await client.get("/api/v1/api-keys/")
+        resp.raise_for_status()
+        return resp.json()
+
+    async def create_api_key(self, name: str) -> Dict[str, Any]:
+        """Create a new API key."""
+        client = await self._get_client()
+        resp = await client.post("/api/v1/api-keys/", json={"name": name})
+        resp.raise_for_status()
+        return resp.json()
+
+    async def revoke_api_key(self, key_id: str) -> Dict[str, Any]:
+        """Revoke an API key."""
+        client = await self._get_client()
+        resp = await client.delete(f"/api/v1/api-keys/{key_id}")
+        resp.raise_for_status()
+        return resp.json()
+
+    # ── Credits & Templates ──────────────────────────────────────────────
+
+    async def get_credits(self) -> Dict[str, Any]:
+        """Get current credit balance."""
+        client = await self._get_client()
+        resp = await client.get("/api/v1/credits/")
+        resp.raise_for_status()
+        return resp.json()
+
+    async def list_templates(self) -> Dict[str, Any]:
+        """List available bot templates."""
+        client = await self._get_client()
+        resp = await client.get("/api/v1/templates")
+        resp.raise_for_status()
+        return resp.json()
+