agentpool 2.1.9__py3-none-any.whl → 2.5.0__py3-none-any.whl

agentpool_commands/text_sharing/opencode.py

@@ -0,0 +1,462 @@
+"""OpenCode text sharing provider.
+
+This implementation "fakes" OpenCode sessions by exploiting the fact that OpenCode's
+share API doesn't validate session existence. The workflow is:
+
+1. Generate random UUIDs for session, message, and part IDs
+2. Call `/share_create` with the session ID to get a secret and share URL
+3. Use `/share_sync` to populate the session with:
+   - Session info (title, timestamps, metadata)
+   - Messages (with different roles: user, assistant, system)
+   - Text parts (containing the actual shared content)
+
+The OpenCode web interface then displays this as if it were a real session.
+
+Two sharing modes are supported:
+- `share()`: Simple string sharing (single user message)
+- `share_conversation()`: Structured multi-turn conversations with different roles
+
+Note: This is not the intended use of OpenCode's API, which is designed to share
+actual OpenCode sessions created via their desktop app/CLI. However, since no
+authentication is required and sessions aren't validated, this works for sharing
+text and conversations.
+"""
+
+from __future__ import annotations
+
+import time
+from typing import TYPE_CHECKING, Any, Self
+import uuid
+
+import httpx
+
+from agentpool_commands.text_sharing.base import ShareResult, TextSharer
+
+
+if TYPE_CHECKING:
+    from anyenv.text_sharing.base import Visibility
+
+    from agentpool.messaging.message_history import MessageHistory
+    from agentpool.messaging.messages import ChatMessage
+
+
+class OpenCodeSharer(TextSharer):
+    """OpenCode text sharing service.
+
+    Creates fake sessions by generating UUIDs and syncing content via the share API.
+    OpenCode doesn't validate session existence, allowing us to create ad-hoc shares.
+
+    Examples:
+        Share a simple string:
+        ```python
+        async with OpenCodeSharer() as sharer:
+            result = await sharer.share("Hello, World!", title="My Share")
+            print(result.url)  # https://opencode.ai/s/abc12345
+        ```
+
+        Share a conversation:
+        ```python
+        # Use share_conversation() with a Conversation object from agentpool
+        async with OpenCodeSharer() as sharer:
+            result = await sharer.share_conversation(
+                conversation,
+                title="Python Discussion"
+            )
+            print(result.url)
+        ```
+    """
+
+    def __init__(
+        self,
+        *,
+        api_url: str | None = None,
+        timeout: float = 30.0,
+    ) -> None:
+        """Initialize OpenCode sharer.
+
+        Args:
+            api_url: OpenCode API URL (defaults to production)
+            timeout: Request timeout in seconds
+        """
+        self.api_url = api_url or "https://api.opencode.ai"
+        self.timeout = timeout
+        self._client = httpx.AsyncClient(timeout=timeout)
+
+    @property
+    def name(self) -> str:
+        """Name of the sharing service."""
+        return "OpenCode"
+
+    async def share(
+        self,
+        content: str,
+        *,
+        title: str | None = None,
+        syntax: str | None = None,
+        visibility: Visibility = "unlisted",
+        expires_in: int | None = None,
+    ) -> ShareResult:
+        """Share text content via OpenCode.
+
+        Args:
+            content: The text content to share
+            title: Optional title for the shared content
+            syntax: Syntax highlighting hint (ignored - OpenCode handles this)
+            visibility: Visibility level (ignored - OpenCode uses private shares)
+            expires_in: Expiration time (ignored - OpenCode doesn't support expiration)
+
+        Returns:
+            ShareResult with OpenCode share URL
+        """
+        try:
+            session_id = str(uuid.uuid4())
+            message_id = str(uuid.uuid4())
+            part_id = str(uuid.uuid4())
+            current_time = int(time.time() * 1000)
+
+            # Create share (returns secret and URL)
+            resp = await self._client.post(
+                f"{self.api_url}/share_create",
+                json={"sessionID": session_id},
+            )
+            resp.raise_for_status()
+            share_data = resp.json()
+            secret = share_data["secret"]
+            share_url = share_data["url"]
+
+            # Sync session info
+            info_key = f"session/info/{session_id}"
+            info_content = {
+                "id": session_id,
+                "projectID": "shared-content",
+                "directory": "/tmp",
+                "title": title or "Shared Content",
+                "version": "1.0.0",
+                "time": {
+                    "created": current_time,
+                    "updated": current_time,
+                },
+            }
+
+            resp = await self._client.post(
+                f"{self.api_url}/share_sync",
+                json={
+                    "sessionID": session_id,
+                    "secret": secret,
+                    "key": info_key,
+                    "content": info_content,
+                },
+            )
+            resp.raise_for_status()
+
+            # Sync message
+            msg_key = f"session/message/{session_id}/{message_id}"
+            msg_content = {
+                "id": message_id,
+                "sessionID": session_id,
+                "role": "user",
+                "time": {"created": current_time},
+            }
+
+            resp = await self._client.post(
+                f"{self.api_url}/share_sync",
+                json={
+                    "sessionID": session_id,
+                    "secret": secret,
+                    "key": msg_key,
+                    "content": msg_content,
+                },
+            )
+            resp.raise_for_status()
+
+            # Sync text part with actual content
+            part_key = f"session/part/{session_id}/{message_id}/{part_id}"
+            part_content = {
+                "id": part_id,
+                "sessionID": session_id,
+                "messageID": message_id,
+                "type": "text",
+                "text": content,
+            }
+
+            resp = await self._client.post(
+                f"{self.api_url}/share_sync",
+                json={
+                    "sessionID": session_id,
+                    "secret": secret,
+                    "key": part_key,
+                    "content": part_content,
+                },
+            )
+            resp.raise_for_status()
+
+            # Store secret in delete_url for later deletion
+            delete_url = f"{self.api_url}/share_delete#{secret}"
+
+            return ShareResult(
+                url=share_url,
+                raw_url=f"{self.api_url}/share_data?id={session_id[-8:]}",
+                delete_url=delete_url,
+                id=session_id,
+            )
+
+        except httpx.HTTPStatusError as e:
+            if e.response.status_code == 404:  # noqa: PLR2004
+                msg = "OpenCode API endpoint not found - service may be unavailable"
+                raise RuntimeError(msg) from e
+            if e.response.status_code == 429:  # noqa: PLR2004
+                msg = "Rate limited by OpenCode API"
+                raise RuntimeError(msg) from e
+            msg = f"OpenCode API error (HTTP {e.response.status_code}): {e.response.text}"
+            raise RuntimeError(msg) from e
+        except httpx.RequestError as e:
+            msg = f"Failed to connect to OpenCode API: {e}"
+            raise RuntimeError(msg) from e
+
+    async def _share_chat_messages(
+        self,
+        messages: list[ChatMessage[Any]],
+        *,
+        title: str | None = None,
+        visibility: Visibility = "unlisted",
+        expires_in: int | None = None,
+    ) -> ShareResult:
+        """Share ChatMessages using the OpenCode API.
+
+        Args:
+            messages: List of ChatMessage objects to share
+
+        This allows sharing AI chat sessions, multi-turn conversations,
+        or any structured dialogue with different roles.
+
+        Args:
+            messages: List of messages to share (must have at least one)
+            title: Optional title for the conversation
+            visibility: Visibility level (ignored - OpenCode uses private shares)
+            expires_in: Expiration time (ignored - OpenCode doesn't support expiration)
+
+        Returns:
+            ShareResult with OpenCode share URL
+
+        Example:
+            ```python
+            messages = [
+                Message(role="user", parts=[MessagePart(type="text", text="Hello!")]),
+                Message(role="assistant", parts=[MessagePart(type="text", text="Hi!")]),
+            ]
+            result = await sharer.share_conversation(messages, title="My Chat")
+            ```
+        """
+        if not messages:
+            msg = "Must provide at least one message"
+            raise ValueError(msg)
+
+        try:
+            session_id = str(uuid.uuid4())
+            current_time = int(time.time() * 1000)
+
+            # Create share
+            resp = await self._client.post(
+                f"{self.api_url}/share_create",
+                json={"sessionID": session_id},
+            )
+            resp.raise_for_status()
+            share_data = resp.json()
+            secret = share_data["secret"]
+            share_url = share_data["url"]
+
+            # Sync session info
+            info_key = f"session/info/{session_id}"
+            info_content = {
+                "id": session_id,
+                "projectID": "shared-conversation",
+                "directory": "/tmp",
+                "title": title or "Shared Conversation",
+                "version": "1.0.0",
+                "time": {
+                    "created": current_time,
+                    "updated": current_time,
+                },
+            }
+
+            resp = await self._client.post(
+                f"{self.api_url}/share_sync",
+                json={
+                    "sessionID": session_id,
+                    "secret": secret,
+                    "key": info_key,
+                    "content": info_content,
+                },
+            )
+            resp.raise_for_status()
+
+            # Sync each message and its parts
+            for chat_msg in messages:
+                from agentpool_server.opencode_server.converters import (
+                    chat_message_to_opencode,
+                )
+
+                # Convert ChatMessage to OpenCode format using the full converter
+                message_with_parts = chat_message_to_opencode(
+                    chat_msg,
+                    session_id=session_id,
+                    working_dir="/tmp",
+                    agent_name=chat_msg.name or "default",
+                    model_id=chat_msg.model_name or "unknown",
+                    provider_id="agentpool",
+                )
+
+                # Serialize to dicts with camelCase and no None values
+                msg_info = message_with_parts.info.model_dump(by_alias=True, exclude_none=True)
+                msg_parts = [
+                    p.model_dump(by_alias=True, exclude_none=True) for p in message_with_parts.parts
+                ]
+
+                # Sync message
+                msg_key = f"session/message/{session_id}/{msg_info['id']}"
+                resp = await self._client.post(
+                    f"{self.api_url}/share_sync",
+                    json={
+                        "sessionID": session_id,
+                        "secret": secret,
+                        "key": msg_key,
+                        "content": msg_info,
+                    },
+                )
+                resp.raise_for_status()
+
+                # Sync parts
+                for part in msg_parts:
+                    part_key = f"session/part/{session_id}/{msg_info['id']}/{part['id']}"
+
+                    resp = await self._client.post(
+                        f"{self.api_url}/share_sync",
+                        json={
+                            "sessionID": session_id,
+                            "secret": secret,
+                            "key": part_key,
+                            "content": part,
+                        },
+                    )
+                    resp.raise_for_status()
+
+            # Store secret in delete_url for later deletion
+            delete_url = f"{self.api_url}/share_delete#{secret}"
+
+            return ShareResult(
+                url=share_url,
+                raw_url=f"{self.api_url}/share_data?id={session_id[-8:]}",
+                delete_url=delete_url,
+                id=session_id,
+            )
+
+        except httpx.HTTPStatusError as e:
+            if e.response.status_code == 404:  # noqa: PLR2004
+                msg = "OpenCode API endpoint not found - service may be unavailable"
+                raise RuntimeError(msg) from e
+            if e.response.status_code == 429:  # noqa: PLR2004
+                msg = "Rate limited by OpenCode API"
+                raise RuntimeError(msg) from e
+            msg = f"OpenCode API error (HTTP {e.response.status_code}): {e.response.text}"
+            raise RuntimeError(msg) from e
+        except httpx.RequestError as e:
+            msg = f"Failed to connect to OpenCode API: {e}"
+            raise RuntimeError(msg) from e
+
+    async def share_conversation(
+        self,
+        conversation: MessageHistory,
+        *,
+        title: str | None = None,
+        visibility: Visibility = "unlisted",
+        expires_in: int | None = None,
+        num_messages: int | None = None,
+    ) -> ShareResult:
+        """Share conversation using OpenCode's native structured format.
+
+        Args:
+            conversation: Conversation object to share
+            title: Optional title for the conversation
+            visibility: Visibility level (ignored)
+            expires_in: Expiration time (ignored)
+            num_messages: Number of messages to include (None = all)
+
+        Returns:
+            ShareResult with OpenCode share URL
+
+        Note:
+            System prompts are stored as metadata on messages (UserMessage.system field
+            in OpenCode, ModelRequest.instructions in pydantic-ai), not as separate
+            "system" role messages. ChatMessage.role only supports "user" and "assistant".
+        """
+        # Get messages to share
+        messages_to_share = list(
+            conversation.chat_messages[-num_messages:]
+            if num_messages
+            else conversation.chat_messages
+        )
+
+        return await self._share_chat_messages(
+            list(messages_to_share),
+            title=title,
+            visibility=visibility,
+            expires_in=expires_in,
+        )
+
+    async def delete_share(self, result: ShareResult) -> bool:
+        """Delete a shared session.
+
+        Args:
+            result: The ShareResult containing session ID and secret
+
+        Returns:
+            True if deletion was successful
+        """
+        if not result.delete_url or "#" not in result.delete_url:
+            return False
+
+        # Extract secret from delete_url
+        secret = result.delete_url.split("#", 1)[1]
+
+        try:
+            resp = await self._client.post(
+                f"{self.api_url}/share_delete",
+                json={
+                    "sessionID": result.id,
+                    "secret": secret,
+                },
+            )
+            resp.raise_for_status()
+        except httpx.HTTPError:
+            return False
+        else:
+            return True
+
+    async def close(self) -> None:
+        """Close the HTTP client."""
+        await self._client.aclose()
+
+    async def __aenter__(self) -> Self:
+        return self
+
+    async def __aexit__(self, *args: object) -> None:
+        await self.close()
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    async def main() -> None:
+        """Test OpenCode sharing functionality."""
+        async with OpenCodeSharer() as sharer:
+            # Test simple string sharing
+            print("=== Simple String Share ===")
+            result = await sharer.share(
+                "Hello, World!\n\nThis is a test of OpenCode sharing.",
+                title="Test Share",
+            )
+            print(f"Share URL: {result.url}")
+            print(f"Raw API URL: {result.raw_url}")
+            print(f"Session ID: {result.id}")
+
+    asyncio.run(main())

