Merge pull request #7 from shadowsaber16/feat/remote-ssh-support

feat: support remote Claude Code sessions via BUDDI_HOST env var

Author: grayashh

buddi/Buddi/Resources/buddi-hook.py

@@ -1,17 +1,59 @@
 #!/usr/bin/env python3
 """
 Buddi Hook
-- Sends session state to Buddi.app via Unix socket
+- Sends session state to Buddi.app via Unix socket (local) or TCP (remote)
 - For PermissionRequest: waits for user decision from the app
+
+Environment variables:
+  BUDDI_HOST    If set (e.g. "localhost:9999"), connect over TCP instead of
+                a Unix socket. Intended for Claude Code running on a remote
+                VM with an SSH reverse port-forward back to the Mac running
+                Buddi. Always tunnel over SSH — events may include prompts,
+                tool calls, and file paths.
+  BUDDI_SOCKET  Override the default Unix socket path (/tmp/buddi.sock).
+                Ignored when BUDDI_HOST is set.
 """
 import json
 import os
 import socket
 import sys
 
-SOCKET_PATH = "/tmp/buddi.sock"
+BUDDI_SOCKET = os.environ.get("BUDDI_SOCKET", "/tmp/buddi.sock")
+BUDDI_HOST = os.environ.get("BUDDI_HOST")
 TIMEOUT_SECONDS = 300  # 5 minutes for permission decisions
 
+if BUDDI_HOST:
+    _host = BUDDI_HOST.rpartition(":")[0].strip("[]")
+    if _host not in ("localhost", "127.0.0.1", "::1"):
+        print(
+            f"buddi-hook: warning: BUDDI_HOST={BUDDI_HOST!r} is not a loopback "
+            "address; events contain prompts and tool inputs — only use over "
+            "an SSH tunnel.",
+            file=sys.stderr,
+        )
+
+
+def _connect_to_buddi():
+    """Open a connection to Buddi (TCP if BUDDI_HOST is set, else Unix socket)."""
+    if BUDDI_HOST:
+        host, sep, port = BUDDI_HOST.rpartition(":")
+        if not sep or not host:
+            raise OSError(f"BUDDI_HOST must be host:port, got {BUDDI_HOST!r}")
+        try:
+            port_num = int(port)
+        except ValueError:
+            raise OSError(f"BUDDI_HOST port must be an integer, got {port!r}")
+        if not 0 < port_num <= 65535:
+            raise OSError(f"BUDDI_HOST port {port_num} out of range (1-65535)")
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        sock.settimeout(TIMEOUT_SECONDS)
+        sock.connect((host, port_num))
+    else:
+        sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+        sock.settimeout(TIMEOUT_SECONDS)
+        sock.connect(BUDDI_SOCKET)
+    return sock
+
 
 def get_tty():
     """Get the TTY of the Claude process (parent)"""
@@ -83,9 +125,7 @@ def get_cmux_surface():
 def send_event(state):
     """Send event to app, return response if any"""
     try:
-        sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
-        sock.settimeout(TIMEOUT_SECONDS)
-        sock.connect(SOCKET_PATH)
+        sock = _connect_to_buddi()
         sock.sendall(json.dumps(state).encode())
 
         # For permission requests, wait for response