@scemoon/cdh 1.0.0

package/tui/acp/api.py

@@ -0,0 +1,55 @@
+# mypy: disable-error-code="empty-body"
+"""
+ACP remote API
+"""
+
+from tui import jsonrpc
+from tui.acp import protocol
+
+API = jsonrpc.API()
+
+
+@API.method()
+def initialize(
+    protocolVersion: int,
+    clientCapabilities: protocol.ClientCapabilities,
+    clientInfo: protocol.Implementation,
+) -> protocol.InitializeResponse:
+    """https://agentclientprotocol.com/protocol/initialization"""
+    ...
+
+
+@API.method(name="session/new")
+def session_new(
+    cwd: str, mcpServers: list[protocol.McpServer]
+) -> protocol.NewSessionResponse:
+    """https://agentclientprotocol.com/protocol/session-setup#session-id"""
+    ...
+
+
+@API.method(name="session/load")
+def session_load(
+    cwd: str, mcpServers: list[protocol.McpServer], sessionId: str
+) -> protocol.LoadSessionResponse:
+    """https://agentclientprotocol.com/protocol/session-setup#loading-a-session"""
+    ...
+
+
+@API.notification(name="session/cancel")
+def session_cancel(sessionId: str, _meta: dict):
+    """https://agentclientprotocol.com/protocol/prompt-turn#cancellation"""
+    ...
+
+
+@API.method(name="session/prompt")
+def session_prompt(
+    prompt: list[protocol.ContentBlock], sessionId: str
+) -> protocol.SessionPromptResponse:
+    """https://agentclientprotocol.com/protocol/prompt-turn#1-user-message"""
+    ...
+
+
+@API.method(name="session/set_mode")
+def session_set_mode(sessionId: str, modeId: str) -> protocol.SetSessionModeResponse:
+    """https://agentclientprotocol.com/protocol/session-modes#from-the-client"""
+    ...

package/tui/acp/encode_tool_call_id.py

@@ -0,0 +1,12 @@
+def encode_tool_call_id(tool_call_id: str) -> str:
+    """Encode the tool call id so that it fits within a TCSS id.
+
+    Args:
+        tool_call_id: Raw tool call id.
+
+    Returns:
+        Tool call usable as widget id.
+    """
+    hex_tool_call_id = "".join(f"{ord(character):2X}" for character in tool_call_id)
+    encoded_tool_call_id = f"tool-call-{hex_tool_call_id}"
+    return encoded_tool_call_id

package/tui/acp/messages.py

@@ -0,0 +1,190 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+
+from asyncio import Future
+from typing import Any, Mapping, TYPE_CHECKING
+from textual.message import Message
+
+import rich.repr
+
+from tui.answer import Answer
+from tui.acp import protocol
+from tui.acp.encode_tool_call_id import encode_tool_call_id
+
+if TYPE_CHECKING:
+    from tui.acp.agent import Mode
+    from tui.widgets.terminal_tool import ToolState
+
+
+class AgentMessage(Message):
+    pass
+
+
+@dataclass
+class ContextUpdate(AgentMessage):
+    """Context token usage update from the engine."""
+    used: int
+    size: int
+
+
+@dataclass
+class Thinking(AgentMessage):
+    type: str
+    text: str
+
+
+@dataclass
+class SessionReplay(AgentMessage):
+    """Marks the start/end of a session replay (loading historical messages)."""
+    active: bool
+
+
+@dataclass
+class UpdateStatusLine(AgentMessage):
+    status_line: str
+
+
+@dataclass
+class Update(AgentMessage):
+    type: str
+    text: str
+
+
+@dataclass
+class UserMessage(Message):
+    type: str
+    text: str
+
+
+@dataclass
+@rich.repr.auto
+class RequestPermission(AgentMessage):
+    options: list[protocol.PermissionOption]
+    tool_call: protocol.ToolCallUpdatePermissionRequest
+    result_future: Future[Answer]
+
+
+@dataclass
+class AskUser(AgentMessage):
+    question: str = ""
+    context: str = ""
+    options: list[dict] = field(default_factory=list)
+    questions: list[dict] = field(default_factory=list)
+    tool_id: str = ""
+
+
+@dataclass
+class Plan(AgentMessage):
+    entries: list[protocol.PlanEntry]
+
+
+@dataclass
+class ToolCall(AgentMessage):
+    tool_call: protocol.ToolCall
+
+    @property
+    def tool_id(self) -> str:
+        """An id suitable for use as a TCSS ID."""
+        return encode_tool_call_id(self.tool_call["toolCallId"])
+
+
+@dataclass
+class ToolCallUpdate(AgentMessage):
+    tool_call: protocol.ToolCall
+    update: protocol.ToolCallUpdate
+
+    @property
+    def tool_id(self) -> str:
+        """An id suitable for use as a TCSS ID."""
+        return encode_tool_call_id(self.tool_call["toolCallId"])
+
+
+@dataclass
+class AvailableCommandsUpdate(AgentMessage):
+    """The agent is reporting its slash commands."""
+
+    commands: list[protocol.AvailableCommand]
+
+
+@dataclass
+class CreateTerminal(AgentMessage):
+    """Request a terminal in the conversation."""
+
+    terminal_id: str
+    command: str
+    result_future: Future[bool]
+    args: list[str] | None = None
+    cwd: str | None = None
+    env: Mapping[str, str] | None = None
+    output_byte_limit: int | None = None
+
+
+@dataclass
+class KillTerminal(AgentMessage):
+    """Kill a terminal process."""
+
+    terminal_id: str
+
+
+@dataclass
+class GetTerminalState(AgentMessage):
+    """Get the state of the terminal."""
+
+    terminal_id: str
+    result_future: Future[ToolState]
+
+
+@dataclass
+class ReleaseTerminal(AgentMessage):
+    """Release the terminal."""
+
+    terminal_id: str
+
+
+@dataclass
+class WaitForTerminalExit(AgentMessage):
+    """Wait for the terminal to exit."""
+
+    terminal_id: str
+    result_future: Future[tuple[int, str | None]]
+
+
+@rich.repr.auto
+@dataclass
+class SetModes(AgentMessage):
+    """Set modes from agent."""
+
+    current_mode: str
+    modes: dict[str, Mode]
+
+
+@dataclass
+class ModeUpdate(AgentMessage):
+    """Agent informed us about a mode change."""
+
+    current_mode: str
+
+
+@dataclass
+class SubAgentStart(AgentMessage):
+    """Sub-agent task started."""
+
+    subagent_id: str
+    agent_type: str
+
+
+@dataclass
+class SubAgentChunk(AgentMessage):
+    """Sub-agent streaming text chunk."""
+
+    subagent_id: str
+    text: str
+
+
+@dataclass
+class SubAgentEnd(AgentMessage):
+    """Sub-agent task completed."""
+
+    subagent_id: str
+    agent_type: str

