#!/usr/bin/env python3
"""
cmux Python Client

A client library for programmatically controlling cmux via Unix socket.

Usage:
    from cmux import cmux

    client = cmux()
    client.connect()

    # Send text to terminal
    client.send("echo hello\\n")

    # Send special keys
    client.send_key("ctrl-c")
    client.send_key("ctrl-d")

    # Tab management
    client.new_tab()
    client.list_tabs()
    client.select_tab(0)
    client.new_split("right")
    client.list_surfaces()
    client.focus_surface(0)

    client.close()
"""

import socket
import select
import os
import time
import errno
import json
import base64
import glob
import re
from typing import Optional, List, Tuple, Union


class cmuxError(Exception):
    """Exception raised for cmux errors"""
    pass


_APP_SUPPORT_DIR = os.path.expanduser("~/Library/Application Support/cmux")
_STABLE_SOCKET_PATH = os.path.join(_APP_SUPPORT_DIR, "cmux.sock")
_LEGACY_STABLE_SOCKET_PATH = "/tmp/cmux.sock"
_LAST_SOCKET_PATH_FILES = [
    os.path.join(_APP_SUPPORT_DIR, "last-socket-path"),
    "/tmp/cmux-last-socket-path",
]
_DEFAULT_DEBUG_BUNDLE_ID = "com.cmuxterm.app.debug"


def _sanitize_tag_slug(raw: str) -> str:
    cleaned = re.sub(r"[^a-z0-9]+", "-", (raw or "").strip().lower())
    cleaned = re.sub(r"-+", "-", cleaned).strip("-")
    return cleaned or "agent"


def _sanitize_bundle_suffix(raw: str) -> str:
    # Must match scripts/reload.sh sanitize_bundle() so tagged tests can
    # reliably target the correct app via AppleScript.
    cleaned = re.sub(r"[^a-z0-9]+", ".", (raw or "").strip().lower())
    cleaned = re.sub(r"\.+", ".", cleaned).strip(".")
    return cleaned or "agent"


def _quote_option_value(value: str) -> str:
    # Must match TerminalController.parseOptions() quoting rules.
    escaped = (value or "").replace("\\", "\\\\").replace('"', '\\"')
    return f"\"{escaped}\""


def _default_bundle_id() -> str:
    override = os.environ.get("CMUX_BUNDLE_ID")
    if override:
        return override

    tag = os.environ.get("CMUX_TAG")
    if tag:
        suffix = _sanitize_bundle_suffix(tag)
        return f"{_DEFAULT_DEBUG_BUNDLE_ID}.{suffix}"

    return _DEFAULT_DEBUG_BUNDLE_ID


def _read_last_socket_path() -> Optional[str]:
    for marker_path in _LAST_SOCKET_PATH_FILES:
        try:
            with open(marker_path, "r", encoding="utf-8") as f:
                path = f.read().strip()
            if path:
                return path
        except OSError:
            continue
    return None


def _can_connect(path: str, timeout: float = 0.15, retries: int = 4) -> bool:
    # Best-effort check to avoid getting stuck on stale socket files.
    for _ in range(max(1, retries)):
        s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        try:
            s.settimeout(timeout)
            s.connect(path)
            return True
        except OSError:
            time.sleep(0.05)
        finally:
            try:
                s.close()
            except Exception:
                pass
    return False


def _default_socket_path() -> str:
    tag = os.environ.get("CMUX_TAG")
    if tag:
        slug = _sanitize_tag_slug(tag)
        tagged_candidates = [
            f"/tmp/cmux-debug-{slug}.sock",
            f"/tmp/cmux-{slug}.sock",
        ]
        for path in tagged_candidates:
            if os.path.exists(path) and _can_connect(path):
                return path
        # If nothing is connectable yet (e.g. the app is still starting),
        # fall back to the first existing candidate.
        for path in tagged_candidates:
            if os.path.exists(path):
                return path
        # Prefer the debug naming convention when we have to guess.
        return tagged_candidates[0]

    override = os.environ.get("CMUX_SOCKET_PATH")
    if override:
        if os.path.exists(override) and _can_connect(override):
            return override
        # Treat stable defaults as implicit so old env values still migrate cleanly.
        if not os.path.exists(override) and override not in {_STABLE_SOCKET_PATH, _LEGACY_STABLE_SOCKET_PATH}:
            return override

    last_socket = _read_last_socket_path()
    if last_socket:
        if os.path.exists(last_socket) and _can_connect(last_socket):
            return last_socket

    # Prefer the non-tagged sockets when present.
    candidates = ["/tmp/cmux-debug.sock", _STABLE_SOCKET_PATH, _LEGACY_STABLE_SOCKET_PATH]
    for path in candidates:
        if os.path.exists(path) and _can_connect(path):
            return path

    # Otherwise, fall back to the newest discovered socket if there is one.
    tagged = glob.glob("/tmp/cmux-debug-*.sock")
    tagged.extend(glob.glob(os.path.join(_APP_SUPPORT_DIR, "cmux*.sock")))
    tagged = [p for p in tagged if os.path.exists(p)]
    if tagged:
        tagged.sort(key=lambda p: os.path.getmtime(p), reverse=True)
        for p in tagged:
            if _can_connect(p, timeout=0.1, retries=2):
                return p

    return candidates[0]