agentpool_commands/text_sharing/paste_rs.py

@@ -0,0 +1,59 @@
+"""paste.rs text sharing provider."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from agentpool_commands.text_sharing.base import ShareResult, TextSharer
+
+
+if TYPE_CHECKING:
+    from agentpool_commands.text_sharing.base import Visibility
+
+
+class PasteRsSharer(TextSharer):
+    """Share text via paste.rs."""
+
+    @property
+    def name(self) -> str:
+        """Name of the sharing service."""
+        return "paste.rs"
+
+    async def share(
+        self,
+        content: str,
+        *,
+        title: str | None = None,
+        syntax: str | None = None,
+        visibility: Visibility = "unlisted",
+        expires_in: int | None = None,
+    ) -> ShareResult:
+        """Share content on paste.rs.
+
+        Args:
+            content: The text content to share
+            title: Ignored (paste.rs doesn't support titles)
+            syntax: Used as file extension for syntax highlighting
+            visibility: Ignored (all pastes are unlisted)
+            expires_in: Ignored (paste.rs doesn't support expiration)
+        """
+        import anyenv
+
+        headers = {"Content-Type": "text/plain; charset=utf-8"}
+        data = content.encode()
+        response = await anyenv.request("POST", "https://paste.rs/", data=data, headers=headers)
+        url = (await response.text()).strip()
+        ext = syntax or "txt"
+        return ShareResult(url=f"{url}.{ext}", raw_url=url)
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    async def main() -> None:
+        sharer = PasteRsSharer()
+        result = await sharer.share("# Test Paste\n\nHello from anyenv!", syntax="md")
+        print(f"URL: {result.url}")
+        print(f"Raw: {result.raw_url}")
+
+    asyncio.run(main())

agentpool_commands/text_sharing/pastebin.py

@@ -0,0 +1,116 @@
+"""Pastebin.com text sharing provider."""
+
+from __future__ import annotations
+
+import os
+from typing import TYPE_CHECKING, Any
+
+from agentpool_commands.text_sharing.base import ShareResult, TextSharer
+
+
+if TYPE_CHECKING:
+    from agentpool_commands.text_sharing.base import Visibility
+
+
+# Pastebin expiry values
+EXPIRY_MAP = {
+    None: "N",  # Never
+    600: "10M",  # 10 Minutes
+    3600: "1H",  # 1 Hour
+    86400: "1D",  # 1 Day
+    604800: "1W",  # 1 Week
+    1209600: "2W",  # 2 Weeks
+    2592000: "1M",  # 1 Month
+    15552000: "6M",  # 6 Months
+    31536000: "1Y",  # 1 Year
+}
+
+# Map visibility to Pastebin's api_paste_private values
+VISIBILITY_MAP = {"public": "0", "unlisted": "1", "private": "2"}
+
+
+def _get_expiry_code(seconds: int | None) -> str:
+    """Convert seconds to Pastebin expiry code, rounding to nearest valid value."""
+    if seconds is None:
+        return "N"
+
+    valid_seconds = sorted(k for k in EXPIRY_MAP if k is not None)
+    closest = min(valid_seconds, key=lambda x: abs(x - seconds))
+    return EXPIRY_MAP[closest]
+
+
+class PastebinSharer(TextSharer):
+    """Share text via Pastebin.com."""
+
+    def __init__(self, api_key: str | None = None) -> None:
+        """Initialize the Pastebin sharer.
+
+        Args:
+            api_key: Pastebin API developer key. If not provided,
+                     reads from PASTEBIN_API_KEY environment variable.
+        """
+        self._api_key = api_key or os.environ.get("PASTEBIN_API_KEY")
+        if not self._api_key:
+            msg = "Pastebin API key required. Set PASTEBIN_API_KEY or pass api_key parameter."
+            raise ValueError(msg)
+
+    @property
+    def name(self) -> str:
+        """Name of the sharing service."""
+        return "Pastebin"
+
+    async def share(
+        self,
+        content: str,
+        *,
+        title: str | None = None,
+        syntax: str | None = None,
+        visibility: Visibility = "unlisted",
+        expires_in: int | None = None,
+    ) -> ShareResult:
+        """Share content on Pastebin.com.
+
+        Args:
+            content: The text content to share
+            title: Title/name of the paste
+            syntax: Syntax highlighting (e.g. "python", "markdown", "javascript")
+            visibility: "public", "unlisted", or "private"
+            expires_in: Expiration time in seconds (rounded to nearest valid value)
+        """
+        import anyenv
+
+        data: dict[str, Any] = {
+            "api_dev_key": self._api_key,
+            "api_option": "paste",
+            "api_paste_code": content,
+            "api_paste_private": VISIBILITY_MAP.get(visibility, "1"),
+            "api_paste_expire_date": _get_expiry_code(expires_in),
+        }
+
+        if title:
+            data["api_paste_name"] = title
+
+        if syntax:
+            data["api_paste_format"] = syntax
+
+        response = await anyenv.post("https://pastebin.com/api/api_post.php", data=data)
+        url = (await response.text()).strip()
+        if url.startswith("Bad API request"):
+            msg = f"Pastebin API error: {url}"
+            raise RuntimeError(msg)
+        paste_key = url.split("/")[-1]
+        raw_url = f"https://pastebin.com/raw/{paste_key}"
+        return ShareResult(url=url, raw_url=raw_url, id=paste_key)
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    async def main() -> None:
+        sharer = PastebinSharer()
+        result = await sharer.share("# Test Paste\n\nHello from anyenv!", title="test.md")
+        print(f"URL: {result.url}")
+        print(f"Raw: {result.raw_url}")
+        print(f"ID: {result.id}")
+
+    asyncio.run(main())