package/tui/acp/prompt.py

@@ -0,0 +1,110 @@
+import base64
+from pathlib import Path
+
+from tui.acp import protocol
+from tui.prompt.extract import extract_paths_from_prompt
+from tui.prompt.resource import load_resource, ResourceError
+
+
+CLOUD_SPEC_SKILL_PATH = Path(__file__).resolve().parent.parent.parent / "cloud-spec-skill" / "SKILL.md"
+
+
+def _get_cloud_spec_content() -> str:
+    """Load cloud-spec-skill content."""
+    if CLOUD_SPEC_SKILL_PATH.exists():
+        return CLOUD_SPEC_SKILL_PATH.read_text(encoding="utf-8")
+    return ""
+
+
+def cloud_spec_skill_loaded() -> bool:
+    """Check if cloud-spec-skill SKILL.md exists."""
+    return CLOUD_SPEC_SKILL_PATH.exists()
+
+
+def ensure_cloud_spec_skill_installed() -> str | None:
+    """Ensure cloud-spec-skill is accessible to agents.
+
+    Checks if SKILL.md exists in repo and copies it to ~/.onecode/skills/ if needed,
+    or updates if the repo version is newer.
+    Returns error message if installation fails, None on success.
+    """
+    from onecode.skills.loader import USER_SKILLS_DIR
+
+    if not CLOUD_SPEC_SKILL_PATH.exists():
+        return None
+
+    skill_dest = USER_SKILLS_DIR / "cloud-spec-skill"
+    dest_skills_md = skill_dest / "SKILL.md"
+
+    if skill_dest.exists() and dest_skills_md.exists():
+        repo_mtime = CLOUD_SPEC_SKILL_PATH.stat().st_mtime
+        dest_mtime = dest_skills_md.stat().st_mtime
+        if dest_mtime >= repo_mtime:
+            return None
+
+    try:
+        import shutil
+
+        if skill_dest.exists():
+            shutil.rmtree(skill_dest)
+        skill_dest.parent.mkdir(parents=True, exist_ok=True)
+        shutil.copytree(CLOUD_SPEC_SKILL_PATH.parent, skill_dest)
+        return None
+    except Exception as e:
+        return str(e)
+
+
+def build(project_path: Path, prompt: str) -> list[protocol.ContentBlock]:
+    """Build the prompt structure and extract paths with the @ syntax.
+
+    Args:
+        project_path: The project root.
+        prompt: The prompt text.
+
+    Returns:
+        A list of content blocks.
+    """
+    prompt_content: list[protocol.ContentBlock] = []
+
+    # Prepend cloud-spec-skill as system guidance
+    cloud_spec_content = _get_cloud_spec_content()
+    if cloud_spec_content:
+        prompt_content.append({
+            "type": "text",
+            "text": f"[System Guidance - Always follow these development standards]\n\n{cloud_spec_content}\n\n---\n\n"
+        })
+
+    prompt_content.append({"type": "text", "text": prompt})
+    for path, _, _ in extract_paths_from_prompt(prompt):
+        if path.endswith("/"):
+            continue
+        try:
+            resource = load_resource(project_path, Path(path))
+        except ResourceError:
+            # TODO: How should this be handled?
+            continue
+        uri = f"file://{resource.path.absolute().resolve()}"
+        if resource.text is not None:
+            prompt_content.append(
+                {
+                    "type": "resource",
+                    "resource": {
+                        "uri": uri,
+                        "text": resource.text,
+                        "mimeType": resource.mime_type,
+                    },
+                }
+            )
+        elif resource.data is not None:
+            prompt_content.append(
+                {
+                    "type": "resource",
+                    "resource": {
+                        "uri": uri,
+                        "blob": base64.b64encode(resource.data).decode("utf-8"),
+                        "mimeType": resource.mime_type,
+                    },
+                }
+            )
+
+    return prompt_content