class cmux:
    """Client for controlling cmux via Unix socket"""

    DEFAULT_SOCKET_PATH = _default_socket_path()
    DEFAULT_BUNDLE_ID = _default_bundle_id()

    @staticmethod
    def default_socket_path() -> str:
        return _default_socket_path()

    @staticmethod
    def default_bundle_id() -> str:
        return _default_bundle_id()

    def __init__(self, socket_path: str = None):
        # Resolve at init time so imports don't "lock in" a stale path.
        self.socket_path = socket_path or _default_socket_path()
        self._socket: Optional[socket.socket] = None
        self._recv_buffer: str = ""

    def connect(self) -> None:
        """Connect to the cmux socket"""
        if self._socket is not None:
            return

        start = time.time()
        while not os.path.exists(self.socket_path):
            if time.time() - start >= 2.0:
                raise cmuxError(
                    f"Socket not found at {self.socket_path}. "
                    "Is cmux running?"
                )
            time.sleep(0.1)

        last_error: Optional[socket.error] = None
        while True:
            self._socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
            try:
                self._socket.connect(self.socket_path)
                self._socket.settimeout(5.0)
                return
            except socket.error as e:
                last_error = e
                self._socket.close()
                self._socket = None
                if e.errno in (errno.ECONNREFUSED, errno.ENOENT) and time.time() - start < 2.0:
                    time.sleep(0.1)
                    continue
                raise cmuxError(f"Failed to connect: {e}")

    def close(self) -> None:
        """Close the connection"""
        if self._socket is not None:
            self._socket.close()
            self._socket = None

    def __enter__(self):
        self.connect()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.close()
        return False

    def _send_command(self, command: str) -> str:
        """Send a command and receive response"""
        if self._socket is None:
            raise cmuxError("Not connected")

        try:
            self._socket.sendall((command + "\n").encode())
            data = self._recv_buffer
            self._recv_buffer = ""
            saw_newline = "\n" in data
            start = time.time()
            while True:
                if saw_newline:
                    ready, _, _ = select.select([self._socket], [], [], 0.1)
                    if not ready:
                        break
                try:
                    chunk = self._socket.recv(8192)
                except socket.timeout:
                    if saw_newline:
                        break
                    if time.time() - start >= 5.0:
                        raise cmuxError("Command timed out")
                    continue
                if not chunk:
                    break
                data += chunk.decode()
                if "\n" in data:
                    saw_newline = True
            if data.endswith("\n"):
                data = data[:-1]
            return data
        except socket.timeout:
            raise cmuxError("Command timed out")
        except socket.error as e:
            raise cmuxError(f"Socket error: {e}")

    def ping(self) -> bool:
        """Check if the server is responding"""
        response = self._send_command("ping")
        return response == "PONG"

    def list_tabs(self) -> List[Tuple[int, str, str, bool]]:
        """
        List all tabs.
        Returns list of (index, id, title, is_selected) tuples.
        """
        response = self._send_command("list_tabs")
        if response.startswith("ERROR: Unknown command"):
            response = self._send_command("list_workspaces")
        if response in ("No tabs", "No workspaces"):
            return []

        tabs = []
        for line in response.split("\n"):
            if not line.strip():
                continue
            selected = line.startswith("*")
            parts = line.lstrip("* ").split(" ", 2)
            if len(parts) >= 3:
                index = int(parts[0].rstrip(":"))
                tab_id = parts[1]
                title = parts[2] if len(parts) > 2 else ""
                tabs.append((index, tab_id, title, selected))
        return tabs

    def new_tab(self) -> str:
        """Create a new tab. Returns the new tab's ID."""
        response = self._send_command("new_tab")
        if response.startswith("ERROR: Unknown command"):
            response = self._send_command("new_workspace")
        if response.startswith("OK "):
            return response[3:]
        raise cmuxError(response)

    def new_split(self, direction: str) -> str:
        """Create a split in the given direction (left/right/up/down). Returns new panel ID when available."""
        response = self._send_command(f"new_split {direction}")
        if response.startswith("OK "):
            return response[3:]
        if response.startswith("OK"):
            return ""
        if not response.startswith("OK"):
            raise cmuxError(response)

    def close_tab(self, tab_id: str) -> None:
        """Close a tab by ID"""
        response = self._send_command(f"close_tab {tab_id}")
        if response.startswith("ERROR: Unknown command"):
            response = self._send_command(f"close_workspace {tab_id}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def select_tab(self, tab: Union[str, int]) -> None:
        """Select a tab by ID or index"""
        response = self._send_command(f"select_tab {tab}")
        if response.startswith("ERROR: Unknown command"):
            response = self._send_command(f"select_workspace {tab}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def list_surfaces(self, tab: Union[str, int, None] = None) -> List[Tuple[int, str, bool]]:
        """
        List surfaces for a tab. Returns list of (index, id, is_focused) tuples.
        If tab is None, uses the current tab.
        """
        arg = "" if tab is None else str(tab)
        response = self._send_command(f"list_surfaces {arg}".rstrip())
        if response in ("No surfaces", "ERROR: Tab not found"):
            return []

        surfaces = []
        for line in response.split("\n"):
            if not line.strip():
                continue
            selected = line.startswith("*")
            parts = line.lstrip("* ").split(" ", 1)
            if len(parts) >= 2:
                index = int(parts[0].rstrip(":"))
                surface_id = parts[1]
                surfaces.append((index, surface_id, selected))
        return surfaces

    def focus_surface(self, surface: Union[str, int]) -> None:
        """Focus a surface by ID or index in the current tab."""
        response = self._send_command(f"focus_surface {surface}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def current_tab(self) -> str:
        """Get the current tab's ID"""
        response = self._send_command("current_tab")
        if response.startswith("ERROR: Unknown command"):
            response = self._send_command("current_workspace")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response

    def current_workspace(self) -> str:
        """Get the current workspace's ID."""
        response = self._send_command("current_workspace")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response

    def send(self, text: str) -> None:
        """
        Send text to the current terminal.
        Use \\n for newline (Enter), \\t for tab, etc.

        Note: The text is sent as-is. Use actual escape sequences:
            client.send("echo hello\\n")  # Sends: echo hello<Enter>
            client.send("echo hello" + "\\n")  # Same thing
        """
        # Escape actual newlines/tabs to their backslash forms for protocol
        # The server will unescape them
        escaped = text.replace("\n", "\\n").replace("\r", "\\r").replace("\t", "\\t")
        response = self._send_command(f"send {escaped}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def send_surface(self, surface: Union[str, int], text: str) -> None:
        """Send text to a specific surface by ID or index in the current tab."""
        escaped = text.replace("\n", "\\n").replace("\r", "\\r").replace("\t", "\\t")
        response = self._send_command(f"send_surface {surface} {escaped}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def send_key(self, key: str) -> None:
        """
        Send a special key to the current terminal.

        Supported keys:
            ctrl-c, ctrl-d, ctrl-z, ctrl-\\
            enter, tab, escape, backspace
            ctrl-<letter> for any letter
        """
        response = self._send_command(f"send_key {key}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def send_key_surface(self, surface: Union[str, int], key: str) -> None:
        """Send a special key to a specific surface by ID or index in the current tab."""
        response = self._send_command(f"send_key_surface {surface} {key}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def send_line(self, text: str) -> None:
        """Send text followed by Enter"""
        self.send(text + "\\n")

    def send_ctrl_c(self) -> None:
        """Send Ctrl+C (SIGINT)"""
        self.send_key("ctrl-c")

    def send_ctrl_d(self) -> None:
        """Send Ctrl+D (EOF)"""
        self.send_key("ctrl-d")

    def help(self) -> str:
        """Get help text from server"""
        return self._send_command("help")

    def notify(self, title: str, subtitle: str = "", body: str = "") -> None:
        """Create a notification for the focused surface."""
        if subtitle or body:
            payload = f"{title}|{subtitle}|{body}"
        else:
            payload = title
        response = self._send_command(f"notify {payload}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def notify_surface(self, surface: Union[str, int], title: str, subtitle: str = "", body: str = "") -> None:
        """Create a notification for a specific surface by ID or index."""
        if subtitle or body:
            payload = f"{title}|{subtitle}|{body}"
        else:
            payload = title
        response = self._send_command(f"notify_surface {surface} {payload}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def list_notifications(self) -> list[dict]:
        """
        List notifications.
        Returns list of dicts with keys: id, tab_id/workspace_id, surface_id, is_read, title, subtitle, body.
        """
        response = self._send_command("list_notifications")
        if response == "No notifications":
            return []

        items = []
        for line in response.split("\n"):
            if not line.strip():
                continue
            _, payload = line.split(":", 1)
            parts = payload.split("|", 6)
            if len(parts) < 7:
                continue
            notif_id, tab_id, surface_id, read_text, title, subtitle, body = parts
            items.append({
                "id": notif_id,
                "tab_id": tab_id,
                "workspace_id": tab_id,
                "surface_id": None if surface_id == "none" else surface_id,
                "is_read": read_text == "read",
                "title": title,
                "subtitle": subtitle,
                "body": body,
            })
        return items

    def clear_notifications(self) -> None:
        """Clear all notifications."""
        response = self._send_command("clear_notifications")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def set_app_focus(self, active: Union[bool, None]) -> None:
        """Override app focus state. Use None to clear override."""
        if active is None:
            value = "clear"
        else:
            value = "active" if active else "inactive"
        response = self._send_command(f"set_app_focus {value}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def simulate_app_active(self) -> None:
        """Trigger the app active handler."""
        response = self._send_command("simulate_app_active")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def set_status(
        self,
        key: str,
        value: str,
        icon: str = None,
        color: str = None,
        url: str = None,
        priority: int = None,
        format: str = None,
        tab: str = None,
    ) -> None:
        """Set a sidebar status entry."""
        # Put options before `--` so value can contain arbitrary tokens like `--tab`.
        cmd = f"set_status {key}"
        if icon:
            cmd += f" --icon={icon}"
        if color:
            cmd += f" --color={color}"
        if url:
            cmd += f" --url={_quote_option_value(url)}"
        if priority is not None:
            cmd += f" --priority={priority}"
        if format:
            cmd += f" --format={format}"
        if tab:
            cmd += f" --tab={tab}"
        cmd += f" -- {_quote_option_value(value)}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def clear_status(self, key: str, tab: str = None) -> None:
        """Remove a sidebar status entry."""
        cmd = f"clear_status {key}"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def report_meta(
        self,
        key: str,
        value: str,
        icon: str = None,
        color: str = None,
        url: str = None,
        priority: int = None,
        format: str = None,
        tab: str = None,
    ) -> None:
        """Report a sidebar metadata entry."""
        cmd = f"report_meta {key}"
        if icon:
            cmd += f" --icon={icon}"
        if color:
            cmd += f" --color={color}"
        if url:
            cmd += f" --url={_quote_option_value(url)}"
        if priority is not None:
            cmd += f" --priority={priority}"
        if format:
            cmd += f" --format={format}"
        if tab:
            cmd += f" --tab={tab}"
        cmd += f" -- {_quote_option_value(value)}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def clear_meta(self, key: str, tab: str = None) -> None:
        """Remove a sidebar metadata entry."""
        cmd = f"clear_meta {key}"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def list_meta(self, tab: str = None) -> str:
        """List sidebar metadata entries."""
        cmd = "list_meta"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response

    def report_meta_block(self, key: str, markdown: str, priority: int = None, tab: str = None) -> None:
        """Report a freeform sidebar markdown metadata block."""
        cmd = f"report_meta_block {key}"
        if priority is not None:
            cmd += f" --priority={priority}"
        if tab:
            cmd += f" --tab={tab}"
        cmd += f" -- {_quote_option_value(markdown)}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def clear_meta_block(self, key: str, tab: str = None) -> None:
        """Remove a sidebar markdown metadata block."""
        cmd = f"clear_meta_block {key}"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def list_meta_blocks(self, tab: str = None) -> str:
        """List sidebar markdown metadata blocks."""
        cmd = "list_meta_blocks"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response

    def log(self, message: str, level: str = None, source: str = None, tab: str = None) -> None:
        """Append a sidebar log entry."""
        # TerminalController.parseOptions treats any --* token as an option until
        # a `--` separator. Put options first and then use `--` so messages can
        # contain arbitrary tokens like `--force`.
        cmd = "log"
        if level:
            cmd += f" --level={level}"
        if source:
            cmd += f" --source={source}"
        if tab:
            cmd += f" --tab={tab}"
        cmd += f" -- {_quote_option_value(message)}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def set_progress(self, value: float, label: str = None, tab: str = None) -> None:
        """Set sidebar progress bar (0.0-1.0)."""
        cmd = f"set_progress {value}"
        if label:
            cmd += f" --label={_quote_option_value(label)}"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def clear_progress(self, tab: str = None) -> None:
        """Clear sidebar progress bar."""
        cmd = "clear_progress"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def report_git_branch(self, branch: str, status: str = None, tab: str = None) -> None:
        """Report git branch for sidebar display."""
        cmd = f"report_git_branch {branch}"
        if status:
            cmd += f" --status={status}"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def report_pr(
        self,
        number: int,
        url: str,
        label: str = None,
        state: str = None,
        tab: str = None,
        panel: str = None,
    ) -> None:
        """Report pull-request metadata for sidebar display."""
        cmd = f"report_pr {number} {url}"
        if label:
            cmd += f" --label={_quote_option_value(label)}"
        if state:
            cmd += f" --state={state}"
        if tab:
            cmd += f" --tab={tab}"
        if panel:
            cmd += f" --panel={panel}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def report_review(
        self,
        number: int,
        url: str,
        label: str = None,
        state: str = None,
        tab: str = None,
        panel: str = None,
    ) -> None:
        """Report provider-specific review metadata (GitLab MR, Bitbucket PR, etc.)."""
        cmd = f"report_review {number} {url}"
        if label:
            cmd += f" --label={_quote_option_value(label)}"
        if state:
            cmd += f" --state={state}"
        if tab:
            cmd += f" --tab={tab}"
        if panel:
            cmd += f" --panel={panel}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def clear_pr(self, tab: str = None, panel: str = None) -> None:
        """Clear pull-request metadata for sidebar display."""
        cmd = "clear_pr"
        if tab:
            cmd += f" --tab={tab}"
        if panel:
            cmd += f" --panel={panel}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def report_ports(self, *ports: int, tab: str = None) -> None:
        """Report listening ports for sidebar display."""
        port_str = " ".join(str(p) for p in ports)
        cmd = f"report_ports {port_str}"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def clear_ports(self, tab: str = None) -> None:
        """Clear listening ports for sidebar display."""
        cmd = "clear_ports"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def report_tty(self, tty_name: str, tab: str = None, panel: str = None) -> None:
        """Register a TTY for batched port scanning."""
        cmd = f"report_tty {tty_name}"
        if tab:
            cmd += f" --tab={tab}"
        if panel:
            cmd += f" --panel={panel}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def ports_kick(self, tab: str = None, panel: str = None) -> None:
        """Request a batched port scan for the given panel."""
        cmd = "ports_kick"
        if tab:
            cmd += f" --tab={tab}"
        if panel:
            cmd += f" --panel={panel}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def sidebar_state(self, tab: str = None) -> str:
        """Dump all sidebar metadata for a tab."""
        cmd = "sidebar_state"
        if tab:
            cmd += f" --tab={tab}"
        return self._send_command(cmd)

    def reset_sidebar(self, tab: str = None) -> None:
        """Clear all sidebar metadata for a tab."""
        cmd = "reset_sidebar"
        if tab:
            cmd += f" --tab={tab}"
        response = self._send_command(cmd)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def focus_notification(self, tab: Union[str, int], surface: Union[str, int, None] = None) -> None:
        """Focus tab/surface using the notification flow."""
        if surface is None:
            command = f"focus_notification {tab}"
        else:
            command = f"focus_notification {tab} {surface}"
        response = self._send_command(command)
        if not response.startswith("OK"):
            raise cmuxError(response)

    def flash_count(self, surface: Union[str, int]) -> int:
        """Get flash count for a surface by ID or index."""
        response = self._send_command(f"flash_count {surface}")
        if response.startswith("OK "):
            return int(response.split(" ", 1)[1])
        raise cmuxError(response)

    def reset_flash_counts(self) -> None:
        """Reset flash counters."""
        response = self._send_command("reset_flash_counts")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def read_screen(self) -> str:
        """Read the visible terminal text from the focused surface."""
        return self._send_command("read_screen")

    # Workspace commands
    def list_workspaces(self) -> List[Tuple[int, str, str, bool]]:
        """List all workspaces."""
        response = self._send_command("list_workspaces")
        if response.startswith("ERROR: Unknown command"):
            return self.list_tabs()
        if response in ("No workspaces", "No tabs"):
            return []

        workspaces = []
        for line in response.split("\n"):
            if not line.strip():
                continue
            selected = line.startswith("*")
            parts = line.lstrip("* ").split(" ", 2)
            if len(parts) >= 3:
                index = int(parts[0].rstrip(":"))
                workspace_id = parts[1]
                title = parts[2] if len(parts) > 2 else ""
                workspaces.append((index, workspace_id, title, selected))
        return workspaces

    def new_workspace(self) -> str:
        """Create a new workspace. Returns the new workspace's ID."""
        response = self._send_command("new_workspace")
        if response.startswith("ERROR: Unknown command"):
            return self.new_tab()
        if response.startswith("OK "):
            return response[3:]
        raise cmuxError(response)

    def close_workspace(self, workspace_id: str) -> None:
        """Close a workspace by ID."""
        response = self._send_command(f"close_workspace {workspace_id}")
        if response.startswith("ERROR: Unknown command"):
            self.close_tab(workspace_id)
            return
        if not response.startswith("OK"):
            raise cmuxError(response)

    def select_workspace(self, workspace: Union[str, int]) -> None:
        """Select a workspace by ID or index."""
        response = self._send_command(f"select_workspace {workspace}")
        if response.startswith("ERROR: Unknown command"):
            self.select_tab(workspace)
            return
        if not response.startswith("OK"):
            raise cmuxError(response)

    # Pane commands
    def list_panes(self) -> List[Tuple[int, str, int, bool]]:
        """
        List all panes in the current workspace.
        Returns list of (index, pane_id, surface_count, is_focused) tuples.
        """
        response = self._send_command("list_panes")
        if response in ("No panes", "ERROR: No tab selected", "ERROR: No workspace selected"):
            return []

        panes = []
        for line in response.split("\n"):
            if not line.strip():
                continue
            selected = line.startswith("*")
            parts = line.lstrip("* ").split()
            if len(parts) >= 4:
                index = int(parts[0].rstrip(":"))
                pane_id = parts[1]
                surface_count = int(parts[2].lstrip("["))
                panes.append((index, pane_id, surface_count, selected))
        return panes

    def focus_pane(self, pane: Union[str, int]) -> None:
        """Focus a pane by ID or index in the current workspace."""
        response = self._send_command(f"focus_pane {pane}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def list_pane_surfaces(self, pane: Union[str, int, None] = None) -> List[Tuple[int, str, str, bool]]:
        """
        List surfaces in a pane.
        Returns list of (index, surface_id, title, is_selected) tuples.
        If pane is None, uses the focused pane.
        """
        if pane is not None:
            response = self._send_command(f"list_pane_surfaces --pane={pane}")
        else:
            response = self._send_command("list_pane_surfaces")

        if response in ("No surfaces", "No tabs in pane"):
            return []
        if response.startswith("ERROR:"):
            raise cmuxError(response)

        surfaces = []
        for line in response.split("\n"):
            if not line.strip():
                continue
            selected = line.startswith("*")
            line2 = line.lstrip("* ").strip()
            try:
                idx_part, rest = line2.split(":", 1)
                index = int(idx_part.strip())
                rest = rest.strip()
            except ValueError:
                continue

            panel_id = ""
            title = rest
            marker = " [panel:"
            if marker in rest and rest.endswith("]"):
                title, suffix = rest.split(marker, 1)
                title = title.strip()
                panel_id = suffix[:-1]
            surfaces.append((index, panel_id, title, selected))
        return surfaces

    def focus_surface_by_panel(self, surface_id: str) -> None:
        """Focus a surface by its panel ID."""
        response = self._send_command(f"focus_surface_by_panel {surface_id}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def focus_webview(self, panel_id: str) -> None:
        """Move keyboard focus into a browser panel's WKWebView."""
        response = self._send_command(f"focus_webview {panel_id}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def is_webview_focused(self, panel_id: str) -> bool:
        """Return True if the browser panel's WKWebView is first responder."""
        response = self._send_command(f"is_webview_focused {panel_id}")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response.strip().lower() == "true"

    def wait_for_webview_focus(self, panel_id: str, timeout_s: float = 2.0) -> None:
        """Poll until the browser panel's WKWebView has focus, or raise."""
        start = time.time()
        while time.time() - start < timeout_s:
            if self.is_webview_focused(panel_id):
                return
            time.sleep(0.05)
        raise cmuxError(f"Timed out waiting for webview focus: {panel_id}")

    def set_shortcut(self, name: str, combo: str) -> None:
        """Set a keyboard shortcut via the debug socket."""
        response = self._send_command(f"set_shortcut {name} {combo}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def simulate_shortcut(self, combo: str) -> None:
        """Simulate a keyDown shortcut via the debug socket."""
        response = self._send_command(f"simulate_shortcut {combo}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def simulate_type(self, text: str) -> None:
        """Insert text into the current first responder (debug builds only)."""
        escaped = (
            text
            .replace("\\", "\\\\")
            .replace("\r", "\\r")
            .replace("\n", "\\n")
            .replace("\t", "\\t")
        )
        response = self._send_command(f"simulate_type {escaped}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def simulate_file_drop(self, surface: Union[str, int], paths: Union[str, List[str]]) -> None:
        """Simulate dropping file path(s) onto a terminal surface (debug builds only)."""
        payload = paths if isinstance(paths, str) else "|".join(paths)
        response = self._send_command(f"simulate_file_drop {surface} {payload}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def seed_drag_pasteboard_fileurl(self) -> None:
        """Seed NSDrag pasteboard with public.file-url in the app process (debug builds only)."""
        response = self._send_command("seed_drag_pasteboard_fileurl")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def seed_drag_pasteboard_tabtransfer(self) -> None:
        """Seed NSDrag pasteboard with tab transfer type in the app process (debug builds only)."""
        response = self._send_command("seed_drag_pasteboard_tabtransfer")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def seed_drag_pasteboard_sidebar_reorder(self) -> None:
        """Seed NSDrag pasteboard with sidebar reorder type in the app process (debug builds only)."""
        response = self._send_command("seed_drag_pasteboard_sidebar_reorder")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def seed_drag_pasteboard_types(self, types: List[str]) -> None:
        """Seed NSDrag pasteboard with comma/space-separated types in app process."""
        if not types:
            raise cmuxError("seed_drag_pasteboard_types requires at least one type")
        payload = ",".join(t.strip() for t in types if t and t.strip())
        if not payload:
            raise cmuxError("seed_drag_pasteboard_types requires at least one non-empty type")
        response = self._send_command(f"seed_drag_pasteboard_types {payload}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def clear_drag_pasteboard(self) -> None:
        """Clear NSDrag pasteboard in the app process (debug builds only)."""
        response = self._send_command("clear_drag_pasteboard")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def overlay_hit_gate(self, event_type: str) -> bool:
        """Return whether FileDropOverlayView would capture hit-testing for event_type."""
        response = self._send_command(f"overlay_hit_gate {event_type}")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response.strip().lower() == "true"

    def overlay_drop_gate(self, source: str = "external") -> bool:
        """Return whether FileDropOverlayView would capture drag-destination routing."""
        response = self._send_command(f"overlay_drop_gate {source}")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response.strip().lower() == "true"

    def portal_hit_gate(self, event_type: str) -> bool:
        """Return whether terminal portal hit-testing should pass through to SwiftUI drag targets."""
        response = self._send_command(f"portal_hit_gate {event_type}")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response.strip().lower() == "true"

    def sidebar_overlay_gate(self, state: str = "active") -> bool:
        """Return whether sidebar outside-drop overlay would capture for drag state."""
        response = self._send_command(f"sidebar_overlay_gate {state}")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response.strip().lower() == "true"

    def drop_hit_test(self, x: float, y: float) -> Optional[str]:
        """Hit-test the file-drop overlay at normalised (0-1) coords.

        Returns the surface UUID string if a terminal is under the point, or None.
        """
        response = self._send_command(f"drop_hit_test {x} {y}")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        val = response.strip()
        return None if val == "none" else val

    def drag_hit_chain(self, x: float, y: float) -> str:
        """Return hit-view chain at normalised (0-1) coordinates."""
        response = self._send_command(f"drag_hit_chain {x} {y}")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response.strip()

    def activate_app(self) -> None:
        """Bring app + main window to front (debug builds only)."""
        response = self._send_command("activate_app")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def is_terminal_focused(self, panel: Union[str, int]) -> bool:
        """Return True if the terminal panel's Ghostty view is first responder."""
        response = self._send_command(f"is_terminal_focused {panel}")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        return response.strip().lower() == "true"

    def identify(self) -> dict:
        """Best-effort legacy identify helper."""
        response = self._send_command("identify")
        if response.startswith("ERROR"):
            raise cmuxError(response)
        try:
            return json.loads(response)
        except Exception:
            return {}

    def layout_debug(self) -> dict:
        """Return bonsplit layout snapshot + selected panel bounds."""
        response = self._send_command("layout_debug")
        if not response.startswith("OK "):
            raise cmuxError(response)
        payload = response[3:].strip()
        try:
            return json.loads(payload)
        except json.JSONDecodeError as e:
            raise cmuxError(f"layout_debug JSON decode failed: {e}: {payload[:200]}")

    def read_terminal_text(self, panel: Union[str, int, None] = None) -> str:
        """
        Read visible terminal text for a panel.
        Returns UTF-8 decoded text.
        """
        cmd = "read_terminal_text"
        if panel is not None:
            cmd += f" {panel}"
        response = self._send_command(cmd)
        if not response.startswith("OK "):
            raise cmuxError(response)
        b64 = response[3:].strip()
        raw = base64.b64decode(b64) if b64 else b""
        return raw.decode("utf-8", errors="replace")

    def render_stats(self, panel: Union[str, int, None] = None) -> dict:
        """Return terminal render stats (debug builds only)."""
        cmd = "render_stats"
        if panel is not None:
            cmd += f" {panel}"
        response = self._send_command(cmd)
        if not response.startswith("OK "):
            raise cmuxError(response)
        payload = response[3:].strip()
        try:
            return json.loads(payload)
        except json.JSONDecodeError as e:
            raise cmuxError(f"render_stats JSON decode failed: {e}: {payload[:200]}")

    def panel_snapshot_reset(self, panel: Union[str, int]) -> None:
        """Reset the stored snapshot for a panel (debug builds only)."""
        response = self._send_command(f"panel_snapshot_reset {panel}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def panel_snapshot(self, panel: Union[str, int], label: str = "") -> dict:
        """
        Capture a screenshot of a panel and return pixel-diff info.
        Returns: panel_id, changed_pixels, width, height, path.
        """
        cmd = f"panel_snapshot {panel}"
        if label:
            cmd += f" {label}"
        response = self._send_command(cmd)
        if not response.startswith("OK "):
            raise cmuxError(response)
        payload = response[3:].strip()
        parts = payload.split(" ", 4)
        if len(parts) != 5:
            raise cmuxError(f"panel_snapshot parse failed: {response}")
        panel_id, changed, width, height, path = parts
        return {
            "panel_id": panel_id,
            "changed_pixels": int(changed),
            "width": int(width),
            "height": int(height),
            "path": path,
        }

    def bonsplit_underflow_count(self) -> int:
        """Return bonsplit arranged-subview underflow counter."""
        response = self._send_command("bonsplit_underflow_count")
        if response.startswith("OK "):
            return int(response.split(" ", 1)[1])
        raise cmuxError(response)

    def reset_bonsplit_underflow_count(self) -> None:
        """Reset bonsplit arranged-subview underflow counter."""
        response = self._send_command("reset_bonsplit_underflow_count")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def empty_panel_count(self) -> int:
        """Return the number of EmptyPanelView appearances."""
        response = self._send_command("empty_panel_count")
        if response.startswith("OK "):
            return int(response.split(" ", 1)[1])
        raise cmuxError(response)

    def reset_empty_panel_count(self) -> None:
        """Reset the EmptyPanelView appearance counter."""
        response = self._send_command("reset_empty_panel_count")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def new_surface(self, pane: Union[str, int, None] = None,
                    panel_type: str = "terminal", url: str = None) -> str:
        """
        Create a new surface in a pane.
        Returns the new surface ID.
        """
        args = []
        if panel_type != "terminal":
            args.append(f"--type={panel_type}")
        if pane is not None:
            args.append(f"--pane={pane}")
        if url:
            args.append(f"--url={url}")

        cmd = "new_surface"
        if args:
            cmd += " " + " ".join(args)

        response = self._send_command(cmd)
        if response.startswith("OK "):
            return response[3:]
        raise cmuxError(response)

    def new_pane(self, direction: str = "right", panel_type: str = "terminal",
                 url: str = None) -> str:
        """
        Create a new pane (split).
        Returns the new surface/panel ID created in the new pane.
        """
        args = [f"--direction={direction}"]
        if panel_type != "terminal":
            args.append(f"--type={panel_type}")
        if url:
            args.append(f"--url={url}")

        cmd = "new_pane " + " ".join(args)
        response = self._send_command(cmd)
        if response.startswith("OK "):
            return response[3:]
        raise cmuxError(response)

    def close_surface(self, surface: Union[str, int, None] = None) -> None:
        """
        Close a surface (collapse split) by ID or index.
        If surface is None, closes the focused surface.
        """
        if surface is None:
            response = self._send_command("close_surface")
        else:
            response = self._send_command(f"close_surface {surface}")
        if not response.startswith("OK"):
            raise cmuxError(response)

    def surface_health(self, workspace: Union[str, int, None] = None) -> List[dict]:
        """
        Check view health of all surfaces in a workspace.
        Returns list of dicts with keys: index, id, type, in_window, plus any
        extra key=value fields returned by the daemon.
        """
        arg = "" if workspace is None else str(workspace)
        response = self._send_command(f"surface_health {arg}".rstrip())
        if response.startswith("ERROR") or response == "No panels":
            return []

        surfaces = []
        for line in response.split("\n"):
            if not line.strip():
                continue
            parts = line.strip().split()
            if len(parts) < 4:
                continue
            index = int(parts[0].rstrip(":"))
            surface_id = parts[1]
            kv: dict[str, str] = {}
            for token in parts[2:]:
                if "=" not in token:
                    continue
                key, value = token.split("=", 1)
                kv[key] = value

            panel_type = kv.get("type", "unknown")
            in_window = kv.get("in_window", "false") == "true"

            row: dict = {
                "index": index,
                "id": surface_id,
                "type": panel_type,
                "in_window": in_window,
            }

            for key, value in kv.items():
                if key in {"type", "in_window"}:
                    continue
                if value == "true":
                    row[key] = True
                elif value == "false":
                    row[key] = False
                elif value.isdigit() or (value.startswith("-") and value[1:].isdigit()):
                    row[key] = int(value)
                else:
                    row[key] = value

            surfaces.append(row)
        return surfaces


def main():
    """CLI interface for cmux"""
    import sys
    import argparse

    parser = argparse.ArgumentParser(description="cmux CLI")
    parser.add_argument("command", nargs="?", help="Command to send")
    parser.add_argument("args", nargs="*", help="Command arguments")
    parser.add_argument("-s", "--socket", default=None,
                        help="Socket path (default: auto-detect)")

    args = parser.parse_args()

    try:
        with cmux(args.socket) as client:
            if not args.command:
                # Interactive mode
                print("cmux CLI (type 'help' for commands, 'quit' to exit)")
                while True:
                    try:
                        line = input("> ").strip()
                        if line.lower() in ("quit", "exit"):
                            break
                        if line:
                            response = client._send_command(line)
                            print(response)
                    except EOFError:
                        break
                    except KeyboardInterrupt:
                        print()
                        break
            else:
                # Single command mode
                command = args.command
                if args.args:
                    command += " " + " ".join(args.args)
                response = client._send_command(command)
                print(response)
    except cmuxError as e:
        print(f"Error: {e}", file=sys.stderr)
        sys.exit(1)


if __name__ == "__main__":
    main()